Object.pm 39.8 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14
# -*- 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 Original Code is the Bugzilla Bug Tracking System.
#
15 16 17 18
# The Initial Developer of the Original Code is Everything Solved.
# Portions created by Everything Solved are Copyright (C) 2006 
# Everything Solved. All Rights Reserved.
#
19
# Contributor(s): Max Kanat-Alexander <mkanat@bugzilla.org>
20
#                 Frédéric Buclin <LpSolit@gmail.com>
21 22 23 24 25

use strict;

package Bugzilla::Object;

26
use Bugzilla::Constants;
27
use Bugzilla::Hook;
28 29 30
use Bugzilla::Util;
use Bugzilla::Error;

31
use Date::Parse;
32
use List::MoreUtils qw(part);
33
use Scalar::Util qw(blessed);
34

35 36 37
use constant NAME_FIELD => 'name';
use constant ID_FIELD   => 'id';
use constant LIST_ORDER => NAME_FIELD;
38

39
use constant UPDATE_VALIDATORS => {};
40
use constant NUMERIC_COLUMNS   => ();
41
use constant DATE_COLUMNS      => ();
42
use constant VALIDATOR_DEPENDENCIES => {};
43 44
# XXX At some point, this will be joined with FIELD_MAP.
use constant REQUIRED_FIELD_MAP  => {};
45
use constant EXTRA_REQUIRED_FIELDS => ();
46

47 48 49 50 51
# This allows the JSON-RPC interface to return Bugzilla::Object instances
# as though they were hashes. In the future, this may be modified to return
# less information.
sub TO_JSON { return { %{ $_[0] } }; }

52 53 54 55 56 57 58 59 60 61 62 63
###############################
####    Initialization     ####
###############################

sub new {
    my $invocant = shift;
    my $class    = ref($invocant) || $invocant;
    my $object   = $class->_init(@_);
    bless($object, $class) if $object;
    return $object;
}

64 65 66 67 68

# Note: Because this uses sql_istrcmp, if you make a new object use
# Bugzilla::Object, make sure that you modify bz_setup_database
# in Bugzilla::DB::Pg appropriately, to add the right LOWER
# index. You can see examples already there.
69 70 71 72
sub _init {
    my $class = shift;
    my ($param) = @_;
    my $dbh = Bugzilla->dbh;
73
    my $columns = join(',', $class->_get_db_columns);
74
    my $table   = $class->DB_TABLE;
75 76
    my $name_field = $class->NAME_FIELD;
    my $id_field   = $class->ID_FIELD;
77

78 79 80 81
    my $id = $param;
    if (ref $param eq 'HASH') {
        $id = $param->{id};
    }
82 83 84
    my $object;

    if (defined $id) {
85 86
        # We special-case if somebody specifies an ID, so that we can
        # validate it as numeric.
87 88 89 90
        detaint_natural($id)
          || ThrowCodeError('param_must_be_numeric',
                            {function => $class . '::_init'});

91 92 93
        # Too large integers make PostgreSQL crash.
        return if $id > MAX_INT_32;

94 95
        $object = $dbh->selectrow_hashref(qq{
            SELECT $columns FROM $table
96
             WHERE $id_field = ?}, undef, $id);
97
    } else {
98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122
        unless (defined $param->{name} || (defined $param->{'condition'} 
                                           && defined $param->{'values'}))
        {
            ThrowCodeError('bad_arg', { argument => 'param',
                                        function => $class . '::new' });
        }

        my ($condition, @values);
        if (defined $param->{name}) {
            $condition = $dbh->sql_istrcmp($name_field, '?');
            push(@values, $param->{name});
        }
        elsif (defined $param->{'condition'} && defined $param->{'values'}) {
            caller->isa('Bugzilla::Object')
                || ThrowCodeError('protection_violation',
                       { caller    => caller, 
                         function  => $class . '::new',
                         argument  => 'condition/values' });
            $condition = $param->{'condition'};
            push(@values, @{$param->{'values'}});
        }

        map { trick_taint($_) } @values;
        $object = $dbh->selectrow_hashref(
            "SELECT $columns FROM $table WHERE $condition", undef, @values);
123 124 125 126 127
    }

    return $object;
}

128 129 130 131 132 133 134
sub check {
    my ($invocant, $param) = @_;
    my $class = ref($invocant) || $invocant;
    # If we were just passed a name, then just use the name.
    if (!ref $param) {
        $param = { name => $param };
    }
135

136
    # Don't allow empty names or ids.
137 138 139 140 141 142 143 144 145
    my $check_param = exists $param->{id} ? 'id' : 'name';
    $param->{$check_param} = trim($param->{$check_param});
    # If somebody passes us "0", we want to throw an error like
    # "there is no X with the name 0". This is true even for ids. So here,
    # we only check if the parameter is undefined or empty.
    if (!defined $param->{$check_param} or $param->{$check_param} eq '') {
        ThrowUserError('object_not_specified', { class => $class });
    }

146 147 148 149 150
    my $obj = $class->new($param);
    if (!$obj) {
        # We don't want to override the normal template "user" object if
        # "user" is one of the params.
        delete $param->{user};
151 152 153 154 155 156
        if (my $error = delete $param->{_error}) {
            ThrowUserError($error, { %$param, class => $class });
        }
        else {
            ThrowUserError('object_does_not_exist', { %$param, class => $class });
        }
157
    }
158 159 160
    return $obj;
}

161
sub new_from_list {
162 163
    my $invocant = shift;
    my $class = ref($invocant) || $invocant;
164
    my ($id_list) = @_;
165
    my $id_field = $class->ID_FIELD;
166

167 168 169 170 171
    my @detainted_ids;
    foreach my $id (@$id_list) {
        detaint_natural($id) ||
            ThrowCodeError('param_must_be_numeric',
                          {function => $class . '::new_from_list'});
172 173
        # Too large integers make PostgreSQL crash.
        next if $id > MAX_INT_32;
174
        push(@detainted_ids, $id);
175
    }
176 177 178 179 180
    # We don't do $invocant->match because some classes have
    # their own implementation of match which is not compatible
    # with this one. However, match() still needs to have the right $invocant
    # in order to do $class->DB_TABLE and so on.
    return match($invocant, { $id_field => \@detainted_ids });
181 182
}

183 184 185 186 187 188 189 190 191 192
# Note: Future extensions to this could be:
#  * Add a MATCH_JOIN constant so that we can join against
#    certain other tables for the WHERE criteria.
sub match {
    my ($invocant, $criteria) = @_;
    my $class = ref($invocant) || $invocant;
    my $dbh   = Bugzilla->dbh;

    return [$class->get_all] if !$criteria;

193
    my (@terms, @values, $postamble);
194 195
    foreach my $field (keys %$criteria) {
        my $value = $criteria->{$field};
196 197 198 199 200
        
        # allow for LIMIT and OFFSET expressions via the criteria.
        next if $field eq 'OFFSET';
        if ( $field eq 'LIMIT' ) {
            next unless defined $value;
201 202 203 204 205 206 207 208 209 210 211 212 213
            detaint_natural($value)
              or ThrowCodeError('param_must_be_numeric', 
                                { param    => 'LIMIT', 
                                  function => "${class}::match" });
            my $offset;
            if (defined $criteria->{OFFSET}) {
                $offset = $criteria->{OFFSET};
                detaint_signed($offset)
                  or ThrowCodeError('param_must_be_numeric', 
                                    { param    => 'OFFSET',
                                      function => "${class}::match" });
            }
            $postamble = $dbh->sql_limit($value, $offset);
214 215 216 217 218
            next;
        }
        elsif ( $field eq 'WHERE' ) {
            # the WHERE value is a hashref where the keys are
            # "column_name operator ?" and values are the placeholder's
219 220 221 222 223 224
            # value (either a scalar or an array of values).
            foreach my $k (keys %$value) {
                push(@terms, $k);
                my @this_value = ref($value->{$k}) ? @{ $value->{$k} } 
                                                   : ($value->{$k});
                push(@values, @this_value);
225 226 227
            }            
            next;
        }
228
        
229 230
        $class->_check_field($field, 'match');

231 232 233 234 235 236 237 238 239 240 241
        if (ref $value eq 'ARRAY') {
            # IN () is invalid SQL, and if we have an empty list
            # to match against, we're just returning an empty
            # array anyhow.
            return [] if !scalar @$value;

            my @qmarks = ("?") x @$value;
            push(@terms, $dbh->sql_in($field, \@qmarks));
            push(@values, @$value);
        }
        elsif ($value eq NOT_NULL) {
242 243 244 245 246 247 248 249 250 251 252
            push(@terms, "$field IS NOT NULL");
        }
        elsif ($value eq IS_NULL) {
            push(@terms, "$field IS NULL");
        }
        else {
            push(@terms, "$field = ?");
            push(@values, $value);
        }
    }

253 254
    my $where = join(' AND ', @terms) if scalar @terms;
    return $class->_do_list_select($where, \@values, $postamble);
255 256 257
}

sub _do_list_select {
258
    my ($class, $where, $values, $postamble) = @_;
259
    my $table = $class->DB_TABLE;
260
    my $cols  = join(',', $class->_get_db_columns);
261 262 263 264 265 266 267
    my $order = $class->LIST_ORDER;

    my $sql = "SELECT $cols FROM $table";
    if (defined $where) {
        $sql .= " WHERE $where ";
    }
    $sql .= " ORDER BY $order";
268 269 270
    
    $sql .= " $postamble" if $postamble;
        
271
    my $dbh = Bugzilla->dbh;
272 273 274 275 276 277
    # Sometimes the values are tainted, but we don't want to untaint them
    # for the caller. So we copy the array. It's safe to untaint because
    # they're only used in placeholders here.
    my @untainted = @{ $values || [] };
    trick_taint($_) foreach @untainted;
    my $objects = $dbh->selectall_arrayref($sql, {Slice=>{}}, @untainted);
278 279
    bless ($_, $class) foreach @$objects;
    return $objects
280 281
}

282 283 284 285
###############################
####      Accessors      ######
###############################

286 287
sub id   { return $_[0]->{$_[0]->ID_FIELD};   }
sub name { return $_[0]->{$_[0]->NAME_FIELD}; }
288

289 290 291 292 293 294 295 296
###############################
####        Methods        ####
###############################

sub set {
    my ($self, $field, $value) = @_;

    # This method is protected. It's used to help implement set_ functions.
297 298
    my $caller = caller;
    $caller->isa('Bugzilla::Object') || $caller->isa('Bugzilla::Extension')
299 300 301 302 303
        || ThrowCodeError('protection_violation', 
                          { caller     => caller,
                            superclass => __PACKAGE__,
                            function   => 'Bugzilla::Object->set' });

304
    Bugzilla::Hook::process('object_before_set',
305 306 307
                            { object => $self, field => $field,
                              value => $value });

308
    my %validators = (%{$self->_get_validators}, %{$self->UPDATE_VALIDATORS});
309 310
    if (exists $validators{$field}) {
        my $validator = $validators{$field};
311
        $value = $self->$validator($value, $field);
312
        trick_taint($value) if (defined $value && !ref($value));
313 314 315 316

        if ($self->can('_set_global_validator')) {
            $self->_set_global_validator($value, $field);
        }
317 318 319
    }

    $self->{$field} = $value;
320 321 322

    Bugzilla::Hook::process('object_end_of_set',
                            { object => $self, field => $field });
323 324
}

325 326
sub set_all {
    my ($self, $params) = @_;
327

328 329 330 331
    # Don't let setters modify the values in $params for the caller.
    my %field_values = %$params;

    my @sorted_names = $self->_sort_by_dep(keys %field_values);
332
    foreach my $key (@sorted_names) {
333 334 335 336
        # It's possible for one set_ method to delete a key from $params
        # for another set method, so if that's happened, we don't call the
        # other set method.
        next if !exists $field_values{$key};
337
        my $method = "set_$key";
338
        $self->$method($field_values{$key}, \%field_values);
339
    }
340 341
    Bugzilla::Hook::process('object_end_of_set_all', 
                            { object => $self, params => \%field_values });
342 343
}

344 345 346 347 348 349
sub update {
    my $self = shift;

    my $dbh      = Bugzilla->dbh;
    my $table    = $self->DB_TABLE;
    my $id_field = $self->ID_FIELD;
350

351 352
    $dbh->bz_start_transaction();

353
    my $old_self = $self->new($self->id);
354 355 356 357 358 359 360
   
    my @all_columns = $self->UPDATE_COLUMNS;
    my @hook_columns;
    Bugzilla::Hook::process('object_update_columns',
                            { object => $self, columns => \@hook_columns });
    push(@all_columns, @hook_columns);

361
    my %numeric = map { $_ => 1 } $self->NUMERIC_COLUMNS;
362
    my %date    = map { $_ => 1 } $self->DATE_COLUMNS;
363
    my (@update_columns, @values, %changes);
364
    foreach my $column (@all_columns) {
365 366 367 368 369 370
        my ($old, $new) = ($old_self->{$column}, $self->{$column});
        # This has to be written this way in order to allow us to set a field
        # from undef or to undef, and avoid warnings about comparing an undef
        # with the "eq" operator.
        if (!defined $new || !defined $old) {
            next if !defined $new && !defined $old;
371
        }
372 373 374
        elsif ( ($numeric{$column} && $old == $new) 
                || ($date{$column} && str2time($old) == str2time($new))
                || $old eq $new ) {
375 376 377 378 379 380 381 382 383
            next;
        }

        trick_taint($new) if defined $new;
        push(@values, $new);
        push(@update_columns, $column);
        # We don't use $new because we don't want to detaint this for
        # the caller.
        $changes{$column} = [$old, $self->{$column}];
384 385
    }

386 387
    my $columns = join(', ', map {"$_ = ?"} @update_columns);

388
    $dbh->do("UPDATE $table SET $columns WHERE $id_field = ?", undef, 
389 390
             @values, $self->id) if @values;

391
    Bugzilla::Hook::process('object_end_of_update',
392 393
                            { object => $self, old_object => $old_self,
                              changes => \%changes });
394

395 396
    $dbh->bz_commit_transaction();

397 398 399 400
    if (wantarray) {
        return (\%changes, $old_self);
    }

401
    return \%changes;
402 403
}

404 405
sub remove_from_db {
    my $self = shift;
406
    Bugzilla::Hook::process('object_before_delete', { object => $self });
407 408 409 410 411 412 413
    my $table = $self->DB_TABLE;
    my $id_field = $self->ID_FIELD;
    Bugzilla->dbh->do("DELETE FROM $table WHERE $id_field = ?",
                      undef, $self->id);
    undef $self;
}

414 415 416 417
###############################
####      Subroutines    ######
###############################

418 419 420 421 422 423 424 425 426
sub any_exist {
    my $class = shift;
    my $table = $class->DB_TABLE;
    my $dbh = Bugzilla->dbh;
    my $any_exist = $dbh->selectrow_array(
        "SELECT 1 FROM $table " . $dbh->sql_limit(1));
    return $any_exist ? 1 : 0;
}

427 428 429 430
sub create {
    my ($class, $params) = @_;
    my $dbh = Bugzilla->dbh;

431
    $dbh->bz_start_transaction();
432 433
    $class->check_required_create_fields($params);
    my $field_values = $class->run_create_validators($params);
434 435 436 437
    my $object = $class->insert_create_data($field_values);
    $dbh->bz_commit_transaction();

    return $object;
438 439
}

440 441 442 443 444 445 446 447 448 449 450
# Used to validate that a field name is in fact a valid column in the
# current table before inserting it into SQL.
sub _check_field {
    my ($invocant, $field, $function) = @_;
    my $class = ref($invocant) || $invocant;
    if (!Bugzilla->dbh->bz_column_info($class->DB_TABLE, $field)) {
        ThrowCodeError('param_invalid', { param    => $field,
                                          function => "${class}::$function" });
    }
}

451 452 453
sub check_required_create_fields {
    my ($class, $params) = @_;

454 455
    # This hook happens here so that even subclasses that don't call
    # SUPER::create are still affected by the hook.
456
    Bugzilla::Hook::process('object_before_create', { class => $class,
457 458
                                                      params => $params });

459 460 461
    my @check_fields = $class->_required_create_fields();
    foreach my $field (@check_fields) {
        $params->{$field} = undef if !exists $params->{$field};
462
    }
463 464 465
}

sub run_create_validators {
466
    my ($class, $params, $options) = @_;
467

468
    my $validators = $class->_get_validators;
469
    my %field_values = %$params;
470

471 472 473 474
    # Make a hash skiplist for easier searching later
    my %skip_list = map { $_ => 1 } @{ $options->{skip} || [] };

    # Get the sorted field names
475
    my @sorted_names = $class->_sort_by_dep(keys %field_values);
476 477 478 479 480

    # Remove the skipped names
    my @unskipped = grep { !$skip_list{$_} } @sorted_names;

    foreach my $field (@unskipped) {
481 482
        my $value;
        if (exists $validators->{$field}) {
483
            my $validator = $validators->{$field};
484 485
            $value = $class->$validator($field_values{$field}, $field,
                                        \%field_values);
486 487
        }
        else {
488
            $value = $field_values{$field};
489
        }
490

491 492
        # We want people to be able to explicitly set fields to NULL,
        # and that means they can be set to undef.
493 494 495 496
        trick_taint($value) if defined $value && !ref($value);
        $field_values{$field} = $value;
    }

497
    Bugzilla::Hook::process('object_end_of_create_validators',
498 499
                            { class => $class, params => \%field_values });

500 501 502 503 504 505 506 507 508
    return \%field_values;
}

sub insert_create_data {
    my ($class, $field_values) = @_;
    my $dbh = Bugzilla->dbh;

    my (@field_names, @values);
    while (my ($field, $value) = each %$field_values) {
509
        $class->_check_field($field, 'create');
510 511 512 513
        push(@field_names, $field);
        push(@values, $value);
    }

514 515 516 517 518 519
    my $qmarks = '?,' x @field_names;
    chop($qmarks);
    my $table = $class->DB_TABLE;
    $dbh->do("INSERT INTO $table (" . join(', ', @field_names)
             . ") VALUES ($qmarks)", undef, @values);
    my $id = $dbh->bz_last_key($table, $class->ID_FIELD);
520 521 522 523 524 525

    my $object = $class->new($id);

    Bugzilla::Hook::process('object_end_of_create', { class => $class,
                                                      object => $object });
    return $object;
526 527
}

528 529
sub get_all {
    my $class = shift;
530
    return @{$class->_do_list_select()};
531 532
}

533 534 535 536 537 538
###############################
####      Validators     ######
###############################

sub check_boolean { return $_[1] ? 1 : 0 }

539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556
sub check_time {
    my ($invocant, $value, $field, $params, $allow_negative) = @_;

    # If we don't have a current value default to zero
    my $current = blessed($invocant) ? $invocant->{$field}
                                     : 0;
    $current ||= 0;

    # Get the new value or zero if it isn't defined
    $value = trim($value) || 0;

    # Make sure the new value is well formed
    _validate_time($value, $field, $allow_negative);

    return $value;
}


557 558 559 560
###################
# General Helpers #
###################

561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583
sub _validate_time {
    my ($time, $field, $allow_negative) = @_;

    # regexp verifies one or more digits, optionally followed by a period and
    # zero or more digits, OR we have a period followed by one or more digits
    # (allow negatives, though, so people can back out errors in time reporting)
    if ($time !~ /^-?(?:\d+(?:\.\d*)?|\.\d+)$/) {
        ThrowUserError("number_not_numeric",
                       {field => $field, num => "$time"});
    }

    # Callers can optionally allow negative times
    if ( ($time < 0) && !$allow_negative ) {
        ThrowUserError("number_too_small",
                       {field => $field, num => "$time", min_num => "0"});
    }

    if ($time > 99999.99) {
        ThrowUserError("number_too_large",
                       {field => $field, num => "$time", max_num => "99999.99"});
    }
}

584 585 586 587 588 589 590
# Sorts fields according to VALIDATOR_DEPENDENCIES. This is not a
# traditional topological sort, because a "dependency" does not
# *have* to be in the list--it just has to be earlier than its dependent
# if it *is* in the list.
sub _sort_by_dep {
    my ($invocant, @fields) = @_;

591
    my $dependencies = $invocant->VALIDATOR_DEPENDENCIES;
592 593 594 595 596
    my ($has_deps, $no_deps) = part { $dependencies->{$_} ? 0 : 1 } @fields;

    # For fields with no dependencies, we sort them alphabetically,
    # so that validation always happens in a consistent order.
    # Fields with no dependencies come at the start of the list.
597
    my @result = sort @{ $no_deps || [] };
598 599 600 601 602

    # Fields with dependencies all go at the end of the list, and if
    # they have dependencies on *each other*, then they have to be
    # sorted properly. We go through $has_deps in sorted order to be
    # sure that fields always validate in a consistent order.
603
    foreach my $field (sort @{ $has_deps || [] }) {
604 605 606
        if (!grep { $_ eq $field } @result) {
            _insert_dep_field($field, $has_deps, $dependencies, \@result);
        }
607
    }
608 609
    return @result;
}
610

611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636
sub _insert_dep_field {
    my ($field, $insert_me, $dependencies, $result, $loop_tracking) = @_;

    if ($loop_tracking->{$field}) {
        ThrowCodeError('object_dep_sort_loop', 
                       { field => $field, 
                         considered => [keys %$loop_tracking] });
    }
    $loop_tracking->{$field} = 1;

    my $required_fields = $dependencies->{$field};
    # Imagine Field A requires field B...
    foreach my $required_field (@$required_fields) {
        # If our dependency is already satisfied, we're good.
        next if grep { $_ eq $required_field } @$result;

        # If our dependency is not in the remaining fields to insert,
        # then we're also OK.
        next if !grep { $_ eq $required_field } @$insert_me;

        # So, at this point, we know that Field B is in $insert_me.
        # So let's put the required field into the result.
        _insert_dep_field($required_field, $insert_me, $dependencies,
                          $result, $loop_tracking);
    }
    push(@$result, $field);
637 638
}

639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665
####################
# Constant Helpers #
####################

# For some classes, some constants take time to generate, so we cache them
# and only access them through the below methods. This also allows certain
# hooks to only run once per request instead of multiple times on each
# page.

sub _get_db_columns {
    my $invocant = shift;
    my $class = ref($invocant) || $invocant;
    my $cache = Bugzilla->request_cache;
    my $cache_key = "object_${class}_db_columns";
    return @{ $cache->{$cache_key} } if $cache->{$cache_key};
    # Currently you can only add new columns using object_columns, not
    # remove or modify existing columns, because removing columns would
    # almost certainly cause Bugzilla to function improperly.
    my @add_columns;
    Bugzilla::Hook::process('object_columns',
                            { class => $class, columns => \@add_columns });
    my @columns = ($invocant->DB_COLUMNS, @add_columns);
    $cache->{$cache_key} = \@columns;
    return @{ $cache->{$cache_key} };
}

# This method is private and should only be called by Bugzilla::Object.
666 667 668 669 670 671 672 673 674 675 676 677 678 679 680
sub _get_validators {
    my $invocant = shift;
    my $class = ref($invocant) || $invocant;
    my $cache = Bugzilla->request_cache;
    my $cache_key = "object_${class}_validators";
    return $cache->{$cache_key} if $cache->{$cache_key};
    # We copy this into a hash so that the hook doesn't modify the constant.
    # (That could be bad in mod_perl.)
    my %validators = %{ $invocant->VALIDATORS };
    Bugzilla::Hook::process('object_validators', 
                            { class => $class, validators => \%validators });
    $cache->{$cache_key} = \%validators;
    return $cache->{$cache_key};
}

681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701
# These are all the fields that need to be checked, always, when
# calling create(), because they have no DEFAULT and they are marked
# NOT NULL.
sub _required_create_fields {
    my $class = shift;
    my $dbh = Bugzilla->dbh;
    my $table = $class->DB_TABLE;

    my @columns = $dbh->bz_table_columns($table);
    my @required;
    foreach my $column (@columns) {
        my $def = $dbh->bz_column_info($table, $column);
        if ($def->{NOTNULL} and !defined $def->{DEFAULT}
            # SERIAL fields effectively have a DEFAULT, but they're not
            # listed as having a DEFAULT in DB::Schema.
            and $def->{TYPE} !~ /serial/i) 
        {
            my $field = $class->REQUIRED_FIELD_MAP->{$column} || $column;
            push(@required, $field);
        }
    }
702
    push(@required, $class->EXTRA_REQUIRED_FIELDS);
703 704 705
    return @required;
}

706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751
1;

__END__

=head1 NAME

Bugzilla::Object - A base class for objects in Bugzilla.

=head1 SYNOPSIS

 my $object = new Bugzilla::Object(1);
 my $object = new Bugzilla::Object({name => 'TestProduct'});

 my $id          = $object->id;
 my $name        = $object->name;

=head1 DESCRIPTION

Bugzilla::Object is a base class for Bugzilla objects. You never actually
create a Bugzilla::Object directly, you only make subclasses of it.

Basically, Bugzilla::Object exists to allow developers to create objects
more easily. All you have to do is define C<DB_TABLE>, C<DB_COLUMNS>,
and sometimes C<LIST_ORDER> and you have a whole new object.

You should also define accessors for any columns other than C<name>
or C<id>.

=head1 CONSTANTS

Frequently, these will be the only things you have to define in your
subclass in order to have a fully-functioning object. C<DB_TABLE>
and C<DB_COLUMNS> are required.

=over

=item C<DB_TABLE>

The name of the table that these objects are stored in. For example,
for C<Bugzilla::Keyword> this would be C<keyworddefs>.

=item C<DB_COLUMNS>

The names of the columns that you want to read out of the database
and into this object. This should be an array.

752 753 754 755 756 757
I<Note>: Though normally you will never need to access this constant's data 
directly in your subclass, if you do, you should access it by calling the
C<_get_db_columns> method instead of accessing the constant directly. (The
only exception to this rule is calling C<SUPER::DB_COLUMNS> from within
your own C<DB_COLUMNS> subroutine in a subclass.)

758 759 760 761 762 763 764 765 766 767 768 769 770
=item C<NAME_FIELD>

The name of the column that should be considered to be the unique
"name" of this object. The 'name' is a B<string> that uniquely identifies
this Object in the database. Defaults to 'name'. When you specify 
C<{name => $name}> to C<new()>, this is the column that will be 
matched against in the DB.

=item C<ID_FIELD>

The name of the column that represents the unique B<integer> ID
of this object in the database. Defaults to 'id'.

771 772 773 774
=item C<LIST_ORDER>

The order that C<new_from_list> and C<get_all> should return objects
in. This should be the name of a database column. Defaults to
775
L</NAME_FIELD>.
776

777 778 779
=item C<VALIDATORS>

A hashref that points to a function that will validate each param to
780 781 782 783 784 785 786 787 788 789 790 791
L</create>. 

Validators are called both by L</create> and L</set>. When
they are called by L</create>, the first argument will be the name
of the class (what we normally call C<$class>).

When they are called by L</set>, the first argument will be
a reference to the current object (what we normally call C<$self>).

The second argument will be the value passed to L</create> or 
L</set>for that field. 

792 793 794
The third argument will be the name of the field being validated.
This may be required by validators which validate several distinct fields.

795 796 797 798
These functions should call L<Bugzilla::Error/ThrowUserError> if they fail.

The validator must return the validated value.

799 800 801 802 803 804 805 806
=item C<UPDATE_VALIDATORS>

This is just like L</VALIDATORS>, but these validators are called only
when updating an object, not when creating it. Any validator that appears
here must not appear in L</VALIDATORS>.

L<Bugzilla::Bug> has good examples in its code of when to use this.

807 808 809 810 811 812 813 814 815 816 817 818 819 820
=item C<VALIDATOR_DEPENDENCIES>

During L</create> and L</set_all>, validators are normally called in
a somewhat-random order.  If you need one field to be validated and set
before another field, this constant is how you do it, by saying that
one field "depends" on the value of other fields.

This is a hashref, where the keys are field names and the values are
arrayrefs of field names. You specify what fields a field depends on using
the arrayrefs. So, for example, to say that a C<component> field depends
on the C<product> field being set, you would do:

 component => ['product']

821 822 823 824 825
=item C<UPDATE_COLUMNS>

A list of columns to update when L</update> is called.
If a field can't be changed, it shouldn't be listed here. (For example,
the L</ID_FIELD> usually can't be updated.)
826

827 828 829 830 831 832 833 834 835
=item C<REQUIRED_FIELD_MAP>

This is a hashref that maps database column names to L</create> argument
names. You only need to specify values for fields where the argument passed
to L</create> has a different name in the database than it does in the
L</create> arguments. (For example, L<Bugzilla::Bug/create> takes a
C<product> argument, but the column name in the C<bugs> table is
C<product_id>.)

836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869
=item C<EXTRA_REQUIRED_FIELDS>

Normally, Bugzilla::Object automatically figures out which fields
are required for L</create>. It then I<always> runs those fields' validators,
even if those fields weren't passed as arguments to L</create>. That way,
any default values or required checks can be done for those fields by
the validators.

L</create> figures out which fields are required by looking for database
columns in the L</DB_TABLE> that are NOT NULL and have no DEFAULT set.
However, there are some fields that this check doesn't work for:

=over

=item *

Fields that have database defaults (or are marked NULL in the database)
but actually have different defaults specified by validators. (For example,
the qa_contact field in the C<bugs> table can be NULL, so it won't be
caught as being required. However, in reality it defaults to the
component's initial_qa_contact.)

=item * 

Fields that have defaults that should be set by validators, but are
actually stored in a table different from L</DB_TABLE> (like the "cc"
field for bugs, which defaults to the "initialcc" of the Component, but won't
be caught as a normal required field because it's in a separate table.) 

=back

Any field matching the above criteria needs to have its name listed in
this constant. For an example of use, see the code of L<Bugzilla::Bug>.

870 871 872 873 874 875 876 877
=item C<NUMERIC_COLUMNS>

When L</update> is called, it compares each column in the object to its
current value in the database. It only updates columns that have changed.

Any column listed in NUMERIC_COLUMNS is treated as a number, not as a string,
during these comparisons.

878 879 880 881 882 883
=item C<DATE_COLUMNS>

This is much like L</NUMERIC_COLUMNS>, except that it treats strings as
dates when being compared. So, for example, C<2007-01-01> would be
equal to C<2007-01-01 00:00:00>.

884 885 886 887
=back

=head1 METHODS

888 889
=head2 Constructors

890 891
=over

892
=item C<new>
893

894 895 896 897 898 899
=over

=item B<Description>

The constructor is used to load an existing object from the database,
by id or by name.
900

901
=item B<Params>
902

903 904 905 906
If you pass an integer, the integer is the id of the object, 
from the database, that we  want to read in. (id is defined
as the value in the L</ID_FIELD> column).

907
If you pass in a hashref, you can pass a C<name> key. The 
908
value of the C<name> key is the case-insensitive name of the object 
909 910 911
(from L</NAME_FIELD>) in the DB. You can also pass in an C<id> key
which will be interpreted as the id of the object you want (overriding the 
C<name> key).
912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933

B<Additional Parameters Available for Subclasses>

If you are a subclass of C<Bugzilla::Object>, you can pass
C<condition> and C<values> as hash keys, instead of the above.

C<condition> is a set of SQL conditions for the WHERE clause, which contain
placeholders.

C<values> is a reference to an array. The array contains the values
for each placeholder in C<condition>, in order.

This is to allow subclasses to have complex parameters, and then to
translate those parameters into C<condition> and C<values> when they
call C<$self->SUPER::new> (which is this function, usually).

If you try to call C<new> outside of a subclass with the C<condition>
and C<values> parameters, Bugzilla will throw an error. These parameters
are intended B<only> for use by subclasses.

=item B<Returns>

934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962
A fully-initialized object, or C<undef> if there is no object in the
database matching the parameters you passed in.

=back

=item C<check>

=over

=item B<Description>

Checks if there is an object in the database with the specified name, and
throws an error if you specified an empty name, or if there is no object
in the database with that name.

=item B<Params>

The parameters are the same as for L</new>, except that if you don't pass
a hashref, the single argument is the I<name> of the object, not the id.

=item B<Returns>

A fully initialized object, guaranteed.

=item B<Notes For Implementors>

If you implement this in your subclass, make sure that you also update
the C<object_name> block at the bottom of the F<global/user-error.html.tmpl>
template.
963 964

=back
965 966 967 968 969 970 971 972 973 974 975 976

=item C<new_from_list(\@id_list)>

 Description: Creates an array of objects, given an array of ids.

 Params:      \@id_list - A reference to an array of numbers, database ids.
                          If any of these are not numeric, the function
                          will throw an error. If any of these are not
                          valid ids in the database, they will simply 
                          be skipped.

 Returns:     A reference to an array of objects.
977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000

=item C<match>

=over

=item B<Description>

Gets a list of objects from the database based on certain criteria.

Basically, a simple way of doing a sort of "SELECT" statement (like SQL)
to get objects.

All criteria are joined by C<AND>, so adding more criteria will give you
a smaller set of results, not a larger set.

=item B<Params>

A hashref, where the keys are column names of the table, pointing to the 
value that you want to match against for that column. 

There are two special values, the constants C<NULL> and C<NOT_NULL>,
which means "give me objects where this field is NULL or NOT NULL,
respectively."

1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015
In addition to the column keys, there are a few special keys that
can be used to rig the underlying database queries. These are 
C<LIMIT>, C<OFFSET>, and C<WHERE>.

The value for the C<LIMIT> key is expected to be an integer defining 
the number of objects to return, while the value for C<OFFSET> defines
the position, relative to the number of objects the query would normally 
return, at which to begin the result set. If C<OFFSET> is defined without 
a corresponding C<LIMIT> it is silently ignored.

The C<WHERE> key provides a mechanism for adding arbitrary WHERE
clauses to the underlying query. Its value is expected to a hash 
reference whose keys are the columns, operators and placeholders, and the 
values are the placeholders' bind value. For example:

1016
 WHERE => { 'some_column >= ?' => $some_value }
1017

1018 1019 1020
would constrain the query to only those objects in the table whose
'some_column' column has a value greater than or equal to $some_value.

1021 1022 1023 1024 1025 1026 1027 1028
If you don't specify any criteria, calling this function is the same
as doing C<[$class-E<gt>get_all]>.

=item B<Returns>

An arrayref of objects, or an empty arrayref if there are no matches.

=back
1029 1030 1031

=back

1032
=head2 Database Manipulation
1033 1034 1035

=over

1036
=item C<create>
1037 1038 1039 1040 1041 1042

Description: Creates a new item in the database.
             Throws a User Error if any of the passed-in params
             are invalid.

Params:      C<$params> - hashref - A value to put in each database
1043
             field for this object.
1044 1045 1046 1047

Returns:     The Object just created in the database.

Notes:       In order for this function to work in your subclass,
1048
             your subclass's L</ID_FIELD> must be of C<SERIAL>
1049
             type in the database.
1050

1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063
             Subclass Implementors: This function basically just
             calls L</check_required_create_fields>, then
             L</run_create_validators>, and then finally
             L</insert_create_data>. So if you have a complex system that
             you need to implement, you can do it by calling these
             three functions instead of C<SUPER::create>.

=item C<check_required_create_fields>

=over

=item B<Description>

1064 1065 1066 1067
Part of L</create>. Modifies the incoming C<$params> argument so that
any field that does not have a database default will be checked
later by L</run_create_validators>, even if that field wasn't specified
as an argument to L</create>.
1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081

=item B<Params>

=over

=item C<$params> - The same as C<$params> from L</create>.

=back

=item B<Returns> (nothing)

=back

=item C<run_create_validators>
1082 1083 1084 1085 1086 1087 1088

Description: Runs the validation of input parameters for L</create>.
             This subroutine exists so that it can be overridden
             by subclasses who need to do special validations
             of their input parameters. This method is B<only> called
             by L</create>.

1089 1090 1091 1092 1093
Params:      C<$params> - hashref - A value to put in each database
             field for this object.
             C<$options> - hashref - Processing options. Currently
             the only option supported is B<skip>, which can be
             used to specify a list of fields to not validate.
1094

1095 1096 1097 1098 1099 1100 1101 1102 1103 1104
Returns:     A hash, in a similar format as C<$params>, except that
             these are the values to be inserted into the database,
             not the values that were input to L</create>.

=item C<insert_create_data>

Part of L</create>.

Takes the return value from L</run_create_validators> and inserts the
data into the database. Returns a newly created object. 
1105

1106 1107
=item C<update>

1108 1109 1110 1111
=over

=item B<Description>

1112 1113
Saves the values currently in this object to the database.
Only the fields specified in L</UPDATE_COLUMNS> will be
1114 1115 1116 1117 1118 1119
updated, and they will only be updated if their values have changed.

=item B<Params> (none)

=item B<Returns>

1120 1121
B<In scalar context:>

1122 1123 1124 1125 1126 1127 1128 1129
A hashref showing what changed during the update. The keys are the column
names from L</UPDATE_COLUMNS>. If a field was not changed, it will not be
in the hash at all. If the field was changed, the key will point to an arrayref.
The first item of the arrayref will be the old value, and the second item
will be the new value.

If there were no changes, we return a reference to an empty hash.

1130 1131 1132 1133 1134 1135 1136
B<In array context:>

Returns a list, where the first item is the above hashref. The second item
is the object as it was in the database before update() was called. (This
is mostly useful to subclasses of C<Bugzilla::Object> that are implementing
C<update>.)

1137
=back
1138

1139 1140 1141 1142 1143 1144
=item C<remove_from_db>

Removes this object from the database. Will throw an error if you can't
remove it for some reason. The object will then be destroyed, as it is
not safe to use the object after it has been removed from the database.

1145 1146
=back

1147
=head2 Mutators
1148

1149 1150
These are used for updating the values in objects, before calling
C<update>.
1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165

=over

=item C<set>

=over

=item B<Description>

Sets a certain hash member of this class to a certain value.
Used for updating fields. Calls the validator for this field,
if it exists. Subclasses should use this function
to implement the various C<set_> mutators for their different
fields.

1166 1167 1168 1169 1170
If your class defines a method called C<_set_global_validator>,
C<set> will call it with C<($value, $field)> as arguments, after running
the validator for this particular field. C<_set_global_validator> does not
return anything.

1171 1172
See L</VALIDATORS> for more information.

1173 1174 1175
B<NOTE>: This function is intended only for use by subclasses. If
you call it from anywhere else, it will throw a C<CodeError>.

1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190
=item B<Params>

=over

=item C<$field> - The name of the hash member to update. This should
be the same as the name of the field in L</VALIDATORS>, if it exists there.

=item C<$value> - The value that you're setting the field to.

=back

=item B<Returns> (nothing)

=back

1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211

=item C<set_all>

=over

=item B<Description>

This is a convenience function which is simpler than calling many different
C<set_> functions in a row. You pass a hashref of parameters and it calls
C<set_$key($value)> for every item in the hashref.

=item B<Params>

Takes a hashref of the fields that need to be set, pointing to the value
that should be passed to the C<set_> function that is called.

=item B<Returns> (nothing)

=back


1212 1213
=back

1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227
=head2 Simple Validators

You can use these in your subclass L</VALIDATORS> or L</UPDATE_VALIDATORS>.
Note that you have to reference them like C<\&Bugzilla::Object::check_boolean>,
you can't just write C<\&check_boolean>.

=over

=item C<check_boolean>

Returns C<1> if the passed-in value is true, C<0> otherwise.

=back

1228 1229 1230 1231
=head1 CLASS FUNCTIONS

=over

1232 1233 1234 1235 1236
=item C<any_exist>

Returns C<1> if there are any of these objects in the database,
C<0> otherwise.

1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251
=item C<get_all>

 Description: Returns all objects in this table from the database.

 Params:      none.

 Returns:     A list of objects, or an empty list if there are none.

 Notes:       Note that you must call this as C<$class->get_all>. For 
              example, C<Bugzilla::Keyword->get_all>. 
              C<Bugzilla::Keyword::get_all> will not work.

=back

=cut