/*

  * Copyright 2005-2007 The Kuali Foundation

  * 

  * 

  * Licensed under the Educational Community License, Version 2.0 (the "License");

  * you may not use this file except in compliance with the License.

  * You may obtain a copy of the License at

  * 

  * http://www.opensource.org/licenses/ecl2.php

  * 

  * Unless required by applicable law or agreed to in writing, software

  * distributed under the License is distributed on an "AS IS" BASIS,

  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

  * See the License for the specific language governing permissions and

  * limitations under the License.

  */

 package org.kuali.rice.kew.notes.service.impl;

 import java.io.BufferedOutputStream;

 import java.io.File;

 import java.io.FileOutputStream;

 import java.util.logging.Logger;

 import org.kuali.rice.kew.notes.Attachment;

 import org.kuali.rice.kew.notes.service.AttachmentService;

 import org.kuali.rice.kew.service.KEWServiceLocator;

 /**

  * Implementation of the {@link AttachmentService}.

  *

  * @author Kuali Rice Team (rice.collab@kuali.org)

  */

 public class AttachmentServiceImpl implements AttachmentService {

         protected final org.apache.log4j.Logger LOG = org.apache.log4j.Logger.getLogger(AttachmentServiceImpl.class);

         private static final String ATTACHMENT_PREPEND = "wf_att_";

         private String attachmentDir;

         public void persistAttachedFileAndSetAttachmentBusinessObjectValue(Attachment attachment) throws Exception {

                 createStorageDirIfNecessary();

                 String uniqueId = KEWServiceLocator.getResponsibilityIdService().getNewResponsibilityId();

                 String internalFileIndicator = attachment.getFileName().replace('.', '_');

                 String fileName = ATTACHMENT_PREPEND + attachment.getNote().getDocumentId() + "_" + internalFileIndicator + "_" + uniqueId;

                 File file = File.createTempFile(fileName, null, new File(attachmentDir));

                 LOG.info("Persisting attachment at: " + file.getAbsolutePath());

                 if (!file.canWrite()) {

                         throw new RuntimeException("Do not have permissions to write to attachment file at: " + file.getAbsolutePath());

                 }

                 FileOutputStream streamOut = null;

         BufferedOutputStream bufferedStreamOut = null;

         try {

             streamOut = new FileOutputStream(file);

             bufferedStreamOut = new BufferedOutputStream(streamOut);

             int c;

             while ((c = attachment.getAttachedObject().read()) != -1) 

                 {

                     bufferedStreamOut.write(c);

                 }

         } finally {

                 if (bufferedStreamOut != null) {

                         bufferedStreamOut.close();

                 }

             if (streamOut != null) {

                     streamOut.close();

             }

         }

         attachment.setFileLoc(file.getAbsolutePath());

         }

         public File findAttachedFile(Attachment attachment) throws Exception {

                 return new File(attachment.getFileLoc());

         }

         public void deleteAttachedFile(Attachment attachment) throws Exception {

                 File file = new File(attachment.getFileLoc());

                 if (! file.delete()) {

                         LOG.error("failed to delete file " + attachment.getFileLoc());

                 }

         }

         private void createStorageDirIfNecessary() {

                 if (attachmentDir == null) {

                         throw new RuntimeException("Attachment Directory was not set when configuring workflow");

                 }

                 File attachDir = new File(attachmentDir);

                 if (! attachDir.exists()) {

                         LOG.warn("No attachment directory found.  Attempting to create directory " + attachmentDir);

                         boolean directoriesCreated = attachDir.mkdirs();

                         if (!directoriesCreated) {

                                 throw new RuntimeException("Failed to create directory for attachments at: " + attachDir.getAbsolutePath());

                         }

                 }

                 if (!attachDir.canWrite()) {

                         throw new RuntimeException("Do not have permission to write to: " + attachDir.getAbsolutePath());

                 }

         }

         public String getAttachmentDir() {

                 return attachmentDir;

         }

         public void setAttachmentDir(String attachmentDir) {

                 this.attachmentDir = attachmentDir;

         }

 }