module Merb
  
  # controller class for the merb pocket-framework. All of your
  # controllers will inherit from Merb::Controller. This superclass
  # takes care of parsing the incoming headers and body into params
  # and cookies and headers. If the request is a file upload it will
  # stream it into a tempfile and pass in the filename and tempfile object
  # to your controller via params. It also parses the ?query=string and
  # puts that into params as well.
  class Controller
    attr_accessor :status
    # Stolen from Camping without a twinge of remorse ;)
    def initialize(req, env, method=(env['REQUEST_METHOD']||"GET")) #:nodoc:
      env = MerbHash[env.to_hash]
      puts env.inspect if $DEBUG
      puts req.inspect if $DEBUG
      @status, @method, @env, @headers, @root = 200, method.downcase, env, 
          {'Content-Type'=>'text/html'}, env['SCRIPT_NAME'].sub(/\/$/,'')
      @k = query_parse(env['HTTP_COOKIE'], ';,')
      qs = query_parse(env['QUERY_STRING'])
      @in = req
      if %r|\Amultipart/form-data.*boundary=\"?([^\";,]+)|n.match(env['CONTENT_TYPE'])
        b = /(?:\r?\n|\A)#{Regexp::quote("--#$1")}(?:--)?\r$/
        until @in.eof?
          fh=MerbHash[]
          for l in @in
            case l
            when "\r\n" : break
            when /^Content-Disposition: form-data;/
              fh.update MerbHash[*$'.scan(/(?:\s(\w+)="([^"]+)")/).flatten]
            when /^Content-Type: (.+?)(\r$|\Z)/m
              puts "=> fh[type] = #$1"
              fh[:type] = $1
            end
          end
          fn=fh[:name]
          o=if fh[:filename]
            o=fh[:tempfile]=Tempfile.new(:Merb)
            o.binmode
          else
            fh=""
          end
          while l=@in.read(16384)
            if l=~b
              o<<$`.chomp
              @in.seek(-$'.size,IO::SEEK_CUR)
              break
            end
            o<<l
          end
          qs[fn]=fh if fn
          fh[:tempfile].rewind if fh.is_a?MerbHash
        end
      elsif @method == "post"
        qs.merge!(query_parse(@in.read))
      end
      @cookies, @params = @k.dup, qs.dup
    end

    def params
      @params
    end  
    
    def cookies
      @cookies
    end  

    def headers
      @headers
    end  

    def query_parse(qs, d = '&;')
      m = proc {|_,o,n|o.update(n,&m)rescue([*o]<<n)}
      (qs||'').split(/[#{d}] */n).inject(MerbHash[]) { |h,p| 
              k, v=unescape(p).split('=',2)
              h.update(k.split(/[\]\[]+/).reverse.
                inject(v) { |x,i| MerbHash[i,x] },&m)
          }
    end

    def escape(s)
      Mongrel::HttpRequest.escape(s)
    end

    def unescape(s)
      Mongrel::HttpRequest.unescape(s)
    end

    def self.template_dir(loc)
      @@template_dir = File.expand_path(Merb::Server.config[:merb_root] + "/app/views/#{loc}")
    end  
    
    def current_method_name(depth=0)
      caller[depth] =~ /`(.*)'$/
      $1
    end
    
    def render(template=current_method_name(1),b=binding)
      template = ERB.new( IO.read( @@template_dir + "/#{template}.rhtml" ) ) 
      template.result( b )
    end  
  end  
  
end