Merge pull request #6761 from freelancing-god/db-rake

Get logic out of db rake tasks, and into classes and objects

1 files changed, 37 insertions, 0 deletions

diff --git a/activerecord/lib/active_record/tasks/sqlite_database_tasks.rb b/activerecord/lib/active_record/tasks/sqlite_database_tasks.rb
new file mode 100644
index 0000000000..4d3cba8a50
--- /dev/null
+++ b/activerecord/lib/active_record/tasks/sqlite_database_tasks.rb
@@ -0,0 +1,37 @@
+class ActiveRecord::Tasks::SQLiteDatabaseTasks
+  delegate :connection, :establish_connection, :to => ActiveRecord::Base
+
+  def initialize(configuration, root = Rails.root)
+    @configuration, @root = configuration, root
+  end
+
+  def create
+    if File.exist? configuration['database']
+      $stderr.puts "#{configuration['database']} already exists"
+      return
+    end
+
+    establish_connection configuration
+    connection
+  end
+
+  def drop
+    require 'pathname'
+    path = Pathname.new configuration['database']
+    file = path.absolute? ? path.to_s : File.join(root, path)
+
+    FileUtils.rm(file)
+  end
+
+  alias :purge :drop
+
+  private
+
+  def configuration
+    @configuration
+  end
+
+  def root
+    @root
+  end
+end