SL/DB.pm

   1 package SL::DB;
   2
   3 use strict;
   4
   5 use Carp;
   6 use Data::Dumper;
   7 use English qw(-no_match_vars);
   8 use Rose::DB;
   9
  10 use base qw(Rose::DB);
  11
  12 __PACKAGE__->use_private_registry;
  13
  14 my (%_db_registered, %_initial_sql_executed);
  15
  16 sub dbi_connect {
  17   shift;
  18
  19   return SL::DBConnect->connect(@_);
  20 }
  21
  22 sub create {
  23   my $domain = shift || SL::DB->default_domain;
  24   my $type   = shift || SL::DB->default_type;
  25
  26   my ($domain, $type) = _register_db($domain, $type);
  27
  28   my $db = __PACKAGE__->new_or_cached(domain => $domain, type => $type);
  29
  30   _execute_initial_sql($db);
  31
  32   return $db;
  33 }
  34
  35 my %_dateformats = ( 'yy-mm-dd'   => 'ISO',
  36                      'yyyy-mm-dd' => 'ISO',
  37                      'mm/dd/yy'   => 'SQL, US',
  38                      'mm-dd-yy'   => 'POSTGRES, US',
  39                      'dd/mm/yy'   => 'SQL, EUROPEAN',
  40                      'dd-mm-yy'   => 'POSTGRES, EUROPEAN',
  41                      'dd.mm.yy'   => 'GERMAN'
  42                    );
  43
  44 sub _register_db {
  45   my $domain = shift;
  46   my $type   = shift;
  47
  48   my %connect_settings;
  49   my $initial_sql;
  50
  51   if (!%::myconfig) {
  52     $type = 'LXOFFICE_EMPTY';
  53     %connect_settings = ( driver => 'Pg' );
  54
  55   } elsif ($type eq 'LXOFFICE_AUTH') {
  56     %connect_settings = ( driver          => $::myconfig{dbdriver} || 'Pg',
  57                           database        => $::auth->{DB_config}->{db},
  58                           host            => $::auth->{DB_config}->{host} || 'localhost',
  59                           port            => $::auth->{DB_config}->{port} || 5432,
  60                           username        => $::auth->{DB_config}->{user},
  61                           password        => $::auth->{DB_config}->{password},
  62                           connect_options => { pg_enable_utf8 => $::locale && $::locale->is_utf8,
  63                                              });
  64   } else {
  65     my $european_dates = 0;
  66     if ($::myconfig{dateformat}) {
  67       $european_dates = 1 if $_dateformats{ $::myconfig{dateformat} } =~ m/european/i;
  68     }
  69
  70     %connect_settings = ( driver          => $::myconfig{dbdriver} || 'Pg',
  71                           database        => $::myconfig{dbname},
  72                           host            => $::myconfig{dbhost} || 'localhost',
  73                           port            => $::myconfig{dbport} || 5432,
  74                           username        => $::myconfig{dbuser},
  75                           password        => $::myconfig{dbpasswd},
  76                           connect_options => { pg_enable_utf8 => $::locale && $::locale->is_utf8,
  77                                              },
  78                           european_dates  => $european_dates);
  79   }
  80
  81   my %flattened_settings = _flatten_settings(%connect_settings);
  82
  83   $domain = 'LXOFFICE' if $type =~ m/^LXOFFICE/;
  84   $type  .= join($SUBSCRIPT_SEPARATOR, map { ($_, $flattened_settings{$_}) } sort keys %flattened_settings);
  85   my $idx = "${domain}::${type}";
  86
  87   if (!$_db_registered{$idx}) {
  88     $_db_registered{$idx} = 1;
  89
  90     __PACKAGE__->register_db(domain => $domain,
  91                              type   => $type,
  92                              %connect_settings,
  93                             );
  94   }
  95
  96   return ($domain, $type);
  97 }
  98
  99 sub _execute_initial_sql {
 100   my ($db) = @_;
 101
 102   return if $_initial_sql_executed{$db} || !%::myconfig || !$::myconfig{dateformat};
 103
 104   $_initial_sql_executed{$db} = 1;
 105
 106   # Don't rely on dboptions being set properly. Chose them from
 107   # dateformat instead.
 108   my $pg_dateformat = $_dateformats{ $::myconfig{dateformat} };
 109   $db->dbh->do("set DateStyle to '${pg_dateformat}'") if $pg_dateformat;
 110 }
 111
 112 sub _flatten_settings {
 113   my %settings  = @_;
 114   my %flattened = ();
 115
 116   while (my ($key, $value) = each %settings) {
 117     if ('HASH' eq ref $value) {
 118       %flattened = ( %flattened, _flatten_settings(%{ $value }) );
 119     } else {
 120       $flattened{$key} = $value;
 121     }
 122   }
 123
 124   return %flattened;
 125 }
 126
 127 1;