luci/libs/web/luasrc/http.lua

--[[
LuCI - HTTP-Interaction

Description:
HTTP-Header manipulator and form variable preprocessor

FileId:
$Id$

ToDo:
- Cookie handling

License:
Copyright 2008 Steven Barth <steven@midlink.org>

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at 

	http://www.apache.org/licenses/LICENSE-2.0 

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.

]]--

module("luci.http", package.seeall)
require("luci.util")
context = luci.util.threadlocal()


Request = luci.util.class()
function Request.__init__(self)
	self.headers = {}
	self.request = {}
	self.uploads = {}
	self.env = {}
	self.data = ""
end

function Request.formvalue(self, name, default)
	return self.request[name] or default
end

function Request.formvalues(self)
	return self.request
end

function Request.formvaluetable(self, prefix)
	local vals = {}
	prefix = prefix and prefix .. "." or "."
	
	for k, v in pairs(self.request) do
		if k:find(prefix, 1, true) == 1 then
			vals[k:sub(#prefix + 1)] = v
		end
	end
	
	return vals
end

function Request.getenv(self, name)
	return self.env[name]
end

function Request.upload(self, name)
	return self.uploads[name]
end


function close()
	if not context.eoh then
		context.eoh = true
		coroutine.yield(3)
	end
	
	if not context.closed then
		context.closed = true
		coroutine.yield(5)
	end
end

function formvalue(...)
	return context.request:formvalue(...)
end

function formvalues(...)
	return context.request:formvalues(...)
end

function formvaluetable(...)
	return context.request:formvaluetable(...)
end

function getenv(...)
	return context.request:getenv(...)
end

function header(key, value)
	if not context.status then
		status()
	end
	if not context.headers then
		context.headers = {}
	end
	context.headers[key:lower()] = value
	coroutine.yield(2, key, value)
end

function prepare_content(mime)
	header("Content-Type", mime)
end

function status(code, message)
	code = code or 200
	message = message or "OK"
	context.status = code
	coroutine.yield(1, code, message)
end

function write(content)
	if not content or #content == 0 then
		return
	end
	if not context.eoh then
		if not context.status then
			status()
		end
		if not context.headers or not context.headers["content-type"] then
			header("Content-Type", "text/html; charset=utf-8")
		end
		
		context.eoh = true
		coroutine.yield(3)
	end
	coroutine.yield(4, content)
end


function basic_auth(realm, errorpage)
	header("Status", "401 Unauthorized")
	header("WWW-Authenticate", string.format('Basic realm="%s"', realm or ""))
	
	if errorpage then
		errorpage()
	end
	
	close()
end

function redirect(url)
	header("Status", "302 Found")
	header("Location", url)
	close()
end

function upload(...)
	return context.request:upload(...)
end



function build_querystring(table)
	local s="?"
	
	for k, v in pairs(table) do
		s = s .. k .. "=" .. v .. "&"
	end
	
	return s
end

function urldecode(str)
	str = str:gsub("+", " ")
	str = str:gsub("%%(%x%x)",
		function(h) return string.char(tonumber(h,16)) end)
	str = str:gsub("\r\n", "\n")
	return str	
end

function urlencode(str)
	str = str:gsub("\n", "\r\n")
	str = str:gsub("([^%w ])",
		function (c) return string.format ("%%%02X", string.byte(c)) end)
	str = str:gsub(" ", "+")
	return str	
end