search tickets by customer advertising source, #16490

[freeside.git] / rt / lib / RT / URI / freeside / Internal.pm

diff --git a/rt/lib/RT/URI/freeside/Internal.pm b/rt/lib/RT/URI/freeside/Internal.pm
index bd7c42c..4069b87 100644 (file)
--- a/rt/lib/RT/URI/freeside/Internal.pm
+++ b/rt/lib/RT/URI/freeside/Internal.pm
@@ -33,7 +33,7 @@ no warnings qw(redefine);
 use FS;
 use FS::UID qw(dbh);
 use FS::CGI qw(popurl);
-use FS::UI::Web::small_custview qw(small_custview);
+use FS::UI::Web::small_custview;
 use FS::Conf;
 use FS::Record qw(qsearchs qsearch dbdef);
 use FS::cust_main;
@@ -53,7 +53,7 @@ See L<RT::URI::freeside> for public/private interface documentation.
 
 
 
-sub _FreesideGetRecord {
+sub _FreesideGetRecord { # cache this?
 
   my $self = shift;
   my ($table, $pkey) = ($self->{'fstable'}, $self->{'fspkey'});
@@ -104,13 +104,13 @@ sub FreesideGetConfig {
 
 sub smart_search { #Subroutine
 
-  return map { { $_->hash } } &FS::cust_main::smart_search(@_);
+  return map { { $_->hash } } &FS::cust_main::Search::smart_search(@_);
 
 }
 
 sub email_search { #Subroutine
 
-  return map { { $_->hash } } &FS::cust_main::email_search(@_);
+  return map { { $_->hash } } &FS::cust_main::Search::email_search(@_);
 
 }
 
@@ -142,4 +142,36 @@ sub _FreesideURILabelLong {
 
 }
 
+sub AgentName {
+  my $self = shift;
+  my $rec = $self->_FreesideGetRecord() or return;
+  my $agent = $rec->{'_object'}->agent or return;
+  return $agent->agentnum . ': ' . $agent->agent;
+}
+
+sub CustomerClass {
+  my $self = shift;
+  my $rec = $self->_FreesideGetRecord() or return;
+  my $cust_class = $rec->{'_object'}->cust_class or return;
+  return $cust_class->classname;
+}
+  
+sub CustomerTags {
+  my $self = shift;
+  my $rec = $self->_FreesideGetRecord() or return;
+  my @part_tag = $rec->{'_object'}->part_tag;
+  return map { 
+    { 'name'  => $_->tagname,
+      'desc'  => $_->tagdesc,
+      'color' => $_->tagcolor }
+  } @part_tag;
+}
+
+sub Referral {
+  my $self = shift;
+  my $rec = $self->_FreesideGetRecord() or return;
+  my $ref = qsearchs('part_referral', { refnum => $rec->{'_object'}->refnum });
+  $ref ? $ref->referral : ''
+}
+
 1;