classes/file/TemporaryFileManager.inc.php

<?php

// $Id: TemporaryFileManager.inc.php,v 1.12 2008/07/01 01:16:09 asmecher Exp $


import('file.FileManager');

class TemporaryFileManager extends FileManager {

   var $filesDir;

   function TemporaryFileManager() {
      $this->filesDir = Config::getVar('files', 'files_dir') . '/temp/';

      $this->_performPeriodicCleanup();
   }

   function &getFile($fileId, $userId) {
      $temporaryFileDao = &DAORegistry::getDAO('TemporaryFileDAO');
      $temporaryFile = &$temporaryFileDao->getTemporaryFile($fileId, $userId);
      return $temporaryFile;
   }

   function readFile($fileId, $userId, $output = false) {
      $temporaryFile = &$this->getFile($fileId, $userId);

      if (isset($temporaryFile)) {
         $filePath = $this->filesDir . $temporaryFile->getFileName();
         return parent::readFile($filePath, $output);
      } else {
         return false;
      }
   }

   function deleteFile($fileId, $userId) {
      $temporaryFile = &$this->getFile($fileId, $userId);

      parent::deleteFile($this->filesDir . $temporaryFile->getFileName());

      $temporaryFileDao = &DAORegistry::getDAO('TemporaryFileDAO');
      $temporaryFileDao->deleteTemporaryFileById($fileId, $userId);
   }

   function downloadFile($fileId, $userId, $inline = false) {
      $temporaryFile = &$this->getFile($fileId, $userId);
      if (isset($temporaryFile)) {
         $filePath = $this->filesDir . $temporaryFile->getFileName();
         return parent::downloadFile($filePath, null, $inline);
      } else {
         return false;
      }
   }

   function viewFile($fileId) {
      $this->downloadFile($fileId, true);
   }

   function parseFileExtension($fileName) {
      $fileParts = explode('.', $fileName);
      if (is_array($fileParts)) {
         $fileExtension = $fileParts[count($fileParts) - 1];
      }

      // FIXME Check for evil
      if (!isset($fileExtension) || strstr($fileExtension, 'php') || strlen($fileExtension) > 6 || !preg_match('/^\w+$/', $fileExtension)) {
         $fileExtension = 'txt';
      }

      return $fileExtension;
   }

   function handleUpload($fileName, $userId) {
      // Get the file extension, then rename the file.
      $fileExtension = $this->parseFileExtension($this->getUploadedFileName($fileName));        

      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 ($this->uploadFile($fileName, $this->filesDir . $newFileName)) {
         $temporaryFileDao = &DAORegistry::getDAO('TemporaryFileDAO');
         $temporaryFile = &new TemporaryFile();

         $temporaryFile->setUserId($userId);
         $temporaryFile->setFileName($newFileName);
         $temporaryFile->setFileType($_FILES[$fileName]['type']);
         $temporaryFile->setFileSize($_FILES[$fileName]['size']);
         $temporaryFile->setOriginalFileName(TemporaryFileManager::truncateFileName($_FILES[$fileName]['name'], 127));
         $temporaryFile->setDateUploaded(Core::getCurrentDate());

         $temporaryFileDao->insertTemporaryFile($temporaryFile);

         return $temporaryFile;

      } else {
         return false;
      }
   }

   function articleToTemporaryFile($articleFile, $userId) {
      // Get the file extension, then rename the file.
      $fileExtension = $this->parseFileExtension($articleFile->getFileName());         

      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($articleFile->getFilePath(), $this->filesDir . $newFileName)) {
         $temporaryFileDao = &DAORegistry::getDAO('TemporaryFileDAO');
         $temporaryFile = &new TemporaryFile();

         $temporaryFile->setUserId($userId);
         $temporaryFile->setFileName($newFileName);
         $temporaryFile->setFileType($articleFile->getFileType());
         $temporaryFile->setFileSize($articleFile->getFileSize());
         $temporaryFile->setOriginalFileName($articleFile->getOriginalFileName());
         $temporaryFile->setDateUploaded(Core::getCurrentDate());

         $temporaryFileDao->insertTemporaryFile($temporaryFile);

         return $temporaryFile;

      } else {
         return false;
      }
   }

   function _performPeriodicCleanup() {
      if (time() % 100 == 0) {
         $temporaryFileDao = &DAORegistry::getDAO('TemporaryFileDAO');
         $expiredFiles = $temporaryFileDao->getExpiredFiles();
         foreach ($expiredFiles as $expiredFile) {
            $this->deleteFile($expiredFile->getFileId(), $expiredFile->getUserId());
         }
      }
   }
}

?>

---