Constantize using Consty gem

gabynaiman · Sep 16, 2016 · e3e5384 · e3e5384
1 parent 2304639
commit e3e5384
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 2 deletions.
diff --git a/lib/rasti/db.rb b/lib/rasti/db.rb
@@ -1,4 +1,5 @@
 require 'sequel'
+require 'consty'
 
 require_relative 'db/version'
 require_relative 'db/collection'

diff --git a/lib/rasti/db/collection.rb b/lib/rasti/db/collection.rb
@@ -15,7 +15,7 @@ def primary_key
         end
 
         def model
-          @model ||= constantize(singularize(name))
+          @model ||= Consty.get(demodulize(singularize(name)), self)
         end
 
         def relations

diff --git a/lib/rasti/db/model.rb b/lib/rasti/db/model.rb
@@ -19,6 +19,10 @@ class << self
         def [](*attributes)
           Class.new(self) do
             attributes.each { |name| attribute name }
+
+            def self.inherited(subclass)
+              subclass.instance_variable_set :@attributes, attributes
+            end
           end
         end
 

diff --git a/lib/rasti/db/relations.rb b/lib/rasti/db/relations.rb
@@ -37,7 +37,7 @@ def initialize(name, source_collection_class, options={})
         end
 
         def target_collection_class
-          @target_collection_class ||= @options[:collection].is_a?(Class) ? @options[:collection] : constantize(@options[:collection] || camelize(pluralize(name)))
+          @target_collection_class ||= @options[:collection].is_a?(Class) ? @options[:collection] : Consty.get(@options[:collection] || camelize(pluralize(name)), self.class)
         end
 
         private

diff --git a/rasti-db.gemspec b/rasti-db.gemspec
@@ -19,6 +19,7 @@ Gem::Specification.new do |spec|
   spec.require_paths = ['lib']
 
   spec.add_runtime_dependency 'sequel', '~> 4.38'
+  spec.add_runtime_dependency 'consty', '~> 1.0'
 
   spec.add_development_dependency 'bundler', '~> 1.12'
   spec.add_development_dependency 'rake', '~> 11.0'