summaryrefslogtreecommitdiffstats
path: root/extensions/PhabBugz/lib
diff options
context:
space:
mode:
Diffstat (limited to 'extensions/PhabBugz/lib')
-rw-r--r--extensions/PhabBugz/lib/Util.pm277
-rw-r--r--extensions/PhabBugz/lib/WebService.pm112
2 files changed, 389 insertions, 0 deletions
diff --git a/extensions/PhabBugz/lib/Util.pm b/extensions/PhabBugz/lib/Util.pm
new file mode 100644
index 000000000..d86cd01a5
--- /dev/null
+++ b/extensions/PhabBugz/lib/Util.pm
@@ -0,0 +1,277 @@
+# 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.
+
+package Bugzilla::Extension::PhabBugz::Util;
+
+use 5.10.1;
+use strict;
+use warnings;
+
+use Bugzilla::Error;
+
+use Data::Dumper;
+use JSON qw(encode_json decode_json);
+use LWP::UserAgent;
+
+use base qw(Exporter);
+
+our @EXPORT = qw(
+ create_revision_attachment
+ create_private_revision_policy
+ create_project
+ edit_revision_policy
+ get_bug_role_phids
+ get_members_by_bmo_id
+ get_project_phid
+ get_revision_by_id
+ intersect
+ make_revision_public
+ request
+ set_project_members
+);
+
+sub get_revision_by_id {
+ my $id = shift;
+
+ my $data = {
+ queryKey => 'all',
+ constraints => {
+ ids => [ int($id) ]
+ }
+ };
+
+ my $result = request('differential.revision.search', $data);
+
+ ThrowUserError('invalid_phabricator_revision_id')
+ unless (exists $result->{result}{data} && @{ $result->{result}{data} });
+
+ return $result->{result}{data}[0];
+}
+
+sub create_revision_attachment {
+ my ($bug, $revision_id, $revision_title) = @_;
+
+ my $dbh = Bugzilla->dbh;
+ $dbh->bz_start_transaction;
+
+ my ($timestamp) = $dbh->selectrow_array("SELECT NOW()");
+
+ my $attachment = Bugzilla::Attachment->create({
+ bug => $bug,
+ creation_ts => $timestamp,
+ data => 'http://phabricator.test/D' . $revision_id,
+ description => $revision_title,
+ filename => 'phabricator-D' . $revision_id . '-url.txt',
+ ispatch => 0,
+ isprivate => 0,
+ mimetype => 'text/x-phabricator-request',
+ });
+
+ $bug->update($timestamp);
+ $attachment->update($timestamp);
+
+ $dbh->bz_commit_transaction;
+
+ return $attachment;
+}
+
+sub intersect {
+ my ($list1, $list2) = @_;
+ my %e = map { $_ => undef } @{$list1};
+ return grep { exists( $e{$_} ) } @{$list2};
+}
+
+sub get_bug_role_phids {
+ my ($bug) = @_;
+
+ my @bug_users = ( $bug->reporter );
+ push(@bug_users, $bug->assigned_to)
+ if !$bug->assigned_to->email !~ /^nobody\@mozilla\.org$/;
+ push(@bug_users, $bug->qa_contact) if $bug->qa_contact;
+ push(@bug_users, @{ $bug->cc_users }) if @{ $bug->cc_users };
+
+ return get_members_by_bmo_id(\@bug_users);
+}
+
+sub create_private_revision_policy {
+ my ($bug, $groups) = @_;
+
+ my $project_phids = [];
+ foreach my $group (@$groups) {
+ my $phid = get_project_phid('bmo-' . $group);
+ push(@$project_phids, $phid) if $phid;
+ }
+
+ ThrowUserError('invalid_phabricator_sync_groups') unless @$project_phids;
+
+ my $data = {
+ objectType => 'DREV',
+ default => 'deny',
+ policy => [
+ {
+ action => 'allow',
+ rule => 'PhabricatorProjectsPolicyRule',
+ value => $project_phids,
+ },
+ {
+ action => 'allow',
+ rule => 'PhabricatorSubscriptionsSubscribersPolicyRule',
+ }
+ ]
+ };
+
+ my $result = request('policy.create', $data);
+ return $result->{result}{phid};
+}
+
+sub make_revision_public {
+ my ($revision_phid) = @_;
+ return request('differential.revision.edit', {
+ transactions => [
+ {
+ type => "view",
+ value => "users"
+ }
+ ],
+ objectIdentifier => $revision_phid
+ });
+}
+
+sub edit_revision_policy {
+ my ($revision_phid, $policy_phid, $subscribers) = @_;
+
+ my $data = {
+ transactions => [
+ {
+ type => 'view',
+ value => $policy_phid
+ },
+ {
+ type => 'edit',
+ value => $policy_phid
+ }
+ ],
+ objectIdentifier => $revision_phid
+ };
+
+ if (@$subscribers) {
+ push(@{ $data->{transactions} }, {
+ type => 'subscribers.add',
+ value => $subscribers
+ });
+ }
+
+ return request('differential.revision.edit', $data);
+}
+
+sub get_project_phid {
+ my $project = shift;
+
+ my $data = {
+ queryKey => 'all',
+ constraints => {
+ name => $project
+ }
+ };
+
+ my $result = request('project.search', $data);
+ return undef
+ unless (exists $result->{result}{data} && @{ $result->{result}{data} });
+
+ return $result->{result}{data}[0]{phid};
+}
+
+sub create_project {
+ my ($project, $description, $members) = @_;
+
+ my $data = {
+ transactions => [
+ { type => 'name', value => $project },
+ { type => 'description', value => $description },
+ { type => 'edit', value => 'admin' },
+ { type => 'join', value => 'admin' },
+ { type => 'view', value => 'admin' },
+ { type => 'icon', value => 'group' },
+ { type => 'color', value => 'red' }
+ ]
+ };
+
+ my $result = request('project.edit', $data);
+ return $result->{result}{object}{phid};
+}
+
+sub set_project_members {
+ my ($project_id, $phab_user_ids) = @_;
+
+ my $data = {
+ objectIdentifier => $project_id,
+ transactions => [
+ { type => 'members.set', value => $phab_user_ids }
+ ]
+ };
+
+ my $result = request('project.edit', $data);
+ return $result->{result}{object}{phid};
+}
+
+sub get_members_by_bmo_id {
+ my $users = shift;
+
+ my $data = {
+ accountids => [ map { $_->id } @$users ]
+ };
+
+ my $result = request('bmoexternalaccount.search', $data);
+ return [] if (!$result->{result});
+
+ my @phab_ids;
+ foreach my $user (@{ $result->{result} }) {
+ push(@phab_ids, $user->{phid});
+ }
+
+ return \@phab_ids;
+}
+
+sub request {
+ my ($method, $data) = @_;
+ my $request_cache = Bugzilla->request_cache;
+ my $params = Bugzilla->params;
+
+ my $ua = $request_cache->{phabricato_ua};
+ unless ($ua) {
+ $ua = $request_cache->{phabricator_ua} = LWP::UserAgent->new(timeout => 10);
+ if ($params->{proxy_url}) {
+ $ua->proxy('https', $params->{proxy_url});
+ }
+ $ua->default_header('Content-Type' => 'application/x-www-form-urlencoded');
+ }
+
+ my $phab_api_key = $params->{phabricator_api_key};
+ my $phab_base_uri = $params->{phabricator_base_uri};
+ ThrowUserError('invalid_phabricator_uri') unless $phab_base_uri;
+ ThrowUserError('invalid_phabricator_api_key') unless $phab_api_key;
+
+ my $full_uri = $phab_base_uri . '/api/' . $method;
+
+ $data->{__conduit__} = { token => $phab_api_key };
+
+ my $response = $ua->post($full_uri, { params => encode_json($data) });
+
+ ThrowCodeError('phabricator_api_error', { reason => $response->message })
+ if $response->is_error;
+
+ my $result = decode_json($response->content);
+ if ($result->{error_code}) {
+ ThrowCodeError('phabricator_api_error',
+ { code => $result->{error_code},
+ reason => $result->{error_info} });
+ }
+
+ return $result;
+}
+
+1;
diff --git a/extensions/PhabBugz/lib/WebService.pm b/extensions/PhabBugz/lib/WebService.pm
new file mode 100644
index 000000000..0e2574582
--- /dev/null
+++ b/extensions/PhabBugz/lib/WebService.pm
@@ -0,0 +1,112 @@
+# 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.
+
+package Bugzilla::Extension::PhabBugz::WebService;
+
+use 5.10.1;
+use strict;
+use warnings;
+
+use base qw(Bugzilla::WebService);
+
+use Bugzilla::Attachment;
+use Bugzilla::Bug;
+use Bugzilla::BugMail;
+use Bugzilla::Error;
+use Bugzilla::User;
+use Bugzilla::Util qw(correct_urlbase detaint_natural);
+use Bugzilla::WebService::Constants;
+
+use Bugzilla::Extension::PhabBugz::Util qw(
+ create_revision_attachment
+ create_private_revision_policy
+ edit_revision_policy
+ get_bug_role_phids
+ get_project_phid
+ get_revision_by_id
+ intersect
+ make_revision_public
+ request
+);
+
+use Data::Dumper;
+
+use constant PUBLIC_METHODS => qw(
+ revision
+);
+
+sub revision {
+ my ($self, $params) = @_;
+
+ unless (defined $params->{revision} && detaint_natural($params->{revision})) {
+ ThrowCodeError('param_required', { param => 'revision' })
+ }
+
+ my $user = Bugzilla->set_user(Bugzilla::User->new({ name => 'conduit@mozilla.bugs' }));
+
+ # Obtain more information about the revision from Phabricator
+ my $revision_id = $params->{revision};
+ my $revision = get_revision_by_id($revision_id);
+
+ my $revision_phid = $revision->{phid};
+ my $revision_title = $revision->{fields}{title} || 'Unknown Description';
+ my $bug_id = $revision->{fields}{'bugzilla.bug-id'};
+
+ my $bug = Bugzilla::Bug->check($bug_id);
+
+ # If bug is public then remove privacy policy
+ my $result;
+ if (!@{ $bug->groups_in }) {
+ $result = make_revision_public($revision_id);
+ }
+ # Else bug is private
+ else {
+ my $phab_sync_groups = Bugzilla->params->{phabricator_sync_groups}
+ || ThrowUserError('invalid_phabricator_sync_groups');
+ my $sync_group_names = [ split('[,\s]+', $phab_sync_groups) ];
+
+ my $bug_groups = $bug->groups_in;
+ my $bug_group_names = [ map { $_->name } @$bug_groups ];
+
+ my @set_groups = intersect($bug_group_names, $sync_group_names);
+
+ # If bug privacy groups do not have any matching synchronized groups,
+ # then leave revision private and it will have be dealt with manually.
+ if (!@set_groups) {
+ ThrowUserError('invalid_phabricator_sync_groups');
+ }
+
+ my $policy_phid = create_private_revision_policy($bug, \@set_groups);
+ my $subscribers = get_bug_role_phids($bug);
+ $result = edit_revision_policy($revision_phid, $policy_phid, $subscribers);
+ }
+
+ my $attachment = create_revision_attachment($bug, $revision_id, $revision_title);
+
+ Bugzilla::BugMail::Send($bug_id, { changer => $user });
+
+ return {
+ result => $result,
+ attachment_id => $attachment->id,
+ attachment_link => correct_urlbase() . "attachment.cgi?id=" . $attachment->id
+ };
+}
+
+sub rest_resources {
+ return [
+ qr{^/phabbugz/revision/([^/]+)$}, {
+ POST => {
+ method => 'revision',
+ params => sub {
+ return { revision => $_[0] };
+ }
+ }
+ }
+ ];
+}
+
+1;