SL/TransNumber.pm

   1 package SL::TransNumber;
   2
   3 use strict;
   4
   5 use parent qw(Rose::Object);
   6
   7 use Carp;
   8 use List::MoreUtils qw(any none);
   9 use SL::DBUtils;
  10 use SL::PrefixedNumber;
  11 use SL::DB;
  12
  13 use Rose::Object::MakeMethods::Generic
  14 (
  15  scalar => [ qw(type id number save dbh dbh_provided business_id) ],
  16 );
  17
  18 my @SUPPORTED_TYPES = qw(invoice credit_note customer vendor sales_delivery_order purchase_delivery_order sales_order purchase_order sales_quotation request_quotation part service assembly letter);
  19
  20 sub new {
  21   my $class = shift;
  22   my $self  = $class->SUPER::new(@_);
  23
  24   croak "Invalid type " . $self->type if none { $_ eq $self->type } @SUPPORTED_TYPES;
  25
  26   $self->dbh_provided($self->dbh);
  27   $self->dbh(SL::DB->client->dbh) if !$self->dbh;
  28   $self->save(1) unless defined $self->save;
  29   $self->business_id(undef) if $self->type ne 'customer';
  30
  31   return $self;
  32 }
  33
  34 sub _get_filters {
  35   my $self    = shift;
  36
  37   my $type    = $self->type;
  38   my %filters = ( where => '' );
  39
  40   if (any { $_ eq $type } qw(invoice credit_note)) {
  41     $filters{trans_number}  = "invnumber";
  42     $filters{numberfield}   = $type eq 'credit_note' ? "cnnumber" : "invnumber";
  43     $filters{table}         = "ar";
  44
  45   } elsif (any { $_ eq $type } qw(customer vendor)) {
  46     $filters{trans_number}  = "${type}number";
  47     $filters{numberfield}   = "${type}number";
  48     $filters{table}         = $type;
  49
  50   } elsif ($type =~ /_delivery_order$/) {
  51     $filters{trans_number}  = "donumber";
  52     $filters{numberfield}   = $type eq 'sales_delivery_order' ? "sdonumber" : "pdonumber";
  53     $filters{table}         = "delivery_orders";
  54     $filters{where}         = $type =~ /^sales/ ? '(customer_id IS NOT NULL)' : '(vendor_id IS NOT NULL)';
  55
  56   } elsif ($type =~ /_order$/) {
  57     $filters{trans_number}  = "ordnumber";
  58     $filters{numberfield}   = $type eq 'sales_order' ? "sonumber" : "ponumber";
  59     $filters{table}         = "oe";
  60     $filters{where}         = 'NOT COALESCE(quotation, FALSE)';
  61     $filters{where}        .= $type =~ /^sales/ ? ' AND (customer_id IS NOT NULL)' : ' AND (vendor_id IS NOT NULL)';
  62
  63   } elsif ($type =~ /_quotation$/) {
  64     $filters{trans_number}  = "quonumber";
  65     $filters{numberfield}   = $type eq 'sales_quotation' ? "sqnumber" : "rfqnumber";
  66     $filters{table}         = "oe";
  67     $filters{where}         = 'COALESCE(quotation, FALSE)';
  68     $filters{where}        .= $type =~ /^sales/ ? ' AND (customer_id IS NOT NULL)' : ' AND (vendor_id IS NOT NULL)';
  69
  70   } elsif ($type =~ /part|service|assembly/) {
  71     $filters{trans_number}  = "partnumber";
  72     $filters{numberfield}   = $type eq 'service' ? 'servicenumber' : 'articlenumber';
  73     $filters{numberfield}   = $type eq 'assembly' ? 'assemblynumber' : $filters{numberfield};
  74     $filters{table}         = "parts";
  75   } elsif ($type =~ /letter/) {
  76     $filters{trans_number}  = "letternumber";
  77     $filters{numberfield}   = "letternumber";
  78     $filters{table}         = "letter";
  79   }
  80
  81   return %filters;
  82 }
  83
  84 sub is_unique {
  85   my $self = shift;
  86
  87   return undef if !$self->number;
  88
  89   my %filters = $self->_get_filters();
  90
  91   my @where;
  92   my @values = ($self->number);
  93
  94   push @where, $filters{where} if $filters{where};
  95
  96   if ($self->id) {
  97     push @where,  qq|id <> ?|;
  98     push @values, conv_i($self->id);
  99   }
 100
 101   my $where_str = @where ? ' AND ' . join(' AND ', map { "($_)" } @where) : '';
 102   my $query     = <<SQL;
 103     SELECT $filters{trans_number}
 104     FROM $filters{table}
 105     WHERE ($filters{trans_number} = ?)
 106       $where_str
 107     LIMIT 1
 108 SQL
 109   my ($existing_number) = selectfirst_array_query($main::form, $self->dbh, $query, @values);
 110
 111   return $existing_number ? 0 : 1;
 112 }
 113
 114 sub create_unique {
 115   my $self    = shift;
 116
 117   my $form    = $main::form;
 118   my %filters = $self->_get_filters();
 119   my $number;
 120
 121   SL::DB->client->with_transaction(sub {
 122     my $where = $filters{where} ? ' WHERE ' . $filters{where} : '';
 123     my $query = <<SQL;
 124       SELECT DISTINCT $filters{trans_number}, 1 AS in_use
 125       FROM $filters{table}
 126       $where
 127 SQL
 128
 129     do_query($form, $self->dbh, "LOCK TABLE " . $filters{table}) || die $self->dbh->errstr;
 130     my %numbers_in_use = selectall_as_map($form, $self->dbh, $query, $filters{trans_number}, 'in_use');
 131
 132     my $business_number;
 133     ($business_number) = selectfirst_array_query($form, $self->dbh, qq|SELECT customernumberinit FROM business WHERE id = ? FOR UPDATE|, $self->business_id) if $self->business_id;
 134     $number         = $business_number;
 135     ($number)          = selectfirst_array_query($form, $self->dbh, qq|SELECT $filters{numberfield} FROM defaults FOR UPDATE|)                               if !$number;
 136     if ($filters{numberfield} eq 'assemblynumber' and length($number) < 1) {
 137       $filters{numberfield} = 'articlenumber';
 138       ($number)        = selectfirst_array_query($form, $self->dbh, qq|SELECT $filters{numberfield} FROM defaults FOR UPDATE|)                               if !$number;
 139     }
 140     $number          ||= '';
 141     my $sequence       = SL::PrefixedNumber->new(number => $number);
 142
 143     do {
 144       $number = $sequence->get_next;
 145     } while ($numbers_in_use{$number});
 146
 147     if ($self->save) {
 148       if ($self->business_id && $business_number) {
 149         do_query($form, $self->dbh, qq|UPDATE business SET customernumberinit = ? WHERE id = ?|, $number, $self->business_id);
 150       } else {
 151         do_query($form, $self->dbh, qq|UPDATE defaults SET $filters{numberfield} = ?|, $number);
 152       }
 153     }
 154
 155     1;
 156   }) or do { die SL::DB->client->error };
 157
 158   return $number;
 159 }
 160
 161 1;