more debugging
[freeside.git] / 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_schema upgrade_config 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 upgrade_config
37
38 =cut
39
40 #config upgrades
41 sub upgrade_config {
42   my %opt = @_;
43
44   my $conf = new FS::Conf;
45
46   $conf->touch('payment_receipt')
47     if $conf->exists('payment_receipt_email')
48     || $conf->config('payment_receipt_msgnum');
49
50 }
51
52 =item upgrade
53
54 =cut
55
56 sub upgrade {
57   my %opt = @_;
58
59   my $data = upgrade_data(%opt);
60
61   my $oldAutoCommit = $FS::UID::AutoCommit;
62   local $FS::UID::AutoCommit = 0;
63   local $FS::UID::AutoCommit = 0;
64
65   foreach my $table ( keys %$data ) {
66
67     my $class = "FS::$table";
68     eval "use $class;";
69     die $@ if $@;
70
71     if ( $class->can('_upgrade_data') ) {
72       warn "Upgrading $table...\n";
73
74       my $start = time;
75
76       $class->_upgrade_data(%opt);
77
78       if ( $oldAutoCommit ) {
79         warn "  committing\n";
80         dbh->commit or die dbh->errstr;
81       }
82       
83       #warn "\e[1K\rUpgrading $table... done in ". (time-$start). " seconds\n";
84       warn "  done in ". (time-$start). " seconds\n";
85
86     } else {
87       warn "WARNING: asked for upgrade of $table,".
88            " but FS::$table has no _upgrade_data method\n";
89     }
90
91 #    my @records = @{ $data->{$table} };
92 #
93 #    foreach my $record ( @records ) {
94 #      my $args = delete($record->{'_upgrade_args'}) || [];
95 #      my $object = $class->new( $record );
96 #      my $error = $object->insert( @$args );
97 #      die "error inserting record into $table: $error\n"
98 #        if $error;
99 #    }
100
101   }
102
103 }
104
105 =item upgrade_data
106
107 =cut
108
109 sub upgrade_data {
110   my %opt = @_;
111
112   tie my %hash, 'Tie::IxHash', 
113
114     #cust_main (remove paycvv from history)
115     'cust_main' => [],
116
117     #msgcat
118     'msgcat' => [],
119
120     #reason type and reasons
121     'reason_type'     => [],
122     'cust_pkg_reason' => [],
123
124     #need part_pkg before cust_credit...
125     'part_pkg' => [],
126
127     #customer credits
128     'cust_credit' => [],
129
130     #duplicate history records
131     'h_cust_svc'  => [],
132
133     #populate cust_pay.otaker
134     'cust_pay'    => [],
135
136     #populate part_pkg_taxclass for starters
137     'part_pkg_taxclass' => [],
138
139     #remove bad pending records
140     'cust_pay_pending' => [],
141
142     #replace invnum and pkgnum with billpkgnum
143     'cust_bill_pkg_detail' => [],
144
145     #usage_classes if we have none
146     'usage_class' => [],
147
148     #phone_type if we have none
149     'phone_type' => [],
150
151     #fixup access rights
152     'access_right' => [],
153
154     #change recur_flat and enable_prorate
155     'part_pkg_option' => [],
156
157     #add weights to pkg_category
158     'pkg_category' => [],
159
160     #cdrbatch fixes
161     'cdr' => [],
162
163     #otaker->usernum
164     'cust_attachment' => [],
165     #'cust_credit' => [],
166     #'cust_main' => [],
167     'cust_main_note' => [],
168     #'cust_pay' => [],
169     'cust_pay_void' => [],
170     'cust_pkg' => [],
171     #'cust_pkg_reason' => [],
172     'cust_pkg_discount' => [],
173     'cust_refund' => [],
174     'banned_pay' => [],
175
176     #default namespace
177     'payment_gateway' => [],
178
179     #migrate to templates
180     'msg_template' => [],
181
182     #return unprovisioned numbers to availability
183     'phone_avail' => [],
184
185     #insert scripcondition
186     'TicketSystem' => [],
187
188   ;
189
190   \%hash;
191
192 }
193
194 =item upgrade_schema
195
196 =cut
197
198 sub upgrade_schema {
199   my %opt = @_;
200
201   my $data = upgrade_schema_data(%opt);
202
203   my $oldAutoCommit = $FS::UID::AutoCommit;
204   local $FS::UID::AutoCommit = 0;
205   local $FS::UID::AutoCommit = 0;
206
207   foreach my $table ( keys %$data ) {
208
209     my $class = "FS::$table";
210     eval "use $class;";
211     die $@ if $@;
212
213     if ( $class->can('_upgrade_schema') ) {
214       warn "Upgrading $table schema...\n";
215
216       my $start = time;
217
218       $class->_upgrade_schema(%opt);
219
220       if ( $oldAutoCommit ) {
221         warn "  committing\n";
222         dbh->commit or die dbh->errstr;
223       }
224       
225       #warn "\e[1K\rUpgrading $table... done in ". (time-$start). " seconds\n";
226       warn "  done in ". (time-$start). " seconds\n";
227
228     } else {
229       warn "WARNING: asked for schema upgrade of $table,".
230            " but FS::$table has no _upgrade_schema method\n";
231     }
232
233   }
234
235 }
236
237 =item upgrade_schema_data
238
239 =cut
240
241 sub upgrade_schema_data {
242   my %opt = @_;
243
244   tie my %hash, 'Tie::IxHash', 
245
246     #fix classnum character(1)
247     'cust_bill_pkg_detail' => [],
248
249   ;
250
251   \%hash;
252
253 }
254
255 sub upgrade_sqlradius {
256   #my %opt = @_;
257
258   my $conf = new FS::Conf;
259
260   my @part_export = FS::part_export::sqlradius->all_sqlradius_withaccounting();
261
262   foreach my $part_export ( @part_export ) {
263
264     my $errmsg = 'Error adding FreesideStatus to '.
265                  $part_export->option('datasrc'). ': ';
266
267     my $dbh = DBI->connect(
268       ( map $part_export->option($_), qw ( datasrc username password ) ),
269       { PrintError => 0, PrintWarn => 0 }
270     ) or do {
271       warn $errmsg.$DBI::errstr;
272       next;
273     };
274
275     my $str2time = str2time_sql( $dbh->{Driver}->{Name} );
276     my $group = "UserName";
277     $group .= ",Realm"
278       if ref($part_export) =~ /withdomain/
279       || $dbh->{Driver}->{Name} =~ /^Pg/; #hmm
280
281     my $sth_alter = $dbh->prepare(
282       "ALTER TABLE radacct ADD COLUMN FreesideStatus varchar(32) NULL"
283     );
284     if ( $sth_alter ) {
285       if ( $sth_alter->execute ) {
286         my $sth_update = $dbh->prepare(
287          "UPDATE radacct SET FreesideStatus = 'done' WHERE FreesideStatus IS NULL"
288         ) or die $errmsg.$dbh->errstr;
289         $sth_update->execute or die $errmsg.$sth_update->errstr;
290       } else {
291         my $error = $sth_alter->errstr;
292         warn $errmsg.$error
293           unless $error =~ /Duplicate column name/i  #mysql
294               || $error =~ /already exists/i;        #Pg
295 ;
296       }
297     } else {
298       my $error = $dbh->errstr;
299       warn $errmsg.$error; #unless $error =~ /exists/i;
300     }
301
302     my $sth_index = $dbh->prepare(
303       "CREATE INDEX FreesideStatus ON radacct ( FreesideStatus )"
304     );
305     if ( $sth_index ) {
306       unless ( $sth_index->execute ) {
307         my $error = $sth_index->errstr;
308         warn $errmsg.$error
309           unless $error =~ /Duplicate key name/i #mysql
310               || $error =~ /already exists/i;    #Pg
311       }
312     } else {
313       my $error = $dbh->errstr;
314       warn $errmsg.$error. ' (preparing statement)';#unless $error =~ /exists/i;
315     }
316
317     my $times = ($dbh->{Driver}->{Name} =~ /^mysql/)
318       ? ' AcctStartTime != 0 AND AcctStopTime != 0 '
319       : ' AcctStartTime IS NOT NULL AND AcctStopTime IS NOT NULL ';
320
321     my $sth = $dbh->prepare("SELECT UserName,
322                                     Realm,
323                                     $str2time max(AcctStartTime)),
324                                     $str2time max(AcctStopTime))
325                               FROM radacct
326                               WHERE FreesideStatus = 'done'
327                                 AND $times
328                               GROUP BY $group
329                             ")
330       or die $errmsg.$dbh->errstr;
331     $sth->execute() or die $errmsg.$sth->errstr;
332   
333     while (my $row = $sth->fetchrow_arrayref ) {
334       my ($username, $realm, $start, $stop) = @$row;
335   
336       $username = lc($username) unless $conf->exists('username-uppercase');
337
338       my $exportnum = $part_export->exportnum;
339       my $extra_sql = " AND exportnum = $exportnum ".
340                       " AND exportsvcnum IS NOT NULL ";
341
342       if ( ref($part_export) =~ /withdomain/ ) {
343         $extra_sql = " AND '$realm' = ( SELECT domain FROM svc_domain
344                          WHERE svc_domain.svcnum = svc_acct.domsvc ) ";
345       }
346   
347       my $svc_acct = qsearchs({
348         'select'    => 'svc_acct.*',
349         'table'     => 'svc_acct',
350         'addl_from' => 'LEFT JOIN cust_svc   USING ( svcnum )'.
351                        'LEFT JOIN export_svc USING ( svcpart )',
352         'hashref'   => { 'username' => $username },
353         'extra_sql' => $extra_sql,
354       });
355
356       if ($svc_acct) {
357         $svc_acct->last_login($start)
358           if $start && (!$svc_acct->last_login || $start > $svc_acct->last_login);
359         $svc_acct->last_logout($stop)
360           if $stop && (!$svc_acct->last_logout || $stop > $svc_acct->last_logout);
361       }
362     }
363   }
364
365 }
366
367 =back
368
369 =head1 BUGS
370
371 Sure.
372
373 =head1 SEE ALSO
374
375 =cut
376
377 1;
378