refactor

Author: pjurewicz

ruby_event_store-active_record/lib/ruby_event_store/active_record/tasks/migration_tasks.rake

@@ -2,33 +2,52 @@ require_relative "../generators/migration_generator"
 
 desc "Generate migration"
 task "db:migrations:copy" do
-  data_type =
-    ENV["DATA_TYPE"] || raise("Specify data type (binary, json, jsonb): rake db:migrations:copy DATA_TYPE=json")
-  ::ActiveRecord::Base.establish_connection(ENV["DATABASE_URL"])
-  database_adapter =
-    RubyEventStore::ActiveRecord::DatabaseAdapter.from_string(::ActiveRecord::Base.connection.adapter_name, data_type)
+  task = MigrationTask.new("db:migrations:copy")
+  task.establish_connection
 
   path =
-    RubyEventStore::ActiveRecord::MigrationGenerator.new.call(database_adapter, ENV["MIGRATION_PATH"] || "db/migrate")
+    RubyEventStore::ActiveRecord::MigrationGenerator.new.call(task.adapter, task.migration_path)
 
   puts "Migration file created #{path}"
 end
 
 desc "Generate migration for missing event_id index"
 task "db:migrations:fix_missing_event_id_index" do
-  ::ActiveRecord::Base.establish_connection(ENV["DATABASE_URL"])
+  task = MigrationTask.new("db:migrations:fix_missing_event_id_index")
+  task.establish_connection
 
-  path = RubyEventStore::ActiveRecord::EventIdIndexMigrationGenerator.new.call(ENV["MIGRATION_PATH"] || "db/migrate")
+  path = RubyEventStore::ActiveRecord::EventIdIndexMigrationGenerator.new.call(task.migration_path)
 
   puts "Migration file created #{path}"
 end
 
 desc "Generate migration for adding foreign key on event_store_events_in_streams.event_id"
 task "db:migrations:add_foreign_key_on_event_id" do
-  ::ActiveRecord::Base.establish_connection(ENV["DATABASE_URL"])
+  task = MigrationTask.new("db:migrations:add_foreign_key_on_event_id")
+  task.establish_connection
 
   path =
-    RubyEventStore::ActiveRecord::ForeignKeyOnEventIdMigrationGenerator.new.call(ENV["MIGRATION_PATH"] || "db/migrate")
+    RubyEventStore::ActiveRecord::ForeignKeyOnEventIdMigrationGenerator.new.call(task.migration_path)
 
   puts "Migration file created #{path}"
 end
+
+class MigrationTask
+  def initialize(task)
+    @task = task
+  end
+
+  def establish_connection
+    ::ActiveRecord::Base.establish_connection(ENV["DATABASE_URL"])
+  end
+
+  def adapter
+    data_type = ENV["DATA_TYPE"] || raise("Specify data type (binary, json, jsonb): rake #{@task} DATA_TYPE=json")
+
+    RubyEventStore::ActiveRecord::DatabaseAdapter.from_string(::ActiveRecord::Base.connection.adapter_name, data_type)
+  end
+
+  def migration_path
+    ENV["MIGRATION_PATH"] || "db/migrate"
+  end
+end