# File lib/webby/apps/generator.rb, line 27
  def initialize( output = $stdout, input = $stdin )
    @options = {}
    @site = @template = nil
    @output, @input = output, input
    @journal = journal
  end

# File lib/webby/apps/generator.rb, line 18
  def self.run( args )
    self.new.run args
  end

# File lib/webby/apps/generator.rb, line 219
  def abort( msg )
    puts "\nAborting!"
    puts "    #{msg}"
    puts
    exit 1
  end

# File lib/webby/apps/generator.rb, line 189
  def cp( file )
    src = template / file
    dst = site / file

    if test(?e, dst)
      if identical?(src, dst)
        identical(dst)
        return
      end

      choice = case options[:collision]
        when :force then :force
        when :skip  then :skip
        else force_file_collision?( dst ) end

      case choice
        when :force then force(dst)
        when :skip  then skip(dst); return
        else raise "Invalid collision choice: #{choice.inspect}" end
    else
      create(dst)
    end
    return if pretend?

    FileUtils.cp(src, dst)
  end

# File lib/webby/apps/generator.rb, line 145
  def create_site
    files = site_files

    # in update mode we only want to update the tasks directory
    if options[:update]
      FileUtils.mkdir_p site unless pretend?
      mkdir 'tasks'
      files['tasks'].sort.each {|file| cp file}
    else
      dirs = files.keys.concat %w[content layouts lib tasks templates]
      dirs.sort!
      dirs.uniq!

      # create the directories first
      dirs.each do |dir|
        next if dir =~ %r/^output\/.*$/
        mkdir dir
      end

      # and the create the files under each directory
      dirs.each do |dir|
        next if dir =~ %r/^output(\/.*)?$/
        files[dir].sort.each {|file| cp file}
      end
    end
  end

# File lib/webby/apps/generator.rb, line 260
  def force_file_collision?( dst )
    dst = dst.sub(%r/#{site}\/?/, '')
    print "overwrite #{dst}? [(Y)es (n)o (q)uit] "
    case gets
      when %r/q/i  then abort 'user asked to quit'
      when %r/n/i  then :skip
      when %r/y/i  then :force
      when %r/\s*/ then :force
      else force_file_collision?(dst) end
  rescue
    retry
  end

# File lib/webby/apps/generator.rb, line 53
  def gets
    @input.gets
  end

# File lib/webby/apps/generator.rb, line 252
  def identical?( src, dst )
    source      = IO.read(src)
    destination = IO.read(dst)
    source == destination
  end

# File lib/webby/apps/generator.rb, line 175
  def mkdir( dir )
    dir = dir.empty? ? site : site / dir
    if test ?d, dir
      exists dir
    else
      create dir
      FileUtils.mkdir_p dir unless pretend?
    end
  end

# File lib/webby/apps/generator.rb, line 67
  def parse( args )
    opts = OptionParser.new
    opts.banner = 'Usage: webby-gen [options] template site'

    opts.separator ''
    opts.separator 'The webby-gen command is used to generate a site from a standard template.'
    opts.separator 'A new site can be created, or an existing site can be added to or updated.'

    opts.separator ''
    opts.on('-f', '--force',
            'overwrite files that already exist') {options[:collision] = :force}
    opts.on('-s', '--skip',
            'skip files that already exist') {options[:collision] = :skip}
    opts.on('-u', '--update',
            'update rake tasks for the site') {options[:update] = true}
    opts.on('-p', '--pretend',
            'run but do not make any changes') {options[:pretend] = true}

    opts.separator ''
    opts.on('-t', '--templates', 'list available templates') {
      ary = templates.map {|t| ::File.basename(t)}
      ary.delete 'webby'
      puts "\nAvailable Templates"
      puts "    #{ary.join(', ')}"
      puts
      exit
    }

    opts.separator ''
    opts.separator 'common options:'

    opts.on( '-h', '--help', 'show this message' ) {puts opts; exit}
    opts.on( '--version', 'show version' ) do
      puts "Webby #{::Webby::VERSION}"
      exit
    end

    # parse the command line arguments
    opts.parse! args
    tmpl, @site = args

    # if no site was given, see if there is a Sitefile in the current
    # directory
    if site.nil?
      self.site = '.' if test(?f, 'Sitefile')
    end

    # exit if comand line args are missing
    if site.nil? or tmpl.nil?
      puts opts
      exit 1
    end

    templates.each {|t| self.template = t if t =~ %r/\/#{tmpl}$/}
    if template.nil?
      puts opts
      abort "Could not find template '#{tmpl}'"
    end

    nil
  end

# File lib/webby/apps/generator.rb, line 133
  def pretend?
    options[:pretend] == true
  end

# File lib/webby/apps/generator.rb, line 47
  def print( *args )
    @output.print(*args)
  end

# File lib/webby/apps/generator.rb, line 41
  def puts( *args )
    @output.puts(*args)
  end

# File lib/webby/apps/generator.rb, line 60
  def run( args )
    parse args
    create_site
  end

# File lib/webby/apps/generator.rb, line 229
  def site_files
    exclude = %r/tmp$|bak$|~$|CVS|\.svn/o

    rgxp = %r/\A#{template}\/?/
    paths = Hash.new {|h,k| h[k] = []}

    Find.find(template) do |p|
      next if exclude =~ p

      if test(?d, p)
        paths[p.sub(rgxp, '')]
        next
      end
      dir = ::File.dirname(p).sub(rgxp, '')
      paths[dir] << p.sub(rgxp, '')
    end

    paths
  end

# File lib/webby/apps/generator.rb, line 139
  def templates
    Dir.glob(::Webby.path('examples') / '*').sort
  end