enable CardFortress in test database, #71513

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

diff --git a/FS/FS/prepay_credit.pm b/FS/FS/prepay_credit.pm
index 86274aa..a0fd7f9 100644 (file)
--- a/FS/FS/prepay_credit.pm
+++ b/FS/FS/prepay_credit.pm
@@ -1,11 +1,12 @@
 package FS::prepay_credit;
+use base qw(FS::Record);
 
 use strict;
-use vars qw( @ISA );
-#use FS::Record qw( qsearch qsearchs );
-use FS::Record qw();
+use vars qw( $DEBUG $me );
+use FS::Record qw(qsearchs dbh);
 
-@ISA = qw(FS::Record);
+$DEBUG = 0;
+$me = '[FS::prepay_credit]';
 
 =head1 NAME
 
@@ -21,6 +22,12 @@ FS::prepay_credit - Object methods for prepay_credit records
     'amount'     => '19.95',
   };
 
+  $record = new FS::prepay_credit {
+    'identifier' => '4198123455512121'
+    'seconds'    => '7200',
+  };
+
+
   $error = $record->insert;
 
   $error = $new_record->replace($old_record);
@@ -31,8 +38,8 @@ FS::prepay_credit - Object methods for prepay_credit records
 
 =head1 DESCRIPTION
 
-An FS::table_name object represents an pre--paid credit, such as a pre-paid
-"calling card".  FS::prepay_credit inherits from FS::Record.  The following
+An FS::prepay_credit object represents a pre-paid card.  FS::prepay_credit
+inherits from FS::Record.  The following
 fields are currently supported:
 
 =over 4
@@ -43,6 +50,10 @@ fields are currently supported:
 
 =item amount - amount of the credit
 
+=item seconds - time amount of credit (see L<FS::svc_acct/seconds>)
+
+=item agentnum - optional agent (see L<FS::agent>) for this prepaid card
+
 =back
 
 =head1 METHODS
@@ -51,7 +62,7 @@ fields are currently supported:
 
 =item new HASHREF
 
-Creates a new pre-paid credit.  To add the example to the database, see
+Creates a new pre-paid credit.  To add the pre-paid credit to the database, see
 L<"insert">.
 
 Note that this stores the hash reference, not a distinct copy of the hash it
@@ -96,31 +107,98 @@ sub check {
   $identifier =~ s/\W//g; #anything else would just confuse things
   $self->identifier($identifier);
 
-  $self->ut_number('prepaynum')
+  $self->ut_numbern('prepaynum')
   || $self->ut_alpha('identifier')
   || $self->ut_money('amount')
+  || $self->ut_numbern('seconds')
+  || $self->ut_numbern('upbytes')
+  || $self->ut_numbern('downbytes')
+  || $self->ut_numbern('totalbytes')
+  || $self->ut_foreign_keyn('agentnum', 'agent', 'agentnum')
+  || $self->SUPER::check
   ;
 
 }
 
+=item agent
+
+Returns the agent (see L<FS::agent>) for this prepaid card, if any.
+
 =back
 
-=head1 VERSION
+=head1 SUBROUTINES
 
-$Id: prepay_credit.pm,v 1.1 2000-01-31 05:22:23 ivan Exp $
+=over 4
 
-=head1 BUGS
+=item generate NUM TYPE LENGTH HASHREF
 
-=head1 SEE ALSO
+Generates the specified number of prepaid cards.  Returns an array reference of
+the newly generated card identifiers, or a scalar error message.
 
-L<FS::Record>, schema.html from the base documentation.
+=cut
 
-=head1 HISTORY
+#false laziness w/agent::generate_reg_codes
+sub generate {
+  my( $num, $type, $length, $hashref ) = @_;
+
+  my @codeset = ();
+  push @codeset, ( 'A'..'Z' ) if $type =~ /alpha/;
+  push @codeset, ( '1'..'9' ) if $type =~ /numeric/;
+  $length ||= 8;
+
+  local $SIG{HUP} = 'IGNORE';
+  local $SIG{INT} = 'IGNORE';
+  local $SIG{QUIT} = 'IGNORE';
+  local $SIG{TERM} = 'IGNORE';
+  local $SIG{TSTP} = 'IGNORE';
+  local $SIG{PIPE} = 'IGNORE';
+
+  my $oldAutoCommit = $FS::UID::AutoCommit;
+  local $FS::UID::AutoCommit = 0;
+  my $dbh = dbh;
+
+  my $condup = 0; #don't retry forever
+  
+  my @cards = ();
+  for ( 1 ... $num ) {
+
+    my $identifier = join('', map($codeset[int(rand $#codeset)], (1..$length) ) );
+
+    if ( qsearchs('prepay_credit',{identifier=>$identifier}) ) {
+      if ( $condup++ < 54 ) {
+        warn "$me generate: duplicate identifier $identifier; retrying\n"
+          if $DEBUG;
+        redo;
+      } else {
+        warn "$me generate: giving up after 54 tries"
+          if $DEBUG;
+      }
+    }
+    $condup = 0;
+
+    my $prepay_credit = new FS::prepay_credit {
+      'identifier' => $identifier,
+      %$hashref,
+    };
+    my $error = $prepay_credit->check || $prepay_credit->insert;
+    if ( $error ) {
+      $dbh->rollback if $oldAutoCommit;
+      return "(inserting prepay_credit) $error";
+    }
+    push @cards, $prepay_credit->identifier;
+  }
+
+  $dbh->commit or die $dbh->errstr if $oldAutoCommit;
+
+  \@cards;
 
-$Log: prepay_credit.pm,v $
-Revision 1.1  2000-01-31 05:22:23  ivan
-prepaid "internet cards"
+}
+
+=head1 BUGS
+
+=head1 SEE ALSO
 
+L<FS::svc_acct>, L<FS::Record>, schema.html from the base documentation.
 
 =cut