summaryrefslogtreecommitdiffstats
path: root/extensions/Push/lib/Connector/Phabricator.pm
blob: be0ea9b58cf62e2c4f1dc3144466e6d2a1f6de34 (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
103
104
105
106
107
108
109
110
111
# 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::Push::Connector::Phabricator;

use 5.10.1;
use strict;
use warnings;

use base 'Bugzilla::Extension::Push::Connector::Base';

use Bugzilla::Bug;
use Bugzilla::Constants;
use Bugzilla::Extension::PhabBugz::Util qw(intersect make_revision_public get_revisions_by_ids);
use Bugzilla::Extension::Push::Constants;
use Bugzilla::Extension::Push::Util;
use Bugzilla::User;
use List::Util qw(any);

use constant PHAB_CONTENT_TYPE => 'text/x-phabricator-request';
use constant PHAB_ATTACHMENT_PATTERN => qr/^phabricator-D(\d+)/;

sub options {
    return (
        {   name     => 'phabricator_url',
            label    => 'Phabricator URL',
            type     => 'string',
            default  => '',
            required => 1,
        }
    );
}

sub should_send {
    my ( $self, $message ) = @_;

    return 0 unless Bugzilla->params->{phabricator_enabled};

    return 0 unless $message->routing_key =~ /^(?:attachment|bug)\.modify:.*\bbug_group\b/;

    my $data = $message->payload_decoded;
    my $bug_data = $self->_get_bug_data($data) || return 0;
    my $bug = Bugzilla::Bug->new( { id => $bug_data->{id}, cache => 1 } );

    return $bug->has_attachment_with_mimetype(PHAB_CONTENT_TYPE);
}

sub send {
    my ( $self, $message ) = @_;

    my $logger = Bugzilla->push_ext->logger;

    my $data = $message->payload_decoded;
    my $bug_data = $self->_get_bug_data($data) || return 0;
    my $bug = Bugzilla::Bug->new( { id => $bug_data->{id}, cache => 1 } );

    if(!is_public($bug)) {
        $logger->info('Bailing on send because the bug is not public');
        return PUSH_RESULT_OK;
    }

    my @attachments = grep {
        $_->isobsolete == 0 &&
        $_->contenttype eq PHAB_CONTENT_TYPE &&
        $_->attacher->login eq 'phab-bot@bmo.tld'
    } @{ $bug->attachments() };

    if(@attachments){
        my @rev_ids;
        foreach my $attachment (@attachments) {
            my ($rev_id) = ($attachment->filename =~ PHAB_ATTACHMENT_PATTERN);
            next if !$rev_id;
            push(@rev_ids, int($rev_id));
        }

        if(@rev_ids) {
            $logger->info('Getting info for revisions: ');
            $logger->info(@rev_ids);

            my @rev_details = get_revisions_by_ids(\@rev_ids);
            foreach my $rev_detail (@rev_details) {
                my $rev_phid = $rev_detail->{phid};
                $logger->info('Making revision $rev_phid public:');
                $logger->info($rev_phid);
                make_revision_public($rev_phid);
            }
        }
    }

    return PUSH_RESULT_OK;
}

sub _get_bug_data {
    my ( $self, $data ) = @_;
    my $target = $data->{event}->{target};
    if ( $target eq 'bug' ) {
        return $data->{bug};
    }
    elsif ( exists $data->{$target}->{bug} ) {
        return $data->{$target}->{bug};
    }
    else {
        return;
    }
}

1;