Merge 12e3011 into 107ba07

palan-inc · Jul 4, 2018 · 8054894 · 8054894
2 parents 107ba07 + 12e3011
commit 8054894
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git a/lib/generators/schemate/templates/schemate.rb b/lib/generators/schemate/templates/schemate.rb
@@ -1,4 +1,5 @@
 # frozen_string_literal: true
 Schemate.configure do |config|
   config.auto_generate = false
+  config.auto_generate_type = 'md'
 end
diff --git a/lib/schemate/config.rb b/lib/schemate/config.rb
@@ -11,9 +11,11 @@ def self.config
   class Configuration
     include ActiveSupport::Configurable
     config_accessor :auto_generate
+    config_accessor :auto_generate_type
   end
 
   configure do |config|
     config.auto_generate = false
+    config.auto_generate_type = 'md'
   end
 end
diff --git a/lib/tasks/schemate_migrate.rake b/lib/tasks/schemate_migrate.rake
@@ -3,13 +3,21 @@ require "schemate/config"
 namespace :db do
   [:migrate, :rollback].each do |cmd|
     task cmd do
-      Rake::Task['schemate:export_md'].invoke if Schemate.config.auto_generate
+      if Schemate.config.auto_generate_type == 'md'
+        Rake::Task['schemate:export_md'].invoke if Schemate.config.auto_generate
+      elsif Schemate.config.auto_generate_type == 'csv'
+        Rake::Task['schemate:export_csv'].invoke if Schemate.config.auto_generate
+      end
     end
 
     namespace cmd do
       [:change, :up, :down, :reset, :redo].each do |t|
         task t do
-          Rake::Task['schemate:export_md'].invoke if Schemate.config.auto_generate
+          if Schemate.config.auto_generate_type == 'md'
+            Rake::Task['schemate:export_md'].invoke if Schemate.config.auto_generate
+          elsif Schemate.config.auto_generate_type == 'csv'
+            Rake::Task['schemate:export_csv'].invoke if Schemate.config.auto_generate
+          end
         end
       end
     end