TemporaryFileManager.inc.php

<?php
 
import('lib.pkp.classes.file.PrivateFileManager');
 
class TemporaryFileManager extends PrivateFileManager {
   function __construct() {
      parent::__construct();
 
      $this->_performPeriodicCleanup();
   }
 
   function getBasePath() {
      return parent::getBasePath() . '/temp/';
   }
 
   function getFile($fileId, $userId) {
      $temporaryFileDao = DAORegistry::getDAO('TemporaryFileDAO'); /* @var $temporaryFileDao TemporaryFileDAO */
      return $temporaryFileDao->getTemporaryFile($fileId, $userId);
   }
 
   function deleteById($fileId, $userId) {
      $temporaryFile = $this->getFile($fileId, $userId);
 
      parent::deleteByPath($this->getBasePath() . $temporaryFile->getServerFileName());
 
      $temporaryFileDao = DAORegistry::getDAO('TemporaryFileDAO'); /* @var $temporaryFileDao TemporaryFileDAO */
      $temporaryFileDao->deleteTemporaryFileById($fileId, $userId);
   }
 
   function downloadById($fileId, $userId, $inline = false) {
      $temporaryFile = $this->getFile($fileId, $userId);
      if (isset($temporaryFile)) {
         $filePath = $this->getBasePath() . $temporaryFile->getServerFileName();
         return parent::downloadByPath($filePath, null, $inline);
      } else {
         return false;
      }
   }
 
   function handleUpload($fileName, $userId) {
      // Get the file extension, then rename the file.
      $fileExtension = $this->parseFileExtension($this->getUploadedFileName($fileName));
 
      if (!$this->fileExists($this->getBasePath(), 'dir')) {
         // Try to create destination directory
         $this->mkdirtree($this->getBasePath());
      }
 
      $newFileName = basename(tempnam($this->getBasePath(), $fileExtension));
      if (!$newFileName) return false;
 
      if ($this->uploadFile($fileName, $this->getBasePath() . $newFileName)) {
         $temporaryFileDao = DAORegistry::getDAO('TemporaryFileDAO'); /* @var $temporaryFileDao TemporaryFileDAO */
         $temporaryFile = $temporaryFileDao->newDataObject();
 
         $temporaryFile->setUserId($userId);
         $temporaryFile->setServerFileName($newFileName);
         $exploded = explode('.', $_FILES[$fileName]['name']);
         $temporaryFile->setFileType(PKPString::mime_content_type($this->getBasePath() . $newFileName, array_pop($exploded)));
         $temporaryFile->setFileSize($_FILES[$fileName]['size']);
         $temporaryFile->setOriginalFileName($this->truncateFileName($_FILES[$fileName]['name'], 127));
         $temporaryFile->setDateUploaded(Core::getCurrentDate());
 
         $temporaryFileDao->insertObject($temporaryFile);
 
         return $temporaryFile;
 
      } else {
         return false;
      }
   }
 
   function submissionToTemporaryFile($submissionFile, $userId) {
      // Get the file extension, then rename the file.
      $fileExtension = $this->parseFileExtension($submissionFile->getServerFileName());
 
      if (!$this->fileExists($this->filesDir, 'dir')) {
         // Try to create destination directory
         $this->mkdirtree($this->filesDir);
      }
 
      $newFileName = basename(tempnam($this->filesDir, $fileExtension));
      if (!$newFileName) return false;
 
      if (copy($submissionFile->getFilePath(), $this->filesDir . $newFileName)) {
         $temporaryFileDao = DAORegistry::getDAO('TemporaryFileDAO'); /* @var $temporaryFileDao TemporaryFileDAO */
         $temporaryFile = $temporaryFileDao->newDataObject();
 
         $temporaryFile->setUserId($userId);
         $temporaryFile->setServerFileName($newFileName);
         $temporaryFile->setFileType($submissionFile->getFileType());
         $temporaryFile->setFileSize($submissionFile->getFileSize());
         $temporaryFile->setOriginalFileName($submissionFile->getOriginalFileName());
         $temporaryFile->setDateUploaded(Core::getCurrentDate());
 
         $temporaryFileDao->insertObject($temporaryFile);
 
         return $temporaryFile;
 
      } else {
         return false;
      }
   }
 
   function _performPeriodicCleanup() {
      if (time() % 100 == 0) {
         $temporaryFileDao = DAORegistry::getDAO('TemporaryFileDAO'); /* @var $temporaryFileDao TemporaryFileDAO */
         $expiredFiles = $temporaryFileDao->getExpiredFiles();
         foreach ($expiredFiles as $expiredFile) {
            $this->deleteById($expiredFile->getId(), $expiredFile->getUserId());
         }
      }
   }
}