lib/ferry/utilities.rb in ferry-1.0.0 vs lib/ferry/utilities.rb in ferry-1.0.1

@@ -1,18 +1,22 @@
 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
-        puts "Unsupported db type or no database associated with this application."
-        return false
+        raise "#{db_type} is not supported by ferry at this time"
       end
     end
 
     def continue?(prompt = "Are you sure", default = false)
       a = ''
@@ -23,17 +27,15 @@
         a = d if a.length == 0
       end
       a == 'y'
     end
 
-    def init
+    def make_starter_file
       if !File.exist?("lib/tasks/ferry.rake")
-        File.open("lib/tasks/ferry.rake", 'w') {|f| f.write("# this is your ferry init file
-# in this file you can write rake tasks that are easily tailored to more case-by-case user implementations
-
-namespace :ferry do
-  # your code here!
-end")}
+        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