diff options
Diffstat (limited to 'scripts')
| -rw-r--r-- | scripts/undo.pl | 203 |
1 files changed, 203 insertions, 0 deletions
diff --git a/scripts/undo.pl b/scripts/undo.pl new file mode 100644 index 000000000..852c786fd --- /dev/null +++ b/scripts/undo.pl @@ -0,0 +1,203 @@ +#!/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 5.10.1; +use strict; +use warnings; + +use File::Basename qw(dirname); +use File::Spec::Functions qw(catfile catdir rel2abs); +use Cwd qw(realpath); +BEGIN { + require lib; + my $dir = rel2abs(catdir(dirname(__FILE__), '..')); + lib->import($dir, catdir($dir, 'lib'), catdir($dir, qw(local lib perl5))); +} + +use Bugzilla; +BEGIN { Bugzilla->extensions }; +use Bugzilla::Extension::UserProfile::Util qw(tag_for_recount_from_bug); + +use Try::Tiny; + +# not involved with kmag +my $query = q{ + resolution = 'INACTIVE' + AND NOT ( + triage_owner.login_name = 'mak77@bonardo.net' + AND bugs.priority IN ('P4', 'P5', '--') + ) + AND NOT ( + product.name = 'Toolkit' AND component.name = 'Add-ons Manager' + ) + AND NOT ( + product.name = 'Toolkit' AND component.name LIKE 'WebExtensions%' + ) + AND NOT ( + product.name = 'Core' AND component.name = 'XPConnect' + ) +}; +undo($query); + +sub undo { + my @args = @_; + my $changes = get_changes(@args); + my $comments = get_comments(@args); + + my %action; + while ($_ = $changes->()) { + push @{ $action{$_->{bug_id}}{$_->{bug_when}}{remove_activities} }, { id => $_->{change_id} }; + $action{ $_->{bug_id} }{ $_->{bug_when} }{change}{ $_->{field_name} } = { + replace => $_->{added}, + with => $_->{removed}, + }; + } + + while ($_ = $comments->()) { + push @{ $action{ $_->{bug_id} }{$_->{bug_when}}{remove_comments} }, { + id => $_->{comment_id}, + }; + } + + my $dbh = Bugzilla->dbh; + my @bug_ids = sort { $b <=> $a } keys %action; + say 'Found ', 0 + @bug_ids, ' bugs'; + foreach my $bug_id (@bug_ids) { + $dbh->bz_start_transaction; + say "Fix $bug_id"; + try { + my ($delta_ts) = $dbh->selectrow_array( + 'SELECT delta_ts FROM bugs WHERE bug_id = ?', + undef, + $bug_id); + my ($previous_last_ts) = $dbh->selectrow_array( + q{ + SELECT MAX(bug_when) FROM ( + SELECT bug_when FROM bugs_activity WHERE bug_id = ? AND bug_when < ? + UNION + SELECT bug_when FROM longdescs WHERE bug_id = ? AND bug_when < ? + UNION + SELECT creation_ts AS bug_when FROM bugs WHERE bug_id = ? + ) as changes ORDER BY bug_when DESC + }, + undef, + $bug_id, $delta_ts, + $bug_id, $delta_ts, + $bug_id, + ); + die 'cannot find previous last updated time' unless $previous_last_ts; + my $action = delete $action{$bug_id}{$delta_ts}; + if (keys %{ $action{$bug_id}{$delta_ts}}) { + die "skipping because more than one change\n"; + } + elsif (!$action) { + die "skipping because most recent change newer than automation change\n"; + } + foreach my $field (keys %{ $action->{change} }) { + my $change = $action->{change}{$field}; + if ($field eq 'cf_last_resolved' && !$change->{with}) { + $change->{with} = undef; + } + my $did = $dbh->do( + "UPDATE bugs SET $field = ? WHERE bug_id = ? AND $field = ?", + undef, $change->{with}, $bug_id, $change->{replace}, + ); + die "Failed to set $field to $change->{with}" unless $did; + } + $dbh->do('UPDATE bugs SET delta_ts = ?, lastdiffed = ? WHERE bug_id = ?', undef, + $previous_last_ts, $previous_last_ts, $bug_id ); + my $del_comments = $dbh->prepare('DELETE FROM longdescs WHERE comment_id = ?'); + my $del_activity = $dbh->prepare('DELETE FROM bugs_activity WHERE id = ?'); + foreach my $c (@{ $action->{remove_comments}}) { + $del_comments->execute($c->{id}) or die "failed to delete comment $c->{id}"; + } + foreach my $a (@{ $action->{remove_activities}}) { + $del_activity->execute($a->{id}) or die "failed to delete comment $a->{id}"; + } + tag_for_recount_from_bug($bug_id); + $dbh->bz_commit_transaction; + sleep 1; + } catch { + chomp $_; + say "Error updating $bug_id: $_"; + $dbh->bz_rollback_transaction; + }; + } + say 'Done.'; +} + +sub get_changes { + my ($where, @bind) = @_; + + my $sql = qq{ + SELECT + BA.id AS change_id, + BA.bug_id, + FD.name AS field_name, + BA.removed, + BA.added, + BA.bug_when + FROM + bugs_activity AS BA + JOIN + fielddefs AS FD ON BA.fieldid = FD.id + JOIN + profiles AS changer ON changer.userid = BA.who + JOIN + (SELECT + bug_id + FROM + bugs + JOIN products AS product ON product.id = product_id + JOIN components AS component ON component.id = component_id + LEFT JOIN profiles AS triage_owner ON triage_owner.userid = component.triage_owner_id + WHERE + $where + ) target_bugs ON BA.bug_id = target_bugs.bug_id + WHERE + changer.login_name = 'automation\@bmo.tld' + AND BA.bug_when BETWEEN '2018-05-22' AND '2018-05-26' + }; + my $sth = Bugzilla->dbh->prepare($sql); + $sth->execute(@bind); + + return sub { $sth->fetchrow_hashref }; +} + +sub get_comments { + my ($where, @bind) = @_; + + my $sql = qq{ + SELECT + C.comment_id AS comment_id, + C.bug_id AS bug_id, + C.bug_when + FROM + longdescs AS C + JOIN + profiles AS commenter ON commenter.userid = C.who + JOIN + (SELECT + bug_id + FROM + bugs + JOIN products AS product ON product.id = product_id + JOIN components AS component ON component.id = component_id + LEFT JOIN profiles AS triage_owner ON triage_owner.userid = component.triage_owner_id + WHERE + $where + ) target_bugs ON C.bug_id = target_bugs.bug_id + WHERE + commenter.login_name = 'automation\@bmo.tld' + AND C.bug_when BETWEEN '2018-05-22' AND '2018-05-26' + }; + my $sth = Bugzilla->dbh->prepare($sql); + $sth->execute(@bind); + + return sub { $sth->fetchrow_hashref }; +} |