[IsleOfWight] correctly filter categories on report page

If it's a standard user only display the Triage category list but if one
of these is selected then look up all the categories in that group and
use those to select the reports to display.

It it's a staff user then ignore the Triage categories.

1 files changed, 74 insertions, 0 deletions

diff --git a/perllib/FixMyStreet/Cobrand/FixMyStreet.pm b/perllib/FixMyStreet/Cobrand/FixMyStreet.pm
index 377253955..bfa991b09 100644
--- a/perllib/FixMyStreet/Cobrand/FixMyStreet.pm
+++ b/perllib/FixMyStreet/Cobrand/FixMyStreet.pm
@@ -38,6 +38,24 @@ sub restriction {
     return {};
 }
 
+sub munge_reports_categories_list {
+    my ($self, $categories) = @_;
+
+    my %bodies = map { $_->body->name => $_->body } @$categories;
+    if ( $bodies{'Isle of Wight Council'} ) {
+        my $user = $self->{c}->user;
+        my $b = $bodies{'Isle of Wight Council'};
+
+        if ( $user && ( $user->is_superuser || $user->belongs_to_body( $b->id ) ) ) {
+            @$categories = grep { !$_->send_method || $_->send_method ne 'Triage' } @$categories;
+            return @$categories;
+        }
+
+        @$categories = grep { $_->send_method && $_->send_method eq 'Triage' } @$categories;
+        return @$categories;
+    }
+}
+
 sub munge_category_list {
     my ($self, $options, $contacts, $extras) = @_;
 
@@ -62,6 +80,62 @@ sub munge_category_list {
     }
 }
 
+sub munge_load_and_group_problems {
+    my ($self, $where, $filter) = @_;
+
+    return unless $where->{category} && $self->{c}->stash->{body}->name eq 'Isle of Wight Council';
+
+    my $cat_names = $self->expand_triage_cat_list($where->{category});
+
+    $where->{category} = $cat_names;
+    my $problems = $self->problems->search($where, $filter);
+    return $problems;
+}
+
+sub expand_triage_cat_list {
+    my ($self, $categories) = @_;
+
+    my $b = $self->{c}->stash->{body};
+
+    my $all_cats = $self->{c}->model('DB::Contact')->not_deleted->search(
+        {
+            body_id => $b->id,
+            send_method => [{ '!=', 'Triage'}, undef]
+        }
+    );
+
+    my %group_to_category;
+    while ( my $cat = $all_cats->next ) {
+        next unless $cat->get_extra_metadata('group');
+        my $groups = $cat->get_extra_metadata('group');
+        $groups = ref $groups eq 'ARRAY' ? $groups : [ $groups ];
+        for my $group ( @$groups ) {
+            $group_to_category{$group} //= [];
+            push @{ $group_to_category{$group} }, $cat->category;
+        }
+    }
+
+    my $cats = $self->{c}->model('DB::Contact')->not_deleted->search(
+        {
+            body_id => $b->id,
+            category => $categories
+        }
+    );
+
+    my @cat_names;
+    while ( my $cat = $cats->next ) {
+        if ( $cat->send_method && $cat->send_method eq 'Triage' ) {
+            # include the category itself
+            push @cat_names, $cat->category;
+            push @cat_names, @{ $group_to_category{$cat->category} } if $group_to_category{$cat->category};
+        } else {
+            push @cat_names, $cat->category;
+        }
+    }
+
+    return \@cat_names;
+}
+
 sub title_list {
     my $self = shift;
     my $areas = shift;