lib/bitters/generator.rb in bitters-0.9.3 vs lib/bitters/generator.rb in bitters-0.9.4

@@ -1,22 +1,24 @@
 require 'bitters/version'
 require 'fileutils'
+require 'pathname'
 require 'thor'
 
 module Bitters
   class Generator < Thor
     map ['-v', '--version'] => :version
     map ['scaffold'] => :install
     map ['delete'] => :remove
 
     desc 'install', 'Install Bitters into your project'
+    method_options path: :string
     def install
       if bitters_files_already_exist?
         puts "Bitters files already installed, doing nothing."
       else
         install_files
-        puts "Bitters files installed to /bitters"
+        puts "Bitters files installed to #{install_path}/bitters"
       end
     end
 
     desc 'reset', 'Reset Bitters'
     def update
@@ -45,15 +47,19 @@
     end
 
     private
 
     def bitters_files_already_exist?
-      File.directory?("bitters")
+      File.directory?(install_path)
     end
 
+    def install_path
+      Pathname.new(options[:path].to_s).join('bitters')
+    end
+
     def install_files
-      FileUtils.mkdir_p("bitters")
-      FileUtils.cp_r(all_stylesheets, "bitters/")
+      FileUtils.mkdir_p(install_path)
+      FileUtils.cp_r(all_stylesheets, install_path)
     end
 
     def remove_bitters_directory
       FileUtils.rm_rf("bitters")
     end