lib/seed_gimmick/options.rb in seed_gimmick-0.0.1 vs lib/seed_gimmick/options.rb in seed_gimmick-0.0.2

@@ -1,23 +1,79 @@
 module SeedGimmick
   class Options
-    %i(seed_dir).each do |key|
+    VALID_OPTIONS_KEYS = %i(
+    seed_dir
+    tables
+    models
+    default_ext
+    exclude_columns
+    ).freeze
+
+    VALID_OPTIONS_KEYS.each do |key|
       define_method "#{key}=" do |value|
         @options[key] = value
       end
     end
 
     def initialize(options = {})
-      @options = options.symbolize_keys
+      @options = load_config.merge(options).symbolize_keys
     end
 
     def seed_dir
-      @options[:seed_dir] || default_seed_dir
+      root_dir.join(@options[:seed_dir] || default_seed_dir)
     end
 
+    def tables
+      @options[:tables] || ENV["TABLES"].try(:split, ",") || []
+    end
+
+    def tables!
+      tables.presence || (raise SeedGimmickError)
+    end
+
+    def models
+      @options[:models] || ENV["MODELS"].try(:split, ",") || []
+    end
+
+    def models!
+      models.presence || (raise SeedGimmickError)
+    end
+
+    def default_ext
+      @options[:default_ext] || ENV["FORMAT"] || "yml"
+    end
+
+    def exclude_columns
+      @options[:exclude_columns] || default_exclude_columns
+    end
+
+    def load_config
+      config.exist? ? YAML.load_file(config)[environment] : {}
+    end
+
+    def environment
+      ENV["RAILS_ENV"] || ENV["APP_ENV"] || default_env
+    end
+
     private
       def default_seed_dir
-        (defined?(Rails) ? Rails.root : Pathname.pwd).join("db", "seeds")
+        Pathname.new("db/seed_gimmick")
+      end
+
+      def default_exclude_columns
+        ["created_at", "updated_at"]
+      end
+
+      def default_env
+        defined?(Rails) ? Rails.env : "development".inquiry
+      end
+
+      def config
+        root_dir.join("config", "seed_gimmick.yml")
+      end
+
+      def root_dir
+        defined?(Rails) ? Rails.root : Pathname.pwd
       end
   end
 end