chassis_repo 0.1.0

data/lib/chassis/array_utils.rb

@@ -0,0 +1,8 @@
+module Chassis
+  module ArrayUtils
+    def extract_options!(array)
+      array.last.is_a?(Hash) ? array.pop : { }
+    end
+    module_function :extract_options!
+  end
+end

data/lib/chassis/core_ext/array.rb

@@ -0,0 +1,5 @@
+class Array
+  def extract_options!
+    Chassis::ArrayUtils.extract_options! self
+  end
+end

data/lib/chassis/core_ext/hash.rb

@@ -0,0 +1,5 @@
+class Hash
+  def symbolize
+    Chassis::HashUtils.symbolize self
+  end
+end

data/lib/chassis/core_ext/string.rb

@@ -0,0 +1,13 @@
+class String
+  def constantize
+    Chassis::StringUtils.constantize self
+  end
+
+  def demodulize
+    Chassis::StringUtils.demodulize self
+  end
+
+  def underscore
+    Chassis::StringUtils.underscore self
+  end
+end

data/lib/chassis/delegate.rb

@@ -0,0 +1,29 @@
+module Chassis
+  DelegationError = Chassis.error do |method, delegate|
+    "Cannot delegate #{method} without #{delegate}"
+  end
+
+  class Delegation < Module
+    def initialize(*methods)
+      options = methods.last.is_a?(Hash) ? methods.pop : { }
+
+      delegate = options.fetch :to do
+        fail ArgumentError, ":to not given"
+      end
+
+      methods.each do |method|
+        define_method method do |*args, &block|
+          object = send delegate
+          fail DelegationError.new method, delegate unless object
+          object.send(method, *args, &block)
+        end
+      end
+    end
+  end
+
+  class << self
+    def delegate(*methods)
+      Delegation.new *methods
+    end
+  end
+end

data/lib/chassis/error.rb

@@ -0,0 +1,7 @@
+module Chassis
+  class << self
+    def error(*args, &block)
+      Tnt.boom *args, &block
+    end
+  end
+end

data/lib/chassis/hash_utils.rb

@@ -0,0 +1,16 @@
+module Chassis
+  module HashUtils
+    def symbolize(hash)
+      hash.inject({}) do |memo, pair|
+        key, value = pair
+
+        if value.is_a? Hash
+          memo.merge! key.to_sym => symbolize(value)
+        else
+          memo.merge! key.to_sym => value
+        end
+      end
+    end
+    module_function :symbolize
+  end
+end

data/lib/chassis/initializable.rb

@@ -0,0 +1,3 @@
+module Chassis
+  Initializable = ::Lift
+end

data/lib/chassis/logger.rb

@@ -0,0 +1,8 @@
+module Chassis
+  class Logger < ::Logger
+    def initialize(logdev = Chassis.stream, shift_age = 0, shift_size = 1048576)
+      super
+      self.level = ENV['LOG_LEVEL'].to_sym if ENV['LOG_LEVEL']
+    end
+  end
+end

data/lib/chassis/persistence.rb

@@ -0,0 +1,84 @@
+module Chassis
+  module Persistence
+    module ClassMethods
+      def create(*args, &block)
+        record = new(*args, &block)
+        record.save
+        record
+      end
+
+      def repo
+        begin
+          @repo ||= StringUtils.constantize "#{name}Repo"
+        rescue NameError
+          fail "#{name}Repo not defined! Define this method to specify a different repo"
+        end
+      end
+    end
+
+    class << self
+      def included(base)
+        base.class_eval do
+          include Initializable
+          #include Equalizer.new(:id)
+
+          # Compare the object with other object for equality
+          #
+          # @example
+          #   object.eql?(other)  # => true or false
+          #
+          # @param [Object] other
+          #   the other object to compare with
+          #
+          # @return [Boolean]
+          #
+          # @api public
+          def eql?(other)
+            instance_of?(other.class) && id.eql?(other.id)
+          end
+
+          # Compare the object with other object for equivalency
+          #
+          # @example
+          #   object == other  # => true or false
+          #
+          # @param [Object] other
+          #   the other object to compare with
+          #
+          # @return [Boolean]
+          #
+          # @api public
+          def ==(other)
+            other = coerce(other).first if respond_to?(:coerce, true)
+            other.kind_of?(self.class) && id == other.id
+          end
+
+          def hash
+            id.hash
+          end
+
+          attr_accessor :id
+        end
+
+        base.extend ClassMethods
+      end
+    end
+
+    def save
+      yield self if block_given?
+      repo.save self
+    end
+
+    def delete
+      repo.delete self
+    end
+
+    def new_record?
+      id.nil?
+    end
+
+    def repo
+      self.class.repo
+    end
+  end
+end

data/lib/chassis/repo.rb

@@ -0,0 +1,71 @@
+module Chassis
+  RecordNotFoundError = Chassis.error do |klass, id|
+    "Could not locate #{klass} with id #{id}"
+  end
+
+  QueryNotImplementedError = Chassis.error do |selector|
+    "Adapter does not support #{selector.class}!"
+  end
+
+  GraphQueryNotImplementedError = Chassis.error do |selector|
+    "Adapter does not know how to graph #{selector.class}!"
+  end
+
+  NoQueryResultError = Chassis.error do |selector|
+    "Query #{selector.class} must return results!"
+  end
+
+  class Repo
+    include Interchange.new(*[
+      :all, :find, :create, :update, :delete,
+      :first, :last, :query, :graph_query,
+      :sample, :empty?, :count, :clear,
+      :initialize_storage
+    ])
+
+    class << self
+      def default
+        @default ||= new
+      end
+    end
+
+    def find(klass, id)
+      raise ArgumentError, "id cannot be nil!" if id.nil?
+      super
+    end
+
+    def save(record)
+      if record.id
+        update record
+      else
+        create record
+      end
+    end
+
+    def query!(klass, selector)
+      result = query klass, selector
+      no_results = result.respond_to?(:empty?) ? result.empty? : result.nil?
+
+      if no_results && block_given?
+        yield klass, selector
+      elsif no_results
+        fail NoQueryResultError, selector
+      end
+
+      result
+    end
+  end
+
+  class << self
+    def repo
+      Repo.default
+    end
+  end
+end
+
+require_relative 'repo/delegation'
+require_relative 'repo/record_map'
+require_relative 'repo/base_repo'
+require_relative 'repo/null_repo'
+require_relative 'repo/memory_repo'
+require_relative 'repo/maglev_repo'

data/lib/chassis/repo/base_repo.rb

@@ -0,0 +1,99 @@
+module Chassis
+  class BaseRepo
+    def create(record)
+      record.id ||= next_id
+      map.set record
+    end
+
+    def update(record)
+      map.set record
+    end
+
+    def delete(record)
+      map.delete record
+    end
+
+    def clear
+      map.clear
+    end
+
+    def all(klass)
+      map.all klass
+    end
+
+    def find(klass, id)
+      map.get klass, id
+    end
+
+    def first(klass)
+      all(klass).first
+    end
+
+    def last(klass)
+      all(klass).last
+    end
+
+    def sample(klass)
+      all(klass).sample
+    end
+
+    def empty?(klass)
+      all(klass).empty?
+    end
+
+    def count(klass)
+      all(klass).count
+    end
+
+    def query(klass, selector)
+      if query_implemented? klass, selector
+        send query_method(klass, selector), klass, selector
+      else
+        raise QueryNotImplementedError, selector
+      end
+    end
+
+    def graph_query(klass, selector)
+      if graph_query_implemented? klass, selector
+        send graph_query_method(klass, selector), klass, selector
+      else
+        raise GraphQueryNotImplementedError, selector
+      end
+    end
+
+    def initialize_storage
+
+    end
+
+    private
+    def map
+      @map
+    end
+
+    def next_id
+      @counter ||= 0
+      @counter = @counter + 1
+      @counter
+    end
+
+    def query_method(klass, selector)
+      "query_#{selector_key(selector)}"
+    end
+
+    def query_implemented?(klass, selector)
+      respond_to? query_method(klass, selector)
+    end
+
+    def graph_query_method(klass, selector)
+      "graph_query_#{selector_key(selector)}"
+    end
+
+    def graph_query_implemented?(klass, selector)
+      respond_to? graph_query_method(klass, selector)
+    end
+
+    def selector_key(selector)
+      StringUtils.underscore(StringUtils.demodulize(selector.class.name))
+    end
+  end
+end

data/lib/chassis/repo/delegation.rb

@@ -0,0 +1,82 @@
+module Chassis
+  UnknownObjectClassError = Chassis.error do
+    "Rename class to end in Repo or define object_class"
+  end
+
+  class Repo
+    module Delegation
+      def all
+        backend.all object_class
+      end
+
+      def each(&block)
+        backend.each object_class, &block
+      end
+
+      def count
+        backend.count object_class
+      end
+
+      def find(id)
+        backend.find object_class, id
+      end
+
+      def save(record)
+        backend.save(record)
+      end
+
+      def delete(record)
+        backend.delete record
+      end
+
+      def first
+        backend.first object_class
+      end
+
+      def last
+        backend.last object_class
+      end
+
+      def query(selector)
+        backend.query object_class, selector
+      end
+
+      def query!(selector, &block)
+        backend.query! object_class, selector, &block
+      end
+
+      def sample
+        backend.sample object_class
+      end
+
+      def empty?
+        backend.empty? object_class
+      end
+
+      def graph(id)
+        backend.graph object_class, id
+      end
+
+      def graph_query(selector)
+        backend.graph_query object_class, selector
+      end
+
+      def lazy(id)
+        LazyAssociation.new self, id
+      end
+
+      def object_class
+        @object_class ||= begin
+          fail UnknownObjectClassError unless name
+          match = name.match(/^(.+)Repo$/)
+          fail UnknownObjectClassError unless match
+          StringUtils.constantize match[1]
+        end
+      end
+
+      def backend
+        Repo.default
+      end
+    end
+  end
+end