autoload methods returning foreign records, RT#13971
[freeside.git] / FS / FS / otaker_Mixin.pm
1 package FS::otaker_Mixin;
2
3 use strict;
4 use Carp qw( croak ); #confess );
5 use FS::Record qw( qsearch qsearchs );
6 use FS::access_user;
7 use FS::UID qw( dbh );
8
9 sub otaker {
10   my $self = shift;
11   if ( scalar(@_) ) { #set
12     my $otaker = shift;
13     my $access_user = qsearchs('access_user', { 'username' => $otaker } );
14     if ( !$access_user && $otaker =~ /^(.+), (.+)$/ ) { #same as below..
15       my($lastname, $firstname) = ($1, $2);
16       $otaker = lc($firstname.$lastname);
17       $otaker =~ s/ //g;
18       $access_user =  qsearchs('access_user', { 'first' => $firstname, 
19                                                 'last'  => $lastname  } )
20                    || qsearchs('access_user', { 'username' => $otaker } );
21     }
22     croak "can't set otaker: $otaker not found!" unless $access_user; #confess?
23     $self->usernum( $access_user->usernum );
24     $otaker; #not sure return is used anywhere, but just in case
25   } else { #get
26     if ( $self->usernum ) {
27       $self->access_user->username;
28     } elsif ( length($self->get('otaker')) ) {
29       $self->get('otaker');
30     } else {
31       '';
32     }
33   }
34 }
35
36 sub _upgrade_otaker {
37   my $class = shift;
38   my $table = $class->table;
39
40   my $limit = ( $table eq 'cust_attachment' ? 10 : 1000 );
41
42   while ( 1 ) {
43     my @records = qsearch({
44                     'table'     => $table,
45                     'hashref'   => {},
46                     'extra_sql' => "WHERE otaker IS NOT NULL LIMIT $limit",
47                   });
48     last unless @records;
49
50     foreach my $record (@records) {
51       eval { $record->otaker($record->otaker) };
52       if ( $@ ) {
53         my $username = $record->otaker;
54         my($lastname, $firstname) = ( 'User', 'Legacy' );
55         if ( $username =~ /^(.+), (.+)$/ ) {
56           ($lastname, $firstname) = ($1, $2);
57           $username = lc($firstname.$lastname);
58           $username =~ s/ //g;
59         }
60         my $access_user = new FS::access_user {
61           'username'  => $username,
62           '_password' => 'CHANGEME',
63           'first'     => $firstname,
64           'last'      => $lastname,
65           'disabled'  => 'Y',
66         };
67         my $error = $access_user->insert;
68         die $error if $error;
69         $record->otaker($record->otaker);
70       }
71       $record->set('otaker', '');
72       my $error = $record->replace;
73       die $error if $error;
74     }
75         
76     if ( $table eq 'cust_attachment' ) {
77         warn "  committing (cust_attachment) \n";
78         dbh->commit or die dbh->errstr;
79     }
80
81   }
82
83 }
84
85 1;