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         if (!$url || @parse_url($url) === false) {
  93             trigger_error('malformed URL given', E_USER_NOTICE);
  94             return false;
  95         }
  96         $data = file_get_contents($url);
  97         if (!$data) {
  98             return false;
  99         }
 100         if (!file_put_contents($this->filename, $data)) {
 101             return false;
 102         }
 103         $this->checkContentType();
 104         return true;
 105     }
 106
 107     static public function &get(array &$file, $forlife, $category, $uniq = false)
 108     {
 109         $upload = new PlUpload($forlife, $category, $uniq ? null : $file['name']);
 110         if (!$upload->upload($file)) {
 111             $upload = null;
 112         }
 113         return $upload;
 114     }
 115
 116     public function rm()
 117     {
 118         @unlink($this->filename);
 119         clearstatcache();
 120     }
 121
 122     public function rename($fn)
 123     {
 124         if (!$this->file_id) {
 125             return false;
 126         }
 127         $filename = $this->makeFilename($fn);
 128         if (rename($this->filename)) {
 129             $this->filename = $filename;
 130             $this->file_id  = $fn;
 131             clearstatcache();
 132             return true;
 133         }
 134         return false;
 135     }
 136
 137     public function exists()
 138     {
 139         return file_exists($this->filename);
 140     }
 141
 142     static public function listFiles($forlife = '*', $category = '*', $basename = false)
 143     {
 144         global $globals;
 145         $filename = $globals->spoolroot . '/spool/uploads/temp/';
 146         $filename .= $forlife . '-' . $category;
 147         $files = glob($filename);
 148         if ($basename) {
 149             array_walk($files, 'basename');
 150         }
 151         return $files;
 152     }
 153
 154     static public function clear($user = '*', $category = '*')
 155     {
 156         $files = PlUpload::listFiles($user, $category, false);
 157         array_walk($files, 'unlink');
 158     }
 159
 160     public function contentType()
 161     {
 162         return $this->type;
 163     }
 164
 165     public function isType($type, $subtype = null)
 166     {
 167         list($mytype, $mysubtype) = explode('/', $this->type);
 168         if ($mytype != $type || ($subtype && $mysubtype != $subtype)) {
 169             return false;
 170         }
 171         return true;
 172     }
 173
 174     public function imageInfo()
 175     {
 176         static $map;
 177         if (!isset($map)) {
 178             $map = array (1 => 'gif', 2 => 'jpeg', 3 => 'png');
 179         }
 180         $array = getimagesize($this->filename);
 181         $array[2] = @$map[$array[2]];
 182         if (!$array[2]) {
 183             trigger_error('unknown image type', E_USER_NOTICE);
 184             return null;
 185         }
 186         return $array;
 187     }
 188
 189     public function resizeImage($max_x = -1, $max_y = -1, $min_x = 0, $min_y = 0, $maxsize = -1)
 190     {
 191         if (!$this->exists() || strpos($this->type, 'image/') !== 0) {
 192             trigger_error('not an image', E_USER_NOTICE);
 193             return false;
 194         }
 195         $image_infos = $this->imageInfo();
 196         if (!$image_infos) {
 197             trigger_error('invalid image', E_USER_NOTICE);
 198             return false;
 199         }
 200         list($this->x, $this->y, $mimetype) = $image_infos;
 201         if ($max_x == -1) {
 202             $max_x = $this->x;
 203         }
 204         if ($max_y == -1) {
 205             $max_y = $this->y;
 206         }
 207         if ($maxsize == -1) {
 208             $maxsize = filesize($this->filename);
 209         }
 210         if (filesize($this->filename) > $maxsize || $this->x > $max_x || $this->y > $max_y
 211                                                  || $this->x < $min_x || $this->y < $min_y) {
 212             $img = imagecreatefromstring(file_get_contents($this->filename));
 213             if (!$img) {
 214                 trigger_error('too large image, can\'t be resized', E_USER_NOTICE);
 215                 return false;
 216             }
 217
 218             $nx = $this->x;
 219             $ny = $this->y;
 220             if ($nx > $max_x) {
 221                 $ny = intval($ny*$max_x/$nx);
 222                 $nx = $max_x;
 223             }
 224             if ($ny > $max_y) {
 225                 $nx = intval($nx*$max_y/$ny);
 226                 $ny = $max_y;
 227             }
 228             if ($nx < $min_x) {
 229                 $ny = intval($ny*$min_x/$nx);
 230                 $nx = $min_x;
 231             }
 232             if ($ny < $min_y) {
 233                 $nx = intval($nx * $min_y/$ny);
 234                 $ny = $min_y;
 235             }
 236
 237             $comp = 90;
 238             do {
 239                 $img2 = imagecreatetruecolor($nx, $ny);
 240                 imagecopyresampled($img2, $img, 0, 0, 0, 0, $nx, $ny, $this->x, $this->y);
 241                 imagejpeg($img2, $this->filename, $comp);
 242                 $comp --;
 243                 clearstatcache();
 244             } while (filesize($this->filename) > $maxsize && $comp > 0);
 245             $this->type = 'image/jpeg';
 246             $this->x    = $nx;
 247             $this->y    = $ny;
 248         }
 249         return true;
 250     }
 251
 252     public function getContents()
 253     {
 254         if ($this->exists()) {
 255             return file_get_contents($this->filename);
 256         }
 257         return null;
 258     }
 259 }
 260 // vim:set et sw=4 sts=4 sws=4 foldmethod=marker enc=utf-8:
 261 ?>