chiark / gitweb /
mason/.perl-lib/TrivGal.pm (Image): Use `Image::Size' to find image size.
[tgal] / mason / .perl-lib / TrivGal.pm
CommitLineData
6ac5dde2
MW
1### -*-cperl-*-
2###
3### Main output for Trivial Gallery.
4###
5### (c) 2021 Mark Wooding
6###
7
8###----- Licensing notice ---------------------------------------------------
9###
10### This file is part of Trivial Gallery.
11###
12### Trivial Gallery is free software: you can redistribute it and/or modify
13### it under the terms of the GNU Affero General Public License as
14### published by the Free Software Foundation; either version 3 of the
15### License, or (at your option) any later version.
16###
17### Trivial Gallery is distributed in the hope that it will be useful, but
18### WITHOUT ANY WARRANTY; without even the implied warranty of
19### MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
20### Affero General Public License for more details.
21###
22### You should have received a copy of the GNU Affero General Public
23### License along with Trivial Gallery. If not, see
24### <https://www.gnu.org/licenses/>.
25
26package TrivGal;
27
28use autodie qw{:all};
29
30use Errno;
31use Exporter qw{import};
6ac5dde2 32use File::stat;
bda837fc 33use Image::ExifTool qw{};
6ac5dde2 34use Image::Imlib2;
8b97a4e3 35use Image::Size qw{};
6ac5dde2
MW
36use User::pwent;
37use POSIX;
38
39our @EXPORT;
40sub export (@) { push @EXPORT, @_; }
41
42###--------------------------------------------------------------------------
43### Internal utilities.
44
45sub read_or_set ($\$@) {
46 my ($me, $ref, @args) = @_;
47 if (@args == 0) { return $$ref; }
48 elsif (@args == 1) { $$ref = $args[0]; return $me; }
49 elsif (@args > 1) { die "too many arguments"; }
50}
51
52###--------------------------------------------------------------------------
53### Random utilities.
54
55export qw{join_paths};
56sub join_paths (@) {
57 my @p = @_;
58 my $p = "";
59 ELT: for my $e (@p) {
e9eac06d
MW
60 $e =~ s#^/{2,}#/#;
61 $e =~ s#([^/])/+$#$1#;
6ac5dde2 62 if ($e eq "") { next ELT; }
e9eac06d 63 elsif ($p eq "" || $e =~ m#^/#) { $p = $e; }
6ac5dde2
MW
64 else { $p = "$p/$e"; }
65 }
66 return $p;
67}
68
69export qw{split_path};
70sub split_path ($) {
71 my ($path) = @_;
72
9fc5e8c1
MW
73 my ($dir, $base, $ext) =
74 $path =~ m#^ (?: (.*) /)?
75 (?: ([^/]*) \.)?
76 ([^./]*) $#x;
6ac5dde2
MW
77 if (defined $base) { $ext = ".$ext"; }
78 else { $base = $ext; $ext = ""; }
79 return ($dir, $base, $ext);
80}
81
82export qw{urlencode};
83sub urlencode ($) {
84 my ($u) = @_;
e9eac06d 85 $u =~ s#([^0-9a-zA-Z_./~-])#sprintf "%%%02x", ord $1#eg;
6ac5dde2
MW
86 return $u;
87}
88
89export qw{urldecode};
90sub urldecode ($) {
91 my ($u) = @_;
e9eac06d 92 $u =~ s#\%([0-9a-fA-F]{2})#chr hex $1#eg;
6ac5dde2
MW
93 return $u;
94}
95
96###--------------------------------------------------------------------------
97### Image types.
98
99our %TYPE;
100
101package TrivGal::ImageType {
102 sub new ($$) {
103 my ($cls, $ext) = @_;
104 return $TYPE{$ext} = bless { ext => $ext }, $cls;
105 }
106 sub ext ($) {
107 my ($me, @args) = @_;
108 return $me->{ext};
109 }
110 sub mimetype ($@) {
111 my ($me, @args) = @_;
112 return TrivGal::read_or_set $me, $me->{mimetype}, @args;
113 }
114 sub imlibfmt ($@) {
115 my ($me, @args) = @_;
116 return TrivGal::read_or_set $me, $me->{imlibfmt}, @args;
117 }
65e4ebfd 118}
6ac5dde2
MW
119
120TrivGal::ImageType->new(".jpg")->mimetype("image/jpeg")->imlibfmt("jpeg");
121TrivGal::ImageType->new(".png")->mimetype("image/png")->imlibfmt("png");
122
123###--------------------------------------------------------------------------
124### Configuration.
125
126export qw{$SCOPE $SUFFIX};
127our $SCOPE //= $::SCOPE;
128our $SUFFIX //= $::SUFFIX;
129
130export qw{$IMGROOT $CACHE $TMP};
131our $IMGROOT //= "$ENV{HOME}/publish/$SCOPE-html$SUFFIX/tgal-images";
132our $CACHE //=
133 ($ENV{XDG_CACHE_HOME} // "$ENV{HOME}/.cache") .
134 "/tgal/$SCOPE$SUFFIX";
135our $TMP //= "$CACHE/tmp";
136
137export qw{$ROOTURL $IMGURL $CACHEURL $STATICURL $SCRIPTURL};
138my $user = getpwuid($>)->name;
139our $ROOTURL //= "/~$user";
140our $IMGURL //= "$ROOTURL/tgal-images";
141our $CACHEURL //= "$ROOTURL/tgal-cache";
142our $STATICURL //= "$ROOTURL/tgal-static";
143our $SCRIPTURL;
144
a38a867d
MW
145export qw{$SRCURL};
146our $SRCURL = "https://git.distorted.org.uk/~mdw/tgal/";
147
6ac5dde2 148export qw{%SIZE};
1408b7a2
MW
149our %SIZE = (smallthumb => 96,
150 medthumb => 144,
151 bigthumb => 228,
40c686c2
MW
152 small => 480,
153 embed => 720,
1408b7a2 154 view => 1200);
6ac5dde2
MW
155
156export qw{init};
157my $initp = 0;
158sub init () {
159 my $m = HTML::Mason::Request->instance;
160 my $r = $m->cgi_request;
161
162 $m->interp->set_escape(u => sub { my ($r) = @_; $$r = urlencode $$r; });
163
164 return unless !$initp;
165
166 $SCRIPTURL //= $r->subprocess_env("SCRIPT_NAME");
167 $initp = 1;
168}
169
170###--------------------------------------------------------------------------
171### Temporary files.
172
173export qw{clean_temp_files};
174sub clean_temp_files () {
175 my $d;
176
177 eval { opendir $d, $TMP; };
178 if ($@) {
179 if ($@->isa("autodie::exception") && $@->errno == ENOENT) { return; }
180 else { die $@; }
181 }
182 my $now = time;
183 FILE: while (my $name = readdir $d) {
184 next FILE unless $name =~ /^t(\d+)\-/;
185 my $pid = $1;
186 next FILE if kill 0, $pid;
187 my $f = "$TMP/$name";
188 my $st = stat $name;
189 next FILE if $now - $st->mtime() < 300;
190 unlink $f;
191 }
192 closedir $d;
193}
194
195###--------------------------------------------------------------------------
196### Scaled images.
197
bda837fc
MW
198my %ORIENT =
199 (1 => [0, 0],
200 2 => [0, 1],
201 3 => [2, 0],
202 4 => [2, 1],
203 5 => [3, 1],
204 6 => [1, 0],
205 7 => [1, 1],
206 8 => [3, 0]);
207
4e74ddf4
MW
208package TrivGal::Image {
209 use File::Path qw{make_path};
210 use File::stat;
211
212 sub new ($$) {
213 my ($cls, $path) = @_;
214 my $imgpath = "$IMGROOT/$path";
215 my $st = stat $imgpath or die "no image `$path'";
216 return bless {
4fcd273f 217 path => $path, imgpath => $imgpath,
4e74ddf4 218 mtime => $st->mtime,
8b97a4e3
MW
219 img => undef,
220 _wd => undef, _ht => undef,
221 sz => undef
4e74ddf4
MW
222 }, $cls;
223 }
224
8b97a4e3
MW
225 sub _getsz ($) {
226 my ($me) = @_;
227 return if defined $me->{_wd};
228
229 my ($wd, $ht, $err) = Image::Size::imgsize $me->{imgpath};
230 defined $wd or die "failed to read size of `$me->{path}': $err";
231 my $sz = $wd; if ($sz < $ht) { $sz = $ht; }
232 @$me{"_wd", "_ht", "sz"} = ($wd, $ht, $sz);
233 }
234
235 sub sz ($) { my ($me) = @_; $me->_getsz; return $me->{sz}; }
236
784bdf8f
MW
237 sub scale ($$;$) {
238 my ($me, $scale, $forcep) = @_;
1b3d6791 239 my $m = HTML::Mason::Request->instance;
4e74ddf4
MW
240
241 my $path = $me->{path};
242 my $sz = $SIZE{$scale} or die "unknown scale `$scale'";
243 my $thumb = "$CACHE/scale.$sz/$path";
1b3d6791
MW
244 my $thumburl =
245 $m->interp->apply_escapes("$CACHEURL/scale.$sz/$path", "u");
4e74ddf4
MW
246 my $st = stat $thumb;
247 if (defined $st && $st->mtime > $me->{mtime}) { return $thumburl; }
784bdf8f
MW
248 return
249 $m->interp->apply_escapes("$SCRIPTURL/$path", "u") . "?scale=$scale"
250 unless $forcep;
4e74ddf4
MW
251
252 my ($dir, $base, $ext) = TrivGal::split_path $thumb;
253 my $ty = $TYPE{lc $ext} or die "unknown type `$ext'";
254
255 my $img = $me->{img};
256 unless (defined $img) {
bda837fc 257 my $exif = new Image::ExifTool;
4fcd273f 258 $exif->ExtractInfo($me->{imgpath});
bda837fc 259 my $orient = $exif->GetValue("Orientation", "ValueConv");
4fcd273f 260 $img = $me->{img} = Image::Imlib2->load($me->{imgpath});
bda837fc
MW
261 if (defined $orient) {
262 my ($rot, $flip) = @{$ORIENT{$orient}};
263 if ($rot) { $img->image_orientate($rot); }
f6158607 264 if ($flip) { $img->flip_horizontal(); }
bda837fc 265 }
4e74ddf4
MW
266 }
267
8b97a4e3 268 if ($me->sz <= $sz)
33b14649 269 { return $m->interp->apply_escapes("$IMGURL/$path", "u"); }
8b97a4e3 270 my $sc = $sz/$me->sz;
4e74ddf4
MW
271 my $scaled = $img->create_scaled_image($sc*$wd, $sc*$ht);
272
273 $scaled->image_set_format($ty->imlibfmt);
274 $scaled->set_quality(90);
227d23a0 275 my $new = "$TMP/t$$-$ext";
2306963b 276 make_path $TMP, { mode => 0771 };
4e74ddf4 277 $scaled->save($new);
2306963b 278 make_path $dir, { mode => 0771 };
4e74ddf4
MW
279 rename $new, $thumb;
280 return $thumburl;
281 }
6ac5dde2
MW
282}
283
284###--------------------------------------------------------------------------
285### Directory listings.
286
287package TrivGal::Item {
288 sub new ($$) {
289 my ($cls, $name) = @_;
290 return bless { name => $name }, $cls;
291 }
292 sub name ($@) {
293 my ($me, @args) = @_;
294 return TrivGal::read_or_set $me, $me->{name}, @args;
295 }
296 sub comment ($@) {
297 my ($me, @args) = @_;
298 return TrivGal::read_or_set $me, $me->{comment}, @args;
299 }
65e4ebfd 300}
6ac5dde2
MW
301
302export qw{listdir};
303sub listdir ($) {
304 my ($path) = @_;
305 my (@d, @f);
306 my $ix = undef;
307
24f4eac3 308 $path =~ s#/$##;
6ac5dde2
MW
309 if (-f "$path/.tgal.index") {
310 open my $f, "<", "$path/.tgal.index";
311 my $item = undef;
312 my $comment = undef;
313 LINE: while (<$f>) {
314 chomp;
315 next LINE if /^\s*(\#|$)/;
316 if (s/^\s+//) {
317 die "no item" unless $item;
318 $comment = defined $comment ? $comment . "\n" . $_ : $_;
319 } else {
320 if ($item && $comment) { $item->comment($comment); }
9940cc5a 321 my ($flags, $name, $c) =
6e51720c 322 /^ (?: ([-!]+) \s+)? # flags
9fc5e8c1
MW
323 (\S+) \s* # filename
324 (\S | \S.*\S )? # start of the comment
325 \s*
326 $/x;
9940cc5a 327 my $indexp = $flags =~ /!/;
6e51720c 328 my $hidep = $flags =~ /-/;
6ac5dde2
MW
329 $name = urldecode $name;
330 my $list;
0485371e 331 $item = TrivGal::Item->new($name);
e9eac06d 332 if ($name =~ m#/$#) {
6ac5dde2
MW
333 $list = \@d;
334 die "can't index a folder" if $indexp;
335 } else {
336 $list = \@f;
b39b2681 337 my ($dir, $base, $ext) = split_path $name;
6ac5dde2
MW
338 die "unknown image type" unless $TYPE{lc $ext};
339 if ($indexp) {
340 die "two index images" if defined $ix;
341 $ix = $item;
342 }
343 }
6ac5dde2 344 $comment = $c;
6e51720c 345 push @$list, $item unless $hidep;
6ac5dde2
MW
346 }
347 }
348 if ($item && $comment) { $item->comment($comment); }
349 close $f;
350 } else {
ebe19115 351 my $st = stat $path;
32fc7001
MW
352 unless ($st->mode&0004) { return ([], [], undef); }
353
6ac5dde2
MW
354 opendir $d, $path;
355 my @e = readdir $d;
356 closedir $d;
357
358 ENT: for my $e (sort @e) {
359 my ($dir, $base, $ext) = split_path $e;
360 my $dotp = $e =~ /^\./;
361 my $st = stat "$path/$e";
362 my $list = undef;
32fc7001 363 if ($dotp) { }
24f4eac3 364 elsif (-d $st) { $list = \@d; $e .= "/"; }
6ac5dde2
MW
365 elsif ($TYPE{lc $ext} && -f $st) { $list = \@f; }
366 $list and push @$list, TrivGal::Item->new($e);
367 }
368 $ix = $f[0] if @f;
369 }
370
371 return (\@d, \@f, $ix);
372}
373
374export qw{contents};
375sub contents ($) {
376 my ($file) = @_;
377 my $contents = "";
378 my $f;
379 local $@;
380 eval { open $f, "<", "$file"; };
381 if ($@) {
382 if ($@->isa("autodie::exception") && $@->errno == ENOENT)
383 { return undef; }
384 die $@;
385 }
386 while (sysread $f, $buf, 16384) { $contents .= $buf; }
387 close $f;
388 return $contents;
389}
390
391export qw{find_covering_file};
392sub find_covering_file ($$$) {
393 my ($top, $path, $name) = @_;
394 for (;;) {
395 my $stuff = contents "$top/$path/$name"; return $stuff if defined $stuff;
396 if ($path eq "") { return undef; }
e9eac06d 397 if ($path =~ m#^(.*)/[^/]+/?#) { $path = $1; }
6ac5dde2
MW
398 else { $path = ""; }
399 }
400}
401
402###----- That's all, folks --------------------------------------------------
403
404clean_temp_files;
405
4061;