check if dir can be created
[surrogator.git] / surrogator.php
1 #!/usr/bin/env php
2 <?php
3 namespace surrogator;
4 $cfgFile = __DIR__ . '/data/surrogator.config.php';
5 if (!file_exists($cfgFile)) {
6     $cfgFile = '/etc/surrogator.config.php';
7     if (!file_exists($cfgFile)) {
8         logErr(
9             "Configuration file does not exist.\n"
10             . "Copy data/surrogator.config.php.dist to data/surrogator.config.php"
11         );
12         exit(2);
13     }
14 }
15 require $cfgFile;
16
17 array_shift($argv);
18 $files = array();
19 foreach ($argv as $arg) {
20     if ($arg == '-v' || $arg == '--verbose') {
21         ++$logLevel;
22     } else if ($arg == '-vv') {
23         $logLevel += 2;
24     } else if ($arg == '-q' || $arg == '--quiet') {
25         $logLevel = 0;
26     } else if ($arg == '-f' || $arg == '--force') {
27         $forceUpdate = true;
28     } else if ($arg == '-h' || $arg == '--help') {
29         showHelp();
30         exit(4);
31     } else if ($arg == '--version') {
32         echo "surrogator 0.0.1\n";
33         exit();
34     } else if (file_exists($arg)) {
35         $files[] = $arg;
36     } else {
37         logErr('Unknown argument: ' . $arg);
38         exit(3);
39     }
40 }
41
42 function showHelp()
43 {
44     echo <<<HLP
45 Usage: php surrogator.php [options] [filename(s)]
46
47 surrogator - a simple libravatar server
48  Put files in raw/ dir and run surrogator.php to generate different sizes
49
50 Options:
51
52  -h, --help     Show help
53  -v, --verbose  Be verbose (more log messages, also -vv)
54  -q, --quiet    Be quiet (no log messages)
55  -f, --force    Force update of all files
56      --version  Show program version
57
58 filenames       One or several files whose small images shall get generated.
59                 If none given, all will be checked
60
61 HLP;
62 }
63
64 if (!isset($rawDir)) {
65     logErr('$rawDir not set');
66     exit(1);
67 }
68 if (!isset($varDir)) {
69     logErr('$varDir not set');
70     exit(1);
71 }
72 if (!isset($sizes)) {
73     logErr('$sizes not set');
74     exit(1);
75 }
76 if (!isset($maxSize)) {
77     logErr('$maxSize not set');
78     exit(1);
79 }
80 if (!isset($logLevel)) {
81     logErr('$logLevel not set');
82     exit(1);
83 }
84
85
86 if (!is_dir($varDir . '/square')) {
87     log('creating square dir: ' . $varDir . '/square');
88     if (!mkdir($varDir . '/square', 0755, true)) {
89         logErr('cannot create square dir');
90         exit(5);
91     }
92 }
93 log('sizes: ' . implode(', ', $sizes), 2);
94 foreach ($sizes as $size) {
95     if (!is_dir($varDir . '/' . $size)) {
96         log('creating size dir: ' . $varDir . '/' . $size);
97         mkdir($varDir . '/' . $size, 0755);
98     }
99 }
100
101 if (count($files)) {
102     $fileInfos = array();
103     foreach ($files as $file) {
104         $fileInfos[] = new \SplFileInfo($file);
105     }
106 } else {
107     $fileInfos = new \RegexIterator(
108         new \DirectoryIterator($rawDir),
109         '#^.+\.(png|jpg)$#'
110     );
111 }
112 foreach ($fileInfos as $fileInfo) {
113     $origPath   = $fileInfo->getPathname();
114     $fileName   = $fileInfo->getFilename();
115     $ext        = strtolower(substr($fileName, strrpos($fileName, '.') + 1));
116     $squarePath = $varDir . '/square/'
117         . substr($fileName, 0, -strlen($ext)) . 'png';
118
119     log('processing ' . $fileName, 1);
120     if (image_uptodate($origPath, $squarePath)) {
121         log(' image up to date', 2);
122         continue;
123     }
124
125     if (!createSquare($origPath, $ext, $squarePath, $maxSize)) {
126         continue;
127     }
128
129     if ($fileName == 'default.png') {
130         $md5 = $sha256 = 'default';
131     } else {
132         list($md5, $sha256) = getHashes($fileName);
133     }
134
135     log(' creating sizes for ' . $fileName, 2);
136     log(' md5:    ' . $md5, 3);
137     log(' sha256: ' . $sha256, 3);
138     $imgSquare = imagecreatefrompng($squarePath);
139     foreach ($sizes as $size) {
140         log(' size ' . $size, 3);
141         $sizePathMd5    = $varDir . '/' . $size . '/' . $md5 . '.png';
142         $sizePathSha256 = $varDir . '/' . $size . '/' . $sha256 . '.png';
143
144         $imgSize = imagecreatetruecolor($size, $size);
145         imagealphablending($imgSize, false);
146         imagefilledrectangle(
147             $imgSize, 0, 0, $size - 1, $size - 1,
148             imagecolorallocatealpha($imgSize, 0, 0, 0, 127)
149         );
150         imagecopyresampled(
151             $imgSize, $imgSquare, 
152             0, 0, 0, 0,
153             $size, $size, $maxSize, $maxSize
154         );
155         imagesavealpha($imgSize, true);
156         imagepng($imgSize, $sizePathMd5);
157         imagepng($imgSize, $sizePathSha256);
158         imagedestroy($imgSize);
159         
160     }
161     imagedestroy($imgSquare);
162 }
163
164 function getHashes($fileName)
165 {
166     $fileNameNoExt = substr($fileName, 0, -strlen(strrpos($fileName, '.')) - 2);
167     $emailAddress  = trim(strtolower($fileNameNoExt));
168
169     return array(
170         md5($emailAddress), hash('sha256', $emailAddress)
171     );
172 }
173
174
175 function createSquare($origPath, $ext, $targetPath, $maxSize)
176 {
177     if ($ext == 'png') {
178         $imgOrig = imagecreatefrompng($origPath);
179     } else if ($ext == 'jpg' || $ext == 'jpeg') {
180         $imgOrig = imagecreatefromjpeg($origPath);
181     } else {
182         //unsupported format
183         return false;
184     }
185
186     if ($imgOrig === false) {
187         logErr('Error loading image file: ' . $origPath);
188         return false;
189     }
190
191     $imgSquare = imagecreatetruecolor($maxSize, $maxSize);
192     imagealphablending($imgSquare, false);
193     imagefilledrectangle(
194         $imgSquare, 0, 0, $maxSize - 1, $maxSize - 1,
195         imagecolorallocatealpha($imgSquare, 0, 0, 0, 127)
196     );
197     imagealphablending($imgSquare, true);
198
199     $oWidth    = imagesx($imgOrig);
200     $oHeight   = imagesy($imgOrig);
201     if ($oWidth > $oHeight) {
202         $flScale = $maxSize / $oWidth;
203     } else {
204         $flScale = $maxSize / $oHeight;
205     }
206     $nWidth  = (int)($oWidth * $flScale);
207     $nHeight = (int)($oHeight * $flScale);
208
209     imagecopyresampled(
210         $imgSquare, $imgOrig, 
211         ($maxSize - $nWidth) / 2, ($maxSize - $nHeight) / 2,
212         0, 0,
213         $nWidth, $nHeight,
214         $oWidth, $oHeight
215     );
216
217     imagesavealpha($imgSquare, true);
218     imagepng($imgSquare, $targetPath);
219
220     imagedestroy($imgSquare);
221     imagedestroy($imgOrig);
222     return true;
223 }
224
225 function image_uptodate($sourcePath, $targetPath)
226 {
227     global $forceUpdate;
228     if ($forceUpdate) {
229         return false;
230     }
231     if (!file_exists($targetPath)) {
232         return false;
233     }
234     if (filemtime($sourcePath) > filemtime($targetPath)) {
235         //source newer
236         return false;
237     }
238
239     return true;
240 }
241
242 function log($msg, $level = 1)
243 {
244     global $logLevel;
245     if ($level <= $logLevel) {
246         echo $msg . "\n";
247     }
248 }
249
250 function logErr($msg)
251 {
252     file_put_contents('php://stderr', $msg . "\n");
253 }
254
255 ?>