Choice.pm 10.3 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
# -*- Mode: perl; indent-tabs-mode: nil -*-
#
# The contents of this file are subject to the Mozilla Public
# License Version 1.1 (the "License"); you may not use this file
# except in compliance with the License. You may obtain a copy of
# the License at http://www.mozilla.org/MPL/
#
# Software distributed under the License is distributed on an "AS
# IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
# implied. See the License for the specific language governing
# rights and limitations under the License.
#
# The Initial Developer of the Original Code is NASA.
# Portions created by NASA are Copyright (C) 2006 San Jose State
# University Foundation. All Rights Reserved.
#
# The Original Code is the Bugzilla Bug Tracking System.
#
# Contributor(s): Max Kanat-Alexander <mkanat@bugzilla.org>
20
#                 Greg Hendricks <ghendricks@novell.com>
21 22 23 24 25

use strict;

package Bugzilla::Field::Choice;

26
use base qw(Bugzilla::Field::ChoiceInterface Bugzilla::Object);
27

28
use Bugzilla::Config qw(SetParam write_params);
29 30
use Bugzilla::Constants;
use Bugzilla::Error;
31 32
use Bugzilla::Field;
use Bugzilla::Util qw(trim detaint_natural);
33 34 35 36 37 38 39 40 41 42 43

use Scalar::Util qw(blessed);

##################
# Initialization #
##################

use constant DB_COLUMNS => qw(
    id
    value
    sortkey
44
    isactive
45
    visibility_value_id
46 47
);

48 49 50
use constant UPDATE_COLUMNS => qw(
    value
    sortkey
51
    isactive
52
    visibility_value_id
53 54
);

55 56 57
use constant NAME_FIELD => 'value';
use constant LIST_ORDER => 'sortkey, value';

58 59 60
use constant VALIDATORS => {
    value   => \&_check_value,
    sortkey => \&_check_sortkey,
61
    visibility_value_id => \&_check_visibility_value_id,
62
    isactive => \&_check_isactive,
63 64 65 66
};

use constant CLASS_MAP => {
    bug_status => 'Bugzilla::Status',
67
    classification => 'Bugzilla::Classification',
68
    component  => 'Bugzilla::Component',
69
    product    => 'Bugzilla::Product',
70 71
};

72 73 74 75 76 77 78
use constant DEFAULT_MAP => {
    op_sys       => 'defaultopsys',
    rep_platform => 'defaultplatform',
    priority     => 'defaultpriority',
    bug_severity => 'defaultseverity',
};

79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94
#################
# Class Factory #
#################

# Bugzilla::Field::Choice is actually an abstract base class. Every field
# type has its own dynamically-generated class for its values. This allows
# certain fields to have special types, like how bug_status's values
# are Bugzilla::Status objects.

sub type {
    my ($class, $field) = @_;
    my $field_obj = blessed $field ? $field : Bugzilla::Field->check($field);
    my $field_name = $field_obj->name;

    if ($class->CLASS_MAP->{$field_name}) {
        return $class->CLASS_MAP->{$field_name};
95
    }
96 97 98 99

    # For generic classes, we use a lowercase class name, so as
    # not to interfere with any real subclasses we might make some day.
    my $package = "Bugzilla::Field::Choice::$field_name";
100
    Bugzilla->request_cache->{"field_$package"} = $field_obj;
101

102 103 104 105 106
    # This package only needs to be created once. We check if the DB_TABLE
    # glob for this package already exists, which tells us whether or not
    # we need to create the package (this works even under mod_perl, where
    # this package definition will persist across requests)).
    if (!defined *{"${package}::DB_TABLE"}) {
107 108 109 110 111 112 113 114
        eval <<EOC;
            package $package;
            use base qw(Bugzilla::Field::Choice);
            use constant DB_TABLE => '$field_name';
EOC
    }

    return $package;
115 116
}

117 118 119 120 121 122 123
################
# Constructors #
################

# We just make new() enforce this, which should give developers 
# the understanding that you can't use Bugzilla::Field::Choice
# without calling type().
124 125
sub new {
    my $class = shift;
126 127 128 129
    if ($class eq 'Bugzilla::Field::Choice') {
        ThrowCodeError('field_choice_must_use_type');
    }
    $class->SUPER::new(@_);
130 131
}

132 133 134
#########################
# Database Manipulation #
#########################
135

136 137 138 139 140
# Our subclasses can take more arguments than we normally accept.
# So, we override create() to remove arguments that aren't valid
# columns. (Normally Bugzilla::Object dies if you pass arguments
# that aren't valid columns.)
sub create {
141 142
    my $class = shift;
    my ($params) = @_;
143
    foreach my $key (keys %$params) {
144
        if (!grep {$_ eq $key} $class->_get_db_columns) {
145 146 147 148
            delete $params->{$key};
        }
    }
    return $class->SUPER::create(@_);
149 150
}

151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177
sub update {
    my $self = shift;
    my $dbh = Bugzilla->dbh;
    my $fname = $self->field->name;

    $dbh->bz_start_transaction();

    my ($changes, $old_self) = $self->SUPER::update(@_);
    if (exists $changes->{value}) {
        my ($old, $new) = @{ $changes->{value} };
        if ($self->field->type == FIELD_TYPE_MULTI_SELECT) {
            $dbh->do("UPDATE bug_$fname SET value = ? WHERE value = ?",
                     undef, $new, $old);
        }
        else {
            $dbh->do("UPDATE bugs SET $fname = ? WHERE $fname = ?",
                     undef, $new, $old);
        }

        if ($old_self->is_default) {
            my $param = $self->DEFAULT_MAP->{$self->field->name};
            SetParam($param, $self->name);
            write_params();
        }
    }

    $dbh->bz_commit_transaction();
178
    return wantarray ? ($changes, $old_self) : $changes;
179 180
}

181 182 183 184
sub remove_from_db {
    my $self = shift;
    if ($self->is_default) {
        ThrowUserError('fieldvalue_is_default',
185
                       { field => $self->field, value => $self,
186 187 188 189 190
                         param_name => $self->DEFAULT_MAP->{$self->field->name},
                       });
    }
    if ($self->is_static) {
        ThrowUserError('fieldvalue_not_deletable', 
191
                       { field => $self->field, value => $self });
192 193 194
    }
    if ($self->bug_count) {
        ThrowUserError("fieldvalue_still_has_bugs",
195
                       { field => $self->field, value => $self });
196
    }
197
    $self->_check_if_controller(); # From ChoiceInterface.
198 199 200
    $self->SUPER::remove_from_db();
}

201 202 203 204
############
# Mutators #
############

205 206 207
sub set_is_active { $_[0]->set('isactive', $_[1]); }
sub set_name      { $_[0]->set('value', $_[1]);    }
sub set_sortkey   { $_[0]->set('sortkey', $_[1]);  }
208 209 210 211 212
sub set_visibility_value {
    my ($self, $value) = @_;
    $self->set('visibility_value_id', $value);
    delete $self->{visibility_value};
}
213

214 215 216 217
##############
# Validators #
##############

218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236
sub _check_isactive {
    my ($invocant, $value) = @_;
    $value = Bugzilla::Object::check_boolean($invocant, $value);
    if (!$value and ref $invocant) {
        if ($invocant->is_default) {
            my $field = $invocant->field;
            ThrowUserError('fieldvalue_is_default', 
                           { value => $invocant, field => $field,
                             param_name => $invocant->DEFAULT_MAP->{$field->name}
                           });
        }
        if ($invocant->is_static) {
            ThrowUserError('fieldvalue_not_deletable',
                           { value => $invocant, field => $invocant->field });
        }
    }
    return $value;
}

237 238 239 240 241 242
sub _check_value {
    my ($invocant, $value) = @_;

    my $field = $invocant->field;

    $value = trim($value);
243 244 245 246 247 248

    # Make sure people don't rename static values
    if (blessed($invocant) && $value ne $invocant->name 
        && $invocant->is_static) 
    {
        ThrowUserError('fieldvalue_not_editable',
249
                       { field => $field, old_value => $invocant });
250 251
    }

252 253 254 255 256
    ThrowUserError('fieldvalue_undefined') if !defined $value || $value eq "";
    ThrowUserError('fieldvalue_name_too_long', { value => $value })
        if length($value) > MAX_FIELD_VALUE_SIZE;

    my $exists = $invocant->type($field)->new({ name => $value });
257
    if ($exists && (!blessed($invocant) || $invocant->id != $exists->id)) {
258
        ThrowUserError('fieldvalue_already_exists', 
259
                       { field => $field, value => $exists });
260 261
    }

262
    return $value;
263 264
}

265 266 267 268 269 270 271 272 273 274 275 276
sub _check_sortkey {
    my ($invocant, $value) = @_;
    $value = trim($value);
    return 0 if !$value;
    # Store for the error message in case detaint_natural clears it.
    my $orig_value = $value;
    detaint_natural($value)
        || ThrowUserError('fieldvalue_sortkey_invalid',
                          { sortkey => $orig_value,
                            field   => $invocant->field });
    return $value;
}
277

278 279 280 281 282 283 284 285 286
sub _check_visibility_value_id {
    my ($invocant, $value_id) = @_;
    $value_id = trim($value_id);
    my $field = $invocant->field->value_field;
    return undef if !$field || !$value_id;
    my $value_obj = Bugzilla::Field::Choice->type($field)
                    ->check({ id => $value_id });
    return $value_obj->id;
}
287 288 289 290 291 292 293 294 295 296 297 298 299

1;

__END__

=head1 NAME

Bugzilla::Field::Choice - A legal value for a <select>-type field.

=head1 SYNOPSIS

 my $field = new Bugzilla::Field({name => 'bug_status'});

300
 my $choice = new Bugzilla::Field::Choice->type($field)->new(1);
301

302 303 304
 my $choices = Bugzilla::Field::Choice->type($field)->new_from_list([1,2,3]);
 my $choices = Bugzilla::Field::Choice->type($field)->get_all();
 my $choices = Bugzilla::Field::Choice->type($field->match({ sortkey => 10 }); 
305 306 307 308

=head1 DESCRIPTION

This is an implementation of L<Bugzilla::Object>, but with a twist.
309 310 311 312 313 314 315
You can't call any class methods (such as C<new>, C<create>, etc.) 
directly on C<Bugzilla::Field::Choice> itself. Instead, you have to
call C<Bugzilla::Field::Choice-E<gt>type($field)> to get the class
you're going to instantiate, and then you call the methods on that.

We do that because each field has its own database table for its values, so
each value type needs its own class.
316

317
See the L</SYNOPSIS> for examples of how this works.
318

319 320 321
This class implements L<Bugzilla::Field::ChoiceInterface>, and so all
methods of that class are also available here.

322 323
=head1 METHODS

324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343
=head2 Class Factory

In object-oriented design, a "class factory" is a method that picks
and returns the right class for you, based on an argument that you pass.

=over

=item C<type>

Takes a single argument, which is either the name of a field from the
C<fielddefs> table, or a L<Bugzilla::Field> object representing a field.

Returns an appropriate subclass of C<Bugzilla::Field::Choice> that you
can now call class methods on (like C<new>, C<create>, C<match>, etc.)

B<NOTE>: YOU CANNOT CALL CLASS METHODS ON C<Bugzilla::Field::Choice>. You
must call C<type> to get a class you can call methods on.

=back

344
=head2 Mutators
345

346 347
This class implements mutators for all of the settable accessors in
L<Bugzilla::Field::ChoiceInterface>.