base.rb in harvesting-0.3.0

- old
+ new
@@ -1,37 +1,65 @@
 module Harvesting
   module Models
     class Base
       attr_accessor :attributes
-      attr_reader :client
+      attr_reader :harvest_client
 
       def initialize(attrs, opts = {})
+        @models = {}
         @attributes = attrs.dup
-        @client = opts[:client] || Harvesting::Client.new(opts)
+        @harvest_client = opts[:client] || Harvesting::Client.new(opts)
       end
 
       def self.attributed(*attribute_names)
         attribute_names.each do |attribute_name|
-          Harvesting::Models::Base.send :define_method, attribute_name.to_s do
+          define_method(attribute_name) do
             @attributes[__method__.to_s]
           end
         end
       end
 
+      def self.modeled(opts = {})
+        opts.each do |attribute_name, model|
+          attribute_name_string = attribute_name.to_s
+          Harvesting::Models::Base.send :define_method, attribute_name_string do
+            @models[attribute_name_string] ||= model.new(@attributes[attribute_name_string] || {}, client: harvest_client)
+          end
+        end
+      end
+
       def save
         id.nil? ? create : update
       end
 
       def create
-        @client.create(self)
+        @harvest_client.create(self)
       end
 
       def update
-        @client.update(self)
+        @harvest_client.update(self)
       end
 
+      def delete
+        @harvest_client.delete(self)
+      end
+
       def to_hash
         @attributes
+      end
+
+      def fetch
+        self.class.new(@harvest_client.get(path), client: @harvest_client)
+      end
+
+      # Retrieves an instance of the object by ID
+      #
+      # @param id [Integer] the id of the object to retrieve
+      # @param opts [Hash] options to pass along to the `Harvesting::Client`
+      #   instance
+      def self.get(id, opts = {})
+        client = opts[:client] || Harvesting::Client.new(opts)
+        self.new({ 'id' => id }, opts).fetch
       end
     end
   end
 end