fix user modification?
[freeside.git] / rt / lib / RT / SearchBuilder.pm
index 22c9aff..527952b 100644 (file)
@@ -1,8 +1,14 @@
-# BEGIN LICENSE BLOCK
+# BEGIN BPS TAGGED BLOCK {{{
 # 
-# Copyright (c) 1996-2003 Jesse Vincent <jesse@bestpractical.com>
+# COPYRIGHT:
 # 
-# (Except where explictly superceded by other copyright notices)
+# This software is Copyright (c) 1996-2009 Best Practical Solutions, LLC
+#                                          <jesse@bestpractical.com>
+# 
+# (Except where explicitly superseded by other copyright notices)
+# 
+# 
+# LICENSE:
 # 
 # This work is made available to you under the terms of Version 2 of
 # the GNU General Public License. A copy of that license should have
 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 # General Public License for more details.
 # 
-# Unless otherwise specified, all modifications, corrections or
-# extensions to this work which alter its source code become the
-# property of Best Practical Solutions, LLC when submitted for
-# inclusion in the work.
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
+# 02110-1301 or visit their web page on the internet at
+# http://www.gnu.org/licenses/old-licenses/gpl-2.0.html.
+# 
 # 
+# CONTRIBUTION SUBMISSION POLICY:
 # 
-# END LICENSE BLOCK
+# (The following paragraph is not intended to limit the rights granted
+# to you to modify and distribute this software under the terms of
+# the GNU General Public License and is only of importance to you if
+# you choose to contribute your changes and enhancements to the
+# community by submitting them to Best Practical Solutions, LLC.)
+# 
+# By intentionally submitting any modifications, corrections or
+# derivatives to this work, or any other work intended for use with
+# Request Tracker, to Best Practical Solutions, LLC, you confirm that
+# you are the copyright holder for those contributions and you grant
+# Best Practical Solutions,  LLC a nonexclusive, worldwide, irrevocable,
+# royalty-free, perpetual, license to use, copy, create derivative
+# works based on those contributions, and sublicense and distribute
+# those contributions and any derivatives thereof.
+# 
+# END BPS TAGGED BLOCK }}}
+
 =head1 NAME
 
   RT::SearchBuilder - a baseclass for RT collection objects
 =head1 METHODS
 
 
-=begin testing
-
-ok (require RT::SearchBuilder);
-
-=end testing
 
 
 =cut
@@ -45,13 +65,13 @@ ok (require RT::SearchBuilder);
 package RT::SearchBuilder;
 
 use RT::Base;
-use DBIx::SearchBuilder;
+use DBIx::SearchBuilder "1.50";
 
 use strict;
-use vars qw(@ISA);
-@ISA = qw(DBIx::SearchBuilder RT::Base);
+use warnings;
+
+use base qw(DBIx::SearchBuilder RT::Base);
 
-# {{{ sub _Init 
 sub _Init  {
     my $self = shift;
     
@@ -64,13 +84,10 @@ sub _Init  {
     }
     $self->SUPER::_Init( 'Handle' => $RT::Handle);
 }
-# }}}
-
-# {{{ sub LimitToEnabled
 
 =head2 LimitToEnabled
 
-Only find items that haven\'t been disabled
+Only find items that haven't been disabled
 
 =cut
 
@@ -81,9 +98,6 @@ sub LimitToEnabled {
                  VALUE => '0',
                  OPERATOR => '=' );
 }
-# }}}
-
-# {{{ sub LimitToDisabled
 
 =head2 LimitToDeleted
 
@@ -100,95 +114,214 @@ sub LimitToDeleted {
                  VALUE => '1'
                );
 }
-# }}}
 
-# {{{ sub FindAllRows
+=head2 LimitAttribute PARAMHASH
 
-=head2 FindAllRows
+Takes NAME, OPERATOR and VALUE to find records that has the
+matching Attribute.
 
-Find all matching rows, regardless of whether they are disabled or not
+If EMPTY is set, also select rows with an empty string as
+Attribute's Content.
+
+If NULL is set, also select rows without the named Attribute.
 
 =cut
 
-sub FindAllRows {
-  shift->{'find_disabled_rows'} = 1;
+my %Negate = (
+    '='        => '!=',
+    '!='       => '=',
+    '>'        => '<=',
+    '<'        => '>=',
+    '>='       => '<',
+    '<='       => '>',
+    'LIKE'     => 'NOT LIKE',
+    'NOT LIKE' => 'LIKE',
+    'IS'       => 'IS NOT',
+    'IS NOT'   => 'IS',
+);
+
+sub LimitAttribute {
+    my ($self, %args) = @_;
+    my $clause = 'ALIAS';
+    my $operator = ($args{OPERATOR} || '=');
+    
+    if ($args{NULL} and exists $args{VALUE}) {
+       $clause = 'LEFTJOIN';
+       $operator = $Negate{$operator};
+    }
+    elsif ($args{NEGATE}) {
+       $operator = $Negate{$operator};
+    }
+    
+    my $alias = $self->Join(
+       TYPE   => 'left',
+       ALIAS1 => $args{ALIAS} || 'main',
+       FIELD1 => 'id',
+       TABLE2 => 'Attributes',
+       FIELD2 => 'ObjectId'
+    );
+
+    my $type = ref($self);
+    $type =~ s/(?:s|Collection)$//; # XXX - Hack!
+
+    $self->Limit(
+       $clause    => $alias,
+       FIELD      => 'ObjectType',
+       OPERATOR   => '=',
+       VALUE      => $type,
+    );
+    $self->Limit(
+       $clause    => $alias,
+       FIELD      => 'Name',
+       OPERATOR   => '=',
+       VALUE      => $args{NAME},
+    ) if exists $args{NAME};
+
+    return unless exists $args{VALUE};
+
+    $self->Limit(
+       $clause    => $alias,
+       FIELD      => 'Content',
+       OPERATOR   => $operator,
+       VALUE      => $args{VALUE},
+    );
+
+    # Capture rows with the attribute defined as an empty string.
+    $self->Limit(
+       $clause    => $alias,
+       FIELD      => 'Content',
+       OPERATOR   => '=',
+       VALUE      => '',
+       ENTRYAGGREGATOR => $args{NULL} ? 'AND' : 'OR',
+    ) if $args{EMPTY};
+
+    # Capture rows without the attribute defined
+    $self->Limit(
+       %args,
+       ALIAS      => $alias,
+       FIELD      => 'id',
+       OPERATOR   => ($args{NEGATE} ? 'IS NOT' : 'IS'),
+       VALUE      => 'NULL',
+    ) if $args{NULL};
 }
 
-# {{{ sub Limit 
+=head2 LimitCustomField
 
-=head2 Limit PARAMHASH
+Takes a paramhash of key/value pairs with the following keys:
 
-This Limit sub calls SUPER::Limit, but defaults "CASESENSITIVE" to 1, thus
-making sure that by default lots of things don't do extra work trying to 
-match lower(colname) agaist lc($val);
+=over 4
 
-=cut
+=item CUSTOMFIELD - CustomField id. Optional
 
-sub Limit {
-       my $self = shift;
-       my %args = ( CASESENSITIVE => 1,
-                    @_ );
+=item OPERATOR - The usual Limit operators
+
+=item VALUE - The value to compare against
+
+=back
 
-   return $self->SUPER::Limit(%args);
+=cut
+
+sub _SingularClass {
+    my $self = shift;
+    my $class = ref($self);
+    $class =~ s/s$// or die "Cannot deduce SingularClass for $class";
+    return $class;
 }
 
-# }}}
+sub LimitCustomField {
+    my $self = shift;
+    my %args = ( VALUE        => undef,
+                 CUSTOMFIELD  => undef,
+                 OPERATOR     => '=',
+                 @_ );
+
+    my $alias = $self->Join(
+       TYPE       => 'left',
+       ALIAS1     => 'main',
+       FIELD1     => 'id',
+       TABLE2     => 'ObjectCustomFieldValues',
+       FIELD2     => 'ObjectId'
+    );
+    $self->Limit(
+       ALIAS      => $alias,
+       FIELD      => 'CustomField',
+       OPERATOR   => '=',
+       VALUE      => $args{'CUSTOMFIELD'},
+    ) if ($args{'CUSTOMFIELD'});
+    $self->Limit(
+       ALIAS      => $alias,
+       FIELD      => 'ObjectType',
+       OPERATOR   => '=',
+       VALUE      => $self->_SingularClass,
+    );
+    $self->Limit(
+       ALIAS      => $alias,
+       FIELD      => 'Content',
+       OPERATOR   => $args{'OPERATOR'},
+       VALUE      => $args{'VALUE'},
+    );
+}
 
-# {{{ sub ItemsArrayRef
+=head2 FindAllRows
 
-=item ItemsArrayRef
+Find all matching rows, regardless of whether they are disabled or not
 
-Return this object's ItemsArray.
-If it has a SortOrder attribute, sort the array by SortOrder.
-Otherwise, if it has a "Name" attribute, sort alphabetically by Name
-Otherwise, just give up and return it in the order it came from the db.
+=cut
 
+sub FindAllRows {
+    shift->{'find_disabled_rows'} = 1;
+}
 
-=begin testing
+=head2 Limit PARAMHASH
 
-use_ok(RT::Queues);
-ok(my $queues = RT::Queues->new($RT::SystemUser), 'Created a queues object');
-ok( $queues->UnLimit(),'Unlimited the result set of the queues object');
-my $items = $queues->ItemsArrayRef();
-my @items = @{$items};
+This Limit sub calls SUPER::Limit, but defaults "CASESENSITIVE" to 1, thus
+making sure that by default lots of things don't do extra work trying to 
+match lower(colname) agaist lc($val);
 
-ok($queues->NewItem->_Accessible('Name','read'));
-my @sorted = sort {lc($a->Name) cmp lc($b->Name)} @items;
-ok (@sorted, "We have an array of queues, sorted". join(',',map {$_->Name} @sorted));
+=cut
 
-ok (@items, "We have an array of queues, raw". join(',',map {$_->Name} @items));
-my @sorted_ids = map {$_->id } @sorted;
-my @items_ids = map {$_->id } @items;
+sub Limit {
+    my $self = shift;
+    my %args = ( CASESENSITIVE => 1,
+                 @_ );
 
-is ($#sorted, $#items);
-is ($sorted[0]->Name, $items[0]->Name);
-is ($sorted[-1]->Name, $items[-1]->Name);
-is_deeply(\@items_ids, \@sorted_ids, "ItemsArrayRef sorts alphabetically by name");;
+    return $self->SUPER::Limit(%args);
+}
 
+=head2 ItemsOrderBy
 
-=end testing
+If it has a SortOrder attribute, sort the array by SortOrder.
+Otherwise, if it has a "Name" attribute, sort alphabetically by Name
+Otherwise, just give up and return it in the order it came from the
+db.
 
 =cut
 
-sub ItemsArrayRef {
+sub ItemsOrderBy {
     my $self = shift;
-    my @items;
-    
+    my $items = shift;
+  
     if ($self->NewItem()->_Accessible('SortOrder','read')) {
-        @items = sort { $a->SortOrder <=> $b->SortOrder } @{$self->SUPER::ItemsArrayRef()};
+        $items = [ sort { $a->SortOrder <=> $b->SortOrder } @{$items} ];
     }
     elsif ($self->NewItem()->_Accessible('Name','read')) {
-        @items = sort { lc($a->Name) cmp lc($b->Name) } @{$self->SUPER::ItemsArrayRef()};
+        $items = [ sort { lc($a->Name) cmp lc($b->Name) } @{$items} ];
     }
-    else {
-        @items = @{$self->SUPER::ItemsArrayRef()};
-    }
-
-    return(\@items);
 
+    return $items;
 }
 
-# }}}
+=head2 ItemsArrayRef
+
+Return this object's ItemsArray, in the order that ItemsOrderBy sorts
+it.
+
+=cut
+
+sub ItemsArrayRef {
+    my $self = shift;
+    return $self->ItemsOrderBy($self->SUPER::ItemsArrayRef());
+}
 
 eval "require RT::SearchBuilder_Vendor";
 die $@ if ($@ && $@ !~ qr{^Can't locate RT/SearchBuilder_Vendor.pm});
@@ -196,5 +329,3 @@ eval "require RT::SearchBuilder_Local";
 die $@ if ($@ && $@ !~ qr{^Can't locate RT/SearchBuilder_Local.pm});
 
 1;
-
-