add prepaid support which sets RADIUS Expiration attribute, update customer view...
[freeside.git] / FS / bin / freeside-daily
index 99d95d5..603da12 100755 (executable)
@@ -5,31 +5,78 @@ use Fcntl qw(:flock);
 use Date::Parse;
 use Getopt::Std;
 use FS::UID qw(adminsuidsetup driver_name dbh datasrc);
-use FS::Record qw(qsearch qsearchs);
+use FS::Record qw(qsearch qsearchs dbdef);
 use FS::Conf;
 use FS::cust_main;
 
 &untaint_argv; #what it sounds like  (eww)
-use vars qw($opt_d $opt_v $opt_p $opt_s $opt_y);
-getopts("p:d:vsy:");
+use vars qw($opt_d $opt_v $opt_p $opt_a $opt_s $opt_y);
+getopts("p:a:d:vsy:");
 my $user = shift or die &usage;
 
 adminsuidsetup $user;
 
 $FS::cust_main::DEBUG = 1 if $opt_v;
 
-my %search;
-$search{'payby'} = $opt_p if $opt_p;
-
-my @cust_main = @ARGV
-  ? map { qsearchs('cust_main', { custnum => $_, %search } ) } @ARGV
-  : qsearch('cust_main', \%search )
-;
+my %search = ();
+$search{'payby'}    = $opt_p if $opt_p;
+$search{'agentnum'} = $opt_a if $opt_a;
 
 #we're at now now (and later).
 my($time)= $opt_d ? str2time($opt_d) : $^T;
 $time += $opt_y * 86400 if $opt_y;
 
+# select * from cust_main where
+my $where_pkg = <<"END";
+  0 < ( select count(*) from cust_pkg
+          where cust_main.custnum = cust_pkg.custnum
+            and ( cancel is null or cancel = 0 )
+            and (    setup is null or setup =  0
+                  or bill  is null or bill  <= $time 
+                  or ( expire is not null and expire <= $^T )
+                )
+      )
+END
+
+# or
+my $where_bill_event = <<"END";
+  0 < ( select count(*) from cust_bill
+          where cust_main.custnum = cust_bill.custnum
+            and 0 < charged
+                    - coalesce(
+                                ( select sum(amount) from cust_bill_pay
+                                    where cust_bill.invnum = cust_bill_pay.invnum )
+                                ,0
+                              )
+                    - coalesce(
+                                ( select sum(amount) from cust_credit_bill
+                                    where cust_bill.invnum = cust_credit_bill.invnum )
+                                ,0
+                              )
+            and 0 < ( select count(*) from part_bill_event
+                        where payby = cust_main.payby
+                          and ( disabled is null or disabled = '' )
+                          and seconds <= $time - cust_bill._date
+                          and 0 = ( select count(*) from cust_bill_event
+                                     where cust_bill.invnum = cust_bill_event.invnum
+                                       and part_bill_event.eventpart = cust_bill_event.eventpart
+                                       and status = 'done'
+                                  )
+
+                    )
+      )
+END
+
+my $extra_sql = ( scalar(%search) ? ' AND ' : ' WHERE ' ). "( $where_pkg OR $where_bill_event )";
+
+my @cust_main;
+if ( @ARGV ) {
+  @cust_main = map { qsearchs('cust_main', { custnum => $_, %search } ) } @ARGV
+} else {
+  @cust_main = qsearch('cust_main', \%search, '', $extra_sql );
+}
+;
+
 my($cust_main,%saw);
 foreach $cust_main ( @cust_main ) {
 
@@ -42,6 +89,19 @@ foreach $cust_main ( @cust_main ) {
          $cust_main->custnum. ": $error"
       if $error;
   }
+  # $^T not $time because -d is for pre-printing invoices
+  foreach my $cust_pkg (
+    grep { $_->part_pkg->is_prepaid
+           && $_->bill && $_->bill < $^T && ! $_->susp
+         }
+         $cust_main->ncancelled_pkgs
+  ) {
+    my $error = $cust_pkg->suspend;
+    warn "Error suspending package ". $cust_pkg->pkgnum.
+         " for custnum ". $cust_main->custnum.
+         ": $error"
+      if $error;
+  }
 
   my $error = $cust_main->bill( 'time'    => $time,
                                 'resetup' => $opt_s, );
@@ -57,8 +117,8 @@ foreach $cust_main ( @cust_main ) {
 
 if ( driver_name eq 'Pg' ) {
   dbh->{AutoCommit} = 1; #so we can vacuum
-  foreach my $statement ( 'vacuum', 'vacuum analyze' ) {
-    my $sth = dbh->prepare($statement) or die dbh->errstr;
+  foreach my $table ( dbdef->tables ) {
+    my $sth = dbh->prepare("VACUUM ANALYZE $table") or die dbh->errstr;
     $sth->execute or die $sth->errstr;
   }
 }
@@ -81,9 +141,11 @@ if ( $dest ) {
                    output    => "/var/tmp/$database.gpg",
                    recipient => $conf->config('dump-pgpid'),
                  );
+    chmod 0600, '/var/tmp/$database.gpg';
     scp("/var/tmp/$database.gpg", $dest);
     unlink "/var/tmp/$database.gpg" or die $!;
   } else {
+    chmod 0600, '/var/tmp/$database.sql';
     scp("/var/tmp/$database.sql", $dest);
   }
   unlink "/var/tmp/$database.sql" or die $!;
@@ -110,7 +172,7 @@ freeside-daily - Run daily billing and invoice collection events.
 
 =head1 SYNOPSIS
 
-  freeside-daily [ -d 'date' ] [ -y days ] [ -p 'payby' ] [ -s ] [ -v ] user [ custnum custnum ... ]
+  freeside-daily [ -d 'date' ] [ -y days ] [ -p 'payby' ] [ -a agentnum ] [ -s ] [ -v ] user [ custnum custnum ... ]
 
 =head1 DESCRIPTION
 
@@ -132,6 +194,8 @@ the bill and collect methods of a cust_main object.  See L<FS::cust_main>.
 
   -p: Only process customers with the specified payby (I<CARD>, I<DCRD>, I<CHEK>, I<DCHK>, I<BILL>, I<COMP>, I<LECB>)
 
+  -a: Only process customers with the specified agentnum
+
   -s: re-charge setup fees
 
   -v: enable debugging