1 files changed, 96 insertions, 0 deletions
diff --git a/extensions/ProductDashboard/lib/Util.pm b/extensions/ProductDashboard/lib/Util.pm
new file mode 100644
index 000000000..ad46ef8f6
--- /dev/null
+++ b/extensions/ProductDashboard/lib/Util.pm
@@ -0,0 +1,96 @@
+# 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
+    filter_bugs
+    bug_milestone_link_total
+    bug_milestone_link_open
+    bug_milestone_link_closed
+);
+
+use Bugzilla::Status;
+use Bugzilla::Util;
+
+sub open_states {
+    my $dbh = Bugzilla->dbh;
+    return join(",", map { $dbh->quote($_) } BUG_STATE_OPEN);
+}
+
+sub closed_states {
+    my $dbh = Bugzilla->dbh;
+    return join(",", map { $dbh->quote($_->name) } closed_bug_statuses());
+}
+
+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;