# BEGIN BPS TAGGED BLOCK {{{ # # COPYRIGHT: # # This software is Copyright (c) 1996-2005 Best Practical Solutions, LLC # # # (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 # been provided with this software, but in any event can be snarfed # from www.gnu.org. # # This work is distributed in the hope that it will be useful, but # WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU # General Public License for more details. # # 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., 675 Mass Ave, Cambridge, MA 02139, USA. # # # CONTRIBUTION SUBMISSION POLICY: # # (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 }}} package RT::CustomField; use strict; no warnings qw(redefine); use vars qw(%FieldTypes $RIGHTS %FRIENDLY_OBJECT_TYPES); use RT::CustomFieldValues; use RT::ObjectCustomFieldValues; %FieldTypes = ( Select => [ 'Select multiple values', # loc 'Select one value', # loc 'Select up to [_1] values', # loc ], Freeform => [ 'Enter multiple values', # loc 'Enter one value', # loc 'Enter up to [_1] values', # loc ], Text => [ 'Fill in multiple text areas', # loc 'Fill in one text area', # loc 'Fill in up to [_1] text areas',# loc ], Wikitext => [ 'Fill in multiple wikitext areas', # loc 'Fill in one wikitext area', # loc 'Fill in up to [_1] wikitext areas',# loc ], Image => [ 'Upload multiple images', # loc 'Upload one image', # loc 'Upload up to [_1] images', # loc ], Binary => [ 'Upload multiple files', # loc 'Upload one file', # loc 'Upload up to [_1] files', # loc ], ); %FRIENDLY_OBJECT_TYPES = (); RT::CustomField->_ForObjectType( 'RT::Queue-RT::Ticket' => "Tickets", ); #loc RT::CustomField->_ForObjectType( 'RT::Queue-RT::Ticket-RT::Transaction' => "Ticket Transactions", ); #loc RT::CustomField->_ForObjectType( 'RT::User' => "Users", ); #loc RT::CustomField->_ForObjectType( 'RT::Group' => "Groups", ); #loc $RIGHTS = { SeeCustomField => 'See custom fields', # loc_pair AdminCustomField => 'Create, delete and modify custom fields', # loc_pair ModifyCustomField => 'Add, delete and modify custom field values for objects' #loc_pair }; # Tell RT::ACE that this sort of object can get acls granted $RT::ACE::OBJECT_TYPES{'RT::CustomField'} = 1; foreach my $right ( keys %{$RIGHTS} ) { $RT::ACE::LOWERCASERIGHTNAMES{ lc $right } = $right; } sub AvailableRights { my $self = shift; return($RIGHTS); } =head1 NAME RT::CustomField_Overlay =head1 DESCRIPTION =head1 'CORE' METHODS =cut =head2 Create PARAMHASH Create takes a hash of values and creates a row in the database: varchar(200) 'Name'. varchar(200) 'Type'. int(11) 'MaxValues'. varchar(255) 'Pattern'. smallint(6) 'Repeated'. varchar(255) 'Description'. int(11) 'SortOrder'. varchar(255) 'LookupType'. smallint(6) 'Disabled'. 'LookupType' is generally the result of either RT::Ticket->CustomFieldLookupType or RT::Transaction->CustomFieldLookupType =cut sub Create { my $self = shift; my %args = ( Name => '', Type => '', MaxValues => '0', Pattern => '', Description => '', Disabled => '0', LookupType => '', Repeated => '0', @_); unless ($self->CurrentUser->HasRight(Object => $RT::System, Right => 'AdminCustomField')) { return (0, $self->loc('Permission Denied')); } if ($args{TypeComposite}) { @args{'Type', 'MaxValues'} = split(/-/, $args{TypeComposite}, 2); } elsif ($args{Type} =~ s/(?:(Single)|Multiple)$//) { # old style Type string $args{'MaxValues'} = $1 ? 1 : 0; } if ( !exists $args{'Queue'}) { # do nothing -- things below are strictly backward compat } elsif ( ! $args{'Queue'} ) { unless ( $self->CurrentUser->HasRight( Object => $RT::System, Right => 'AssignCustomFields') ) { return ( 0, $self->loc('Permission Denied') ); } $args{'LookupType'} = 'RT::Queue-RT::Ticket'; } else { my $queue = RT::Queue->new($self->CurrentUser); $queue->Load($args{'Queue'}); unless ($queue->Id) { return (0, $self->loc("Queue not found")); } unless ( $queue->CurrentUserHasRight('AssignCustomFields') ) { return ( 0, $self->loc('Permission Denied') ); } $args{'LookupType'} = 'RT::Queue-RT::Ticket'; } my $rv = $self->SUPER::Create( Name => $args{'Name'}, Type => $args{'Type'}, MaxValues => $args{'MaxValues'}, Pattern => $args{'Pattern'}, Description => $args{'Description'}, Disabled => $args{'Disabled'}, LookupType => $args{'LookupType'}, Repeated => $args{'Repeated'}, ); return $rv unless exists $args{'Queue'}; # Compat code -- create a new ObjectCustomField mapping my $OCF = RT::ObjectCustomField->new($self->CurrentUser); $OCF->Create( CustomField => $self->Id, ObjectId => $args{'Queue'}, ); return $rv; } =head2 Load ID/NAME Load a custom field. If the value handed in is an integer, load by custom field ID. Otherwise, Load by name. =cut sub Load { my $self = shift; my $id = shift; if ($id =~ /^\d+$/) { return ($self->SUPER::Load($id)); } else { return($self->LoadByName(Name => $id)); } } # {{{ sub LoadByName =head2 LoadByName (Queue => QUEUEID, Name => NAME) Loads the Custom field named NAME. If a Queue parameter is specified, only look for ticket custom fields tied to that Queue. If the Queue parameter is '0', look for global ticket custom fields. If no queue parameter is specified, look for any and all custom fields with this name. BUG/TODO, this won't let you specify that you only want user or group CFs. =cut # Compatibility for API change after 3.0 beta 1 *LoadNameAndQueue = \&LoadByName; # Change after 3.4 beta. *LoadByNameAndQueue = \&LoadByName; sub LoadByName { my $self = shift; my %args = ( Queue => undef, Name => undef, @_, ); # if we're looking for a queue by name, make it a number if (defined $args{'Queue'} && $args{'Queue'} !~ /^\d+$/) { my $QueueObj = RT::Queue->new($self->CurrentUser); $QueueObj->Load($args{'Queue'}); $args{'Queue'} = $QueueObj->Id; } # XXX - really naive implementation. Slow. - not really. still just one query my $CFs = RT::CustomFields->new($self->CurrentUser); $CFs->Limit( FIELD => 'Name', VALUE => $args{'Name'} ); # Don't limit to queue if queue is 0. Trying to do so breaks # RT::Group type CFs. if (defined $args{'Queue'}) { $CFs->LimitToQueue( $args{'Queue'} ); } # When loading by name, it's ok if they're disabled. That's not a big deal. $CFs->{'find_disabled_rows'}=1; # We only want one entry. $CFs->RowsPerPage(1); unless ($CFs->First) { return(0); } return($self->Load($CFs->First->id)); } # }}} # {{{ Dealing with custom field values =begin testing use_ok(RT::CustomField); ok(my $cf = RT::CustomField->new($RT::SystemUser)); ok(my ($id, $msg)= $cf->Create( Name => 'TestingCF', Queue => '0', SortOrder => '1', Description => 'A Testing custom field', Type=> 'SelectSingle'), 'Created a global CustomField'); ok($id != 0, 'Global custom field correctly created'); ok ($cf->SingleValue); is($cf->Type, 'Select'); is($cf->MaxValues, 1); my ($val, $msg) = $cf->SetMaxValues('0'); ok($val, $msg); is($cf->Type, 'Select'); is($cf->MaxValues, 0); ok(!$cf->SingleValue ); ok(my ($bogus_val, $bogus_msg) = $cf->SetType('BogusType') , "Trying to set a custom field's type to a bogus type"); ok($bogus_val == 0, "Unable to set a custom field's type to a bogus type"); ok(my $bad_cf = RT::CustomField->new($RT::SystemUser)); ok(my ($bad_id, $bad_msg)= $cf->Create( Name => 'TestingCF-bad', Queue => '0', SortOrder => '1', Description => 'A Testing custom field with a bogus Type', Type=> 'SelectSingleton'), 'Created a global CustomField with a bogus type'); ok($bad_id == 0, 'Global custom field correctly decided to not create a cf with a bogus type '); =end testing =cut # {{{ AddValue =head2 AddValue HASH Create a new value for this CustomField. Takes a paramhash containing the elements Name, Description and SortOrder =begin testing ok(my $cf = RT::CustomField->new($RT::SystemUser)); $cf->Load(1); ok($cf->Id == 1); ok(my ($val,$msg) = $cf->AddValue(Name => 'foo' , Description => 'TestCFValue', SortOrder => '6')); ok($val != 0); ok (my ($delval, $delmsg) = $cf->DeleteValue($val)); ok ($delval,"Deleting a cf value: $delmsg"); =end testing =cut sub AddValue { my $self = shift; my %args = ( Name => undef, Description => undef, SortOrder => undef, @_ ); unless ($self->CurrentUserHasRight('AdminCustomField')) { return (0, $self->loc('Permission Denied')); } unless ($args{'Name'}) { return(0, $self->loc("Can't add a custom field value without a name")); } my $newval = RT::CustomFieldValue->new($self->CurrentUser); return($newval->Create( CustomField => $self->Id, Name =>$args{'Name'}, Description => ($args{'Description'} || ''), SortOrder => ($args{'SortOrder'} || '0') )); } # }}} # {{{ DeleteValue =head2 DeleteValue ID Deletes a value from this custom field by id. Does not remove this value for any article which has had it selected =cut sub DeleteValue { my $self = shift; my $id = shift; unless ($self->CurrentUserHasRight('AdminCustomField')) { return (0, $self->loc('Permission Denied')); } my $val_to_del = RT::CustomFieldValue->new($self->CurrentUser); $val_to_del->Load($id); unless ($val_to_del->Id) { return (0, $self->loc("Couldn't find that value")); } unless ($val_to_del->CustomField == $self->Id) { return (0, $self->loc("That is not a value for this custom field")); } my $retval = $val_to_del->Delete(); if ($retval) { return ($retval, $self->loc("Custom field value deleted")); } else { return(0, $self->loc("Custom field value could not be deleted")); } } # }}} # {{{ Values =head2 Values FIELD Return a CustomFieldeValues object of all acceptable values for this Custom Field. =cut *ValuesObj = \&Values; sub Values { my $self = shift; my $cf_values = RT::CustomFieldValues->new($self->CurrentUser); # if the user has no rights, return an empty object if ($self->id && $self->CurrentUserHasRight( 'SeeCustomField') ) { $cf_values->LimitToCustomField($self->Id); } return ($cf_values); } # }}} # }}} # {{{ Ticket related routines # {{{ ValuesForTicket =head2 ValuesForTicket TICKET Returns a RT::ObjectCustomFieldValues object of this Field's values for TICKET. TICKET is a ticket id. This is deprecated -- use ValuesForObject instead. =cut sub ValuesForTicket { my $self = shift; my $ticket_id = shift; $RT::Logger->debug( ref($self) . " -> ValuesForTicket deprecated in favor of ValuesForObject"); my $ticket = RT::Ticket->new($self->CurrentUser); $ticket->Load($ticket_id); return $self->ValuesForObject($ticket); } # }}} # {{{ AddValueForTicket =head2 AddValueForTicket HASH Adds a custom field value for a ticket. Takes a param hash of Ticket and Content This is deprecated -- use AddValueForObject instead. =cut sub AddValueForTicket { my $self = shift; my %args = ( Ticket => undef, Content => undef, @_ ); $RT::Logger->debug( ref($self) . " -> AddValueForTicket deprecated in favor of AddValueForObject"); my $ticket = RT::Ticket->new($self->CurrentUser); $ticket->Load($args{'Ticket'}); return($self->AddValueForObject(Content => $args{'Content'}, Object => $ticket,@_)); } # }}} # {{{ DeleteValueForTicket =head2 DeleteValueForTicket HASH Adds a custom field value for a ticket. Takes a param hash of Ticket and Content This is deprecated -- use DeleteValueForObject instead. =cut sub DeleteValueForTicket { my $self = shift; my %args = ( Ticket => undef, Content => undef, @_ ); $RT::Logger->debug( ref($self) . " -> DeleteValueForTicket deprecated in favor of DeleteValueForObject"); my $ticket = RT::Ticket->new($self->CurrentUser); $ticket->load($args{'Ticket'}); return ($self->DeleteValueForObject(Object => $ticket, Content => $args{'Content'}, @_)); } # }}} # }}} =head2 ValidateQueue Queue Make sure that the queue specified is a valid queue name =cut sub ValidateQueue { my $self = shift; my $id = shift; if ($id eq '0') { # 0 means "Global" null would _not_ be ok. return (1); } my $q = RT::Queue->new($RT::SystemUser); $q->Load($id); unless ($q->id) { return undef; } return (1); } # {{{ Types =head2 Types Retuns an array of the types of CustomField that are supported =cut sub Types { return (keys %FieldTypes); } # }}} =head2 FriendlyType [TYPE, MAX_VALUES] Returns a localized human-readable version of the custom field type. If a custom field type is specified as the parameter, the friendly type for that type will be returned =cut sub FriendlyType { my $self = shift; my $type = @_ ? shift : $self->Type; my $max = @_ ? shift : $self->MaxValues; if (my $friendly_type = $FieldTypes{$type}[$max>2 ? 2 : $max]) { return ( $self->loc( $friendly_type, $max ) ); } else { return ( $self->loc( $type ) ); } } sub FriendlyTypeComposite { my $self = shift; my $composite = shift || $self->TypeComposite; return $self->FriendlyType(split(/-/, $composite, 2)); } =head2 ValidateType TYPE Takes a single string. returns true if that string is a value type of custom field =begin testing ok(my $cf = RT::CustomField->new($RT::SystemUser)); ok($cf->ValidateType('SelectSingle')); ok($cf->ValidateType('SelectMultiple')); ok(!$cf->ValidateType('SelectFooMultiple')); =end testing =cut sub ValidateType { my $self = shift; my $type = shift; if ($type =~ s/(?:Single|Multiple)$//) { $RT::Logger->warning( "Prefix 'Single' and 'Multiple' to Type deprecated, use MaxValues instead"); } if( $FieldTypes{$type}) { return(1); } else { return undef; } } sub SetType { my $self = shift; my $type = shift; if ($type =~ s/(?:(Single)|Multiple)$//) { warn "'Single' and 'Multiple' on SetType deprecated, use SetMaxValues instead"; $self->SetMaxValues($1 ? 1 : 0); } $self->SUPER::SetType($type); } # {{{ SingleValue =head2 SingleValue Returns true if this CustomField only accepts a single value. Returns false if it accepts multiple values =cut sub SingleValue { my $self = shift; if ($self->MaxValues == 1) { return 1; } else { return undef; } } sub UnlimitedValues { my $self = shift; if ($self->MaxValues == 0) { return 1; } else { return undef; } } # }}} # {{{ sub CurrentUserHasRight =head2 CurrentUserHasRight RIGHT Helper function to call the custom field's queue's CurrentUserHasRight with the passed in args. =cut sub CurrentUserHasRight { my $self = shift; my $right = shift; return $self->CurrentUser->HasRight( Object => $self, Right => $right, ); } # }}} # {{{ sub _Set sub _Set { my $self = shift; unless ( $self->CurrentUserHasRight('AdminCustomField') ) { return ( 0, $self->loc('Permission Denied') ); } return ( $self->SUPER::_Set(@_) ); } # }}} # {{{ sub _Value =head2 _Value Takes the name of a table column. Returns its value as a string, if the user passes an ACL check =cut sub _Value { my $self = shift; my $field = shift; # we need to do the rights check unless ( $self->id && $self->CurrentUserHasRight( 'SeeCustomField') ) { return (undef); } return ( $self->__Value($field) ); } # }}} # {{{ sub SetDisabled =head2 SetDisabled Takes a boolean. 1 will cause this custom field to no longer be avaialble for tickets. 0 will re-enable this queue =cut # }}} sub Queue { $RT::Logger->debug( ref($_[0]) . " -> Queue deprecated"); return 0; } sub SetQueue { $RT::Logger->debug( ref($_[0]) . " -> SetQueue deprecated"); return 0; } sub QueueObj { $RT::Logger->debug( ref($_[0]) . " -> QueueObj deprecated"); return undef; } =head2 SetTypeComposite Set this custom field's type and maximum values as a composite value =cut sub SetTypeComposite { my $self = shift; my $composite = shift; my ($type, $max_values) = split(/-/, $composite, 2); $self->SetType($type); $self->SetMaxValues($max_values); } =head2 SetLookupType Autrijus: care to doc how LookupTypes work? =cut sub SetLookupType { my $self = shift; my $lookup = shift; if ($lookup ne $self->LookupType) { # Okay... We need to invalidate our existing relationships my $ObjectCustomFields = RT::ObjectCustomFields->new($self->CurrentUser); $ObjectCustomFields->LimitToCustomField($self->Id); $_->Delete foreach @{$ObjectCustomFields->ItemsArrayRef}; } $self->SUPER::SetLookupType($lookup); } =head2 TypeComposite Returns a composite value composed of this object's type and maximum values =cut sub TypeComposite { my $self = shift; join('-', $self->Type, $self->MaxValues); } =head2 TypeComposites Returns an array of all possible composite values for custom fields. =cut sub TypeComposites { my $self = shift; return grep !/Text-0/, map { ("$_-1", "$_-0") } $self->Types; } =head2 LookupTypes Returns an array of LookupTypes available =cut sub LookupTypes { my $self = shift; return keys %FRIENDLY_OBJECT_TYPES; } my @FriendlyObjectTypes = ( "[_1] objects", # loc "[_1]'s [_2] objects", # loc "[_1]'s [_2]'s [_3] objects", # loc ); =head2 FriendlyTypeLookup =cut sub FriendlyLookupType { my $self = shift; my $lookup = shift || $self->LookupType; return ($self->loc( $FRIENDLY_OBJECT_TYPES{$lookup} )) if (defined $FRIENDLY_OBJECT_TYPES{$lookup} ); my @types = map { s/^RT::// ? $self->loc($_) : $_ } grep { defined and length } split( /-/, $lookup ) or return; return ( $self->loc( $FriendlyObjectTypes[$#types], @types ) ); } =head2 AddToObject OBJECT Add this custom field as a custom field for a single object, such as a queue or group. Takes an object =cut sub AddToObject { my $self = shift; my $object = shift; my $id = $object->Id || 0; unless (index($self->LookupType, ref($object)) == 0) { return ( 0, $self->loc('Lookup type mismatch') ); } unless ( $object->CurrentUserHasRight('AssignCustomFields') ) { return ( 0, $self->loc('Permission Denied') ); } my $ObjectCF = RT::ObjectCustomField->new( $self->CurrentUser ); $ObjectCF->LoadByCols( ObjectId => $id, CustomField => $self->Id ); if ( $ObjectCF->Id ) { return ( 0, $self->loc("That is already the current value") ); } my ( $id, $msg ) = $ObjectCF->Create( ObjectId => $id, CustomField => $self->Id ); return ( $id, $msg ); } =head2 RemoveFromObject OBJECT Remove this custom field for a single object, such as a queue or group. Takes an object =cut sub RemoveFromObject { my $self = shift; my $object = shift; my $id = $object->Id || 0; unless (index($self->LookupType, ref($object)) == 0) { return ( 0, $self->loc('Object type mismatch') ); } unless ( $object->CurrentUserHasRight('AssignCustomFields') ) { return ( 0, $self->loc('Permission Denied') ); } my $ObjectCF = RT::ObjectCustomField->new( $self->CurrentUser ); $ObjectCF->LoadByCols( ObjectId => $id, CustomField => $self->Id ); unless ( $ObjectCF->Id ) { return ( 0, $self->loc("This custom field does not apply to that object") ); } my ( $id, $msg ) = $ObjectCF->Delete; return ( $id, $msg ); } # {{{ AddValueForObject =head2 AddValueForObject HASH Adds a custom field value for a record object of some kind. Takes a param hash of Required: Object Content Optional: LargeContent ContentType =cut sub AddValueForObject { my $self = shift; my %args = ( Object => undef, Content => undef, LargeContent => undef, ContentType => undef, @_ ); my $obj = $args{'Object'} or return; unless ( $self->CurrentUserHasRight('ModifyCustomField') ) { return ( 0, $self->loc('Permission Denied') ); } $RT::Handle->BeginTransaction; my $current_values = $self->ValuesForObject($obj); if ( $self->MaxValues ) { my $extra_values = ( $current_values->Count + 1 ) - $self->MaxValues; # (The +1 is for the new value we're adding) # If we have a set of current values and we've gone over the maximum # allowed number of values, we'll need to delete some to make room. # which former values are blown away is not guaranteed while ($extra_values) { my $extra_item = $current_values->Next; unless ( $extra_item->id ) { $RT::Logger->crit( "We were just asked to delete a custom fieldvalue that doesn't exist!" ); $RT::Handle->Rollback(); return (undef); } $extra_item->Delete; $extra_values--; } } my $newval = RT::ObjectCustomFieldValue->new( $self->CurrentUser ); my $val = $newval->Create( ObjectType => ref($obj), ObjectId => $obj->Id, Content => $args{'Content'}, LargeContent => $args{'LargeContent'}, ContentType => $args{'ContentType'}, CustomField => $self->Id ); unless ($val) { $RT::Handle->Rollback(); return ($val); } $RT::Handle->Commit(); return ($val); } # }}} # {{{ DeleteValueForObject =head2 DeleteValueForObject HASH Deletes a custom field value for a ticket. Takes a param hash of Object and Content Returns a tuple of (STATUS, MESSAGE). If the call succeeded, the STATUS is true. otherwise it's false =cut sub DeleteValueForObject { my $self = shift; my %args = ( Object => undef, Content => undef, Id => undef, @_ ); unless ($self->CurrentUserHasRight('ModifyCustomField')) { return (0, $self->loc('Permission Denied')); } my $oldval = RT::ObjectCustomFieldValue->new($self->CurrentUser); if (my $id = $args{'Id'}) { $oldval->Load($id); } unless ($oldval->id) { $oldval->LoadByObjectContentAndCustomField( Object => $args{'Object'}, Content => $args{'Content'}, CustomField => $self->Id, ); } # check ot make sure we found it unless ($oldval->Id) { return(0, $self->loc("Custom field value [_1] could not be found for custom field [_2]", $args{'Content'}, $self->Name)); } # delete it my $ret = $oldval->Delete(); unless ($ret) { return(0, $self->loc("Custom field value could not be found")); } return($oldval->Id, $self->loc("Custom field value deleted")); } =head2 ValuesForObject OBJECT Return an RT::ObjectCustomFieldValues object containing all of this custom field's values for OBJECT =cut sub ValuesForObject { my $self = shift; my $object = shift; my $values = new RT::ObjectCustomFieldValues($self->CurrentUser); unless ($self->CurrentUserHasRight('SeeCustomField')) { # Return an empty object if they have no rights to see return ($values); } $values->LimitToCustomField($self->Id); $values->LimitToEnabled(); $values->LimitToObject($object); return ($values); } =head2 _ForObjectType PATH FRIENDLYNAME Tell RT that a certain object accepts custom fields Examples: 'RT::Queue-RT::Ticket' => "Tickets", # loc 'RT::Queue-RT::Ticket-RT::Transaction' => "Ticket Transactions", # loc 'RT::User' => "Users", # loc 'RT::Group' => "Groups", # loc This is a class method. =cut sub _ForObjectType { my $self = shift; my $path = shift; my $friendly_name = shift; $FRIENDLY_OBJECT_TYPES{$path} = $friendly_name; } # }}} 1;