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
|
--[[
LuCI - Lua Configuration Interface
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
$Id$
]]--
local pcall, ipairs, tonumber, type, next = pcall, ipairs, tonumber, type, next
local util = require "luci.util"
local http = require "luci.http.protocol"
local ltn12 = require "luci.ltn12"
local table = require "table"
module "luci.ttpd.module"
-- Server handler implementation
Handler = util.class()
-- Constructor
function Handler.__init__(self)
self.handler = {}
self.filters = {}
self.modifiers = {}
end
-- Add a filter
function Handler.setfilter(self, filter, key)
self.filters[(key) or (#self.filters+1)] = filter
end
-- Add a modifier
function Handler.setmodifier(self, modifier, key)
self.modifiers[(pos) or (#self.modifiers+1)] = modifier
end
-- Creates a failure reply
function Handler.failure(self, code, message)
local response = Response(code, { ["Content-Type"] = "text/plain" })
local sourceout = ltn12.source.string(message)
return response, sourceout
end
-- Processes a request
function Handler.process(self, request, sourcein, sinkerr)
local stat, response, sourceout
-- Detect request Method
local hname = "handle_" .. request.request_method
if self[hname] then
local t = {
processor = self[hname],
handler = self,
request = request,
sourcein = sourcein,
sinkerr = sinkerr
}
if next(self.modifiers) then
for _, mod in util.kspairs(self.modifiers) do
mod(t)
end
end
-- Run the handler
stat, response, sourceout = pcall(
t.processor, t.handler, t.request, t.sourcein, t.sinkerr
)
-- Check for any errors
if not stat then
response, sourceout = self:failure(500, response)
elseif next(self.filters) then
local t = {
response = response,
sourceout = sourceout,
sinkerr = t.sinkerr
}
for _, filter in util.kspairs(self.filters) do
filter(t)
end
response = t.response
sourceout = t.sourceout
end
else
response, sourceout = self:failure(405, http.protocol.statusmsg[405])
end
-- Check data
if not util.instanceof(response, Response) then
response, sourceout = self:failure(500, "Core error: Invalid module response!")
end
return response, sourceout
end
-- Handler Response
Response = util.class()
function Response.__init__(self, status, headers)
self.status = tonumber(status) or 200
self.headers = (type(headers) == "table") and headers or {}
end
function Response.addheader(self, key, value)
self.headers[key] = value
end
function Response.setstatus(self, status)
self.status = status
end
|
