X-Git-Url: http://git.freeside.biz/gitweb/?p=freeside.git;a=blobdiff_plain;f=FS%2FFS%2Fprepay_credit.pm;h=c93999c6dffa07c4435d8be568066d916ce79aa6;hp=302ba37c75e62f0049b7f7938d31f91db47f0830;hb=395cc72629d31c8dcd138acf423e66d2d73d89d2;hpb=63a268637b2d51a8766412617724b9436439deb6

diff --git a/FS/FS/prepay_credit.pm b/FS/FS/prepay_credit.pm
index 302ba37c7..c93999c6d 100644
--- a/FS/FS/prepay_credit.pm
+++ b/FS/FS/prepay_credit.pm
@@ -1,12 +1,15 @@
 package FS::prepay_credit;
 
 use strict;
-use vars qw( @ISA );
+use vars qw( @ISA $DEBUG $me );
 use FS::Record qw(qsearchs dbh);
 use FS::agent;
 
 @ISA = qw(FS::Record);
 
+$DEBUG = 0;
+$me = '[FS::prepay_credit]';
+
 =head1 NAME
 
 FS::prepay_credit - Object methods for prepay_credit records
@@ -136,7 +139,7 @@ sub agent {
 
 =over 4
 
-=item generate NUM TYPE HASHREF
+=item generate NUM TYPE LENGTH HASHREF
 
 Generates the specified number of prepaid cards.  Returns an array reference of
 the newly generated card identifiers, or a scalar error message.
@@ -145,11 +148,12 @@ the newly generated card identifiers, or a scalar error message.
 
 #false laziness w/agent::generate_reg_codes
 sub generate {
-  my( $num, $type, $hashref ) = @_;
+  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';
@@ -163,13 +167,22 @@ sub generate {
   my $dbh = dbh;
 
   my $condup = 0; #don't retry forever
-
+  
   my @cards = ();
   for ( 1 ... $num ) {
 
-    my $identifier = join('', map($codeset[int(rand $#codeset)], (0..7) ) );
-
-    redo if qsearchs('prepay_credit',{identifier=>$identifier}) && $condup++<23;
+    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 {