summaryrefslogtreecommitdiffstats
path: root/extensions/BMO/bin/migrate-github-pull-requests.pl
diff options
context:
space:
mode:
Diffstat (limited to 'extensions/BMO/bin/migrate-github-pull-requests.pl')
-rwxr-xr-xextensions/BMO/bin/migrate-github-pull-requests.pl90
1 files changed, 90 insertions, 0 deletions
diff --git a/extensions/BMO/bin/migrate-github-pull-requests.pl b/extensions/BMO/bin/migrate-github-pull-requests.pl
new file mode 100755
index 000000000..de71a7856
--- /dev/null
+++ b/extensions/BMO/bin/migrate-github-pull-requests.pl
@@ -0,0 +1,90 @@
+#!/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;
+BEGIN { Bugzilla->extensions() }
+
+use Bugzilla::Extension::BMO::Data;
+use Bugzilla::Field;
+use Bugzilla::Install::Util qw(indicate_progress);
+use Bugzilla::User;
+use Bugzilla::Util qw(trim);
+
+my $dbh = Bugzilla->dbh;
+my $nobody = Bugzilla::User->check({ name => 'nobody@mozilla.org' });
+my $field = Bugzilla::Field->check({ name => 'attachments.mimetype' });
+
+# grab list of suitable attachments
+
+my $sql = <<EOF;
+SELECT attachments.attach_id,
+ attachments.bug_id,
+ attachments.mimetype,
+ attach_data.thedata
+ FROM attachments
+ INNER JOIN attach_data ON attach_data.id = attachments.attach_id
+ WHERE ispatch = 0
+ AND mimetype = 'text/plain'
+ AND thedata IS NOT NULL
+ AND LENGTH(thedata) > 0
+ AND LENGTH(thedata) <= 256
+EOF
+print "Searching for suitable attachments..\n";
+my $attachments = $dbh->selectall_arrayref($sql, { Slice => {} });
+my ($current, $total, $updated) = (1, scalar(@$attachments), 0);
+
+die "No suitable attachments found\n" unless $total;
+print "About to check $total attachments for github pull requests, and\n";
+print "update content-type if required.\n";
+print "Press <enter> to start, or ^C to cancel...\n";
+<>;
+
+foreach my $attachment (@$attachments) {
+ indicate_progress({ current => $current++, total => $total, every => 25 });
+
+ # check payload
+ my $url = trim($attachment->{thedata});
+ next if $url =~ /\s/;
+ next unless $url =~ m#^https://github\.com/[^/]+/[^/]+/pull/\d+\/?$#i;
+
+ $dbh->bz_start_transaction;
+
+ # set content-type
+ $dbh->do(
+ "UPDATE attachments SET mimetype = ? WHERE attach_id = ?",
+ undef,
+ GITHUB_PR_CONTENT_TYPE, $attachment->{attach_id}
+ );
+
+ # insert into bugs_activity
+ my $timestamp = $dbh->selectrow_array('SELECT LOCALTIMESTAMP(0)');
+ $dbh->do(
+ "INSERT INTO bugs_activity(bug_id, who, bug_when, fieldid, removed, added)
+ VALUES (?, ?, ?, ?, ?, ?)",
+ undef,
+ $attachment->{bug_id}, $nobody->id, $timestamp, $field->id,
+ $attachment->{mimetype}, GITHUB_PR_CONTENT_TYPE
+ );
+ $dbh->do(
+ "UPDATE bugs SET delta_ts = ?, lastdiffed = ? WHERE bug_id = ?",
+ undef,
+ $timestamp, $timestamp, $attachment->{bug_id}
+ );
+
+ $dbh->bz_commit_transaction;
+ $updated++;
+}
+
+print "Attachments updated: $updated\n";