1 files changed, 116 insertions, 0 deletions
diff --git a/extensions/ProductDashboard/lib/Util.pm b/extensions/ProductDashboard/lib/Util.pm
new file mode 100644
index 000000000..d83ddf187
--- /dev/null
+++ b/extensions/ProductDashboard/lib/Util.pm
@@ -0,0 +1,116 @@
+# 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::ProductDashboard::Util;
+
+use strict;
+
+use base qw(Exporter);
+@Bugzilla::Extension::ProductDashboard::Util::EXPORT = qw(
+    bug_link_all
+    bug_link_open
+    bug_link_closed
+    open_states
+    closed_states
+    quoted_open_states
+    quoted_closed_states
+    filter_bugs
+    bug_milestone_link_total
+    bug_milestone_link_open
+    bug_milestone_link_closed
+);
+
+use Bugzilla::Status;
+use Bugzilla::Util;
+
+use Bugzilla::Status;
+
+our $_open_states;
+sub open_states {
+    $_open_states ||= Bugzilla::Status->match({ is_open => 1, isactive => 1 });
+    return wantarray ? @$_open_states : $_open_states;
+}
+
+our $_quoted_open_states;
+sub quoted_open_states {
+    my $dbh = Bugzilla->dbh;
+    $_quoted_open_states ||= [ map { $dbh->quote($_->name) } open_states() ];
+    return wantarray ? @$_quoted_open_states : $_quoted_open_states;
+}
+
+our $_closed_states;
+sub closed_states {
+    $_closed_states ||= Bugzilla::Status->match({ is_open => 0, isactive => 1 });
+    return wantarray ? @$_closed_states : $_closed_states;
+}
+
+our $_quoted_closed_states;
+sub quoted_closed_states {
+    my $dbh = Bugzilla->dbh;
+    $_quoted_closed_states ||= [ map { $dbh->quote($_->name) } closed_states() ];
+    return wantarray ? @$_quoted_closed_states : $_quoted_closed_states;
+}
+
+sub bug_link_all {
+    my $product = shift;
+
+    return correct_urlbase() . 'buglist.cgi?product=' . url_quote($product->name);
+}
+
+sub bug_link_open {
+    my $product = shift;
+
+    return correct_urlbase() . 'buglist.cgi?product=' . url_quote($product->name) . 
+        "&bug_status=__open__";
+}
+
+sub bug_link_closed {
+    my $product = shift;
+
+    return correct_urlbase() . 'buglist.cgi?product=' . url_quote($product->name) . 
+        "&bug_status=__closed__";
+}
+
+sub bug_milestone_link_total {
+    my ($product, $milestone) = @_;
+
+    return correct_urlbase() . 'buglist.cgi?product=' . url_quote($product->name) . 
+        "&target_milestone=" . url_quote($milestone->name);
+}
+
+sub bug_milestone_link_open {
+    my ($product, $milestone) = @_;
+
+    return correct_urlbase() . 'buglist.cgi?product=' . url_quote($product->name) . 
+        "&target_milestone=" . url_quote($milestone->name) . "&bug_status=__open__";
+}
+
+sub bug_milestone_link_closed {
+    my ($product, $milestone) = @_;
+
+    return correct_urlbase() . 'buglist.cgi?product=' . url_quote($product->name) . 
+        "&target_milestone=" . url_quote($milestone->name) . "&bug_status=__closed__";
+}
+
+sub filter_bugs {
+    my ($unfiltered_bugs) = @_;
+    my $dbh = Bugzilla->dbh;
+
+    # Filter out which bugs that cannot be viewed
+    my $params = Bugzilla::CGI->new({ bug_id => [ map { $_->{'id'} } @$unfiltered_bugs ] });
+    my $search = Bugzilla::Search->new(fields => ['bug_id' ], params => $params );
+    my %filtered_bug_ids = map { $_ => 1 } @{$dbh->selectcol_arrayref($search->getSQL())};
+
+    my @filtered_bugs;
+    foreach my $bug (@$unfiltered_bugs) {
+        next if !$filtered_bug_ids{$bug->{'id'}};
+        push(@filtered_bugs, $bug);
+    }
+
+    return \@filtered_bugs;
+}
+
+1;