X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=ex%2FBeerDB.pm;h=778185a0b7e657336dabe63b8c55a2e7cea3eefa;hb=73df58543e31b805fe123e4e818ab863eaac07de;hp=59c7bb210f35a1c8fdf9f53792139872e02262db;hpb=f80621477b1dfe74dc1a18d5453e3f9225022a78;p=maypole.git

diff --git a/ex/BeerDB.pm b/ex/BeerDB.pm
index 59c7bb2..778185a 100644
--- a/ex/BeerDB.pm
+++ b/ex/BeerDB.pm
@@ -1,29 +1,57 @@
 package BeerDB;
-use base 'Maypole::Application';
+use Maypole::Application;
 use Class::DBI::Loader::Relationship;
 
-# This is the sample application. Change this to the path to your
+sub debug { $ENV{BEERDB_DEBUG} }
+# This is the sample application.  Change this to the path to your
 # database. (or use mysql or something)
-BeerDB->setup("dbi:SQLite:t/beerdb.db");
-#BeerDB->setup("dbi:mysql:beerdb");
+use constant DBI_DRIVER => 'SQLite';
+use constant DATASOURCE => $ENV{BEERDB_DATASOURCE} || 't/beerdb.db';
 
-# Change this to the root of the web space.
-BeerDB->config->{uri_base} = "http://localhost/beerdb/";
-#BeerDB->config->{uri_base} = "http://neo.trinity-house.org.uk/beerdb/";
+BEGIN {
+    my $dbi_driver = DBI_DRIVER;
+    if ($dbi_driver =~ /^SQLite/) {
+        die sprintf "SQLite datasource '%s' not found, correct the path or "
+            . "recreate the database by running Makefile.PL", DATASOURCE
+            unless -e DATASOURCE;
+        eval "require DBD::SQLite";
+        if ($@) {
+            eval "require DBD::SQLite2" and $dbi_driver = 'SQLite2';
+        }
+    }
+    BeerDB->setup(join ':', "dbi", $dbi_driver, DATASOURCE);
+}
 
-BeerDB->config->{rows_per_page} = 10;
+# Give it a name.
+BeerDB->config->application_name('The Beer Database');
+
+# Change this to the root of the web site for your maypole application.
+BeerDB->config->uri_base( $ENV{BEERDB_BASE} || "http://localhost/beerdb/" );
+
+# Change this to the htdoc root for your maypole application.
+BeerDB->config->template_root( $ENV{BEERDB_TEMPLATE_ROOT} ) if $ENV{BEERDB_TEMPLATE_ROOT};
+
+# Specify the rows per page in search results, lists, etc : 10 is a nice round number
+BeerDB->config->rows_per_page(10);
 
 # Handpumps should not show up.
-BeerDB->config->{display_tables} = [qw[beer brewery pub style]];
+BeerDB->config->display_tables([qw[beer brewery pub style]]);
 BeerDB::Brewery->untaint_columns( printable => [qw/name notes url/] );
 BeerDB::Style->untaint_columns( printable => [qw/name notes/] );
 BeerDB::Beer->untaint_columns(
-    printable => [qw/abv name price notes/],
+    printable => [qw/abv name price notes url/],
     integer => [qw/style brewery score/],
     date =>[ qw/date/],
 );
+BeerDB::Pub->untaint_columns(printable => [qw/name notes url/]);
+
 BeerDB->config->{loader}->relationship($_) for (
     "a brewery produces beers",
     "a style defines beers",
     "a pub has beers on handpumps");
+
+# For testing classmetadata
+sub BeerDB::Beer::classdata :Exported {};
+sub BeerDB::Beer::list_columns  { return qw/score name price style brewery url/};
+
 1;