add balance_age_under condition and similar referring customer functionality to has_referral_custnum condition, RT#17640

2 files changed, 78 insertions, 7 deletions

diff --git a/FS/FS/part_event/Condition/balance_age_under.pm b/FS/FS/part_event/Condition/balance_age_under.pm
new file mode 100644
index 000000000..ac6d786fc
--- /dev/null
+++ b/FS/FS/part_event/Condition/balance_age_under.pm
@@ -0,0 +1,52 @@
+package FS::part_event::Condition::balance_age_under;
+
+use strict;
+use base qw( FS::part_event::Condition );
+
+sub description { 'Customer balance age (under)'; }
+
+sub option_fields {
+  (
+    'balance' => { 'label'      => 'Balance under (or equal to)',
+                   'type'       => 'money',
+                   'value'      => '0.00', #default
+                 },
+    'age'     => { 'label'      => 'Age',
+                   'type'       => 'freq',
+                 },
+  );
+}
+
+sub condition {
+  my($self, $object, %opt) = @_;
+
+  my $cust_main = $self->cust_main($object);
+
+  my $under = $self->option('balance');
+  $under = 0 unless length($under);
+
+  my $age = $self->option_age_from('age', $opt{'time'} );
+
+  $cust_main->balance_date($age) <= $under;
+}
+
+sub condition_sql {
+  my( $class, $table, %opt ) = @_;
+
+  my $under   = $class->condition_sql_option('balance');
+  my $age     = $class->condition_sql_option_age_from('age', $opt{'time'});
+
+  my $balance_sql = FS::cust_main->balance_date_sql( $age );
+
+  "$balance_sql <= CAST( $under AS DECIMAL(10,2) )";
+}
+
+sub order_sql {
+  shift->condition_sql_option_age('age');
+}
+
+sub order_sql_weight {
+  10;
+}
+
+1;
diff --git a/FS/FS/part_event/Condition/has_referral_custnum.pm b/FS/FS/part_event/Condition/has_referral_custnum.pm
index 70c9c7f8b..dee240fec 100644
--- a/FS/FS/part_event/Condition/has_referral_custnum.pm
+++ b/FS/FS/part_event/Condition/has_referral_custnum.pm
@@ -13,30 +13,49 @@ sub option_fields {
                   'type'  => 'checkbox',
                   'value' => 'Y',
                 },
+    'check_bal' => { 'label' => 'Check referring custoemr balance',
+                     'type'  => 'checkbox',
+                     'value' => 'Y',
+                   },
+    'balance' => { 'label'      => 'Referring customer balance under (or equal to)',
+                   'type'       => 'money',
+                   'value'      => '0.00', #default
+                 },
+    'age'     => { 'label'      => 'Referring customer balance age',
+                   'type'       => 'freq',
+                 },
   );
 }
 
 sub condition {
-  my($self, $object) = @_;
+  my($self, $object, %opt) = @_;
 
   my $cust_main = $self->cust_main($object);
 
   if ( $self->option('active') ) {
-
     return 0 unless $cust_main->referral_custnum;
-
     #check for no cust_main for referral_custnum? (deleted?)
+    return 0 unless $cust_main->referral_custnum_cust_main->status eq 'active';
+  } else {
+    return 0 unless $cust_main->referral_custnum; # ? 1 : 0;
+  }
 
-    $cust_main->referral_custnum_cust_main->status eq 'active';
+  return 1 unless $self->option('check_bal');
 
-  } else {
+  my $referring_cust_main = $cust_main->referral_custnum_cust_main;
 
-    $cust_main->referral_custnum; # ? 1 : 0;
+  #false laziness w/ balance_age_under
+  my $under = $self->option('balance');
+  $under = 0 unless length($under);
 
-  }
+  my $age = $self->option_age_from('age', $opt{'time'} );
+
+  $referring_cust_main->balance_date($age) <= $under;
 
 }
 
+#this is incomplete wrt checking referring customer balances, but that's okay.
+# false positives are acceptable here, its just an optimizaiton
 sub condition_sql {
   my( $class, $table ) = @_;