show credit balance on invoices, #11564

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

diff --git a/FS/FS/phone_avail.pm b/FS/FS/phone_avail.pm
index e436bca..8bb6a5c 100644 (file)
--- a/FS/FS/phone_avail.pm
+++ b/FS/FS/phone_avail.pm
@@ -1,10 +1,14 @@
 package FS::phone_avail;
 
 use strict;
-use vars qw( @ISA );
-use FS::Record qw( qsearch qsearchs );
+use vars qw( @ISA $DEBUG $me );
+use FS::Record qw( qsearch qsearchs dbh );
+use FS::cust_svc;
 
-@ISA = qw(FS::Record);
+@ISA = qw(FS::cust_main_Mixin FS::Record);
+
+$me = '[FS::phone_avail]';
+$DEBUG = 0;
 
 =head1 NAME
 
@@ -61,6 +65,10 @@ nxx
 
 station
 
+=item name
+
+Optional name
+
 =item svcnum
 
 svcnum
@@ -137,13 +145,37 @@ sub check {
     || $self->ut_numbern('nxx')
     || $self->ut_numbern('station')
     || $self->ut_foreign_keyn('svcnum', 'cust_svc', 'svcnum' )
+    || $self->ut_foreign_keyn('ordernum', 'did_order', 'ordernum' )
     || $self->ut_textn('availbatch')
+    || $self->ut_textn('name')
+    || $self->ut_textn('rate_center_abbrev')
   ;
   return $error if $error;
 
   $self->SUPER::check;
 }
 
+=item cust_svc
+
+=cut
+
+sub cust_svc {
+  my $self = shift;
+  return '' unless $self->svcnum;
+  qsearchs('cust_svc', { 'svcnum' => $self->svcnum });
+}
+
+=item part_export
+
+=cut
+
+sub part_export {
+  my $self = shift;
+  return '' unless $self->exportnum;
+  qsearchs('part_export', { 'exportnum' => $self->exportnum });
+}
+
+
 sub process_batch_import {
   my $job = shift;
 
@@ -158,14 +190,59 @@ sub process_batch_import {
   };
 
   my $opt = { 'table'   => 'phone_avail',
-              'params'  => [ 'availbatch', 'exportnum', 'countrycode' ],
-              'formats' => { 'default' => [ 'state', $numsub ] },
+              'params'  => [ 'availbatch', 'exportnum', 'countrycode', 'ordernum' ],
+              'formats' => { 'default' => [ 'state', $numsub, 'name' ],
+                            'bulk' => [ 'state', $numsub, 'name', 'rate_center_abbrev' ],
+                          },
+             'postinsert_callback' => sub {  
+                   my $record = shift;
+                   if($record->ordernum) {
+                       my $did_order = qsearchs('did_order', 
+                                               { 'ordernum' => $record->ordernum } );
+                       if($did_order && !$did_order->received) {
+                           $did_order->received(time);
+                           $did_order->replace;
+                       }
+                   }
+               }, 
             };
 
   FS::Record::process_batch_import( $job, $opt, @_ );
 
 }
 
+sub flush { # evil direct SQL
+    my $opt = shift;
+
+    if ( $opt->{'ratecenter'} =~ /^[\w\s]+$/
+           && $opt->{'state'} =~ /^[A-Z][A-Z]$/ 
+           && $opt->{'exportnum'} =~ /^\d+$/) {
+       my $sth = dbh->prepare('delete from phone_avail where exportnum = ? '.
+                   ' and state = ? and name = ?');
+       $sth->execute($opt->{'exportnum'},$opt->{'state'},$opt->{'ratecenter'})
+           or die $sth->errstr;
+    }
+
+    '';
+}
+
+# Used by FS::Upgrade to migrate to a new database.
+sub _upgrade_data {
+  my ($class, %opts) = @_;
+
+  warn "$me upgrading $class\n" if $DEBUG;
+
+  my $sth = dbh->prepare(
+    'UPDATE phone_avail SET svcnum = NULL
+       WHERE svcnum IS NOT NULL
+         AND 0 = ( SELECT COUNT(*) FROM svc_phone
+                     WHERE phone_avail.svcnum = svc_phone.svcnum )'
+  ) or die dbh->errstr;
+
+  $sth->execute or die $sth->errstr;
+
+}
+
 =back
 
 =head1 BUGS