lib/bitters/generator.rb in bitters-0.0.2 vs lib/bitters/generator.rb in bitters-0.9.0.beta.1

@@ -1,63 +1,63 @@
-require "fileutils"
+require 'bitters/version'
+require 'fileutils'
+require 'thor'
 
 module Bitters
-  class Generator
-    def initialize(arguments)
-      @subcommand = arguments.first
-    end
+  class Generator < Thor
+    map ['-v', '--version'] => :version
+    map ['scaffold'] => :install
+    map ['delete'] => :remove
 
-    def run
-      if @subcommand == "install"
-        install
-      elsif @subcommand == "update"
-        update
-      elsif @subcommand == "remove"
-        remove
+    desc 'install', 'Install Bitters into your project'
+    def install
+      if bitters_files_already_exist?
+        puts "Bitters files already installed, doing nothing."
+      else
+        install_files
+        puts "Bitters files installed to /bitters"
       end
     end
 
+    desc 'reset', 'Reset Bitters'
     def update
       if bitters_files_already_exist?
         remove_bitters_directory
         install_files
         puts "Bitters files updated."
       else
         puts "No existing Bitters installation. Doing nothing."
       end
     end
 
-    def install
-      if bitters_files_already_exist?
-        puts "Bitters files already installed, doing nothing."
-      else
-        install_files
-        puts "Bitters files installed to /base"
-      end
-    end
-
+    desc 'remove', 'Remove Bitters'
     def remove
       if bitters_files_already_exist?
         remove_bitters_directory
         puts "Bitters was successfully removed."
       else
         puts "No existing Bitters installation. Doing nothing."
       end
     end
 
+    desc 'version', 'Show Bitters version'
+    def version
+      say "Bitters #{Bitters::VERSION}"
+    end
+
     private
 
     def bitters_files_already_exist?
-      File.directory?("base")
+      File.directory?("bitters")
     end
 
     def install_files
-      FileUtils.mkdir_p("base")
-      FileUtils.cp_r(all_stylesheets, "base/")
+      FileUtils.mkdir_p("bitters")
+      FileUtils.cp_r(all_stylesheets, "bitters/")
     end
 
     def remove_bitters_directory
-      FileUtils.rm_rf("base")
+      FileUtils.rm_rf("bitters")
     end
 
     def all_stylesheets
       Dir["#{stylesheets_directory}/*"]
     end