summaryrefslogtreecommitdiffstats
path: root/extensions/RestrictComments/Extension.pm
blob: 61a4912b395367d9e3107e614d1221340bdfa7de (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
# 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::RestrictComments;

use strict;
use warnings;

use base qw(Bugzilla::Extension);

use Bugzilla::Constants;
use Bugzilla::Util qw(i_am_webservice);

BEGIN {
    *Bugzilla::Bug::restrict_comments = \&_bug_restrict_comments;
}

sub _bug_restrict_comments {
    my ($self) = @_;
    return $self->{restrict_comments};
}

sub bug_check_can_change_field {
    my ($self, $args) = @_;
    my ($bug, $priv_results) = @$args{qw(bug priv_results)};
    my $user = Bugzilla->user;

    if ($user->id
        && $bug->restrict_comments
        && !$user->in_group(Bugzilla->params->{'restrict_comments_group'}))
    {
        push(@$priv_results, PRIVILEGES_REQUIRED_EMPOWERED);
        return;
    }
}

sub _can_restrict_comments {
    my ($self, $object) = @_;
    return unless $object->isa('Bugzilla::Bug');
    $self->{setter_group} ||= Bugzilla->params->{'restrict_comments_enable_group'};
    return Bugzilla->user->in_group($self->{setter_group});
}

sub object_end_of_set_all {
    my ($self, $args) = @_;
    my $object = $args->{object};
    my $input = Bugzilla->input_params;
    my $update_restrict_comments = !i_am_webservice() || exists $input->{restrict_comments};
    if ($update_restrict_comments && $self->_can_restrict_comments($object)) {
        $object->set('restrict_comments', $input->{restrict_comments} ? 1 : undef);
    }
}

sub object_update_columns {
    my ($self, $args) = @_;
    my ($object, $columns) = @$args{qw(object columns)};
    if ($self->_can_restrict_comments($object)) {
        push(@$columns, 'restrict_comments');
    }
}

sub object_columns {
    my ($self, $args) = @_;
    my ($class, $columns) = @$args{qw(class columns)};
    if ($class->isa('Bugzilla::Bug')) {
        push(@$columns, 'restrict_comments');
    }
}

sub bug_fields {
    my ($self, $args) = @_;
    my $fields = $args->{'fields'};
    push (@$fields, 'restrict_comments')
}

sub config_add_panels {
    my ($self, $args) = @_;
    my $modules = $args->{panel_modules};
    $modules->{RestrictComments} = "Bugzilla::Extension::RestrictComments::Config";
}

sub install_update_db {
    my $dbh = Bugzilla->dbh;

    my $field = new Bugzilla::Field({ name => 'restrict_comments' });
    if (!$field) {
        Bugzilla::Field->create({ name => 'restrict_comments', description => 'Restrict Comments' });
    }

    $dbh->bz_add_column('bugs', 'restrict_comments', { TYPE => 'BOOLEAN' });
}

__PACKAGE__->NAME;