module Ferry
  class Utilities
    def db_connect(environment)
      db_config = YAML::load(IO.read("config/database.yml"))

      if db_config[environment].nil?
        raise "No database associated with #{environment} environment"
      end

      db_type = db_config[environment]["adapter"]

      if ['sqlite3', 'postgresql', 'mysql2'].include?(db_type)
        ActiveRecord::Base.establish_connection(adapter: db_type, database: db_config[environment]['database'])
        puts "operating with "+db_type
        return db_type
      else
        raise "#{db_type} is not supported by ferry at this time"
      end
    end

    def make_starter_file
      if !File.exist?("lib/tasks/ferry.rake")
        install_dir = `bundle show ferry`.chomp
        starter_file_contents = File.open("#{install_dir}/doc/ferry_rake_contents.rb", "rb")
        contents = starter_file_contents.read
        File.open("lib/tasks/ferry.rake", 'w') {|f| f.write(contents)}
        puts "/lib/tasks/ferry.rake created!"
      else
        puts "/lib/tasks/ferry.rake already exists - but you knew that already ... didn't you?"
      end
    end

    def print_version
      puts "Ferry #{Ferry::VERSION}"
    end
  end
end