Support cookbook_versions query param in /cookbooks

1 files changed, 8 insertions, 1 deletions

diff --git a/lib/chef_zero/endpoints/cookbooks_endpoint.rb b/lib/chef_zero/endpoints/cookbooks_endpoint.rb
index 587862d..2c5a8fe 100644
--- a/lib/chef_zero/endpoints/cookbooks_endpoint.rb
+++ b/lib/chef_zero/endpoints/cookbooks_endpoint.rb
@@ -5,7 +5,14 @@ module ChefZero
     # /cookbooks
     class CookbooksEndpoint < CookbooksBase
       def get(request)
-        json_response(200, format_cookbooks_list(request, all_cookbooks_list))
+        if request.query_params['num_versions'] == 'all'
+          num_versions = nil
+        elsif request.query_params['num_versions']
+          num_versions = request.query_params['num_versions'].to_i
+        else
+          num_versions = 1
+        end
+        json_response(200, format_cookbooks_list(request, all_cookbooks_list, {}, num_versions))
       end
     end
   end