Add a counter for how much each equipment combination is used.
[pr0n] / perl / Sesse / pr0n / Index.pm
1 package Sesse::pr0n::Index;
2 use strict;
3 use warnings;
4
5 use Sesse::pr0n::Common qw(error dberror);
6 use Apache2::Request;
7 use POSIX;
8
9 sub handler {
10         my $r = shift;
11         my $apr = Apache2::Request->new($r);
12         my $dbh = Sesse::pr0n::Common::get_dbh();
13
14         # Find the event
15         $r->uri =~ m#^/([a-zA-Z0-9-]+)/?$#
16                 or error($r, "Could not extract event");
17         my $event = $1;
18
19         # Fix common error: pr0n.sesse.net/event -> pr0n.sesse.net/event/
20         if ($r->uri !~ m#/$#) {
21                 $r->headers_out->{'location'} = "/$event/";
22                 return Apache2::Const::REDIRECT;
23         }
24
25         # Internal? (Ugly?) 
26         if ($r->get_server_name =~ /internal/ || $r->get_server_name =~ /skoyen\.bilder\.knatten\.com/) {
27                 my $user = Sesse::pr0n::Common::check_access($r);
28                 if (!defined($user)) {
29                         return Apache2::Const::OK;
30                 }
31         }
32
33         # Read the appropriate settings from the query string into the settings hash
34         my %defsettings = (
35                 thumbxres => 80,
36                 thumbyres => 64,
37                 xres => -1,
38                 yres => -1,
39                 start => 1,
40                 num => -1,
41                 all => 1,
42                 infobox => 1,
43                 rot => 0,
44                 sel => 0,
45                 fullscreen => 0,
46         );
47         
48         # Any NEF files => default to processing
49         my $ref = $dbh->selectrow_hashref('SELECT * FROM images WHERE event=? AND LOWER(filename) LIKE \'%.nef\' LIMIT 1',
50                 undef, $event)
51                 and $defsettings{'xres'} = $defsettings{'yres'} = undef;
52         
53         # Reduce the front page load when in overload mode.
54         if (Sesse::pr0n::Overload::is_in_overload($r)) {
55                 $defsettings{'num'} = 100;
56         }
57                 
58         my %settings = %defsettings;
59
60         for my $s qw(thumbxres thumbyres xres yres start num all infobox rot sel fullscreen) {
61                 my $val = $apr->param($s);
62                 if (defined($val) && $val =~ /^(\d+)$/) {
63                         $settings{$s} = $val;
64                 }
65                 if (($s eq "num" || $s eq "xres" || $s eq "yres") && defined($val) && $val == -1) {
66                         $settings{$s} = $val;
67                 }
68         }
69
70         my $thumbxres = $settings{'thumbxres'};
71         my $thumbyres = $settings{'thumbyres'};
72         my $xres = $settings{'xres'};
73         my $yres = $settings{'yres'};
74         my $start = $settings{'start'};
75         my $num = $settings{'num'};
76         my $all = $settings{'all'};
77         my $infobox = $settings{'infobox'} ? '' : 'nobox/';
78         my $rot = $settings{'rot'};
79         my $sel = $settings{'sel'};
80
81         if (defined($num) && $num == -1) {
82                 $num = undef;
83         }
84
85         $ref = $dbh->selectrow_hashref('SELECT name,date,EXTRACT(EPOCH FROM last_update) AS last_update FROM events WHERE id=? AND vhost=?',
86                 undef, $event, $r->get_server_name)
87                 or error($r, "Could not find event $event", 404, "File not found");
88
89         my $date = HTML::Entities::encode_entities(Encode::decode_utf8($ref->{'date'}));
90         my $name = HTML::Entities::encode_entities(Encode::decode_utf8($ref->{'name'}));
91         $r->set_last_modified($ref->{'last_update'});
92                                 
93         # If the client can use cache, do so
94         if ((my $rc = $r->meets_conditions) != Apache2::Const::OK) {
95                 return $rc;
96         }
97         
98         # Count the number of selected images.
99         $ref = $dbh->selectrow_hashref("SELECT COUNT(*) AS num_selected FROM images WHERE event=? AND selected=\'t\'", undef, $event);
100         my $num_selected = $ref->{'num_selected'};
101
102         # Find all images related to this event.
103         my $q;
104         my $where = ($all == 0) ? ' AND selected=\'t\'' : '';
105
106         if (defined($start) && defined($num) && !$settings{'fullscreen'}) {
107                 $q = $dbh->prepare("SELECT *, (date - INTERVAL '6 hours')::date AS day FROM images WHERE event=? $where ORDER BY (date - INTERVAL '6 hours')::date,takenby,date,filename LIMIT $num OFFSET " . ($start-1))
108                         or dberror($r, "prepare()");
109         } else {
110                 $q = $dbh->prepare("SELECT *, (date - INTERVAL '6 hours')::date AS day FROM images WHERE event=? $where ORDER BY (date - INTERVAL '6 hours')::date,takenby,date,filename")
111                         or dberror($r, "prepare()");
112         }
113         $q->execute($event)
114                 or dberror($r, "image enumeration");
115
116         # Print the page itself
117         if ($settings{'fullscreen'}) {
118                 $r->content_type("text/html; charset=utf-8");
119                 Sesse::pr0n::Templates::print_template($r, "fullscreen-header", { title => "$name [$event]" });
120                 while (my $ref = $q->fetchrow_hashref()) {
121                         $r->print("        \"" . $infobox . $ref->{'filename'} . "\",\n");
122                 }
123
124                 my %settings_no_fullscreen = %settings;
125                 $settings_no_fullscreen{'fullscreen'} = 0;
126
127                 my $returnurl = "http://" . $r->get_server_name . "/" . $event . "/" .
128                         Sesse::pr0n::Common::get_query_string(\%settings_no_fullscreen, \%defsettings);
129
130                 # *whistle*
131                 $returnurl =~ s/&/&/g;
132
133                 Sesse::pr0n::Templates::print_template($r, "fullscreen-footer", {
134                         vhost => $r->get_server_name,
135                         event => $event,
136                         start => $settings{'start'} - 1,
137                         returnurl => $returnurl,
138                         sel => $settings{'sel'},
139                 });
140         } else {
141                 Sesse::pr0n::Common::header($r, "$name [$event]");
142                 Sesse::pr0n::Templates::print_template($r, "date", { date => $date });
143
144                 if (Sesse::pr0n::Overload::is_in_overload($r)) {
145                         Sesse::pr0n::Templates::print_template($r, "overloadmode");
146                 }
147
148                 print_thumbsize($r, $event, \%settings, \%defsettings);
149                 print_viewres($r, $event, \%settings, \%defsettings);
150                 print_pagelimit($r, $event, \%settings, \%defsettings);
151                 print_infobox($r, $event, \%settings, \%defsettings);
152                 print_nextprev($r, $event, \%settings, \%defsettings);
153                 print_selected($r, $event, \%settings, \%defsettings) if ($num_selected > 0);
154                 print_fullscreen($r, $event, \%settings, \%defsettings);
155                 
156                 # Find the equipment used
157                 my $eq = $dbh->prepare('
158                         SELECT 
159                                 model.value AS model,
160                                 coalesce(lens_spec.value, lens.value) AS lens,
161                                 COUNT(*) AS num
162                         FROM images i
163                                 LEFT JOIN exif_info model ON i.id=model.image
164                                 LEFT JOIN ( SELECT * FROM exif_info WHERE tag=\'Lens\' ) lens ON i.id=lens.image
165                                 LEFT JOIN ( SELECT * FROM exif_info WHERE tag=\'LensSpec\') lens_spec ON i.id=lens_spec.image
166                         WHERE event=? AND model.tag=\'Model\'
167                         GROUP BY 1,2
168                         ORDER BY 1,2')
169                         or die "Couldn't prepare to find equipment: $!";
170                 $eq->execute($event)
171                         or die "Couldn't find equipment: $!";
172
173                 my @equipment = ();
174                 my %cameras_seen = ();
175                 while (my $ref = $eq->fetchrow_hashref) {
176                         if (!defined($ref->{'lens'}) && exists($cameras_seen{$ref->{'model'}})) {
177                                 #
178                                 # Some compact cameras seem to add lens info sometimes and not at other
179                                 # times; if we have seen a camera with at least one specific lens earlier,
180                                 # just combine entries without a lens with the previous one.
181                                 #
182                                 $equipment[$#equipment]->{'num'} += $ref->{'num'};
183                                 next;
184                         }
185                         push @equipment, $ref;
186                         $cameras_seen{$ref->{'model'}} = 1;
187                 }
188                 $eq->finish;
189
190                 if (scalar @equipment > 0) {
191                         Sesse::pr0n::Templates::print_template($r, "equipment-start");
192                         for my $e (@equipment) {
193                                 my $eqspec = $e->{'model'};
194                                 $eqspec .= ', ' . $e->{'lens'} if (defined($e->{'lens'}));
195
196                                 # This isn't correct for all languages. Fix if we ever need to care. :-)
197                                 if ($e->{'num'} == 1) {
198                                         Sesse::pr0n::Templates::print_template($r, "equipment-item-singular", { eqspec => $eqspec });
199                                 } else {
200                                         Sesse::pr0n::Templates::print_template($r, "equipment-item", { eqspec => $eqspec, num => $e->{'num'} });
201                                 }
202                         }
203                         Sesse::pr0n::Templates::print_template($r, "equipment-end");
204                 }
205
206
207                 my $toclose = 0;
208                 my $lastupl = "";
209                 
210                 # Print out all thumbnails
211                 if ($rot == 1) {
212                         $r->print("    <form method=\"post\" action=\"/rotate\">\n");
213                         $r->print("      <input type=\"hidden\" name=\"event\" value=\"$event\" />\n");
214                 
215                         while (my $ref = $q->fetchrow_hashref()) {
216                                 my $imgsz = "";
217                                 my $takenby = $ref->{'takenby'};
218                                 if (defined($ref->{'day'})) {
219                                          $takenby .= ", " . $ref->{'day'};
220                                 }
221
222                                 if ($takenby ne $lastupl) {
223                                         $lastupl = $takenby;
224                                         Sesse::pr0n::Templates::print_template($r, "submittedby", { author => $lastupl });
225                                 }
226                                 if ($ref->{'width'} != -1 && $ref->{'height'} != -1) {
227                                         my $width = $ref->{'width'};
228                                         my $height = $ref->{'height'};
229                                                 
230                                         ($width, $height) = Sesse::pr0n::Common::scale_aspect($width, $height, $thumbxres, $thumbyres);
231                                         $imgsz = " width=\"$width\" height=\"$height\"";
232                                 }
233
234                                 my $filename = $ref->{'filename'};
235                                 my $uri = $infobox . $filename;
236                                 if (defined($xres) && defined($yres) && $xres != -1) {
237                                         $uri = "${xres}x$yres/$infobox$filename";
238                                 } elsif (defined($xres) && $xres == -1) {
239                                         $uri = "original/$infobox$filename";
240                                 }
241
242                                 $r->print("    <p><a href=\"$uri\"><img src=\"${thumbxres}x${thumbyres}/$filename\" alt=\"\"$imgsz /></a>\n");
243                                 $r->print("      90 <input type=\"checkbox\" name=\"rot-" .
244                                         $ref->{'id'} . "-90\" />\n");
245                                 $r->print("      180 <input type=\"checkbox\" name=\"rot-" .
246                                         $ref->{'id'} . "-180\" />\n");
247                                 $r->print("      270 <input type=\"checkbox\" name=\"rot-" .
248                                         $ref->{'id'} . "-270\" />\n");
249                                 $r->print("      &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;" .
250                                         "&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;Del <input type=\"checkbox\" name=\"del-" . $ref->{'id'} . "\" /></p>\n");
251                         }
252                         $r->print("      <input type=\"submit\" value=\"Rotate\" />\n");
253                         $r->print("    </form>\n");
254                 } elsif ($sel == 1) {
255                         $r->print("    <form method=\"post\" action=\"/select\">\n");
256                         $r->print("      <input type=\"hidden\" name=\"event\" value=\"$event\" />\n");
257                 
258                         while (my $ref = $q->fetchrow_hashref()) {
259                                 my $imgsz = "";
260                                 my $takenby = $ref->{'takenby'};
261                                 if (defined($ref->{'day'})) {
262                                          $takenby .= ", " . $ref->{'day'};
263                                 }
264
265                                 if ($takenby ne $lastupl) {
266                                         $lastupl = $takenby;
267                                         Sesse::pr0n::Templates::print_template($r, "submittedby", { author => $lastupl });
268                                 }
269                                 if ($ref->{'width'} != -1 && $ref->{'height'} != -1) {
270                                         my $width = $ref->{'width'};
271                                         my $height = $ref->{'height'};
272                                                 
273                                         ($width, $height) = Sesse::pr0n::Common::scale_aspect($width, $height, $thumbxres, $thumbyres);
274                                         $imgsz = " width=\"$width\" height=\"$height\"";
275                                 }
276
277                                 my $filename = $ref->{'filename'};
278                                 my $uri = $infobox . $filename;
279                                 if (defined($xres) && defined($yres) && $xres != -1) {
280                                         $uri = "${xres}x$yres/$infobox$filename";
281                                 } elsif (defined($xres) && $xres == -1) {
282                                         $uri = "original/$infobox$filename";
283                                 }
284
285                                 my $selected = $ref->{'selected'} ? ' checked="checked"' : '';
286
287                                 $r->print("    <p><a href=\"$uri\"><img src=\"${thumbxres}x${thumbyres}/$filename\" alt=\"\"$imgsz /></a>\n");
288                                 $r->print("      <input type=\"checkbox\" name=\"sel-" .
289                                         $ref->{'id'} . "\"$selected /></p>\n");
290                         }
291                         $r->print("      <input type=\"submit\" value=\"Select\" />\n");
292                         $r->print("    </form>\n");
293                 } else {
294                         while (my $ref = $q->fetchrow_hashref()) {
295                                 my $imgsz = "";
296                                 my $takenby = $ref->{'takenby'};
297                                 if (defined($ref->{'day'})) {
298                                          $takenby .= ", " . $ref->{'day'};
299                                 }
300
301                                 if ($takenby ne $lastupl) {
302                                         $r->print("    </p>\n\n") if ($lastupl ne "");
303                                         $lastupl = $takenby;
304                                         Sesse::pr0n::Templates::print_template($r, "submittedby", { author => $lastupl });
305                                         $r->print("    <p>\n");
306                                 }
307                                 if ($ref->{'width'} != -1 && $ref->{'height'} != -1) {
308                                         my $width = $ref->{'width'};
309                                         my $height = $ref->{'height'};
310                                                 
311                                         ($width, $height) = Sesse::pr0n::Common::scale_aspect($width, $height, $thumbxres, $thumbyres);
312                                         $imgsz = " width=\"$width\" height=\"$height\"";
313                                 }
314
315                                 my $filename = $ref->{'filename'};
316                                 my $uri = $infobox . $filename;
317                                 if (defined($xres) && defined($yres) && $xres != -1) {
318                                         $uri = "${xres}x$yres/$infobox$filename";
319                                 } elsif (defined($xres) && $xres == -1) {
320                                         $uri = "original/$infobox$filename";
321                                 }
322                                 
323                                 $r->print("      <a href=\"$uri\"><img src=\"${thumbxres}x${thumbyres}/$filename\" alt=\"\"$imgsz /></a>\n");
324                         }
325                         $r->print("    </p>\n");
326                 }
327
328                 print_nextprev($r, $event, \%settings, \%defsettings);
329                 Sesse::pr0n::Common::footer($r);
330         }
331
332         return Apache2::Const::OK;
333 }
334
335 sub eq_with_undef {
336         my ($a, $b) = @_;
337         
338         return 1 if (!defined($a) && !defined($b));
339         return 0 unless (defined($a) && defined($b));
340         return ($a eq $b);
341 }
342
343 sub print_changes {
344         my ($r, $event, $template, $settings, $defsettings, $var1, $var2, $alternatives) = @_;
345
346         my $title = Sesse::pr0n::Templates::fetch_template($r, $template);
347         chomp $title;
348         $r->print("    <p>$title:\n");
349
350         for my $a (@$alternatives) {
351                 my $text;
352                 my %newsettings = %$settings;
353
354                 if (ref $a) {
355                         my ($v1, $v2);
356                         ($text, $v1, $v2) = @$a;
357                         
358                         $newsettings{$var1} = $v1;
359                         $newsettings{$var2} = $v2;
360                 } else {
361                         $text = $a;
362
363                         # Parse the current alternative
364                         my ($v1, $v2) = split /x/, $a;
365
366                         $newsettings{$var1} = $v1;
367                         $newsettings{$var2} = $v2;
368                 }
369
370                 $r->print("      ");
371
372                 # Check if these settings are current (print only label)
373                 if (eq_with_undef($settings->{$var1}, $newsettings{$var1}) &&
374                     eq_with_undef($settings->{$var2}, $newsettings{$var2})) {
375                         $r->print($text);
376                 } else {
377                         Sesse::pr0n::Common::print_link($r, $text, "/$event/", \%newsettings, $defsettings);
378                 }
379                 $r->print("\n");
380         }
381         $r->print("    </p>\n");
382 }
383
384 sub print_thumbsize {
385         my ($r, $event, $settings, $defsettings) = @_;
386         my @alternatives = qw(80x64 120x96 160x128 240x192 320x256);
387
388         print_changes($r, $event, 'thumbsize', $settings, $defsettings,
389                       'thumbxres', 'thumbyres', \@alternatives);
390 }
391 sub print_viewres {
392         my ($r, $event, $settings, $defsettings) = @_;
393         my @alternatives = qw(320x256 512x384 640x480 800x600 1024x768 1280x960);
394         chomp (my $unlimited = Sesse::pr0n::Templates::fetch_template($r, 'viewres-unlimited'));
395         chomp (my $original = Sesse::pr0n::Templates::fetch_template($r, 'viewres-original'));
396         push @alternatives, [ $unlimited, undef, undef ];
397         push @alternatives, [ $original, -1, -1 ];
398
399         print_changes($r, $event, 'viewres', $settings, $defsettings,
400                       'xres', 'yres', \@alternatives);
401 }
402
403 sub print_pagelimit {
404         my ($r, $event, $settings, $defsettings) = @_;
405         
406         my $title = Sesse::pr0n::Templates::fetch_template($r, 'imgsperpage');
407         chomp $title;
408         $r->print("    <p>$title:\n");
409         
410         # Get choices
411         chomp (my $unlimited = Sesse::pr0n::Templates::fetch_template($r, 'imgsperpage-unlimited'));
412         my @alternatives = qw(10 50 100 500);
413         push @alternatives, $unlimited;
414         
415         for my $num (@alternatives) {
416                 my %newsettings = %$settings;
417
418                 if ($num !~ /^\d+$/) { # unlimited
419                         $newsettings{'num'} = -1;
420                 } else {
421                         $newsettings{'num'} = $num;
422                 }
423
424                 $r->print("      ");
425                 if (eq_with_undef($settings->{'num'}, $newsettings{'num'})) {
426                         $r->print($num);
427                 } else {
428                         Sesse::pr0n::Common::print_link($r, $num, "/$event/", \%newsettings, $defsettings);
429                 }
430                 $r->print("\n");
431         }
432         $r->print("    </p>\n");
433 }
434
435 sub print_infobox {
436         my ($r, $event, $settings, $defsettings) = @_;
437
438         chomp (my $title = Sesse::pr0n::Templates::fetch_template($r, 'infobox'));
439         chomp (my $on = Sesse::pr0n::Templates::fetch_template($r, 'infobox-on'));
440         chomp (my $off = Sesse::pr0n::Templates::fetch_template($r, 'infobox-off'));
441
442         $r->print("    <p>$title:\n");
443
444         my %newsettings = %$settings;
445
446         if ($settings->{'infobox'} == 1) {
447                 $r->print($on);
448         } else {
449                 $newsettings{'infobox'} = 1;
450                 Sesse::pr0n::Common::print_link($r, $on, "/$event/", \%newsettings, $defsettings);
451         }
452
453         $r->print(' ');
454
455         if ($settings->{'infobox'} == 0) {
456                 $r->print($off);
457         } else {
458                 $newsettings{'infobox'} = 0;
459                 Sesse::pr0n::Common::print_link($r, $off, "/$event/", \%newsettings, $defsettings);
460         }
461         
462         $r->print('</p>');
463 }
464
465 sub print_nextprev {
466         my ($r, $event, $settings, $defsettings) = @_;
467         my $start = $settings->{'start'};
468         my $num = $settings->{'num'};
469         my $dbh = Sesse::pr0n::Common::get_dbh();
470
471         $num = undef if (defined($num) && $num == -1);
472         return unless (defined($start) && defined($num));
473
474         # determine total number
475         my $ref = $dbh->selectrow_hashref('SELECT count(*) AS num_images FROM images WHERE event=?',
476                 undef, $event)
477                 or dberror($r, "image enumeration");
478         my $num_images = $ref->{'num_images'};
479
480         return if ($start == 1 && $start + $num >= $num_images);
481
482         my $end = $start + $num - 1;
483         if ($end > $num_images) {
484                 $end = $num_images;
485         }
486
487         $r->print("    <p>\n");
488
489         # Previous
490         if ($start > 1) {
491                 my $newstart = $start - $num;
492                 if ($newstart < 1) {
493                         $newstart = 1;
494                 }
495                 my $newend = $newstart + $num - 1;
496                 if ($newend > $num_images) {
497                         $newend = $num_images;
498                 }
499
500                 my %newsettings = %$settings;
501                 $newsettings{'start'} = $newstart;
502                 chomp (my $title = Sesse::pr0n::Templates::fetch_template($r, 'prevpage'));
503                 chomp (my $accesskey = Sesse::pr0n::Templates::fetch_template($r, 'prevaccesskey'));
504                 Sesse::pr0n::Common::print_link($r, "$title ($newstart-$newend)\n", "/$event/", \%newsettings, $defsettings, $accesskey);
505         }
506
507         # This
508         chomp (my $title = Sesse::pr0n::Templates::fetch_template($r, 'thispage'));
509         $r->print("    $title ($start-$end)\n");
510
511         # Next
512         if ($end < $num_images) {
513                 my $newstart = $start + $num;
514                 my $newend = $newstart + $num - 1;
515                 if ($newend > $num_images) {
516                         $newend = $num_images;
517                 }
518                 
519                 my %newsettings = %$settings;
520                 $newsettings{'start'} = $newstart;
521                 chomp (my $title = Sesse::pr0n::Templates::fetch_template($r, 'nextpage'));
522                 chomp (my $accesskey = Sesse::pr0n::Templates::fetch_template($r, 'nextaccesskey'));
523                 Sesse::pr0n::Common::print_link($r, "$title ($newstart-$newend)", "/$event/", \%newsettings, $defsettings, $accesskey);
524         }
525
526         $r->print("    </p>\n");
527 }
528
529 sub print_selected {
530         my ($r, $event, $settings, $defsettings) = @_;
531
532         chomp (my $title = Sesse::pr0n::Templates::fetch_template($r, 'show'));
533         chomp (my $all = Sesse::pr0n::Templates::fetch_template($r, 'show-all'));
534         chomp (my $sel = Sesse::pr0n::Templates::fetch_template($r, 'show-selected'));
535
536         $r->print("    <p>$title:\n");
537
538         my %newsettings = %$settings;
539
540         if ($settings->{'all'} == 0) {
541                 $r->print($sel);
542         } else {
543                 $newsettings{'all'} = 0;
544                 Sesse::pr0n::Common::print_link($r, $sel, "/$event/", \%newsettings, $defsettings);
545         }
546
547         $r->print(' ');
548
549         if ($settings->{'all'} == 1) {
550                 $r->print($all);
551         } else {
552                 $newsettings{'all'} = 1;
553                 Sesse::pr0n::Common::print_link($r, $all, "/$event/", \%newsettings, $defsettings);
554         }
555         
556         $r->print('</p>');
557 }
558
559 sub print_fullscreen {
560         my ($r, $event, $settings, $defsettings) = @_;
561
562         chomp (my $title = Sesse::pr0n::Templates::fetch_template($r, 'fullscreen'));
563
564         my %newsettings = %$settings;
565         $newsettings{'fullscreen'} = 1;
566
567         $r->print("    <p>");
568         Sesse::pr0n::Common::print_link($r, $title, "/$event/", \%newsettings, $defsettings);
569         $r->print("</p>\n");
570 }
571         
572 1;
573
574