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;