modify rakefile to support both sqlite3 and mysql

Author: parsahonarmand99

Rakefile

@@ -15,34 +15,59 @@ end
 
 namespace :db do
   database_config = YAML.load(File.open("./spec/support/database.yml"))
-  admin_database_config = database_config.merge(database: "mysql")
   migration_path = File.expand_path("./spec/support/migrations")
 
   desc "Create the database"
   task :create do
-    ActiveRecord::Base.establish_connection(admin_database_config)
-    ActiveRecord::Base.connection.create_database(database_config.fetch(:database))
-    puts "Database created."
+    if database_config[:adapter] == 'sqlite3'
+      # For SQLite3, just ensure the directory exists
+      db_file = database_config.fetch(:database)
+      FileUtils.mkdir_p(File.dirname(db_file)) unless File.dirname(db_file) == '.'
+      puts "Database created (SQLite3 will create file on first connection)."
+    else
+      # For other databases (MySQL, PostgreSQL, etc.)
+      admin_config = database_config.merge(database: "mysql")
+      ActiveRecord::Base.establish_connection(admin_config)
+      ActiveRecord::Base.connection.create_database(database_config.fetch(:database))
+      puts "Database created."
+    end
   end
 
   desc "Migrate the database"
   task :migrate do
     ActiveRecord::Base.establish_connection(database_config)
-    ActiveRecord::Migrator.migrate(migration_path)
+    
+    # Handle different Rails versions
+    active_record_version = Gem::Version.new(ActiveRecord::VERSION::STRING)
+    
+    if active_record_version >= Gem::Version.new("6.0")
+      ActiveRecord::MigrationContext.new(migration_path).migrate
+    else
+      ActiveRecord::Migrator.migrate(migration_path)
+    end
+    
     Rake::Task["db:schema"].invoke
     puts "Database migrated."
   end
 
   desc "Drop the database"
   task :drop do
-    ActiveRecord::Base.establish_connection(admin_database_config)
-    ActiveRecord::Base.connection.drop_database(database_config.fetch(:database))
+    if database_config[:adapter] == 'sqlite3'
+      # For SQLite3, just delete the file
+      db_file = database_config.fetch(:database)
+      File.delete(db_file) if File.exist?(db_file)
+    else
+      # For other databases (MySQL, PostgreSQL, etc.)
+      admin_config = database_config.merge(database: "mysql")
+      ActiveRecord::Base.establish_connection(admin_config)
+      ActiveRecord::Base.connection.drop_database(database_config.fetch(:database))
+    end
     puts "Database deleted."
   end
 
   desc "Reset the database"
   task reset: [:drop, :create, :migrate]
-    desc 'Create a db/schema.rb file that is portable against any DB supported by AR'
+  desc 'Create a db/schema.rb file that is portable against any DB supported by AR'
 
   task :schema do
     # Noop to make ActiveRecord happy