summaryrefslogtreecommitdiffstats
path: root/scripts/migrate-cab-review.pl
blob: 2d6a0c209257054d2639143dad0c3adfdbca647e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
#!/usr/bin/perl

# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
#
# This Source Code Form is "Incompatible With Secondary Licenses", as
# defined by the Mozilla Public License, v. 2.0.

use strict;
use warnings;

use FindBin qw($RealBin);
use lib "$RealBin/../../..";

use Bugzilla;
use Bugzilla::Bug;
use Bugzilla::Constants;
use Bugzilla::Field;
use Bugzilla::Group;
use Bugzilla::Install::Util qw(indicate_progress);
use Bugzilla::User;

Bugzilla->usage_mode(USAGE_MODE_CMDLINE);

my $dbh = Bugzilla->dbh;

# Make all changes as the automation user
my $auto_user = Bugzilla::User->check({ name => 'automation@bmo.tld' });
$auto_user->{groups} = [ Bugzilla::Group->get_all ];
$auto_user->{bless_groups} = [ Bugzilla::Group->get_all ];
Bugzilla->set_user($auto_user);

# cab-review flag values to custom field values mapping
my %map = (
    '?' => '?',
    '+' => 'approved',
    '-' => 'denied'
);

# Verify that all of the custom field values in the mapping data are present
my $cab_field = Bugzilla::Field->check({ name => 'cf_cab_review' });
foreach my $value (values %map) {
    unless (grep($_ eq $value, map { $_->name } @{ $cab_field->legal_values })) {
        die "Value '$value' does not exist. Please add to 'cf_cab_review.\n";
    }
}

# Grab list of bugs with cab-review flag set
my $sql = <<EOF;
    SELECT bugs.bug_id
      FROM bugs JOIN flags ON bugs.bug_id = flags.bug_id
           JOIN flagtypes ON flags.type_id = flagtypes.id
     WHERE flagtypes.name = 'cab-review' AND flags.status IN ('?','+', '-')
     ORDER BY bug_id
EOF

print "Searching for matching bugs..\n";
my $bugs = $dbh->selectcol_arrayref($sql);
my ($current, $total, $updated) = (1, scalar(@$bugs), 0);

die "No matching bugs found\n" unless $total;
print "About to update $total bugs with cab-review flags set and migrate to the cf_cab_review custom field.\n";
print "Press <enter> to start, or ^C to cancel...\n";
<>;

foreach my $bug_id (@$bugs) {
    indicate_progress({ current => $current++, total => $total, every => 5 });

    # Load bug object
    my $bug = Bugzilla::Bug->new($bug_id);

    # Find the current cab-review status
    my $cab_flag;
    foreach my $flag (@{ $bug->flags }) {
        next if $flag->type->name ne 'cab-review';
        $cab_flag = $flag;
        last;
    }

    my $timestamp = $dbh->selectrow_array('SELECT LOCALTIMESTAMP(0)');

    $dbh->bz_start_transaction;

    # Set the cab-review custom field to the right status based on the mapped values
    $bug->set_custom_field($cab_field, $map{$cab_flag->status});

    # Clear the old cab-review flag
    $bug->set_flags([{ id => $cab_flag->id, status => 'X' }], []);

    # Update the bug
    $bug->update($timestamp);

    # Do not send email about this change
    $dbh->do("UPDATE bugs SET delta_ts = ?, lastdiffed = ? WHERE bug_id = ?",
        undef, $timestamp, $timestamp, $bug_id);

    $dbh->bz_commit_transaction;
    $updated++;
}

print "Bugs updated: $updated\n";