Source for file KTUploadManager.inc.php

Documentation is available at KTUploadManager.inc.php

  1. <?
  2.  
  3. /**
  4.  *
  5.  * KTUploadManager manages files in the uploaded_files table.
  6.  * @license http://www.knowledgetree.com/KPL KnowledgeTree Public License Version 1.1
  7.  * @package KTWEBSERVICE
  8.  */
  9.  
  10. /*
  11.  *
  12.  * The contents of this file are subject to the KnowledgeTree Public
  13.  * License Version 1.1 ("License"); You may not use this file except in
  14.  * compliance with the License. You may obtain a copy of the License at
  15.  * http://www.knowledgetree.com/KPL
  16.  * 
  17.  * Software distributed under the License is distributed on an "AS IS"
  18.  * basis,
  19.  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  20.  * for the specific language governing rights and limitations under the
  21.  * License.
  22.  * 
  23.  * The Original Code is: KnowledgeTree Open Source
  24.  * 
  25.  * The Initial Developer of the Original Code is The Jam Warehouse Software
  26.  * (Pty) Ltd, trading as KnowledgeTree.
  27.  * Portions created by The Jam Warehouse Software (Pty) Ltd are Copyright
  28.  * (C) 2007 The Jam Warehouse Software (Pty) Ltd;
  29.  * All Rights Reserved.
  30.  *
  31.  */
  32.  
  33. class KTUploadManager
  34. {
  35.     var $userid;
  36.     var $age;
  37.     var $temp_dir;
  38.     var $session;
  39.     
  40.  
  41.     function KTUploadManager()
  42.     {
  43.         $config = KTConfig::getSingleton();         
  44.         
  45.         $this->age = $config->get('webservice/uploadExpiry',60);
  46.         $this->temp_dir= $config->get('webservice/uploadDirectory');
  47.     } 
  48.     
  49.     /**
  50.      * Sets the current session.
  51.      *
  52.      * @param KTAPI_Session $session 
  53.      */
  54.     function set_session($session)
  55.     {
  56.         $user = &$session->get_user();
  57.         $this->userid=$user->getId();
  58.         $this->session = $session->get_session();        
  59.     }
  60.  
  61.     /**
  62.      * This tells the manager to manage a file that has been uploaded.
  63.      *
  64.      * @param string $filename 
  65.      * @param string $tempfile 
  66.      * @param string $action 
  67.      */
  68.     function uploaded($filename, $tempfile, $action, $relatedid = null)
  69.     {
  70.         $filename=basename($filename);
  71.         $now=date('Y-m-d H:i:s');
  72.         $now_str=date('YmdHis');
  73.         
  74.         $tempfile = str_replace('/','\\',$tempfile);
  75.         $newtempfile = str_replace('\\','/',realpath($this->temp_dir) . '/' .  $this->userid  . '-'. $now_str);
  76.         
  77.         DBUtil::startTransaction();
  78.         $id = DBUtil::autoInsert('uploaded_files',
  79.             array(
  80.                 'tempfilename'=>$newtempfile,
  81.                 'filename'=>$filename,
  82.                 'userid'=>$this->userid,
  83.                 'uploaddate'=>$now,
  84.                 'action'=>$action,
  85.                 'related_uploadid'=>$relatedid                
  86.                 ),
  87.                 array('noid'=>true)    
  88.             );
  89.             
  90.         if (PEAR::isError($id))
  91.         {
  92.             DBUtil::rollback();
  93.             return $id;
  94.         }
  95.  
  96.         $result = move_uploaded_file($tempfile, $newtempfile);
  97.         if ($result == false)
  98.         {
  99.  
  100.             DBUtil::rollback();
  101.             return false;
  102.         }
  103.  
  104.         DBUtil::commit();
  105.  
  106.         return $newtempfile;
  107.     }
  108.     
  109.     /**
  110.      * This is a list of all all managed files.
  111.      *
  112.      * @param string $action 
  113.      */
  114.     function get_uploaded_list($action=null)
  115.     {
  116.         $sql = "SELECT id, tempfilename, filename, userid, action, related_uploadid FROM uploaded_files WHERE userid=$this->userid";
  117.         if (!is_null($action))
  118.         {
  119.             $sql .= " AND action='$action'";
  120.         }
  121.         $result = DBUtil::getResultArray($sql);
  122.         return $result;
  123.     }
  124.     
  125.     function imported_file($action, $filename, $documentid)
  126.     {
  127.         DBUtil::startTransaction();
  128.         $filename=basename($filename);
  129.         $sql = "DELETE FROM uploaded_files WHERE action='$action' AND filename='$filename'";
  130.         $rs = DBUtil::runQuery($sql);
  131.         if (PEAR::isError($rs))
  132.         {
  133.             DBUtil::rollback();
  134.             return false;
  135.         }
  136.                 
  137.         $sql = "INSERT INTO index_files(document_id, user_id) VALUES($documentid, $this->userid)";
  138.         DBUtil::runQuery($sql);
  139.         if (PEAR::isError($rs))
  140.         {
  141.             DBUtil::rollback();
  142.             return false;
  143.         }
  144.         
  145.         DBUtil::commit();
  146.         return true;
  147.     }
  148.     
  149.     /**
  150.      * This will remove any temporary files that have not been dealt with in the correct timeframe.
  151.      *
  152.      */
  153.     function cleanup()
  154.     {
  155.         list($year,$mon,$day,$hour, $min) = explode(':', date('Y:m:d:H:i'));
  156.         $expirydate = date('Y-m-d H:i:s', mktime($hour, $min - $this->age, 0, $mon, $day, $year));
  157.         
  158.         $sql = "SELECT tempfilename FROM uploaded_files WHERE uploaddate<'$expirydate'";
  159.         $rows = DBUtil::getResultArray($sql);
  160.         
  161.         foreach($rows as $record)
  162.         {
  163.             $tempfilename=addslashes($record['tempfilename']);
  164.             
  165.             $sql = "DELETE FROM uploaded_files WHERE tempfilename='$tempfilename'";
  166.             $rs = DBUtil::runQuery($sql);
  167.             if (PEAR::isError($rs))
  168.             {
  169.                 continue;    
  170.             }
  171.             
  172.             @unlink($tempfilename);        
  173.         }
  174.     }
  175. }
  176. ?>

Documentation generated on Sun, 22 Apr 2007 02:30:55 +0200 by phpDocumentor 1.3.2