mirror of
https://github.com/lxsang/silk.git
synced 2024-07-06 05:29:47 +02:00
203 lines
4.8 KiB
Lua
203 lines
4.8 KiB
Lua
std = {}
|
|
require("silk.core.mimes")
|
|
|
|
RESPONSE_HEADER = {
|
|
status = 200,
|
|
header = {},
|
|
cookie = {},
|
|
sent = false
|
|
}
|
|
|
|
local http_status = {
|
|
[100] = "Continue",
|
|
[101] = "Switching Protocols",
|
|
[102] = "Processing",
|
|
[103] = "Early Hints",
|
|
|
|
[200] = "OK",
|
|
[201] = "Created",
|
|
[202] = "Accepted",
|
|
[203] = "Non-Authoritative Information",
|
|
[204] = "No Content",
|
|
[205] = "Reset Content",
|
|
[206] = "Partial Content",
|
|
[207] = "Multi-Status",
|
|
[208] = "Already Reported",
|
|
[226] = "IM Used",
|
|
|
|
[300] = "Multiple Choices",
|
|
[301] = "Moved Permanently",
|
|
[302] = "Found",
|
|
[303] = "See Other",
|
|
[304] = "Not Modified",
|
|
[305] = "Use Proxy",
|
|
[306] = "Switch Proxy",
|
|
[307] = "Temporary Redirect",
|
|
[308] = "Permanent Redirect",
|
|
|
|
[400] = "Bad Request",
|
|
[401] = "Unauthorized",
|
|
[402] = "Payment Required",
|
|
[403] = "Forbidden",
|
|
[404] = "Not Found",
|
|
[405] = "Method Not Allowed",
|
|
[406] = "Not Acceptable",
|
|
[407] = "Proxy Authentication Required",
|
|
[408] = "Request Timeout",
|
|
[409] = "Conflict",
|
|
[410] = "Gone",
|
|
[411] = "Length Required",
|
|
[412] = "Precondition Failed",
|
|
[413] = "Payload Too Large",
|
|
[414] = "URI Too Long",
|
|
[415] = "Unsupported Media Type",
|
|
[416] = "Range Not Satisfiable",
|
|
[417] = "Expectation Failed",
|
|
[421] = "Misdirected Request",
|
|
[422] = "Unprocessable Entity",
|
|
[423] = "Locked",
|
|
[424] = "Failed Dependency",
|
|
[425] = "Too Early",
|
|
[426] = "Upgrade Required",
|
|
[428] = "Precondition Required",
|
|
[429] = "Too Many Requests",
|
|
[431] = "Request Header Fields Too Large",
|
|
[451] = "Unavailable For Legal Reasons",
|
|
|
|
[500] = "Internal Server Error",
|
|
[501] = "Not Implemented",
|
|
[502] = "Bad Gateway",
|
|
[503] = "Service Unavailable",
|
|
[504] = "Gateway Timeout",
|
|
[505] = "HTTP Version Not Supported",
|
|
[506] = "Variant Also Negotiates",
|
|
[507] = "Insufficient Storage",
|
|
[508] = "Loop Detected",
|
|
[510] = "Not Extended",
|
|
[511] = "Network Authentication Required"
|
|
}
|
|
setmetatable(http_status, {
|
|
__index = function(this, key)
|
|
return "Unofficial Status"
|
|
end
|
|
})
|
|
|
|
function std.status(code)
|
|
RESPONSE_HEADER.status = code
|
|
end
|
|
|
|
function std.custom_header(k, v)
|
|
std.header(k, v)
|
|
end
|
|
|
|
function std.header(k, v)
|
|
RESPONSE_HEADER.header[k] = v
|
|
end
|
|
|
|
function std.header_flush()
|
|
-- send out status
|
|
echo("Status: ", RESPONSE_HEADER.status, " ", http_status[RESPONSE_HEADER.status], "\r\n")
|
|
-- send out header
|
|
for key, val in pairs(RESPONSE_HEADER.header) do
|
|
echo(key, ": ", val, "\r\n")
|
|
end
|
|
-- send out cookie
|
|
for key, val in ipairs(RESPONSE_HEADER.cookie) do
|
|
echo("Set-Cookie: ", val, "\r\n")
|
|
end
|
|
echo("\r\n")
|
|
RESPONSE_HEADER.sent = true
|
|
RESPONSE_HEADER.header = {}
|
|
RESPONSE_HEADER.cookie = {}
|
|
end
|
|
|
|
function std.setCookie(...)
|
|
local args = table.pack(...)
|
|
cookie = table.concat(args,";")
|
|
RESPONSE_HEADER.cookie[#RESPONSE_HEADER.cookie + 1] = cookie
|
|
end
|
|
|
|
function std.error(status, msg)
|
|
std.status(status)
|
|
std.header("Content-Type", "text/html")
|
|
std.header_flush()
|
|
echo(string.format("<HTML><HEAD><TITLE>%s</TITLE></HEAD><BODY><h2>%s</h2></BODY></HTML>",msg, msg))
|
|
end
|
|
|
|
function std.unknow(s)
|
|
std.error(404, "Unknown request")
|
|
end
|
|
|
|
function std.f(path)
|
|
fcgio:send_file(path)
|
|
end
|
|
|
|
function std.t(...)
|
|
echo(...)
|
|
end
|
|
|
|
|
|
function std.html()
|
|
std.header("Content-Type", "text/html; charset=utf-8")
|
|
std.header_flush()
|
|
end
|
|
|
|
function std.text()
|
|
std.header("Content-Type", "text/plain; charset=utf-8")
|
|
std.header_flush()
|
|
end
|
|
|
|
function std.json()
|
|
std.header("Content-Type", "application/json; charset=utf-8")
|
|
std.header_flush()
|
|
end
|
|
|
|
function std.jpeg()
|
|
std.header("Content-Type", "image/jpeg")
|
|
std.header_flush()
|
|
end
|
|
|
|
function std.octstream(s)
|
|
std.header("Content-Type", "application/octet-stream")
|
|
std.header("Content-Disposition", 'attachment; filename="' .. s .. '"')
|
|
std.header_flush()
|
|
end
|
|
|
|
function std.is_file(f)
|
|
return ulib.is_dir(f) == false
|
|
end
|
|
|
|
-- TODO use coroutine to read socket message
|
|
std.ws = {}
|
|
function std.ws.header()
|
|
if not fcgio:is_ws() then return nil end
|
|
return fcgio:ws_header()
|
|
end
|
|
|
|
function std.ws.read(h)
|
|
if not fcgio:is_ws() then return nil end
|
|
return fcgio:ws_read(h)
|
|
end
|
|
function std.ws.t(...)
|
|
if not fcgio:is_ws() then return nil end
|
|
fcgio:ws_send(false,...)
|
|
end
|
|
function std.ws.f(s)
|
|
if not fcgio:is_ws() then return nil end
|
|
fcgio:send_file(s)
|
|
end
|
|
function std.ws.b(...)
|
|
if not fcgio:is_ws() then return nil end
|
|
fcgio:ws_send(true,...)
|
|
end
|
|
function std.ws.enable()
|
|
return fcgio:is_ws()
|
|
end
|
|
function std.ws.close(code)
|
|
if not fcgio:is_ws() then return nil end
|
|
fcgio:ws_close(code)
|
|
end
|
|
std.ws.TEXT = 1
|
|
std.ws.BIN = 2
|
|
std.ws.CLOSE = 8
|