Bug 920060: Bug.search API doesn't return total results

1 files changed, 61 insertions, 2 deletions

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