ruroku 0.0.1 → 0.0.2

data/lib/ruroku/app.rb

@@ -1,32 +1,53 @@
 module Ruroku
   class App < Base
-    attr_accessor :id, :name, :create_status, :created_at, :stack, :git_url,
-      :requested_stack, :repo_migrate_status, :slug_size, :repo_size,
-      :dynos, :workers
+    attribute :id, Integer
+    attribute :name, String
+    attribute :create_status, String
+    attribute :stack, String
+    attribute :requested_stack, String
+    attribute :git_url, String
+    attribute :repo_migrate_status, String
+    attribute :slug_size, Integer
+    attribute :repo_size, Integer
+    attribute :dynos, Integer
+    attribute :workers, Integer
+    attribute :created_at, Time
+
+    resource_id :name
 
     # Public: Get addons associated with current app.
     def addons
-      AddonSet.new self, *api.get_addons(name).body.map { |addon| Addon.new self, addon }
+      @addons ||= AddonSet.new self
     end
 
     # Public: Get collaborators associated with current app.
     def collaborators
-      CollaboratorSet.new self, *api.get_collaborators(name).body.map { |collaborator| Collaborator.new self, collaborator }
+      @collaborators ||= CollaboratorSet.new self
     end
 
     # Public: Get config vars associated with current app.
     def config_vars
-      ConfigVarSet.new self, *api.get_config_vars(name).body.map { |key, value| ConfigVar.new self, key: key, value: value }
+      @config_vars ||= ConfigVarSet.new self
+    end
+
+    # Public: Get domains associated with current app.
+    def domains
+      @domains ||= DomainSet.new self
     end
 
     # Public: Get processes associated with current app.
     def processes
-      ProcessSet.new self, *api.get_ps(name).body.map { |ps| Process.new self, ps }
+      @processes ||= ProcessSet.new self
     end
 
     # Public: Get releases associated with current app.
     def releases
-      ReleaseSet.new self, *api.get_releases(name).body.map { |release| Release.new self, release }
+      @releases ||= ReleaseSet.new self
+    end
+
+    # Public: Get available stacks for current app.
+    def stacks
+      @stacks ||= StackSet.new self
     end
 
     # Public: Turn the maintenance mode on.
@@ -38,17 +59,5 @@ module Ruroku
     def no_maintenance!
       api.post_app_maintenance name, '0'
     end
-
-    def created_at=(value)
-      @created_at = Time.parse value
-    end
-
-    def slug_size=(value)
-      @slug_size = value.to_i.bytes
-    end
-
-    def repo_size=(value)
-      @repo_size = value.to_i.bytes
-    end
   end
 end

data/lib/ruroku/app_set.rb

@@ -0,0 +1,13 @@
+module Ruroku
+  class AppSet < ResourceSet
+    resource_class App
+    collection_api_selector :get_apps
+
+    # Map API methods to collection methods.
+    #
+    # Examples
+    #
+    #   apps.create 'another-app', 'cedar'
+    map_api create: :post_app
+  end
+end

data/lib/ruroku/base.rb

@@ -1,13 +1,30 @@
 module Ruroku
   class Base
+    include Virtus
+
     attr_accessor :api
 
     def initialize(api, attributes = {})
       self.api = api
 
-      attributes.each do |key, value|
-        method = "#{key}="
-        send method, value if respond_to? method
+      super attributes
+    end
+
+    # Public: Define resource id key.
+    def self.resource_id(id_key = nil)
+      if id_key
+        @_resource_id = id_key
+      else
+        @_resource_id
+      end
+    end
+
+    def inspect
+      if self.class.resource_id
+        resource_id = send self.class.resource_id
+        "<#{self.class} #{resource_id}>"
+      else
+        "<#{self.class}>"
       end
     end
   end

data/lib/ruroku/collaborator.rb

@@ -1,6 +1,9 @@
 module Ruroku
   class Collaborator < NestedBase
-    attr_accessor :email, :access
-    deletable :email
+    attribute :email, String
+    attribute :access, String
+
+    resource_id :email
+    deletable_resource
   end
 end

data/lib/ruroku/collaborator_set.rb

@@ -1,5 +1,8 @@
 module Ruroku
-  class CollaboratorSet < ResourceSet
+  class CollaboratorSet < NestedResourceSet
+    resource_class Collaborator
+    collection_api_selector :get_collaborators
+
     # Map API methods to collection methods.
     #
     # Examples

data/lib/ruroku/config_var.rb

@@ -1,7 +1,10 @@
 module Ruroku
   class ConfigVar < NestedBase
-    attr_accessor :key, :value
-    deletable :key
+    attribute :key, String
+    attribute :value, String
+
+    resource_id :key
+    deletable_resource
 
     def value=(new_value)
       if @value.nil?
@@ -11,5 +14,9 @@ module Ruroku
         @value = new_value
       end
     end
+
+    def inspect
+      "<#{self.class} #{key} => #{value}>"
+    end
   end
 end

data/lib/ruroku/config_var_set.rb

@@ -1,11 +1,14 @@
 module Ruroku
-  class ConfigVarSet < ResourceSet
+  class ConfigVarSet < NestedResourceSet
+    resource_class ConfigVar
+    collection_api_selector :get_config_vars
+
     # Map API methods to collection methods.
     #
     # Examples
     #
     #   config_vars.add 'KEY' => 'value'
-    #   config_vars.delete 'KEY' => 'value'
+    #   config_vars.delete 'KEY'
     map_api add: :post_config_vars,
       delete: :delete_config_var
   end

data/lib/ruroku/domain.rb

@@ -1,14 +1,13 @@
 module Ruroku
   class Domain < NestedBase
-    attr_accessor :id, :domain, :base_domain, :default, :created_at, :updated_at
-    deletable :domain
+    attribute :id, Integer
+    attribute :domain, String
+    attribute :base_domain, String
+    attribute :default, Boolean, default: false
+    attribute :created_at, Time
+    attribute :updated_at, Time
 
-    def created_at=(value)
-      @created_at = Time.parse value
-    end
-
-    def updated_at=(value)
-      @updated_at = Time.parse value
-    end
+    resource_id :domain
+    deletable_resource
   end
 end

data/lib/ruroku/domain_set.rb

@@ -1,5 +1,8 @@
 module Ruroku
-  class DomainSet < ResourceSet
+  class DomainSet < NestedResourceSet
+    resource_class Domain
+    collection_api_selector :get_domains
+
     # Map API methods to collection methods.
     #
     # Examples

data/lib/ruroku/key.rb

@@ -0,0 +1,8 @@
+module Ruroku
+  class Key < Base
+    attribute :email, String
+    attribute :contents, String
+
+    resource_id :email
+  end
+end

data/lib/ruroku/key_set.rb

@@ -0,0 +1,17 @@
+module Ruroku
+  class KeySet < ResourceSet
+    resource_class Key
+    collection_api_selector :get_keys
+
+    # Map API methods to collection methods.
+    #
+    # Examples
+    #
+    #   keys.add 'id rsa pub content here'
+    #   keys.delete 'email@me.com'
+    #   keys.delete_all
+    map_api add: :post_key,
+      delete: :delete_key,
+      delete_all: :delete_keys
+  end
+end

data/lib/ruroku/nested_base.rb

@@ -8,12 +8,13 @@ module Ruroku
       super app.api, attributes
     end
 
-    def self.deletable(resource_id)
+    # Public: Make resource deletable.
+    def self.deletable_resource
       resource_name = name.demodulize.underscore
 
       define_method :delete! do
         api_method = "delete_#{resource_name}"
-        resource_id = send resource_id
+        resource_id = send self.class.resource_id
         api.send api_method, app.name, resource_id
       end
     end

data/lib/ruroku/nested_resource_set.rb

@@ -0,0 +1,28 @@
+module Ruroku
+  class NestedResourceSet < ResourceSet
+    attr_accessor :app
+
+    def initialize(app, *args)
+      self.app = app
+
+      super app.api, *args
+    end
+
+    def build_resource(response)
+      resource_class = self.class.resource_class
+      resource_class.new app, response
+    end
+
+    def collection_query_params
+      [app.name]
+    end
+
+    def self.map_api(methods)
+      methods.each do |method_name, api_mapping|
+        define_method method_name do |*args|
+          api.send api_mapping, app.name, *args
+        end
+      end
+    end
+  end
+end

data/lib/ruroku/process.rb

@@ -1,7 +1,19 @@
 module Ruroku
   class Process < NestedBase
-    attr_accessor :process, :type, :command, :upid, :slug, :action,
-      :pretty_state, :elapsed, :rendezvous_url, :attached, :transitioned_at
+    attribute :process, String
+    attribute :type, String
+    attribute :command, String
+    attribute :upid, String
+    attribute :slug, String
+    attribute :action, String
+    attribute :state, String
+    attribute :pretty_state, String
+    attribute :elapsed, Integer
+    attribute :rendezvous_url, String
+    attribute :attached, Boolean, default: false
+    attribute :transitioned_at, Time
+
+    resource_id :process
 
     # Public: Restart the process.
     def restart
@@ -12,14 +24,5 @@ module Ruroku
     def stop
       api.post_ps_stop app.name, 'ps' => process
     end
-
-    # Public: Check if the process is attached.
-    def attached?
-      !!attached
-    end
-
-    def transitioned_at=(value)
-      @transitioned_at = Time.parse value
-    end
   end
 end

data/lib/ruroku/process_set.rb

@@ -1,37 +1,20 @@
 module Ruroku
-  class ProcessSet < ResourceSet
-    # Publc: Run the command.
-    #
-    # command - The String command.
-    def run(command)
-      api.post_ps app.name, command
-    end
-
-    # Public: Restart all the processes.
-    def restart
-      api.post_ps_restart app.name
-    end
-
-    # Public: Scale processes.
-    #
-    # type - The String process type.
-    # qty  - The Integer process quantity.
-    def scale(type, qty)
-      api.ps_scale app.name, type, qty
-    end
+  class ProcessSet < NestedResourceSet
+    resource_class Process
+    collection_api_selector :get_ps
 
-    # Public: Stop the process(es).
+    # Map API methods to collection methods.
     #
-    # options - The Hash
-    #           either :ps   - The String process name
-    #               or :type - The String process type
-    # 
     # Examples
     #
+    #   processes.run 'rake task'
+    #   processes.restart
+    #   processes.scale 'web', 10
     #   processes.stop 'ps' => 'web.1'
     #   processes.stop 'type' => 'web'
-    def stop(options)
-      api.post_ps_stop app.name, options
-    end
+    map_api run: :post_ps,
+      restart: :post_ps_restart,
+      scale: :ps_scale,
+      stop: :post_ps_stop
   end
 end

data/lib/ruroku/release.rb

@@ -1,10 +1,14 @@
 module Ruroku
   class Release < NestedBase
-    attr_accessor :name, :descr, :user, :commit, :env, :addons, :pstable,
-      :created_at
+    attribute :name, String
+    attribute :descr, String
+    attribute :user, String
+    attribute :commit, String
+    attribute :env, Hash
+    attribute :addons, Array
+    attribute :pstable, Hash
+    attribute :created_at, Time
 
-    def created_at=(value)
-      @created_at = Time.parse value
-    end
+    resource_id :name
   end
 end

data/lib/ruroku/release_set.rb

@@ -1,5 +1,8 @@
 module Ruroku
-  class ReleaseSet < ResourceSet
+  class ReleaseSet < NestedResourceSet
+    resource_class Release
+    collection_api_selector :get_releases
+
     # Map API methods to collection methods.
     #
     # Examples

data/lib/ruroku/resource_set.rb

@@ -1,18 +1,73 @@
 module Ruroku
   class ResourceSet < Array
-    attr_accessor :app, :api
+    attr_accessor :api
 
-    def initialize(app, *args)
-      self.app = app
-      self.api = app.api
+    def initialize(api, *args)
+      self.api = api
 
       super args
+
+      query_collection_objects
+    end
+
+    # Public: Reload collection.
+    #
+    # Examples
+    #
+    #   collection.reload
+    def reload
+      clear
+      query_collection_objects
+    end
+
+    # Public: Query collection objects from API.
+    def query_collection_objects
+      collection_objects = api.send(self.class.collection_api_selector, *collection_query_params).body
+
+      if collection_objects.instance_of? Array
+        collection_objects.each do |response|
+          self << build_resource(response)
+        end
+      elsif collection_objects.instance_of? Hash
+        collection_objects.each do |key, value|
+          self << build_resource(key: key, value: value)
+        end
+      end
+    end
+
+    # Public: Build a resource from response.
+    def build_resource(response)
+      resource_class = self.class.resource_class
+      resource_class.new response
+    end
+
+    # Public: Get params for querying collection.
+    def collection_query_params
+      []
+    end
+
+    # Public: Set or get a collection API selector.
+    def self.collection_api_selector(meth = nil)
+      if meth
+        @_collection_api_selector = meth
+      else
+        @_collection_api_selector
+      end
+    end
+
+    # Public: Set or get a collection resource class.
+    def self.resource_class(klass = nil)
+      if klass
+        @_resource_class = klass
+      else
+        @_resource_class
+      end
     end
 
     def self.map_api(methods)
       methods.each do |method_name, api_mapping|
-        define_method method_name do |resource_name|
-          api.send api_mapping, app.name, resource_name
+        define_method method_name do |*args|
+          api.send api_mapping, *args
         end
       end
     end