From 3d49752f959fd6c2b5e88eef8333b15705614f68 Mon Sep 17 00:00:00 2001
From: Byron Jones <bjones@mozilla.com>
Date: Thu, 26 Sep 2013 00:17:52 +0800
Subject: Bug 920060: Bug.search API doesn't return total results

---
 extensions/Ember/lib/WebService.pm | 63 ++++++++++++++++++++++++++++++++++++--
 1 file changed, 61 insertions(+), 2 deletions(-)

(limited to 'extensions/Ember')

diff --git a/extensions/Ember/lib/WebService.pm b/extensions/Ember/lib/WebService.pm
index 423c8de34..57ce2fe36 100644
--- a/extensions/Ember/lib/WebService.pm
+++ b/extensions/Ember/lib/WebService.pm
@@ -25,7 +25,7 @@ use Bugzilla::Util qw(trick_taint);
 use Bugzilla::Extension::Ember::FakeBug;
 
 use Scalar::Util qw(blessed);
-use Data::Dumper;
+use Storable qw(dclone);
 
 use constant FIELD_TYPE_MAP => {
     0  => 'unknown',
@@ -203,6 +203,23 @@ sub show {
     return $data;
 }
 
+sub search {
+    my ($self, $params) = @_;
+
+    my $total;
+    if (exists $params->{offset} && exists $params->{limit}) {
+        my $count_params = dclone($params);
+        delete $count_params->{offset};
+        delete $count_params->{limit};
+        $count_params->{count_only} = 1;
+        $total = $self->SUPER::search($count_params);
+    }
+
+    my $result = $self->SUPER::search($params);
+    $result->{total} = defined $total ? $total : scalar(@{ $result->{bugs} });
+    return $result;
+}
+
 ###################
 # Private Methods #
 ###################
@@ -529,7 +546,14 @@ sub rest_resources {
             GET => {
                 method => 'show'
             }
-        }
+        },
+        # search - wrapper around SUPER::search which also includes the total
+        # number of bugs when using pagination
+        qr{^/ember/search$}, {
+            GET  => {
+                method => 'search',
+            },
+        },
     ];
 };
 
@@ -636,3 +660,38 @@ You pass a field called C<id> that is the current bug id.
 =back
 
 =back
+
+=head2 search
+
+B<UNSTABLE>
+
+=over
+
+=item B<Description>
+
+A wrapper around Bugzilla's C<search> method which also returns the total of
+bugs matching a query, even if the limit and offset parameters are supplied.
+
+=item B<Params>
+
+As per Bugzilla::WebService::Bug::search()
+
+=item B<Returns>
+
+=over
+
+=back
+
+=item B<Errors>
+
+=over
+
+=back
+
+=item B<History>
+
+=over
+
+=back
+
+=back
-- 
cgit v1.2.3-24-g4f1b