mirror of
https://github.com/JamesonHuang/OpenWrt_Luci_Lua.git
synced 2024-11-23 22:00:11 +00:00
1037 lines
31 KiB
Lua
1037 lines
31 KiB
Lua
--[[
|
||
LuCI - Dispatcher
|
||
|
||
Description:
|
||
The request dispatcher and module dispatcher generators
|
||
|
||
FileId:
|
||
$Id: dispatcher.lua 9018 2012-08-14 15:31:26Z jow $
|
||
|
||
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.
|
||
|
||
]]--
|
||
|
||
--- LuCI web dispatcher.
|
||
|
||
local fs = require "nixio.fs"
|
||
local bit = require("bit")
|
||
local sys = require "luci.sys"
|
||
local init = require "luci.init"
|
||
local util = require "luci.util"
|
||
local http = require "luci.http"
|
||
local nixio = require "nixio", require "nixio.util"
|
||
|
||
local XQSecureUtil = require("xiaoqiang.util.XQSecureUtil")
|
||
|
||
module("luci.dispatcher", package.seeall)
|
||
context = util.threadlocal()
|
||
i18n = require "luci.i18n"
|
||
_M.fs = fs
|
||
|
||
authenticator = {}
|
||
-- Index table
|
||
local index = nil
|
||
|
||
-- Fastindex
|
||
local fi
|
||
|
||
--- Build the URL relative to the server webroot from given virtual path.
|
||
-- @param ... Virtual path
|
||
-- @return Relative URL
|
||
function build_url(...)
|
||
local path = {...}
|
||
local url = { http.getenv("SCRIPT_NAME") or "" }
|
||
|
||
local k, v
|
||
for k, v in pairs(context.urltoken) do
|
||
url[#url+1] = "/;"
|
||
url[#url+1] = http.urlencode(k)
|
||
url[#url+1] = "="
|
||
url[#url+1] = http.urlencode(v)
|
||
end
|
||
|
||
local p
|
||
for _, p in ipairs(path) do
|
||
if p:match("^[a-zA-Z0-9_%-%.%%/,;]+$") then
|
||
url[#url+1] = "/"
|
||
url[#url+1] = p
|
||
end
|
||
end
|
||
|
||
return table.concat(url, "")
|
||
end
|
||
|
||
--- Check whether a dispatch node shall be visible
|
||
-- @param node Dispatch node
|
||
-- @return Boolean indicating whether the node should be visible
|
||
function node_visible(node)
|
||
if node then
|
||
return not (
|
||
(not node.title or #node.title == 0) or
|
||
(not node.target or node.hidden == true) or
|
||
(type(node.target) == "table" and node.target.type == "firstchild" and
|
||
(type(node.nodes) ~= "table" or not next(node.nodes)))
|
||
)
|
||
end
|
||
return false
|
||
end
|
||
|
||
--- Return a sorted table of visible childs within a given node
|
||
-- @param node Dispatch node
|
||
-- @return Ordered table of child node names
|
||
function node_childs(node)
|
||
local rv = { }
|
||
if node then
|
||
local k, v
|
||
for k, v in util.spairs(node.nodes,
|
||
function(a, b)
|
||
return (node.nodes[a].order or 100)
|
||
< (node.nodes[b].order or 100)
|
||
end)
|
||
do
|
||
if node_visible(v) then
|
||
rv[#rv+1] = k
|
||
end
|
||
end
|
||
end
|
||
return rv
|
||
end
|
||
|
||
--- Send a 404 error code and render the "error404" template if available.
|
||
-- @param message Custom error message (optional)
|
||
-- @return false
|
||
function error404(message)
|
||
luci.http.status(404, "Not Found")
|
||
message = message or "Not Found"
|
||
|
||
require("luci.template")
|
||
if not luci.util.copcall(luci.template.render, "error404") then
|
||
luci.http.prepare_content("text/plain")
|
||
luci.http.write(message)
|
||
end
|
||
return false
|
||
end
|
||
|
||
--- Send a 500 error code and render the "error500" template if available.
|
||
-- @param message Custom error message (optional)#
|
||
-- @return false
|
||
function error500(message)
|
||
--luci.util.perror(message)
|
||
local logger = require("xiaoqiang.XQLog")
|
||
logger.log(3, "Internal Server Error", message)
|
||
message = "Internal Server Error"
|
||
if not context.template_header_sent then
|
||
luci.http.status(500, "Internal Server Error")
|
||
luci.http.prepare_content("text/plain")
|
||
luci.http.write(message)
|
||
else
|
||
require("luci.template")
|
||
if not luci.util.copcall(luci.template.render, "error500", {message=message}) then
|
||
luci.http.prepare_content("text/plain")
|
||
luci.http.write(message)
|
||
end
|
||
end
|
||
return false
|
||
end
|
||
|
||
function empower(lan,wan,admin)
|
||
local XQFunction = require("xiaoqiang.common.XQFunction")
|
||
local remoteAddr = luci.http.getenv("REMOTE_ADDR")
|
||
local mac = XQFunction.macFormat(luci.sys.net.ip4mac(remoteAddr))
|
||
if not XQFunction.isStrNil(mac) then
|
||
local XQSysUtil = require("xiaoqiang.util.XQSysUtil")
|
||
if not XQSysUtil.setMacFilter(mac,lan,wan,admin) then
|
||
local XQLog = require("xiaoqiang.XQLog")
|
||
XQLog.log(3,"Empower failed"..mac)
|
||
end
|
||
end
|
||
end
|
||
|
||
function getremotemac()
|
||
local XQFunction = require("xiaoqiang.common.XQFunction")
|
||
local remote_addr = luci.http.getenv("REMOTE_ADDR") or ""
|
||
local mac = luci.sys.net.ip4mac(remote_addr) or ""
|
||
return XQFunction.macFormat(mac)
|
||
end
|
||
|
||
-- TODO auth will be found similar
|
||
function authenticator.jsonauth(validator, accs, default)
|
||
local XQSysUtil = require("xiaoqiang.util.XQSysUtil")
|
||
|
||
local user = luci.http.xqformvalue("username")
|
||
local pass = luci.http.xqformvalue("password")
|
||
local nonce = luci.http.xqformvalue("nonce")
|
||
local uuid = luci.http.xqformvalue("uuid")
|
||
local token = luci.http.xqformvalue("token")
|
||
local isBinded = XQSysUtil.getPassportBindInfo()
|
||
|
||
if isBinded and uuid and token and (uuid == isBinded) then
|
||
local XQDBUtil = require("xiaoqiang.util.XQDBUtil")
|
||
local passport = XQDBUtil.fetchPassport(uuid)[1]
|
||
if passport and token == passport.token then
|
||
empower("1","1",nil)
|
||
local logtype = "1"
|
||
luci.http.header("Set-Cookie", "psp=" .. uuid .. "|||" .. logtype .. "|||" .. token .. ";path=/;")
|
||
return default, logtype
|
||
end
|
||
end
|
||
|
||
if nonce then
|
||
if XQSecureUtil.checkNonce(nonce, getremotemac()) then
|
||
if XQSecureUtil.checkUser(user, nonce, pass) then
|
||
empower("1","1",nil)
|
||
local logtype = "2"
|
||
luci.http.header("Set-Cookie", "psp=" .. user .. "|||" .. logtype .. "|||0;path=/;")
|
||
return user, logtype
|
||
end
|
||
else
|
||
context.path = {}
|
||
luci.http.write([[{"code":1582,"msg":"nonce invalid"}]])
|
||
return false
|
||
end
|
||
else
|
||
if XQSecureUtil.checkPlaintextPwd(user, pass) then
|
||
empower("1","1",nil)
|
||
local logtype = "2"
|
||
luci.http.header("Set-Cookie", "psp=" .. user .. "|||" .. logtype .. "|||0;path=/;")
|
||
return user, logtype
|
||
else
|
||
context.path = {}
|
||
luci.http.write([[{"code":401,"msg":"密码错误"}]])
|
||
return false
|
||
end
|
||
end
|
||
context.path = {}
|
||
luci.http.write([[{"code":401,"msg":"not auth"}]])
|
||
return false
|
||
end
|
||
|
||
function authenticator.htmlauth(validator, accs, default)
|
||
local XQSysUtil = require("xiaoqiang.util.XQSysUtil")
|
||
local redirectKey = luci.http.xqformvalue("redirectKey")
|
||
local isBinded = XQSysUtil.getPassportBindInfo()
|
||
|
||
if redirectKey then
|
||
local check = XQSecureUtil.checkRedirectKey(redirectKey)
|
||
if check then
|
||
if check == "1" and isBinded then
|
||
local XQDBUtil = require("xiaoqiang.util.XQDBUtil")
|
||
local bindUUID = XQSysUtil.getBindUUID()
|
||
local passport = XQDBUtil.fetchPassport(bindUUID)[1]
|
||
if passport then
|
||
luci.http.header("Set-Cookie", "psp=" .. bindUUID .. "|||1|||" .. passport.token .. ";path=/;")
|
||
return default, check
|
||
end
|
||
elseif check == "2" then
|
||
luci.http.header("Set-Cookie", "psp=admin|||2|||0;path=/;")
|
||
return "admin", check
|
||
end
|
||
end
|
||
end
|
||
require("luci.i18n")
|
||
require("luci.template")
|
||
context.path = {}
|
||
luci.template.render("web/sysauth", {duser=default, fuser=user})
|
||
return false
|
||
end
|
||
|
||
function authenticator.htmlauth_moblie(validator, accs, default)
|
||
local user = luci.http.xqformvalue("username")
|
||
local pass = luci.http.xqformvalue("password")
|
||
local nonce = luci.http.xqformvalue("nonce")
|
||
if nonce then
|
||
if XQSecureUtil.checkNonce(nonce, getremotemac()) and XQSecureUtil.checkUser(user, nonce, pass) then
|
||
empower("1","1",nil)
|
||
return user, "2"
|
||
end
|
||
end
|
||
require("luci.i18n")
|
||
require("luci.template")
|
||
context.path = {}
|
||
luci.template.render("mobile/sysauth", {duser=default, fuser=user})
|
||
return false
|
||
end
|
||
|
||
function check_show_syslock(sysauth)
|
||
local XQFunction = require("xiaoqiang.common.XQFunction")
|
||
local XQSysUtil = require("xiaoqiang.util.XQSysUtil")
|
||
|
||
if XQFunction.sysLockStatus() == 1 then
|
||
if XQSysUtil.isUpgrading() then
|
||
require("luci.i18n")
|
||
require("luci.template")
|
||
if type(sysauth) == "string" and sysauth == "htmlauth" then
|
||
context.path = {}
|
||
luci.template.render("web/syslock",{})
|
||
elseif type(sysauth) == "string" and sysauth == "jsonauth" then
|
||
context.path = {}
|
||
luci.http.write([[{"code":403,"msg":"system locked"}]])
|
||
else
|
||
XQFunction.sysUnlock()
|
||
return false
|
||
end
|
||
return true
|
||
else
|
||
XQFunction.sysUnlock()
|
||
end
|
||
end
|
||
return false
|
||
end
|
||
|
||
function http_request_log(request, tag)
|
||
local XQLog = require("xiaoqiang.XQLog")
|
||
local requestUri = request:getenv("REQUEST_URI")
|
||
if requestUri and tag and type(tag) == "string" then
|
||
local uriInfo = luci.util.split(requestUri,"?")
|
||
XQLog.log(6,tag..":"..uriInfo[1])
|
||
if uriInfo[2] then
|
||
XQLog.log(7,uriInfo[2])
|
||
end
|
||
end
|
||
end
|
||
|
||
-- API Permissions
|
||
function _noauthAccessAllowed(flag)
|
||
if flag == nil then
|
||
return false
|
||
end
|
||
if bit.band(flag, 0x01) == 0x01 then
|
||
return true
|
||
else
|
||
return false
|
||
end
|
||
end
|
||
|
||
function _remoteAccessForbidden(flag)
|
||
if flag == nil then
|
||
return false
|
||
end
|
||
if bit.band(flag, 0x02) == 0x02 then
|
||
return true
|
||
else
|
||
return false
|
||
end
|
||
end
|
||
|
||
function _syslockAccessAllowed(flag)
|
||
if flag == nil then
|
||
return false
|
||
end
|
||
if bit.band(flag, 0x04) == 0x04 then
|
||
return true
|
||
else
|
||
return false
|
||
end
|
||
end
|
||
|
||
function _noinitAccessAllowed(flag)
|
||
local xqsys = require("xiaoqiang.util.XQSysUtil")
|
||
if xqsys.getInitInfo() then
|
||
return true
|
||
else
|
||
if flag == nil then
|
||
return false
|
||
end
|
||
if bit.band(flag, 0x08) == 0x08 then
|
||
return true
|
||
else
|
||
return false
|
||
end
|
||
end
|
||
end
|
||
|
||
function _sdkFilter(flag)
|
||
if flag == nil then
|
||
return false
|
||
end
|
||
if bit.band(flag, 0x10) == 0x10 then
|
||
return true
|
||
else
|
||
return false
|
||
end
|
||
end
|
||
|
||
--- Dispatch an HTTP request.
|
||
-- @param request LuCI HTTP Request object
|
||
function httpdispatch(request, prefix)
|
||
http_request_log(request, "request")
|
||
-- 设置全局表,使 _() 使用当前文件内的 _函数,即为标记tag, translate 可以直接使用翻译
|
||
_G._ = _
|
||
_G.translate = i18n.translate
|
||
luci.http.context.request = request
|
||
|
||
local r = {}
|
||
context.request = r
|
||
context.urltoken = {}
|
||
|
||
local pathinfo = http.urldecode(request:getenv("PATH_INFO") or "", true)
|
||
|
||
if prefix then
|
||
for _, node in ipairs(prefix) do
|
||
r[#r+1] = node
|
||
end
|
||
end
|
||
|
||
local tokensok = true
|
||
for node in pathinfo:gmatch("[^/]+") do
|
||
local tkey, tval
|
||
if tokensok then
|
||
tkey, tval = node:match(";(%w+)=([a-fA-F0-9]*)")
|
||
end
|
||
if tkey then
|
||
context.urltoken[tkey] = tval
|
||
else
|
||
tokensok = false
|
||
r[#r+1] = node
|
||
end
|
||
end
|
||
|
||
local stat, err = util.coxpcall(function()
|
||
dispatch(context.request)
|
||
end, error500)
|
||
luci.http.close()
|
||
http_request_log(request, "finished")
|
||
end
|
||
|
||
--- Dispatches a LuCI virtual path.
|
||
-- @param request Virtual path
|
||
function dispatch(request)
|
||
local ctx = context
|
||
ctx.path = request
|
||
|
||
local conf = require "luci.config"
|
||
assert(conf.main,
|
||
"/etc/config/luci seems to be corrupt, unable to find section 'main'")
|
||
|
||
local lang = conf.main.lang or "auto"
|
||
if lang == "auto" then
|
||
local aclang = http.getenv("HTTP_ACCEPT_LANGUAGE") or ""
|
||
for lpat in aclang:gmatch("[%w-]+") do
|
||
lpat = lpat and lpat:gsub("-", "_")
|
||
if conf.languages[lpat] then
|
||
lang = lpat
|
||
break
|
||
end
|
||
end
|
||
end
|
||
require "luci.i18n".setlanguage(lang)
|
||
|
||
local c = ctx.tree
|
||
local stat
|
||
if not c then
|
||
c = createtree()
|
||
end
|
||
|
||
local track = {}
|
||
local args = {}
|
||
ctx.args = args
|
||
ctx.requestargs = ctx.requestargs or args
|
||
local n
|
||
local token = ctx.urltoken
|
||
local preq = {}
|
||
local freq = {}
|
||
|
||
for i, s in ipairs(request) do
|
||
preq[#preq+1] = s
|
||
freq[#freq+1] = s
|
||
c = c.nodes[s]
|
||
n = i
|
||
if not c then
|
||
break
|
||
end
|
||
|
||
util.update(track, c)
|
||
|
||
if c.leaf then
|
||
break
|
||
end
|
||
end
|
||
|
||
if c and c.leaf then
|
||
for j=n+1, #request do
|
||
args[#args+1] = request[j]
|
||
freq[#freq+1] = request[j]
|
||
end
|
||
end
|
||
|
||
ctx.requestpath = ctx.requestpath or freq
|
||
ctx.path = preq
|
||
|
||
if track.i18n then
|
||
i18n.loadc(track.i18n)
|
||
end
|
||
|
||
-- Init template engine
|
||
if (c and c.index) or not track.notemplate then
|
||
local tpl = require("luci.template")
|
||
local media = track.mediaurlbase or luci.config.main.mediaurlbase
|
||
if not pcall(tpl.Template, "themes/%s/header" % fs.basename(media)) then
|
||
media = nil
|
||
for name, theme in pairs(luci.config.themes) do
|
||
if name:sub(1,1) ~= "." and pcall(tpl.Template,
|
||
"themes/%s/header" % fs.basename(theme)) then
|
||
media = theme
|
||
end
|
||
end
|
||
assert(media, "No valid theme found")
|
||
end
|
||
|
||
local function _ifattr(cond, key, val)
|
||
if cond then
|
||
local env = getfenv(3)
|
||
local scope = (type(env.self) == "table") and env.self
|
||
return string.format(
|
||
' %s="%s"', tostring(key),
|
||
luci.util.pcdata(tostring( val
|
||
or (type(env[key]) ~= "function" and env[key])
|
||
or (scope and type(scope[key]) ~= "function" and scope[key])
|
||
or "" ))
|
||
)
|
||
else
|
||
return ''
|
||
end
|
||
end
|
||
tpl.context.viewns = setmetatable({
|
||
write = luci.http.write;
|
||
include = function(name) tpl.Template(name):render(getfenv(2)) end;
|
||
translate = i18n.translate;
|
||
export = function(k, v) if tpl.context.viewns[k] == nil then tpl.context.viewns[k] = v end end;
|
||
striptags = util.striptags;
|
||
pcdata = util.pcdata;
|
||
media = media;
|
||
theme = fs.basename(media);
|
||
resource = luci.config.main.resourcebase;
|
||
ifattr = function(...) return _ifattr(...) end;
|
||
attr = function(...) return _ifattr(true, ...) end;
|
||
}, {__index=function(table, key)
|
||
if key == "controller" then
|
||
return build_url()
|
||
elseif key == "REQUEST_URI" then
|
||
return build_url(unpack(ctx.requestpath))
|
||
else
|
||
return rawget(table, key) or _G[key]
|
||
end
|
||
end})
|
||
end
|
||
|
||
track.dependent = (track.dependent ~= false)
|
||
assert(not track.dependent or not track.auto,
|
||
"Access Violation\nThe page at '" .. table.concat(request, "/") .. "/' " ..
|
||
"has no parent node so the access to this location has been denied.\n" ..
|
||
"This is a software bug, please report this message at " ..
|
||
"http://luci.subsignal.org/trac/newticket"
|
||
)
|
||
if not _syslockAccessAllowed(track.flag) then
|
||
if check_show_syslock(track.sysauth_authenticator) then
|
||
return
|
||
end
|
||
end
|
||
if not _noinitAccessAllowed(track.flag) then
|
||
luci.http.status(403, "Forbidden")
|
||
return
|
||
end
|
||
local isremote = http.getenv("REMOTE_ADDR") == "127.0.0.1"
|
||
if _sdkFilter(track.flag) and not isremote then
|
||
local sdkutil = require("xiaoqiang.util.XQSDKUtil")
|
||
if not sdkutil.checkPermission(getremotemac()) then
|
||
context.path = {}
|
||
luci.http.write([[{"code":1500,"msg":"Permission denied"}]])
|
||
return
|
||
end
|
||
end
|
||
if not isremote and not _noauthAccessAllowed(track.flag) and track.sysauth then
|
||
local sauth = require "luci.sauth"
|
||
local crypto = require "xiaoqiang.util.XQCryptoUtil"
|
||
local sysutil = require "xiaoqiang.util.XQSysUtil"
|
||
local isBinded = sysutil.getPassportBindInfo()
|
||
|
||
local authen = type(track.sysauth_authenticator) == "function"
|
||
and track.sysauth_authenticator
|
||
or authenticator[track.sysauth_authenticator]
|
||
|
||
local def = (type(track.sysauth) == "string") and track.sysauth
|
||
local accs = def and {track.sysauth} or track.sysauth
|
||
local sess = ctx.urltoken.stok
|
||
local sdat = sauth.read(sess)
|
||
local user
|
||
if sdat then
|
||
if ctx.urltoken.stok == sdat.token then
|
||
if (sdat.ltype == "2" or (sdat.ltype == "1" and isBinded)) then
|
||
user = sdat.user
|
||
end
|
||
end
|
||
else
|
||
local eu = http.getenv("HTTP_AUTH_USER")
|
||
local ep = http.getenv("HTTP_AUTH_PASS")
|
||
if eu and ep and luci.sys.user.checkpasswd(eu, ep) then
|
||
authen = function() return eu end
|
||
end
|
||
end
|
||
|
||
if not util.contains(accs, user) then
|
||
if authen then
|
||
ctx.urltoken.stok = nil
|
||
local user, logintype = authen(nil, accs, def)
|
||
if not user or not util.contains(accs, user) then
|
||
return
|
||
else
|
||
local sid = sess or luci.sys.uniqueid(16)
|
||
local ltype = logintype or "2"
|
||
local token = luci.sys.uniqueid(16)
|
||
sauth.reap()
|
||
sauth.write(token, {
|
||
user=user,
|
||
token=token,
|
||
ltype=ltype,
|
||
secret=luci.sys.uniqueid(16)
|
||
})
|
||
ctx.urltoken.stok = token
|
||
ctx.authsession = token
|
||
ctx.authuser = user
|
||
end
|
||
else
|
||
luci.http.status(403, "Forbidden")
|
||
return
|
||
end
|
||
else
|
||
ctx.authsession = sess
|
||
ctx.authuser = user
|
||
end
|
||
end
|
||
|
||
if track.setgroup then
|
||
luci.sys.process.setgroup(track.setgroup)
|
||
end
|
||
if track.setuser then
|
||
luci.sys.process.setuser(track.setuser)
|
||
end
|
||
|
||
local target = nil
|
||
if c then
|
||
if type(c.target) == "function" then
|
||
target = c.target
|
||
elseif type(c.target) == "table" then
|
||
target = c.target.target
|
||
end
|
||
end
|
||
|
||
if c and (c.index or type(target) == "function") then
|
||
ctx.dispatched = c
|
||
ctx.requested = ctx.requested or ctx.dispatched
|
||
end
|
||
|
||
if c and c.index then
|
||
local tpl = require "luci.template"
|
||
|
||
if util.copcall(tpl.render, "indexer", {}) then
|
||
return true
|
||
end
|
||
end
|
||
|
||
if type(target) == "function" then
|
||
util.copcall(function()
|
||
local oldenv = getfenv(target)
|
||
local module = require(c.module)
|
||
local env = setmetatable({}, {__index=
|
||
|
||
function(tbl, key)
|
||
return rawget(tbl, key) or module[key] or oldenv[key]
|
||
end})
|
||
setfenv(target, env)
|
||
end)
|
||
|
||
local ok, err
|
||
if type(c.target) == "table" then
|
||
ok, err = util.copcall(target, c.target, unpack(args))
|
||
else
|
||
ok, err = util.copcall(target, unpack(args))
|
||
end
|
||
assert(ok,
|
||
"Failed to execute " .. (type(c.target) == "function" and "function" or c.target.type or "unknown") ..
|
||
" dispatcher target for entry '/" .. table.concat(request, "/") .. "'.\n" ..
|
||
"The called action terminated with an exception:\n" .. tostring(err or "(unknown)"))
|
||
else
|
||
local root = node()
|
||
if not root or not root.target then
|
||
error404("No root node was registered, this usually happens if no module was installed.\n" ..
|
||
"Install luci-mod-admin-full and retry. " ..
|
||
"If the module is already installed, try removing the /tmp/luci-indexcache file.")
|
||
else
|
||
error404("No page is registered at '/" .. table.concat(request, "/") .. "'.\n" ..
|
||
"If this url belongs to an extension, make sure it is properly installed.\n" ..
|
||
"If the extension was recently installed, try removing the /tmp/luci-indexcache file.")
|
||
end
|
||
end
|
||
end
|
||
|
||
--- Generate the dispatching index using the best possible strategy.
|
||
function createindex()
|
||
local path = luci.util.libpath() .. "/controller/"
|
||
local suff = { ".lua", ".lua.gz" }
|
||
|
||
-- if luci.util.copcall(require, "luci.fastindex") then
|
||
-- createindex_fastindex(path, suff)
|
||
-- else
|
||
createindex_plain(path, suff)
|
||
-- end
|
||
end
|
||
|
||
--- Generate the dispatching index using the fastindex C-indexer.
|
||
-- @param path Controller base directory
|
||
-- @param suffixes Controller file suffixes
|
||
function createindex_fastindex(path, suffixes)
|
||
index = {}
|
||
|
||
if not fi then
|
||
fi = luci.fastindex.new("index")
|
||
for _, suffix in ipairs(suffixes) do
|
||
fi.add(path .. "*" .. suffix)
|
||
fi.add(path .. "*/*" .. suffix)
|
||
end
|
||
end
|
||
fi.scan()
|
||
|
||
for k, v in pairs(fi.indexes) do
|
||
index[v[2]] = v[1]
|
||
end
|
||
end
|
||
|
||
--- Generate the dispatching index using the native file-cache based strategy.
|
||
-- @param path Controller base directory
|
||
-- @param suffixes Controller file suffixes
|
||
function createindex_plain(path, suffixes)
|
||
local controllers = { }
|
||
for _, suffix in ipairs(suffixes) do
|
||
nixio.util.consume((fs.glob(path .. "*" .. suffix)), controllers)
|
||
nixio.util.consume((fs.glob(path .. "*/*" .. suffix)), controllers)
|
||
end
|
||
|
||
if indexcache then
|
||
local cachedate = fs.stat(indexcache, "mtime")
|
||
if cachedate then
|
||
local realdate = 0
|
||
for _, obj in ipairs(controllers) do
|
||
local omtime = fs.stat(obj, "mtime")
|
||
realdate = (omtime and omtime > realdate) and omtime or realdate
|
||
end
|
||
|
||
if cachedate > realdate then
|
||
assert(
|
||
sys.process.info("uid") == fs.stat(indexcache, "uid")
|
||
and fs.stat(indexcache, "modestr") == "rw-------",
|
||
"Fatal: Indexcache is not sane!"
|
||
)
|
||
|
||
index = loadfile(indexcache)()
|
||
return index
|
||
end
|
||
end
|
||
end
|
||
|
||
index = {}
|
||
|
||
for i,c in ipairs(controllers) do
|
||
local modname = "luci.controller." .. c:sub(#path+1, #c):gsub("/", ".")
|
||
for _, suffix in ipairs(suffixes) do
|
||
modname = modname:gsub(suffix.."$", "")
|
||
end
|
||
|
||
local mod = require(modname)
|
||
assert(mod ~= true,
|
||
"Invalid controller file found\n" ..
|
||
"The file '" .. c .. "' contains an invalid module line.\n" ..
|
||
"Please verify whether the module name is set to '" .. modname ..
|
||
"' - It must correspond to the file path!")
|
||
|
||
local idx = mod.index
|
||
assert(type(idx) == "function",
|
||
"Invalid controller file found\n" ..
|
||
"The file '" .. c .. "' contains no index() function.\n" ..
|
||
"Please make sure that the controller contains a valid " ..
|
||
"index function and verify the spelling!")
|
||
|
||
index[modname] = idx
|
||
end
|
||
|
||
if indexcache then
|
||
local f = nixio.open(indexcache, "w", 600)
|
||
f:writeall(util.get_bytecode(index))
|
||
f:close()
|
||
end
|
||
end
|
||
|
||
-- Create the dispatching tree from the index.
|
||
-- Build the index before if it does not exist yet.
|
||
function createtree()
|
||
if not index then
|
||
createindex()
|
||
end
|
||
|
||
local ctx = context
|
||
local tree = {nodes={}, inreq=true}
|
||
local modi = {}
|
||
|
||
ctx.treecache = setmetatable({}, {__mode="v"})
|
||
ctx.tree = tree
|
||
ctx.modifiers = modi
|
||
|
||
-- Load default translation
|
||
require "luci.i18n".loadc("base")
|
||
|
||
local scope = setmetatable({}, {__index = luci.dispatcher})
|
||
|
||
for k, v in pairs(index) do
|
||
scope._NAME = k
|
||
setfenv(v, scope)
|
||
v()
|
||
end
|
||
|
||
local function modisort(a,b)
|
||
return modi[a].order < modi[b].order
|
||
end
|
||
|
||
for _, v in util.spairs(modi, modisort) do
|
||
scope._NAME = v.module
|
||
setfenv(v.func, scope)
|
||
v.func()
|
||
end
|
||
|
||
return tree
|
||
end
|
||
|
||
--- Register a tree modifier.
|
||
-- @param func Modifier function
|
||
-- @param order Modifier order value (optional)
|
||
function modifier(func, order)
|
||
context.modifiers[#context.modifiers+1] = {
|
||
func = func,
|
||
order = order or 0,
|
||
module
|
||
= getfenv(2)._NAME
|
||
}
|
||
end
|
||
|
||
--- Clone a node of the dispatching tree to another position.
|
||
-- @param path Virtual path destination
|
||
-- @param clone Virtual path source
|
||
-- @param title Destination node title (optional)
|
||
-- @param order Destination node order value (optional)
|
||
-- @param flag For extension (optional)
|
||
-- @return Dispatching tree node
|
||
function assign(path, clone, title, order, flag)
|
||
local obj = node(unpack(path))
|
||
obj.nodes = nil
|
||
obj.module = nil
|
||
|
||
obj.title = title
|
||
obj.order = order
|
||
obj.flag = flag
|
||
|
||
setmetatable(obj, {__index = _create_node(clone)})
|
||
return obj
|
||
end
|
||
|
||
--- Create a new dispatching node and define common parameters.
|
||
-- @param path Virtual path
|
||
-- @param target Target function to call when dispatched.
|
||
-- @param title Destination node title
|
||
-- @param order Destination node order value (optional)
|
||
-- @param flag For extension (optional)
|
||
-- @return Dispatching tree node
|
||
function entry(path, target, title, order, flag)
|
||
local c = node(unpack(path))
|
||
|
||
c.target = target
|
||
c.title = title
|
||
c.order = order
|
||
c.flag = flag
|
||
c.module = getfenv(2)._NAME
|
||
|
||
return c
|
||
end
|
||
|
||
--- Fetch or create a dispatching node without setting the target module or
|
||
-- enabling the node.
|
||
-- @param ... Virtual path
|
||
-- @return Dispatching tree node
|
||
function get(...)
|
||
return _create_node({...})
|
||
end
|
||
|
||
--- Fetch or create a new dispatching node.
|
||
-- @param ... Virtual path
|
||
-- @return Dispatching tree node
|
||
function node(...)
|
||
local c = _create_node({...})
|
||
|
||
c.module = getfenv(2)._NAME
|
||
c.auto = nil
|
||
|
||
return c
|
||
end
|
||
|
||
function _create_node(path)
|
||
if #path == 0 then
|
||
return context.tree
|
||
end
|
||
|
||
local name = table.concat(path, ".")
|
||
local c = context.treecache[name]
|
||
|
||
if not c then
|
||
local last = table.remove(path)
|
||
local parent = _create_node(path)
|
||
|
||
c = {nodes={}, auto=true}
|
||
-- the node is "in request" if the request path matches
|
||
-- at least up to the length of the node path
|
||
if parent.inreq and context.path[#path+1] == last then
|
||
c.inreq = true
|
||
end
|
||
parent.nodes[last] = c
|
||
context.treecache[name] = c
|
||
end
|
||
return c
|
||
end
|
||
|
||
-- Subdispatchers --
|
||
|
||
function _firstchild()
|
||
local path = { unpack(context.path) }
|
||
local name = table.concat(path, ".")
|
||
local node = context.treecache[name]
|
||
|
||
local lowest
|
||
if node and node.nodes and next(node.nodes) then
|
||
local k, v
|
||
for k, v in pairs(node.nodes) do
|
||
if not lowest or
|
||
(v.order or 100) < (node.nodes[lowest].order or 100)
|
||
then
|
||
lowest = k
|
||
end
|
||
end
|
||
end
|
||
|
||
assert(lowest ~= nil,
|
||
"The requested node contains no childs, unable to redispatch")
|
||
|
||
path[#path+1] = lowest
|
||
dispatch(path)
|
||
end
|
||
|
||
--- Alias the first (lowest order) page automatically
|
||
function firstchild()
|
||
return { type = "firstchild", target = _firstchild }
|
||
end
|
||
|
||
--- Create a redirect to another dispatching node.
|
||
-- @param ... Virtual path destination
|
||
function alias(...)
|
||
local req = {...}
|
||
return function(...)
|
||
for _, r in ipairs({...}) do
|
||
req[#req+1] = r
|
||
end
|
||
|
||
dispatch(req)
|
||
end
|
||
end
|
||
|
||
--- Rewrite the first x path values of the request.
|
||
-- @param n Number of path values to replace
|
||
-- @param ... Virtual path to replace removed path values with
|
||
function rewrite(n, ...)
|
||
local req = {...}
|
||
return function(...)
|
||
local dispatched = util.clone(context.dispatched)
|
||
|
||
for i=1,n do
|
||
table.remove(dispatched, 1)
|
||
end
|
||
|
||
for i, r in ipairs(req) do
|
||
table.insert(dispatched, i, r)
|
||
end
|
||
|
||
for _, r in ipairs({...}) do
|
||
dispatched[#dispatched+1] = r
|
||
end
|
||
|
||
dispatch(dispatched)
|
||
end
|
||
end
|
||
|
||
local function _call(self, ...)
|
||
local func = getfenv()[self.name]
|
||
assert(func ~= nil,
|
||
'Cannot resolve function "' .. self.name .. '". Is it misspelled or local?')
|
||
|
||
assert(type(func) == "function",
|
||
'The symbol "' .. self.name .. '" does not refer to a function but data ' ..
|
||
'of type "' .. type(func) .. '".')
|
||
|
||
if #self.argv > 0 then
|
||
return func(unpack(self.argv), ...)
|
||
else
|
||
return func(...)
|
||
end
|
||
end
|
||
|
||
--- Create a function-call dispatching target.
|
||
-- @param name Target function of local controller
|
||
-- @param ... Additional parameters passed to the function
|
||
function call(name, ...)
|
||
return {type = "call", argv = {...}, name = name, target = _call}
|
||
end
|
||
|
||
|
||
local _template = function(self, ...)
|
||
require "luci.template".render(self.view)
|
||
end
|
||
|
||
--- Create a template render dispatching target.
|
||
-- @param name Template to be rendered
|
||
function template(name)
|
||
return {type = "template", view = name, target = _template}
|
||
end
|
||
|
||
local function _arcombine(self, ...)
|
||
local argv = {...}
|
||
local target = #argv > 0 and self.targets[2] or self.targets[1]
|
||
setfenv(target.target, self.env)
|
||
target:target(unpack(argv))
|
||
end
|
||
|
||
--- Create a combined dispatching target for non argv and argv requests.
|
||
-- @param trg1 Overview Target
|
||
-- @param trg2 Detail Target
|
||
function arcombine(trg1, trg2)
|
||
return {type = "arcombine", env = getfenv(), target = _arcombine, targets = {trg1, trg2}}
|
||
end
|
||
|
||
--- Access the luci.i18n translate() api.
|
||
-- @class function
|
||
-- @name translate
|
||
-- @param text Text to translate
|
||
translate = i18n.translate
|
||
|
||
--- No-op function used to mark translation entries for menu labels.
|
||
-- This function does not actually translate the given argument but
|
||
-- is used by build/i18n-scan.pl to find translatable entries.
|
||
function _(text)
|
||
return text
|
||
end |