require 'erb'
module Passenger
 
class ConsoleTextTemplate
  TEMPLATE_DIR = "#{File.dirname(__FILE__)}/templates"
 
  def initialize(input, options = {})
    @buffer = ''
    if input[:file]
      data = File.read("#{TEMPLATE_DIR}/#{input[:file]}.txt.erb")
    else
      data = input[:text]
    end
    @template = ERB.new(substitute_color_tags(data),
      nil, nil, '@buffer')
    options.each_pair do |name, value|
      self[name] = value
    end
  end
  
  def []=(name, value)
    instance_variable_set("@#{name}".to_sym, value)
    return self
  end
  
  def result
    return @template.result(binding)
  end
 
private
  DEFAULT_TERMINAL_COLORS = "\e[0m\e[37m\e[40m"
 
  def substitute_color_tags(data)
    data = data.gsub(%r{<b>(.*?)</b>}, "\e[1m\\1#{DEFAULT_TERMINAL_COLORS}")
    data.gsub!(%r{<red>(.*?)</red>}, "\e[1m\e[31m\\1#{DEFAULT_TERMINAL_COLORS}")
    data.gsub!(%r{<green>(.*?)</green>}, "\e[1m\e[32m\\1#{DEFAULT_TERMINAL_COLORS}")
    data.gsub!(%r{<yellow>(.*?)</yellow>}, "\e[1m\e[33m\\1#{DEFAULT_TERMINAL_COLORS}")
    data.gsub!(%r{<banner>(.*?)</banner>}, "\e[33m\e[44m\e[1m\\1#{DEFAULT_TERMINAL_COLORS}")
    return data
  end
end
 
end # module Passenger