summaryrefslogtreecommitdiffstats
path: root/extensions/PhabBugz/bin/update_project_members.pl
blob: 2a305da3d3cf0f6c0aed632915b6f3b78abf0469 (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
#!/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 lib qw(. lib local/lib/perl5);

use Bugzilla;
BEGIN { Bugzilla->extensions() }

use Bugzilla::Constants;
use Bugzilla::Error;
use Bugzilla::Group;

use Bugzilla::Extension::PhabBugz::Project;
use Bugzilla::Extension::PhabBugz::Util qw(
    get_phab_bmo_ids
);

Bugzilla->usage_mode(USAGE_MODE_CMDLINE);

my ($phab_uri, $phab_sync_groups);

if (!Bugzilla->params->{phabricator_enabled}) {
    exit;
}

# Sanity checks
unless ($phab_uri = Bugzilla->params->{phabricator_base_uri}) {
    ThrowUserError('invalid_phabricator_uri');
}

unless ($phab_sync_groups = Bugzilla->params->{phabricator_sync_groups}) {
    ThrowUserError('invalid_phabricator_sync_groups');
}

# Loop through each group and perform the following:
#
# 1. Load flattened list of group members
# 2. Check to see if Phab project exists for 'bmo-<group_name>'
# 3. Create if does not exist with locked down policy.
# 4. Set project members to exact list
# 5. Profit

my $sync_groups = Bugzilla::Group->match({ name => [ split('[,\s]+', $phab_sync_groups) ] });

foreach my $group (@$sync_groups) {
    # Create group project if one does not yet exist
    my $phab_project_name = 'bmo-' . $group->name;
    my $project = Bugzilla::Extension::PhabBugz::Project->new_from_query({
        name => $phab_project_name
    });
    if (!$project) {
        $project = Bugzilla::Extension::PhabBugz::Project->create({
            name        => $phab_project_name,
            description => 'BMO Security Group for ' . $group->name
        });
    }

    if (my @group_members = get_group_members($group)) {
        $project->set_members(\@group_members);
        $project->update();
    }
}

sub get_group_members {
    my ($group) = @_;
    my $group_obj = ref $group ? $group : Bugzilla::Group->check({ name => $group });
    my $members_all = $group_obj->members_complete();
    my %users;
    foreach my $name (keys %$members_all) {
        foreach my $user (@{ $members_all->{$name} }) {
            $users{$user->id} = $user;
        }
    }

    # Look up the phab ids for these users
    my $phab_users = get_phab_bmo_ids({ ids => [ keys %users ] });
    foreach my $phab_user (@{ $phab_users }) {
        $users{$phab_user->{id}}->{phab_phid} = $phab_user->{phid};
    }

    # We only need users who have accounts in phabricator
    return grep { $_->phab_phid } values %users;
}