module Apartment::TaskHelper

def self.create_tenant(tenant_name)
  puts("Creating #{tenant_name} tenant")
  Apartment::Tenant.create(tenant_name)
rescue Apartment::TenantExists => e
  puts "Tried to create already existing tenant: #{e}"
end

def self.each_tenant(&block)
  Parallel.each(tenants_without_default, in_threads: Apartment.parallel_migration_threads) do |tenant|
    block.call(tenant)
  end
end

def self.migrate_tenant(tenant_name)
  strategy = Apartment.db_migrate_tenant_missing_strategy
  create_tenant(tenant_name) if strategy == :create_tenant
  puts("Migrating #{tenant_name} tenant")
  Apartment::Migrator.migrate tenant_name
rescue Apartment::TenantNotFound => e
  raise e if strategy == :raise_exception
  puts e.message
end

def self.tenants
  ENV['DB'] ? ENV['DB'].split(',').map(&:strip) : Apartment.tenant_names || []
end

def self.tenants_without_default
  tenants - [Apartment.default_tenant]
end

def self.warn_if_tenants_empty
  return unless tenants.empty? && ENV['IGNORE_EMPTY_TENANTS'] != 'true'
  puts <<-WARNING
    [WARNING] - The list of tenants to migrate appears to be empty. This could mean a few things:
      1. You may not have created any, in which case you can ignore this message
      2. You've run `apartment:migrate` directly without loading the Rails environment
        * `apartment:migrate` is now deprecated. Tenants will automatically be migrated with `db:migrate`
    Note that your tenants currently haven't been migrated. You'll need to run `db:migrate` to rectify this.
  WARNING
end