Merge pull request #1377 from travis-ci/sc-implement-storage-executions-usage

Implement storage executions usage endpoint

Author: speedywizard

lib/travis/api/v3/billing_client.rb

@@ -54,6 +54,13 @@ def storage_usage(owner_type, owner_id, from, to)
       end
     end
 
+    def storage_executions_usage(owner_type, owner_id)
+      response = connection.get("/usage/#{owner_type.downcase}s/#{owner_id}/storage_executions")
+      body(response).map do |usage_data|
+        usage_data
+      end
+    end
+
     def calculate_credits(users, executions)
       response = connection.post("/usage/credits_calculator", users: users, executions: executions)
       body(response).map do |calculator_data|

lib/travis/api/v3/queries/custom_images.rb

@@ -18,6 +18,10 @@ def current_storage(owner, user_id)
       Models::CustomImageStorage.where(owner_type: owner_type(owner), owner_id: owner.id).order('id desc').limit(1).first
     end
 
+    def storage_executions_usage(owner, user_id)
+      BillingClient.new(user_id).storage_executions_usage(owner_type(owner), owner.id)
+    end
+
     private
 
     def owner_type(owner)

lib/travis/api/v3/renderer/storage_executions_usage.rb

@@ -0,0 +1,6 @@
+module Travis::API::V3
+  class Renderer::StorageExecutionsUsage < ModelRenderer
+    representation :minimal, :estimated_usage
+    representation :standard, *representations[:minimal]
+  end
+end

lib/travis/api/v3/renderer/storage_executions_usages.rb

@@ -0,0 +1,6 @@
+module Travis::API::V3
+  class Renderer::StorageExecutionsUsages < CollectionRenderer
+    type           :storage_executions_usages
+    collection_key :storage_executions_usages
+  end
+end

lib/travis/api/v3/routes.rb

@@ -176,6 +176,7 @@ module Routes
         delete :delete
         get :usage, '/usage'
         get :current_storage, '/current_storage'
+        get :storage_executions_usage, '/storage_executions_usage'
       end
     end
 

lib/travis/api/v3/services/custom_images/storage_executions_usage.rb

@@ -0,0 +1,17 @@
+module Travis::API::V3
+  class Services::CustomImages::StorageExecutionsUsage < Service
+    result_type :storage_executions_usages
+
+    def run!
+      raise MethodNotAllowed if Travis.config.org?
+      raise LoginRequired unless access_control.logged_in?
+
+      owner = query(:owner).find
+
+      raise NotFound unless owner
+      raise InsufficientAccess unless access_control.visible?(owner)
+
+      result query(:custom_images).storage_executions_usage(owner, access_control.user.id)
+    end
+  end
+end