abe7e055 |
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 | { |
02838718 |
92 | if (!$url || @parse_url($url) === false) { |
93 | trigger_error('malformed URL given', E_USER_NOTICE); |
94 | return false; |
95 | } |
abe7e055 |
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 | |
da419622 |
142 | static public function listRawFiles($forlife = '*', $category = '*', $uniq = false, $basename = false) |
abe7e055 |
143 | { |
144 | global $globals; |
145 | $filename = $globals->spoolroot . '/spool/uploads/temp/'; |
146 | $filename .= $forlife . '-' . $category; |
da419622 |
147 | if (!$uniq) { |
148 | $filename .= '-*'; |
149 | } |
abe7e055 |
150 | $files = glob($filename); |
151 | if ($basename) { |
da419622 |
152 | $files = array_map('basename', $files); |
153 | } |
154 | return $files; |
155 | } |
156 | |
157 | static public function listFilenames($forlife = '*', $category = '*') |
158 | { |
159 | $files = PlUpload::listRawFiles($forlife, $category, false, true); |
160 | foreach ($files as &$name) { |
161 | list($forlife, $cat, $fn) = explode('-', $name, 3); |
162 | $name = $fn; |
abe7e055 |
163 | } |
164 | return $files; |
165 | } |
166 | |
da419622 |
167 | static public function &listFiles($forlife = '*', $category = '*', $uniq = false) |
168 | { |
169 | $res = array(); |
170 | $files = PlUpload::listRawFiles($forlife, $category, $uniq, true); |
171 | foreach ($files as $name) { |
172 | list($forlife, $cat, $fn) = explode('-', $name, 3); |
173 | $res[$fn] = new PlUpload($forlife, $cat, $fn); |
174 | } |
175 | return $res; |
176 | } |
177 | |
178 | static public function clear($user = '*', $category = '*', $uniq = false) |
abe7e055 |
179 | { |
da419622 |
180 | $files = PlUpload::listRawFiles($user, $category, $uniq, false); |
181 | array_map('unlink', $files); |
abe7e055 |
182 | } |
183 | |
184 | public function contentType() |
185 | { |
186 | return $this->type; |
187 | } |
188 | |
02838718 |
189 | public function isType($type, $subtype = null) |
190 | { |
191 | list($mytype, $mysubtype) = explode('/', $this->type); |
192 | if ($mytype != $type || ($subtype && $mysubtype != $subtype)) { |
193 | return false; |
194 | } |
195 | return true; |
196 | } |
197 | |
abe7e055 |
198 | public function imageInfo() |
199 | { |
02838718 |
200 | static $map; |
201 | if (!isset($map)) { |
202 | $map = array (1 => 'gif', 2 => 'jpeg', 3 => 'png'); |
203 | } |
204 | $array = getimagesize($this->filename); |
205 | $array[2] = @$map[$array[2]]; |
206 | if (!$array[2]) { |
207 | trigger_error('unknown image type', E_USER_NOTICE); |
208 | return null; |
209 | } |
210 | return $array; |
abe7e055 |
211 | } |
212 | |
213 | public function resizeImage($max_x = -1, $max_y = -1, $min_x = 0, $min_y = 0, $maxsize = -1) |
214 | { |
215 | if (!$this->exists() || strpos($this->type, 'image/') !== 0) { |
216 | trigger_error('not an image', E_USER_NOTICE); |
217 | return false; |
218 | } |
219 | $image_infos = $this->imageInfo(); |
02838718 |
220 | if (!$image_infos) { |
abe7e055 |
221 | trigger_error('invalid image', E_USER_NOTICE); |
222 | return false; |
223 | } |
224 | list($this->x, $this->y, $mimetype) = $image_infos; |
abe7e055 |
225 | if ($max_x == -1) { |
226 | $max_x = $this->x; |
227 | } |
228 | if ($max_y == -1) { |
229 | $max_y = $this->y; |
230 | } |
231 | if ($maxsize == -1) { |
232 | $maxsize = filesize($this->filename); |
233 | } |
234 | if (filesize($this->filename) > $maxsize || $this->x > $max_x || $this->y > $max_y |
235 | || $this->x < $min_x || $this->y < $min_y) { |
236 | $img = imagecreatefromstring(file_get_contents($this->filename)); |
237 | if (!$img) { |
238 | trigger_error('too large image, can\'t be resized', E_USER_NOTICE); |
239 | return false; |
240 | } |
241 | |
242 | $nx = $this->x; |
243 | $ny = $this->y; |
244 | if ($nx > $max_x) { |
245 | $ny = intval($ny*$max_x/$nx); |
246 | $nx = $max_x; |
247 | } |
248 | if ($ny > $max_y) { |
249 | $nx = intval($nx*$max_y/$ny); |
250 | $ny = $max_y; |
251 | } |
252 | if ($nx < $min_x) { |
253 | $ny = intval($ny*$min_x/$nx); |
254 | $nx = $min_x; |
255 | } |
256 | if ($ny < $min_y) { |
257 | $nx = intval($nx * $min_y/$ny); |
258 | $ny = $min_y; |
259 | } |
260 | |
261 | $comp = 90; |
262 | do { |
263 | $img2 = imagecreatetruecolor($nx, $ny); |
264 | imagecopyresampled($img2, $img, 0, 0, 0, 0, $nx, $ny, $this->x, $this->y); |
265 | imagejpeg($img2, $this->filename, $comp); |
266 | $comp --; |
267 | clearstatcache(); |
268 | } while (filesize($this->filename) > $maxsize && $comp > 0); |
269 | $this->type = 'image/jpeg'; |
270 | $this->x = $nx; |
271 | $this->y = $ny; |
272 | } |
273 | return true; |
274 | } |
275 | |
276 | public function getContents() |
277 | { |
278 | if ($this->exists()) { |
279 | return file_get_contents($this->filename); |
280 | } |
281 | return null; |
282 | } |
283 | } |
284 | // vim:set et sw=4 sts=4 sws=4 foldmethod=marker enc=utf-8: |
285 | ?> |