chiark / gitweb /
WIP route value calculation - compiles, although doesn't run the LP
[ypp-sc-tools.db-test.git] / yarrg / CommodsWeb.pm
1 # This is part of the YARRG website.  YARRG is a tool and website
2 # for assisting players of Yohoho Puzzle Pirates.
3 #
4 # Copyright (C) 2009 Ian Jackson <ijackson@chiark.greenend.org.uk>
5 # Copyright (C) 2009 Clare Boothby
6 #
7 #  YARRG's client code etc. is covered by the ordinary GNU GPL (v3 or later).
8 #  The YARRG website is covered by the GNU Affero GPL v3 or later, which
9 #   basically means that every installation of the website will let you
10 #   download the source.
11 #
12 # This program 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 # This program is distributed in the hope that it will be useful,
18 # but WITHOUT ANY WARRANTY; without even the implied warranty of
19 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
20 # GNU Affero General Public License for more details.
21 #
22 # You should have received a copy of the GNU Affero General Public License
23 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
24 #
25 # Yohoho and Puzzle Pirates are probably trademarks of Three Rings and
26 # are used without permission.  This program is not endorsed or
27 # sponsored by Three Rings.
28
29
30 # This Perl module is used by the Mason scripts in yarrg/web/.
31 # We look for a symlink DATA to the actual data to use, so that
32 # the data uploader and website displayer can use different code.
33
34 package CommodsWeb;
35
36 use strict;
37 use warnings;
38
39 use DBI;
40 use POSIX;
41 use JSON;
42
43 use Commods;
44 use CommodsDatabase;
45
46 BEGIN {
47     use Exporter ();
48     our ($VERSION, @ISA, @EXPORT, @EXPORT_OK, %EXPORT_TAGS);
49     $VERSION     = 1.00;
50     @ISA         = qw(Exporter);
51     @EXPORT      = qw(&dbw_connect &ocean_list &sourcebasedir
52                       &to_json_shim &to_json_protecttags
53                       &set_ctype_utf8
54                       &expected_error &dbw_lookup_string
55                       &prettyprint_age &meta_prettyprint_age);
56     %EXPORT_TAGS = ( );
57
58     @EXPORT_OK   = qw();
59 }
60
61 sub dotperllibdir () {
62     my $dir;
63     
64     for my $dir (@INC) {
65         if ($dir =~ m/\.perl-lib$/) {
66             return $dir;
67         }
68     }
69     die "no appropriate dotperllib dir in @INC";
70 }
71
72 sub sourcebasedir () {
73     return dotperllibdir().'/..';
74 }
75
76 sub datadir () {
77     my $edir= $ENV{'YARRG_DATA_DIR'};
78     return $edir if defined $edir;
79     my $dir= dotperllibdir();
80     if (stat "$dir/DATA") {
81         return "$dir/DATA";
82     } elsif ($!==&ENOENT) {
83         return "$dir";
84     } else {
85         die "stat $dir/DATA $!";
86     }
87     return '.';
88 }
89
90 my @ocean_list;
91
92 sub ocean_list () {
93     my $datadir= datadir();
94     if (!@ocean_list) {
95         my $fn= "$datadir/source-info.txt";
96         my $f= new IO::File $fn or die "$fn $!";
97         my @r;
98         while (<$f>) {
99             next unless m/^ocean\s+(\S.*\S)\s*$/;
100             push @r, $1;
101         }
102         $f->error and die $!;
103         close $fn;
104         @ocean_list= @r;
105     }
106     return @ocean_list;
107 }
108
109 sub dbw_connect ($) {
110     my ($ocean) = @_;
111     die "unknown ocean $ocean ?"
112         unless grep { $_ eq $ocean } ocean_list();
113     return dbr_connect(datadir(), $ocean);
114 }
115
116 sub to_json_shim ($) {
117     my ($obj) = @_;
118     # In JSON.pm 2.x, jsonToObj prints a warning to stderr which
119     # our callers don't like at all.
120     if ($JSON::VERSION >= 2.0) {
121         return to_json($obj);
122     } else {
123         return objToJson($obj);
124     }
125 }
126
127 sub to_json_protecttags ($) {
128     my ($v) = @_;
129     my $j= to_json_shim($v);
130     $j =~ s,/,\\/,g;
131     return $j;
132 }
133
134 sub meta_prettyprint_age ($$$) {
135     my ($age,$floor,$plus) = @_;
136     return <<END;
137         $age < 60 ?             'less than a minute'                    :
138         $age < 60*2 ?           '1 minute'                              :
139         $age < 3600*2 ?         $floor ($age/60) $plus' minutes'        :
140         $age < 86400*2 ?        $floor ($age/3600) $plus ' hours'       :
141                                 $floor ($age/86400) $plus ' days';
142 END
143 };
144
145 BEGIN { eval '
146   sub prettyprint_age ($) {
147                 my ($age) = @_;
148                 '.meta_prettyprint_age('$age','floor','.').'
149   };
150   1;
151 ' or die "$@";
152 }
153
154
155 sub dbw_lookup_string ($$$$$$$$) { # => ( $emsg, @dbresults )
156     my ($each,
157         $sth, $stmt_nqs, $abbrev_initials, $maxambig,
158         $em_nomatch, $em_manyambig, $emf_ambiguous) = @_;
159     
160     $each =~ s/^\s*//;  $each =~ s/\s*$//;  $each =~ s/\s+/ /g;
161     my %m;
162     my $results;
163     my @pats= ("$each", "$each \%", "$each\%", "\%$each\%");
164     if ($abbrev_initials) {
165         push @pats, join ' ', map { "$_%" } split //, $each;
166     }
167     foreach my $pat (@pats) {
168         $sth->execute(($pat) x $stmt_nqs);
169         $results= $sth->fetchall_arrayref();
170         last if @$results==1;
171         $m{ $_->[0] }=1 for @$results;
172         $results= undef;
173     }
174     if (!$results) {
175         if (!%m) {
176             return $em_nomatch;
177         } elsif (keys(%m) > $maxambig) {
178             return $em_manyambig;
179         } else {
180             return $emf_ambiguous->($each, join(', ', sort keys %m));
181         }
182     }
183     return (undef, @{ $results->[0] });
184 }
185
186 sub expected_error ($) {
187     my $r= { Emsg => $_[0] };
188     bless $r, 'CommodsWeb::ExpectedError';
189     die $r;
190 }
191
192 package CommodsWeb::ExpectedError;
193
194 sub emsg ($) {
195     my ($self) = @_;
196     return $self->{Emsg};
197 }
198
199 1;