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