FS/FS/Upgrade.pm

   1 package FS::Upgrade;
   2
   3 use strict;
   4 use vars qw( @ISA @EXPORT_OK $DEBUG );
   5 use Exporter;
   6 use Tie::IxHash;
   7 use FS::UID qw( dbh driver_name );
   8 use FS::Conf;
   9 use FS::Record qw(qsearchs str2time_sql);
  10
  11 use FS::svc_domain;
  12 $FS::svc_domain::whois_hack = 1;
  13
  14 @ISA = qw( Exporter );
  15 @EXPORT_OK = qw( upgrade upgrade_sqlradius );
  16
  17 $DEBUG = 1;
  18
  19 =head1 NAME
  20
  21 FS::Upgrade - Database upgrade routines
  22
  23 =head1 SYNOPSIS
  24
  25   use FS::Upgrade;
  26
  27 =head1 DESCRIPTION
  28
  29 Currently this module simply provides a place to store common subroutines for
  30 database upgrades.
  31
  32 =head1 SUBROUTINES
  33
  34 =over 4
  35
  36 =item
  37
  38 =cut
  39
  40 sub upgrade {
  41   my %opt = @_;
  42
  43   my $oldAutoCommit = $FS::UID::AutoCommit;
  44   local $FS::UID::AutoCommit = 0;
  45   $FS::UID::AutoCommit = 0;
  46
  47   my $data = upgrade_data(%opt);
  48
  49   foreach my $table ( keys %$data ) {
  50
  51     my $class = "FS::$table";
  52     eval "use $class;";
  53     die $@ if $@;
  54
  55     if ( $class->can('_upgrade_data') ) {
  56       warn "Upgrading $table...\n";
  57       $class->_upgrade_data(%opt);
  58     } else {
  59       warn "WARNING: asked for upgrade of $table,".
  60            " but FS::$table has no _upgrade_data method\n";
  61     }
  62
  63 #    my @records = @{ $data->{$table} };
  64 #
  65 #    foreach my $record ( @records ) {
  66 #      my $args = delete($record->{'_upgrade_args'}) || [];
  67 #      my $object = $class->new( $record );
  68 #      my $error = $object->insert( @$args );
  69 #      die "error inserting record into $table: $error\n"
  70 #        if $error;
  71 #    }
  72
  73   }
  74
  75   if ( $oldAutoCommit ) {
  76     dbh->commit or die dbh->errstr;
  77   }
  78
  79 }
  80
  81
  82 sub upgrade_data {
  83   my %opt = @_;
  84
  85   tie my %hash, 'Tie::IxHash',
  86
  87     #reason type and reasons
  88     'reason_type'     => [],
  89     'reason'          => [],
  90     'cust_pkg_reason' => [],
  91
  92     #need part_pkg before cust_credit...
  93     'part_pkg' => [],
  94
  95     #customer credits
  96     'cust_credit' => [],
  97
  98     #duplicate history records
  99     'h_cust_svc'  => [],
 100
 101     #populate cust_pay.otaker
 102     'cust_pay'    => [],
 103
 104     #populate part_pkg_taxclass for starters
 105     'part_pkg_taxclass' => [],
 106
 107     #remove bad pending records
 108     'cust_pay_pending' => [],
 109
 110   ;
 111
 112   \%hash;
 113
 114 }
 115
 116 sub upgrade_sqlradius {
 117   #my %opt = @_;
 118
 119   my $conf = new FS::Conf;
 120
 121   my @part_export = FS::part_export::sqlradius->all_sqlradius_withaccounting();
 122
 123   foreach my $part_export ( @part_export ) {
 124
 125     my $errmsg = 'Error adding FreesideStatus to '.
 126                  $part_export->option('datasrc'). ': ';
 127
 128     my $dbh = DBI->connect(
 129       ( map $part_export->option($_), qw ( datasrc username password ) ),
 130       { PrintError => 0, PrintWarn => 0 }
 131     ) or do {
 132       warn $errmsg.$DBI::errstr;
 133       next;
 134     };
 135
 136     my $str2time = str2time_sql( $dbh->{Driver}->{Name} );
 137     my $group = "UserName";
 138     $group .= ",Realm"
 139       if ( ref($part_export) =~ /withdomain/ );
 140
 141     my $sth_alter = $dbh->prepare(
 142       "ALTER TABLE radacct ADD COLUMN FreesideStatus varchar(32) NULL"
 143     );
 144     if ( $sth_alter ) {
 145       if ( $sth_alter->execute ) {
 146         my $sth_update = $dbh->prepare(
 147          "UPDATE radacct SET FreesideStatus = 'done' WHERE FreesideStatus IS NULL"
 148         ) or die $errmsg.$dbh->errstr;
 149         $sth_update->execute or die $errmsg.$sth_update->errstr;
 150       } else {
 151         my $error = $sth_alter->errstr;
 152         warn $errmsg.$error unless $error =~ /Duplicate column name/i;
 153       }
 154     } else {
 155       my $error = $dbh->errstr;
 156       warn $errmsg.$error; #unless $error =~ /exists/i;
 157     }
 158
 159     my $sth_index = $dbh->prepare(
 160       "CREATE INDEX FreesideStatus ON radacct ( FreesideStatus )"
 161     );
 162     if ( $sth_index ) {
 163       unless ( $sth_index->execute ) {
 164         my $error = $sth_index->errstr;
 165         warn $errmsg.$error unless $error =~ /Duplicate key name/i;
 166       }
 167     } else {
 168       my $error = $dbh->errstr;
 169       warn $errmsg.$error; #unless $error =~ /exists/i;
 170     }
 171
 172     my $sth = $dbh->prepare("SELECT UserName,
 173                                     Realm,
 174                                     $str2time max(AcctStartTime)),
 175                                     $str2time max(AcctStopTime))
 176                               FROM radacct
 177                               WHERE FreesideStatus = 'done'
 178                                 AND AcctStartTime != 0
 179                                 AND AcctStopTime  != 0
 180                               GROUP BY $group
 181                             ")
 182       or die $errmsg.$dbh->errstr;
 183     $sth->execute() or die $errmsg.$sth->errstr;
 184
 185     while (my $row = $sth->fetchrow_arrayref ) {
 186       my ($username, $realm, $start, $stop) = @$row;
 187
 188       $username = lc($username) unless $conf->exists('username-uppercase');
 189
 190       my $exportnum = $part_export->exportnum;
 191       my $extra_sql = " AND exportnum = $exportnum ".
 192                       " AND exportsvcnum IS NOT NULL ";
 193
 194       if ( ref($part_export) =~ /withdomain/ ) {
 195         $extra_sql = " AND '$realm' = ( SELECT domain FROM svc_domain
 196                          WHERE svc_domain.svcnum = svc_acct.domsvc ) ";
 197       }
 198
 199       my $svc_acct = qsearchs({
 200         'select'    => 'svc_acct.*',
 201         'table'     => 'svc_acct',
 202         'addl_from' => 'LEFT JOIN cust_svc   USING ( svcnum )'.
 203                        'LEFT JOIN export_svc USING ( svcpart )',
 204         'hashref'   => { 'username' => $username },
 205         'extra_sql' => $extra_sql,
 206       });
 207
 208       if ($svc_acct) {
 209         $svc_acct->last_login($start)
 210           if $start && (!$svc_acct->last_login || $start > $svc_acct->last_login);
 211         $svc_acct->last_logout($stop)
 212           if $stop && (!$svc_acct->last_logout || $stop > $svc_acct->last_logout);
 213       }
 214     }
 215   }
 216
 217 }
 218
 219 =back
 220
 221 =head1 BUGS
 222
 223 Sure.
 224
 225 =head1 SEE ALSO
 226
 227 =cut
 228
 229 1;
 230