earth 0.2.11 → 0.2.12

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    earth (0.2.7)
+    earth (0.2.11)
       activerecord (>= 3.0.0.beta4)
       cohort_scope (>= 0.0.7)
       conversions (>= 1.4.5)

data/lib/earth.rb

@@ -31,9 +31,14 @@ module Earth
   # Default is search all domains
   # For example, <tt>[ 'Aircraft', 'Airline' ]</tt>
   def resource_names(search_domains = nil)
-    (search_domains || domains).map do |domain|
-      Dir[File.join(Earth.gem_root, 'lib', 'earth', domain, '*.rb')]
-    end.flatten.uniq.map { |p| File.basename(p, '.rb').camelcase } - %w{ DataMiner }
+    if search_domains.nil?
+      resources.keys
+    else
+      resources.inject([]) do |list, (name, data)|
+        list << name if search_domains.include? data[:domain]
+        list
+      end
+    end
   end
 
   def gem_root 
@@ -41,7 +46,22 @@ module Earth
   end
 
   def domains
-    %w{air automobile bus diet fuel locality pet rail residence}
+    @domains ||= resources.map { |(name, data)| data[:domain] }.uniq.sort
+  end
+
+  def resources
+    return @resources unless @resources.nil?
+    earth_files = Dir[File.join(Earth.gem_root, 'lib', 'earth', '*')]
+    domain_dirs = earth_files.find_all { |f| File.directory?(f) }
+    @resources = domain_dirs.inject({}) do |hsh, domain_dir|
+      Dir[File.join(domain_dir, '*.rb')].each do |resource_file|
+        resource_camel = File.basename(resource_file, '.rb').camelcase
+        unless resource_camel == 'DataMiner'
+          hsh[resource_camel] = { :domain => File.basename(domain_dir) }
+        end
+      end
+      hsh
+    end
   end
 
   # Earth.init will load any specified domains, any needed ActiveRecord plugins, 

data/spec/earth_spec.rb

@@ -1,25 +1,56 @@
 require 'spec_helper'
 
 describe Earth do
-  before :all do
-    Earth.init :all, :apply_schemas => true
+  describe '.init' do
+    before :all do
+      Earth.init :all, :apply_schemas => true
+    end
+
+    it 'should require all Earth models' do
+      lambda do
+        Earth.resource_names.each { |k| k.constantize }
+      end.should_not raise_error(NameError)
+    end
+
+    it 'should include data_miner definitions' do
+      lambda do
+        Earth.resource_names.each { |k| k.constantize.should_receive(:data_miner) }
+      end
+      require 'earth/data_miner'
+    end
+
+    it 'should create a fallbacks table' do
+      Fallback.should be_table_exists
+    end
   end
 
-  it 'should require all Earth models' do
-    lambda do
-      Earth.resource_names.each { |k| k.constantize }
-    end.should_not raise_error(NameError)
+  describe '.resources' do
+    it 'should get a list of resources' do
+      resources = Earth.resources
+      resources.keys.count.should == 56
+      resources['FuelType'][:domain].should == 'fuel'
+    end
+    it 'should exclude data_miner files' do
+      Earth.resources.keys.should_not include('DataMiner')
+    end
   end
 
-  it 'should include data_miner definitions' do
-    lambda do
-      Earth.resource_names.each { |k| k.constantize.should_receive(:data_miner) }
+  describe '.resource_names' do
+    it 'should get a list of all resource names' do
+      Earth.resource_names.count.should == 56
+      Earth.resource_names.should include('Aircraft')
+      Earth.resource_names.should include('Industry')
+    end
+    it 'should filter resources by domain' do
+      Earth.resource_names(['fuel']).count.should == 2
+      Earth.resource_names(['fuel']).should include('FuelType')
     end
-    require 'earth/data_miner'
   end
 
-  it 'should create a fallbacks table' do
-    Fallback.should be_table_exists
+  describe '.domains' do
+    it 'should return a list of all domains' do
+      Earth.domains.should == %w{air automobile bus diet fuel hospitality industry locality pet rail residence}
+    end
   end
 end
 

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 2
-  - 11
-  version: 0.2.11
+  - 12
+  version: 0.2.12
 platform: ruby
 authors: 
 - Seamus Abshere