classes/plupload.php

   1 <?php
   2 /***************************************************************************
   3  *  Copyright (C) 2003-2007 Polytechnique.org                              *
   4  *  http://opensource.polytechnique.org/                                   *
   5  *                                                                         *
   6  *  This program is free software; you can redistribute it and/or modify   *
   7  *  it under the terms of the GNU General Public License as published by   *
   8  *  the Free Software Foundation; either version 2 of the License, or      *
   9  *  (at your option) any later version.                                    *
  10  *                                                                         *
  11  *  This program is distributed in the hope that it will be useful,        *
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of         *
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the          *
  14  *  GNU General Public License for more details.                           *
  15  *                                                                         *
  16  *  You should have received a copy of the GNU General Public License      *
  17  *  along with this program; if not, write to the Free Software            *
  18  *  Foundation, Inc.,                                                      *
  19  *  59 Temple Place, Suite 330, Boston, MA  02111-1307  USA                *
  20  ***************************************************************************/
  21
  22 /** Class to store per user and per category files
  23  */
  24 class PlUpload
  25 {
  26     private $forlife;
  27     private $category;
  28     private $file_id;
  29
  30     private $filename;
  31     private $type;
  32
  33     /** For images
  34      */
  35     private $x;
  36     private $y;
  37
  38     public function __construct($forlife, $category, $filename = null)
  39     {
  40         $this->file_id  = $filename;
  41         $this->category = $category;
  42         $this->forlife  = $forlife;
  43         $this->filename = $this->makeFilename($this->file_id);
  44         $this->checkContentType();
  45     }
  46
  47     private function makeFilename($file_id)
  48     {
  49         global $globals;
  50         $filename = $globals->spoolroot . '/spool/uploads/temp/';
  51         if (!file_exists($filename)) {
  52             if (!mkdir($filename)) {
  53                 trigger_error('can\'t create upload directory: ' . $filename, E_USER_ERROR);
  54             }
  55         }
  56         $filename .= $this->forlife . '-' . $this->category;
  57         if ($file_id) {
  58             $filename .= '-' . $file_id;
  59         }
  60         return $filename;
  61     }
  62
  63     private function checkContentType()
  64     {
  65         if ($this->exists()) {
  66             $this->type = trim(mime_content_type($this->filename));
  67         }
  68     }
  69
  70     public function upload(array &$file)
  71     {
  72         if (!is_uploaded_file($file['tmp_name'])) {
  73             return false;
  74         } else if (!move_uploaded_file($file['tmp_name'], $this->filename)) {
  75             return false;
  76         }
  77         $this->checkContentType();
  78         return true;
  79     }
  80
  81     public function copyFrom($filename)
  82     {
  83         if (!copy($filename, $this->filename)) {
  84             return false;
  85         }
  86         $this->checkContentType();
  87         return true;
  88     }
  89
  90     public function download($url)
  91     {
  92         $data = file_get_contents($url);
  93         if (!$data) {
  94             return false;
  95         }
  96         if (!file_put_contents($this->filename, $data)) {
  97             return false;
  98         }
  99         $this->checkContentType();
 100         return true;
 101     }
 102
 103     static public function &get(array &$file, $forlife, $category, $uniq = false)
 104     {
 105         $upload = new PlUpload($forlife, $category, $uniq ? null : $file['name']);
 106         if (!$upload->upload($file)) {
 107             $upload = null;
 108         }
 109         return $upload;
 110     }
 111
 112     public function rm()
 113     {
 114         @unlink($this->filename);
 115         clearstatcache();
 116     }
 117
 118     public function rename($fn)
 119     {
 120         if (!$this->file_id) {
 121             return false;
 122         }
 123         $filename = $this->makeFilename($fn);
 124         if (rename($this->filename)) {
 125             $this->filename = $filename;
 126             $this->file_id  = $fn;
 127             clearstatcache();
 128             return true;
 129         }
 130         return false;
 131     }
 132
 133     public function exists()
 134     {
 135         return file_exists($this->filename);
 136     }
 137
 138     static public function listFiles($forlife = '*', $category = '*', $basename = false)
 139     {
 140         global $globals;
 141         $filename = $globals->spoolroot . '/spool/uploads/temp/';
 142         $filename .= $forlife . '-' . $category;
 143         $files = glob($filename);
 144         if ($basename) {
 145             array_walk($files, 'basename');
 146         }
 147         return $files;
 148     }
 149
 150     static public function clear($user = '*', $category = '*')
 151     {
 152         $files = PlUpload::listFiles($user, $category, false);
 153         array_walk($files, 'unlink');
 154     }
 155
 156     public function contentType()
 157     {
 158         return $this->type;
 159     }
 160
 161     public function imageInfo()
 162     {
 163         return getimagesize($this->filename);
 164     }
 165
 166     public function resizeImage($max_x = -1, $max_y = -1, $min_x = 0, $min_y = 0, $maxsize = -1)
 167     {
 168         if (!$this->exists() || strpos($this->type, 'image/') !== 0) {
 169             trigger_error('not an image', E_USER_NOTICE);
 170             return false;
 171         }
 172         $image_infos = $this->imageInfo();
 173         if (empty($image_infos)) {
 174             trigger_error('invalid image', E_USER_NOTICE);
 175             return false;
 176         }
 177         list($this->x, $this->y, $mimetype) = $image_infos;
 178         if ($mimetype < 1 || $mimetype > 3) { // 1 - gif, 2 - jpeg, 3 - png
 179             trigger_error('unknown image type', E_USER_NOTICE);
 180             return false;
 181         }
 182         if ($max_x == -1) {
 183             $max_x = $this->x;
 184         }
 185         if ($max_y == -1) {
 186             $max_y = $this->y;
 187         }
 188         if ($maxsize == -1) {
 189             $maxsize = filesize($this->filename);
 190         }
 191         if (filesize($this->filename) > $maxsize || $this->x > $max_x || $this->y > $max_y
 192                                                  || $this->x < $min_x || $this->y < $min_y) {
 193             $img = imagecreatefromstring(file_get_contents($this->filename));
 194             if (!$img) {
 195                 trigger_error('too large image, can\'t be resized', E_USER_NOTICE);
 196                 return false;
 197             }
 198
 199             $nx = $this->x;
 200             $ny = $this->y;
 201             if ($nx > $max_x) {
 202                 $ny = intval($ny*$max_x/$nx);
 203                 $nx = $max_x;
 204             }
 205             if ($ny > $max_y) {
 206                 $nx = intval($nx*$max_y/$ny);
 207                 $ny = $max_y;
 208             }
 209             if ($nx < $min_x) {
 210                 $ny = intval($ny*$min_x/$nx);
 211                 $nx = $min_x;
 212             }
 213             if ($ny < $min_y) {
 214                 $nx = intval($nx * $min_y/$ny);
 215                 $ny = $min_y;
 216             }
 217
 218             $comp = 90;
 219             do {
 220                 $img2 = imagecreatetruecolor($nx, $ny);
 221                 imagecopyresampled($img2, $img, 0, 0, 0, 0, $nx, $ny, $this->x, $this->y);
 222                 imagejpeg($img2, $this->filename, $comp);
 223                 $comp --;
 224                 clearstatcache();
 225             } while (filesize($this->filename) > $maxsize && $comp > 0);
 226             $this->type = 'image/jpeg';
 227             $this->x    = $nx;
 228             $this->y    = $ny;
 229         }
 230         return true;
 231     }
 232
 233     public function getContents()
 234     {
 235         if ($this->exists()) {
 236             return file_get_contents($this->filename);
 237         }
 238         return null;
 239     }
 240 }
 241 // vim:set et sw=4 sts=4 sws=4 foldmethod=marker enc=utf-8:
 242 ?>