current image storage api

Author: gbarc-dt

lib/travis/api/v3/models/custom_image_storage.rb

@@ -0,0 +1,4 @@
+module Travis::API::V3
+  class Models::CustomImageStorage < Model
+  end
+end

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

@@ -14,6 +14,10 @@ def usage(owner, user_id, from, to)
       client.storage_usage(owner_type(owner), owner.id, from, to)
     end
 
+    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
+
     private
 
     def owner_type(owner)

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

@@ -0,0 +1,6 @@
+module Travis::API::V3
+  class Renderer::CustomImageStorage < ModelRenderer
+    representation :minimal, :id, :owner_id, :owner_type, :current_aggregated_storage, :created_at, :updated_at, :end_date
+    representation :standard, *representations[:minimal]
+  end
+end

lib/travis/api/v3/routes.rb

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

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

@@ -0,0 +1,16 @@
+module Travis::API::V3
+  class Services::CustomImages::CurrentStorage < Service
+    result_type :custom_image_storage
+    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).current_storage(owner, access_control.user.id)
+    end
+  end
+end