Custom images list implementation

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

@@ -0,0 +1,20 @@
+module Travis::API::V3
+  class Models::CustomImage < Model
+    after_initialize :readonly!
+    belongs_to :owner, polymorphic: true
+    has_many :custom_image_logs
+
+    scope :available, -> { where(state: 'available') }
+
+    def created_by
+      user_id = custom_image_logs.where(action: 'created').first&.sender_id
+      return unless user_id
+
+      User.find(user_id)
+    end
+
+    def private
+      true
+    end
+  end
+end

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

@@ -0,0 +1,6 @@
+module Travis::API::V3
+  class Models::CustomImageLog < Model
+    after_initialize :readonly!
+    belongs_to :custom_image
+  end
+end

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

@@ -0,0 +1,13 @@
+module Travis::API::V3
+  class Queries::CustomImages < Query
+    def for_owner(owner)
+      Models::CustomImage.where(owner_id: owner.id, owner_type: owner_type(owner))
+    end
+
+    private
+
+    def owner_type(owner)
+      owner.vcs_type =~ /User/ ? 'User' : 'Organization'
+    end
+  end
+end

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

@@ -0,0 +1,20 @@
+module Travis::API::V3
+  class Renderer::CustomImage < ModelRenderer
+    representation :minimal, :id, :owner_id, :owner_type, :name, :usage, :created_at, :updated_at, :os_version,
+                   :created_by, :private, :size_bytes
+    representation :standard, *representations[:minimal]
+
+    def created_by
+      return nil unless user = model.created_by
+      {
+        '@type' => 'user',
+        '@href' => "/v3/user/#{user.id}",
+        '@representation' => 'minimal'.freeze,
+        'id' => user.id,
+        'login' => user.login,
+        'name' => user.name,
+        'avatar_url' => user.avatar_url
+      }
+    end
+  end
+end

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

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

lib/travis/api/v3/routes.rb

@@ -169,6 +169,11 @@ module Routes
         route '/executions_per_sender'
         get :for_owner_per_sender
       end
+
+      resource :custom_images do
+        route '/custom_images'
+        get :for_owner
+      end
     end
 
     resource :credits_calculator do

lib/travis/api/v3/services.rb

@@ -26,6 +26,7 @@ module Services
     Crons                 = Module.new { extend Services }
     CustomKey             = Module.new { extend Services }
     CustomKeys            = Module.new { extend Services }
+    CustomImages          = Module.new { extend Services }
     EmailSubscription     = Module.new { extend Services }
     EnvVar                = Module.new { extend Services }
     EnvVars               = Module.new { extend Services }

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

@@ -0,0 +1,20 @@
+module Travis::API::V3
+  class Services::CustomImages::ForOwner < Service
+    result_type :custom_images
+
+    def run!
+      raise MethodNotAllowed if Travis.config.org?
+      raise LoginRequired unless access_control.logged_in?
+
+      owner = query(:owner).find
+
+      raise NotFound unless owner
+      repo = owner.repositories.first
+      raise InsufficientAccess unless repo
+      access_control.permissions(repo).build_create!
+
+      results = query(:custom_images).for_owner(owner)
+      result results
+    end
+  end
+end