lib/ferry/utilities.rb in ferry-1.3.0 vs lib/ferry/utilities.rb in ferry-1.3.1

- old
+ new

@@ -1,25 +1,35 @@ module Ferry class Utilities + + def check_valid_db(db) + %w[sqlite3 postgresql mysql2].include?(db) ? true : false + end + + def check_valid_filetype(filepath) + %w[.csv .json .sql .yml].include?(filepath.extname) ? true : false + end + 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 execute(command) + `#{command}` + 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 @@ -31,7 +41,8 @@ end def print_version puts "Ferry #{Ferry::VERSION}" end + end end