start adding package locations, RT#4499

[freeside.git] / FS / FS / cust_main_county.pm

diff --git a/FS/FS/cust_main_county.pm b/FS/FS/cust_main_county.pm
index 172eac9..edf57ab 100644 (file)
--- a/FS/FS/cust_main_county.pm
+++ b/FS/FS/cust_main_county.pm
@@ -157,16 +157,75 @@ sub recurtax {
   return '';
 }
 
-=item taxline CUST_BILL_PKG, ...
+=item sql_taxclass_sameregion
+
+Returns an SQL WHERE fragment or the empty string to search for entries
+with different tax classes.
+
+=cut
+
+#hmm, description above could be better...
+
+sub sql_taxclass_sameregion {
+  my $self = shift;
+
+  my $same_query = 'SELECT taxclass FROM cust_main_county '.
+                   ' WHERE taxnum != ? AND country = ?';
+  my @same_param = ( 'taxnum', 'country' );
+  foreach my $opt_field (qw( state county )) {
+    if ( $self->$opt_field() ) {
+      $same_query .= " AND $opt_field = ?";
+      push @same_param, $opt_field;
+    } else {
+      $same_query .= " AND $opt_field IS NULL";
+    }
+  }
+
+  my @taxclasses = $self->_list_sql( \@same_param, $same_query );
+
+  return '' unless scalar(@taxclasses);
+
+  '( taxclass IS NULL OR ( '.  #only if !$self->taxclass ??
+     join(' AND ', map { 'taxclass != '.dbh->quote($_) } @taxclasses ). 
+  ' ) ) ';
+}
+
+sub _list_sql {
+  my( $self, $param, $sql ) = @_;
+  my $sth = dbh->prepare($sql) or die dbh->errstr;
+  $sth->execute( map $self->$_(), @$param )
+    or die "Unexpected error executing statement $sql: ". $sth->errstr;
+  map $_->[0], @{ $sth->fetchall_arrayref };
+}
+
+=item taxline TAXABLES, [ OPTIONSHASH ]
 
 Returns a listref of a name and an amount of tax calculated for the list of
-packages.  Returns a scalar error message on error.
+packages or amounts referenced by TAXABLES.  Returns a scalar error message
+on error.  
+
+OPTIONSHASH includes custnum and invoice_date and are hints to this method
 
 =cut
 
 sub taxline {
   my $self = shift;
 
+  my $taxables;
+  my %opt = ();
+
+  if (ref($_[0]) eq 'ARRAY') {
+    $taxables = shift;
+    %opt = @_;
+  }else{
+    $taxables = [ @_ ];
+    # exemptions broken in this case
+  }
+
+  my @exemptions = ();
+  push @exemptions, @{ $_->_cust_tax_exempt_pkg }
+    for grep { ref($_) } @$taxables;
+    
   local $SIG{HUP} = 'IGNORE';
   local $SIG{INT} = 'IGNORE';
   local $SIG{QUIT} = 'IGNORE';
@@ -181,10 +240,13 @@ sub taxline {
   my $name = $self->taxname || 'Tax';
   my $amount = 0;
 
-  foreach my $cust_bill_pkg (@_) {
+  foreach my $cust_bill_pkg (@$taxables) {
 
-    my $cust_bill = $cust_bill_pkg->cust_pkg->cust_bill;
-    my $part_pkg = $cust_bill_pkg->part_pkg;
+    my $cust_pkg  = $cust_bill_pkg->cust_pkg;
+    my $cust_bill = $cust_pkg->cust_bill if $cust_pkg;
+    my $custnum   = $cust_pkg ? $cust_pkg->custnum : $opt{custnum};
+    my $part_pkg  = $cust_bill_pkg->part_pkg;
+    my $invoice_date = $cust_bill ? $cust_bill->_date : $opt{invoice_date};
   
     my $taxable_charged = 0;
     $taxable_charged += $cust_bill_pkg->setup
@@ -194,13 +256,12 @@ sub taxline {
       unless $part_pkg->recurtax =~ /^Y$/i
           || $self->recurtax =~ /^Y$/i;
 
-    return [ $name, 0 ]
-      unless $taxable_charged;
+    next unless $taxable_charged;
   
     if ( $self->exempt_amount && $self->exempt_amount > 0 ) {
       #my ($mon,$year) = (localtime($cust_bill_pkg->sdate) )[4,5];
       my ($mon,$year) =
-        (localtime( $cust_bill_pkg->sdate || $cust_bill->_date ) )[4,5];
+        (localtime( $cust_bill_pkg->sdate || $invoice_date ) )[4,5];
       $mon++;
       my $freq = $part_pkg->freq || 1;
       if ( $freq !~ /(\d+)$/ ) {
@@ -214,7 +275,7 @@ sub taxline {
       #call the whole thing off if this customer has any old
       #exemption records...
       my @cust_tax_exempt =
-        qsearch( 'cust_tax_exempt' => { custnum=> $cust_bill->custnum } );
+        qsearch( 'cust_tax_exempt' => { custnum=> $custnum } );
       if ( @cust_tax_exempt ) {
         $dbh->rollback if $oldAutoCommit;
         return
@@ -240,7 +301,7 @@ sub taxline {
           return "fatal: can't lookup exising exemption: ". dbh->errstr;
         };
         $sth->execute(
-          $cust_bill->custnum,
+          $custnum,
           $self->taxnum,
           1900+$year,
           $mon,
@@ -249,6 +310,15 @@ sub taxline {
           return "fatal: can't lookup exising exemption: ". dbh->errstr;
         };
         my $existing_exemption = $sth->fetchrow_arrayref->[0] || 0;
+
+        foreach ( grep { $_->taxnum == $self->taxnum &&
+                         $_->month  == $mon          &&
+                         $_->year   == 1900+$year
+                       } @exemptions
+                )
+        {
+          $existing_exemption += $_->amount;
+        }
         
         my $remaining_exemption =
           $self->exempt_amount - $existing_exemption;
@@ -259,17 +329,22 @@ sub taxline {
           $taxable_charged -= $addl;
 
           my $cust_tax_exempt_pkg = new FS::cust_tax_exempt_pkg ( {
-            'billpkgnum' => $cust_bill_pkg->billpkgnum,
             'taxnum'     => $self->taxnum,
             'year'       => 1900+$year,
             'month'      => $mon,
             'amount'     => sprintf("%.2f", $addl ),
           } );
-          my $error = $cust_tax_exempt_pkg->insert;
-          if ( $error ) {
-            $dbh->rollback if $oldAutoCommit;
-            return "fatal: can't insert cust_tax_exempt_pkg: $error";
-          }
+          if ($cust_bill_pkg->billpkgnum) {
+            $cust_tax_exempt_pkg->billpkgnum($cust_bill_pkg->billpkgnum);
+            my $error = $cust_tax_exempt_pkg->insert;
+            if ( $error ) {
+              $dbh->rollback if $oldAutoCommit;
+              return "fatal: can't insert cust_tax_exempt_pkg: $error";
+            }
+          }else{
+            push @exemptions, $cust_tax_exempt_pkg;
+            push @{ $cust_bill_pkg->_cust_tax_exempt_pkg }, $cust_tax_exempt_pkg;
+          } # if $cust_bill_pkg->billpkgnum
         } # if $remaining_exemption > 0
 
         #++