lua-http-endpoints/nice_server.lua at master · RussellHaley/lua-http-endpoints · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
--- A base server for processing client http/ws requests.
-- @author Daurnimator
-- @copyright 2015-2016
-- @license The MIT License (MIT). See License.txt

local cqueues = require "cqueues"
local cc = require "cqueues.condition"
local ce = require "cqueues.errno"
local new_headers = require "http.headers".new
local server = require "http.server"
local version = require "http.version"
local http_util = require "http.util"

local websocket = require "http.websocket"

--local zlib = require "http.zlib"

local default_server = string.format("%s/%s", version.name, version.version)

local error_text = [[
<html>
<head>
<title>503 Internal Server Error</title>
</head>
<body>
An internal server error occured.
</body>
</html>
]]

local response_methods = {}
local response_mt = {
    __index = response_methods;
    __name = nil;
}

local function new_response(request_headers, stream)
    local headers = new_headers();
    -- Give some defaults
    headers:append(":status", "503")
    headers:append("server", default_server)
    headers:append("date", http_util.imf_date())
    return setmetatable({
        request_headers = request_headers;
        stream = stream;
        -- Record peername upfront, as the client might disconnect before request is completed
        peername = select(2, stream:peername());

        headers = headers;
        body = nil;
    }, response_mt)
end

function response_methods:combined_log()
    -- Log in "Combined Log Format"
    -- https://httpd.apache.org/docs/2.2/logs.html#combined
    return string.format('%s - - [%s] "%s %s HTTP/%g" %s %d "%s" "%s"',
        self.peername or "-",
        os.date("%d/%b/%Y:%H:%M:%S %z"),
        self.request_headers:get(":method") or "",
        self.request_headers:get(":path") or "",
        self.stream.connection.version,
        self.headers:get(":status") or "",
        self.stream.stats_sent,
        self.request_headers:get("referer") or "-",
        self.request_headers:get("user-agent") or "-")
end

function response_methods:set_body(body)
    self.body = body
    local length
    if type(self.body) == "string" then
        length = #body
    end
    if length then
        self.headers:upsert("content-length", string.format("%d", #body))
    end
end

function response_methods:set_503()
    local headers = new_headers()
    headers:append(":status", "503")
    headers:append("server", default_server)
    headers:append("date", http_util.imf_date())
    self.headers = headers
    headers:append("content-type", "text/html")
    self:set_body(error_text)
end

function response_methods:enable_compression()
    if self.headers:has("content-encoding") then
        return false
    end
    local deflater-- = zlib.deflate()
    --local new_body = deflater(self.body, true)
    local new_body = self.body
    self.headers:append("content-encoding", "gzip")
    self.body = new_body
    return true
end

local function default_onerror(...)
    io.stderr:write(string.format(...), "\n\n")
end
local function default_log(response)
    io.stderr:write(response:combined_log(), "\n")
end

local function new(options)
    local reply = assert(options.reply)
    local onerror = options.onerror or default_onerror
    local log = options.log or default_log
    local myserver = server.listen(options)

    local main_cq = cqueues.new()
    main_cq:wrap(function()
        local clients_cq = cqueues.new()
        -- create a thread that waits forever so :loop doesn't return
        local cond = cc.new()
        clients_cq:wrap(function()
            cond:wait()
        end)
        cqueues.running():wrap(function()
            while true do
                local _, err, _, thd = clients_cq:loop()
                if thd == nil then -- non-thread specific error; something is terribly wrong
                error(err)
                end
                onerror("client thread error: %s", debug.traceback(thd, err))
            end
        end)
        assert(myserver:run(function(stream)
            local req_headers, err, errno = stream:get_headers()
            if req_headers == nil then
                -- connection hit EOF before headers arrived
                stream:shutdown()
                if err ~= ce.EPIPE and errno ~= ce.ECONNRESET then
                    onerror("header error: %s", tostring(err))
                end
                return
            end

            local resp = new_response(req_headers, stream)

            local ok, err2 = pcall(reply, resp) -- only works in 5.2+

            if stream.state ~= "closed" and stream.state ~= "half closed (local)" then
                if not ok then
                    resp:set_503()
                end
                local send_body = resp.body and req_headers:get ":method" ~= "HEAD"
                stream:write_headers(resp.headers, not send_body)
                if send_body then
                    stream:write_chunk(resp.body, true)
                end
            end
            stream:shutdown()
            if not ok then
                onerror("stream error: %s", tostring(err2))
            end
            log(resp)
        end, clients_cq))
    end)
    return main_cq
end

return {
    new = new;
}