class Erector::Erect

# File lib/erector/erect/erect.rb, line 11
def initialize(args)
  @verbose = true
  @mode = :to_erector
  @output_dir = nil
  @superklass = 'Erector::Widget'
  @method_name = 'content'

  opts = OptionParser.new do |opts|
    opts.banner = "Usage: erector [options] [file|dir]*"

    opts.separator "Converts from html/rhtml files to erector widgets, or from erector widgets to html files"
    opts.separator ""
    opts.separator "Options:"

    opts.on("-q", "--quiet",
            "Operate silently except in case of error") do |quiet|
      @verbose = !quiet
    end

    opts.on("--to-erector", "(default) Convert from html/rhtml to erector classes") do
      @mode = :to_erector
    end

    opts.on("--to-html", "Convert from erector to html") do
      @mode = :to_html
    end

    opts.on("--superclass SUPERCLASS", "Superclass for new widget (default Erector::Widget)") do |superklass|
      @superklass = superklass
    end

    opts.on("--method METHOD", "Method containing content for widget (default 'content')") do |method_name|
      @method_name = method_name
    end

    opts.on("-o", "--output-dir DIRECTORY", "Output files to DIRECTORY (default: output files go next to input files)") do |dir|
      @output_dir = dir
    end

    opts.on_tail("-h", "--help", "Show this message") do
      @mode = :help
      puts opts
      exit
    end

    opts.on_tail("-v", "--version", "Show version") do
      puts Erector::VERSION
      exit
    end

  end
  opts.parse!(args)
  @files = args
  explode_dirs
end

# File lib/erector/erect/erect.rb, line 151
def camelize(lower_case_and_underscored_word, first_letter_in_uppercase = true)
  if first_letter_in_uppercase
    lower_case_and_underscored_word.to_s.gsub(/\/(.?)/) { "::" + $1.upcase }.gsub(/(^|_)(.)/) { $2.upcase }
  else
    lower_case_and_underscored_word.first + camelize(lower_case_and_underscored_word)[1..-1]
  end
end

# File lib/erector/erect/erect.rb, line 158
def constantize(camel_cased_word)
  unless /\A(?:::)?([A-Z]\w*(?:::[A-Z]\w*)*)\z/ =~ camel_cased_word
    raise NameError, "#{camel_cased_word.inspect} is not a valid constant name!"
  end
  Object.module_eval("::#{$1}", __FILE__, __LINE__)
end

# File lib/erector/erect/erect.rb, line 84
def explode(dir)
  case mode
  when :to_erector
    ["#{dir}/**/*.rhtml",
      "#{dir}/**/*.html",
      "#{dir}/**/*.html.erb"].map do |pattern|
      Dir.glob pattern
    end.flatten
  when :to_html
    Dir.glob "#{dir}/**/*.rb"
  end
end

# File lib/erector/erect/erect.rb, line 72
def explode_dirs
  exploded_files = []
  files.each do |file|
    if File.directory?(file)
      exploded_files << explode(file)
    else
      exploded_files << file
    end
  end
  @files = exploded_files.flatten
end

# File lib/erector/erect/erect.rb, line 97
def run
  @success = true
  self.send(mode)
  @success
end

# File lib/erector/erect/erect.rb, line 67
def say(msg)
  print msg if verbose
end

# File lib/erector/erect/erect.rb, line 103
def to_erector
  files.each do |file|
    say "Erecting #{file}... "
    begin
      e = Erector::Erected.new(file, @superklass, @method_name)
      e.convert
      say " --> #{e.filename}\n"
    rescue => e
      puts e
      puts e.backtrace.join("\n\t")
      @success = false
    end
  end
end

# File lib/erector/erect/erect.rb, line 118
def to_html
  files.each do |file|
    say "Erecting #{file}... "
    #todo: move this into Erected with better tests for the naming methods
    begin
      #todo: fail if file isn't a .rb file
      require file
      filename = file.split('/').last.gsub(/\.rb$/, '')
      widget_name = camelize(filename)
      widget_class = constantize(widget_name)

      if widget_class < Erector::Widget
        widget = widget_class.new
        #todo: skip if it's missing a no-arg constructor
        dir = output_dir || File.dirname(file)
        ::FileUtils.mkdir_p(dir)
        output_file = "#{dir}/#{filename}.html"
        File.open(output_file, "w") do |f|
          f.puts widget.to_html
        end
        say " --> #{output_file}\n"
      else
        say " -- not a widget, skipping\n"
      end
    rescue => e
      puts e
      puts e.backtrace.join("\n\t")
      @success = false
    end
  end
end