generator.rb in shoe-0.6.0

- old
+ new
@@ -1,64 +1,146 @@
 require 'erb'
+require 'optparse'
 require 'pathname'
 
 module Shoe
   class Generator
-    def initialize(root)
-      @root          = Pathname.new(root).expand_path
-      @template_path = Pathname.new(__FILE__).dirname.join('templates')
+    def initialize
+      @options = {}
+      @parser  = OptionParser.new do |opts|
+        opts.extend(Extensions::OptionParser)
+
+        opts.banner   = "Usage: #{File.basename($0)} [-adehtv] [path]"
+        opts.defaults = %w(--no-application --no-data --no-extension --test-unit .)
+        opts.version  = Shoe::VERSION
+
+        opts.on('-a', '--[no-]application', 'Generate a command-line application.') do |application|
+          @options[:application] = application
+        end
+
+        opts.on('-d', '--[no-]data', 'Generate a data directory.') do |data|
+          @options[:data] = data
+        end
+
+        opts.on('-e', '--[no-]extension', 'Generate a C extension.') do |extension|
+          @options[:extension] = extension
+        end
+
+        opts.on('-t', '--[no-]test-unit', 'Generate Test::Unit tests.') do |test_unit|
+          @options[:test_unit] = test_unit
+        end
+      end
     end
 
-    def run
-      path('README.rdoc').install template('readme.erb')
-      path('Rakefile').install    template('rakefile.erb')
-      path(version_path).install  template('version.erb')
-      path(gemspec_path).install  template('gemspec.erb')
+    def run(arguments)
+      @parser.order(arguments) { |path| @project = Project.new(path, @options) }
+
+      @project.generate
     end
 
     private
 
-    def project_name
-      @root.basename.to_s
-    end
 
-    def project_module
-      project_name.capitalize.gsub(/_(\w)/) { $1.upcase }
-    end
+    class Project
+      def initialize(path, options)
+        @path    = Pathname.new(path)
+        @options = options
+      end
 
-    def project_version
-      '0.0.0'
-    end
+      def generate
+        install('gitignore.erb', '.gitignore')
+        install('readme.erb',    'README.rdoc')
+        install('rakefile.erb',  'Rakefile')
+        install('gemspec.erb',   "#{name}.gemspec")
+        install('module.erb',    "lib/#{name}.rb")
 
-    def version_path
-      "lib/#{project_name}/version.rb"
-    end
+        if application?
+          install('executable.erb',  "bin/#{name}", 0755)
+          install('application.erb', "lib/#{name}/application.rb")
+          install('manpage.erb',     "man/#{name}.1.ronn")
+        end
 
-    def gemspec_path
-      "#{project_name}.gemspec"
-    end
+        if data?
+          install('gitkeep.erb', "data/#{name}/.gitkeep")
+        end
 
-    def template(name)
-      ERB.new(template_contents(name)).result(binding)
-    end
+        if extension?
+          install('extconf.erb',   "ext/#{name}/extconf.rb")
+          install('extension.erb', "ext/#{name}/#{extension_name}.c")
+        end
 
-    def template_contents(name)
-      @template_path.join(name).read
-    end
+        if test_unit?
+          install('test_helper.erb', "test/helper.rb")
+          install('module_test.rb',  "test/#{name}_test.rb")
+        end
+      end
 
-    def path(name)
-      path = @root.join(name)
-      path.dirname.mkpath
-      path.extend(PathExtensions)
+      private
+
+      def install(template, path, mode=0644)
+        installable_path(path).install(contents(template), mode)
+      end
+
+      def application?
+        @options[:application]
+      end
+
+      def data?
+        @options[:data]
+      end
+
+      def extension?
+        @options[:extension]
+      end
+
+      def test_unit?
+        @options[:test_unit]
+      end
+
+      def name
+        @path.expand_path.basename.to_s
+      end
+
+      def module_name
+        name.capitalize.gsub(/_(\w)/) { $1.upcase }
+      end
+
+      def version
+        '0.0.0'
+      end
+
+      def extension_name
+        'extension'
+      end
+
+      def extension_module_name
+        extension_name.capitalize.gsub(/_(\w)/) { $1.upcase }
+      end
+
+      def installable_path(name)
+        path = @path.join(name)
+        path.dirname.mkpath
+        path.extend(Extensions::Pathname)
+      end
+
+      def contents(template)
+        Template.new(template).evaluate(binding)
+      end
     end
 
-    module PathExtensions #:nodoc:
-      def install(contents)
-        if exist?
-          $stderr.puts "#{to_s} exists. Not clobbering."
-        else
-          open('w') { |file| file.write(contents) }
-        end
+    class Template
+      def initialize(name)
+        @name = name
+      end
+
+      def evaluate(binding)
+        ERB.new(contents, nil, '<>').result(binding)
+      end
+
+      private
+
+      def contents
+        Shoe.datadir.join('templates', @name).read
       end
     end
   end
 end