GeorgeKaraszi/active_record_extended

# frozen_string_literal: true

require "bundler/gem_tasks"
require "rspec/core/rake_task"

RSpec::Core::RakeTask.new(:spec)

task default: :spec

desc "Initialize the ENV file for connecting to the test database"
task :setup do
  if File.exist?(".env")
    puts "This will overwrite your existing .env file"
  end
  db_name     = fetch_input("Enter your database name: [active_record_extended_test] ")
  db_user     = fetch_input("Enter your database user: [] ")
  db_password = fetch_input("Enter your database password: [] ")
  db_server   = fetch_input("Enter your database server: [localhost] ")

  db_name     = "active_record_extended_test" if db_name.empty?
  db_password = ":#{db_password}"             unless db_password.empty?
  db_server   = "localhost"                   if db_server.empty?
  db_server   = "@#{db_server}"               unless db_user.empty?
  env_path    = File.expand_path("./.env")

  File.open(env_path, "w") do |file|
    file.puts "DATABASE_NAME=#{db_name}"
    file.puts "DATABASE_URL=\"postgres://#{db_user}#{db_password}#{db_server}/#{db_name}\""
  end

  puts ".env file saved"
end

# rubocop:disable Metrics/BlockLength

namespace :db do
  desc "Loads the test database ENV file"
  task :load_db_settings do
    require "active_record"
    unless ENV["DATABASE_URL"]
      require "dotenv"
      Dotenv.load
    end
  end

  desc "Drop test database"
  task drop: :load_db_settings do
    db_config = if ActiveRecord.version > Gem::Version.new("6.1")
      ActiveRecord::Base.configurations.resolve ENV["DATABASE_URL"]
    else
      ActiveRecord::ConnectionAdapters::ConnectionSpecification::ConnectionUrlResolver.new(ENV["DATABASE_URL"]).to_hash
    end

    ActiveRecord::Tasks::PostgreSQLDatabaseTasks.new(db_config).drop
  end

  desc "Create test database"
  task create: :load_db_settings do
    db_config = if ActiveRecord.version > Gem::Version.new("6.1")
      ActiveRecord::Base.configurations.resolve ENV["DATABASE_URL"]
    else
      ActiveRecord::ConnectionAdapters::ConnectionSpecification::ConnectionUrlResolver.new(ENV["DATABASE_URL"]).to_hash
    end

    ActiveRecord::Tasks::PostgreSQLDatabaseTasks.new(db_config).create
  end

  desc "Migrate schema to the test database"
  task migrate: :load_db_settings do
    ActiveRecord::Base.establish_connection(ENV["DATABASE_URL"])

    ActiveRecord::Schema.define do
      enable_extension "hstore"

      create_table :users, force: true do |t|
        t.integer  "tag_ids", array: true
        t.string   "name"
        t.string   "tags",         array: true
        t.integer  "number",       default: 0
        t.integer  "personal_id"
        t.hstore   "data"
        t.jsonb    "jsonb_data"
        t.inet     "ip"
        t.cidr     "subnet"
        t.datetime "created_at"
        t.datetime "updated_at"
      end

      create_table :sti_records, force: true do |t|
        t.string "type", default: "StiRecord"
      end

      create_table :namespaced_records, force: true do |t|
        t.inet "ip"
        t.cidr "subnet"
      end

      create_table :tags, force: true do |t|
        t.belongs_to :user, index: true, foreign_key: true
        t.integer :tag_number, default: 0
      end

      create_table :profile_ls, force: true do |t|
        t.belongs_to :user, index: true, foreign_key: true
        t.integer :likes
      end

      create_table :profile_rs, force: true do |t|
        t.belongs_to :user, index: true, foreign_key: true
        t.integer :dislikes
      end

      create_table :version_controls, force: true do |t|
        t.references :versionable, polymorphic: true, index: true, null: false
        t.jsonb :source, default: {}, null: false
      end

      create_table :groups, force: true

      create_table :groups_users, force: true do |t|
        t.belongs_to :user, index: true, foreign_key: true
        t.belongs_to :group, index: true, foreign_key: true
      end
    end

    puts "Database migrated"
  end

  desc "Creates and migrates the schema to the test database"
  task setup: :load_db_settings do
    unless ENV["DATABASE_URL"]
      Rake::Task["setup"].invoke
      Dotenv.load
    end

    Rake::Task["db:create"].invoke
    Rake::Task["db:migrate"].invoke
  end
end

# rubocop:enable Metrics/BlockLength

def fetch_input(message)
  print message
  $stdin.gets.chomp
end