lib/spontaneous/cli/generate.rb in spontaneous-0.2.0.beta4 vs lib/spontaneous/cli/generate.rb in spontaneous-0.2.0.beta5

@@ -16,12 +16,11 @@
     method_option :database, :type => :string, :default => "mysql", :aliases => "-d", :desc => "Database adapter"
     method_option :user, :type => :string, :default => "root", :aliases => "-u", :desc => "Database admin user"
     method_option :password, :type => :string, :default => "", :aliases => "-p", :desc => "Database admin user"
 
     def site(*args)
-      # require "spontaneous"
-      ::Spontaneous::Generators::Site.start(ARGV.drop_while { |e| %w(generate site).include?(e) })
+      generate_site(args)
     end
 
     def method_missing(method, *args)
       if PublicSuffix.valid?(method.to_s)
         puts method
@@ -29,9 +28,13 @@
         ARGV.unshift("site")
         self.send(:site, *args)
       else
         super
       end
+    end
+
+    def generate_site(args)
+      ::Spontaneous::Generators::Site.start(args.drop_while { |e| %w(generate site).include?(e) })
     end
   end # Generate
 end # Spontaneous::Cli