module Wunderbar

  module CGI

    # produce json
    def self.json(&block)
      builder = JsonBuilder.new
      output = builder.encode($param, &block)
      Kernel.print "Status: 404 Not Found\r\n" if output == {}
    rescue Exception => exception
      Kernel.print "Status: 500 Internal Error\r\n"
      Wunderbar.error exception.inspect
      backtrace = []
      exception.backtrace.each do |frame| 
        next if frame =~ %r{/wunderbar/}
        next if frame =~ %r{/gems/.*/builder/}
        Wunderbar.warn "  #{frame}"
        backtrace << frame 
      end
      builder = JsonBuilder.new
      builder._exception exception.inspect
      builder._backtrace backtrace
    ensure
      out? 'type' => 'application/json', 'Cache-Control' => 'no-cache' do
        builder.target!
      end
    end

    # produce text
    def self.text &block
      builder = TextBuilder.new
      output = builder.encode($param, &block)
      Kernel.print "Status: 404 Not Found\r\n" if output == ''
    rescue Exception => exception
      Wunderbar.error exception.inspect
      Kernel.print "Status: 500 Internal Error\r\n"
      builder.puts unless builder.size == 0
      builder.puts exception.inspect
      exception.backtrace.each do |frame| 
        next if frame =~ %r{/wunderbar/}
        next if frame =~ %r{/gems/.*/builder/}
        Wunderbar.warn "  #{frame}"
        builder.puts "  #{frame}"
      end
    ensure
      out? 'type' => 'text/plain', 'Cache-Control' => 'no-cache' do
        builder.target!
      end
    end

    # Conditionally provide output, based on ETAG
    def self.out?(headers, &block)
      content = block.call
      require 'digest/md5'
      etag = Digest::MD5.hexdigest(content)

      if ENV['HTTP_IF_NONE_MATCH'] == etag.inspect
        Kernel.print "Status: 304 Not Modified\r\n\r\n"
      else
        $cgi.out headers.merge('Etag' => etag.inspect) do
          content
        end
      end
    rescue
    end

    # produce html/xhtml
    def self.html(*args, &block)
      args << {} if args.empty?
      if Hash === args.first
        args.first[:xmlns] ||= 'http://www.w3.org/1999/xhtml'
      end
      mimetype = ($XHTML ? 'application/xhtml+xml' : 'text/html')
      x = HtmlMarkup.new
      x._! "\xEF\xBB\xBF"
      x._.declare :DOCTYPE, :html

      begin
        output = x.html *args, &block
      rescue ::Exception => exception
        Kernel.print "Status: 500 Internal Error\r\n"
        x.clear!
        x._! "\xEF\xBB\xBF"
        x._.declare :DOCTYPE, :html
        output = x.html(*args) do
          _head do
            _title 'Internal Error'
          end
          _body do
            _h1 'Internal Error'
            text = exception.inspect
            Wunderbar.error text
            exception.backtrace.each do |frame| 
              next if frame =~ %r{/wunderbar/}
              next if frame =~ %r{/gems/.*/builder/}
              Wunderbar.warn "  #{frame}"
              text += "\n  #{frame}"
            end
    
            _pre text
          end
        end
      end

      out? 'type' => mimetype, 'charset' => 'UTF-8' do
        output
      end
    end
  end

  @queue = []

  # canonical interface
  def self.html(*args, &block)
    @queue << [:html, args, block]
  end

  def self.xhtml(*args, &block)
    @queue << [:xhtml, args, block]
  end

  def self.json(*args, &block)
    @queue << [:json, args, block]
  end

  def self.text(*args, &block)
    @queue << [:text, args, block]
  end

  def self.evaluate
    queue, @queue = @queue, []
    xhtml = ARGV.delete('--xhtml')
    html  = ARGV.delete('--html')

    queue.each do |type, args, block|
      case type
      when :html
        unless $XHR_JSON or $TEXT
          $XHTML = false unless xhtml
          CGI.html(*args, &block)
          break
        end
      when :xhtml
        unless $XHR_JSON or $TEXT
          $XHTML = false if html
          CGI.html(*args, &block)
          break
        end
      when :json
        if $XHR_JSON
          CGI.json(*args, &block)
          break
        end
      when :text
        if $TEXT
          CGI.text(*args, &block)
          break
        end
      end
    end
  end
end

at_exit do
  Wunderbar.evaluate
end