antd-lua-plugin/APIs/api.lua

math.randomseed(os.clock())
package.cpath = __api__.apiroot..'/?.so'
require("antd")
std = modules.std()
local read_header =function()
    local l
    repeat
        l = std.antd_recv(HTTP_REQUEST.id)
        if l and l ~= '\r'  then
            if l == "HTTP_REQUEST" or l == "request" or l == "COOKIE" or l == "REQUEST_HEADER" or l == "REQUEST_DATA" then
                coroutine.yield(l, "LUA_TABLE")
            else
                local l1 = std.antd_recv(HTTP_REQUEST.id)
                if l1 ~= '\r' then
                    coroutine.yield(l, l1)
                end
                l = l1
            end
        end
    until not l or l == '\r'
end


local read_headers = function()
    local co = coroutine.create(function () read_header() end)
      return function ()   -- iterator
        local code, k, v = coroutine.resume(co)
        return k,v
      end
end

local parse_headers =function()
    local lut = {
        HTTP_REQUEST = HTTP_REQUEST
    }
    local curr_tbl = "HTTP_REQUEST"
    for k,v in read_headers() do
        if v == "LUA_TABLE" then
            if not lut[k] then
                lut[k] = {}
            end
            curr_tbl = k
        else
            lut[curr_tbl][k] = v
        end
    end
    HTTP_REQUEST.request = lut.request
    HTTP_REQUEST.request.COOKIE = lut.COOKIE
    HTTP_REQUEST.request.REQUEST_HEADER = lut.REQUEST_HEADER
    HTTP_REQUEST.request.REQUEST_DATA = lut.REQUEST_DATA
end

-- parsing the header
parse_headers()
-- root dir
__ROOT__ = HTTP_REQUEST.request.SERVER_WWW_ROOT
-- set require path
package.path = __ROOT__ .. '/?.lua;'..__api__.apiroot..'/?.lua'
require("std")
require("utils")
require("extra_mime")
ulib = require("ulib")
-- set session
SESSION = {}

REQUEST = HTTP_REQUEST.request.REQUEST_DATA
REQUEST.method = HTTP_REQUEST.request.METHOD
if HTTP_REQUEST.request.COOKIE then
	SESSION = HTTP_REQUEST.request.COOKIE
end
HEADER = HTTP_REQUEST.request.REQUEST_HEADER
HEADER.mobile = false

if HEADER["User-Agent"] and HEADER["User-Agent"]:match("Mobi") then
    HEADER.mobile = true
end

function LOG_INFO(fmt,...)
    ulib.syslog(5,string.format(fmt or "LOG",...))
end

function LOG_ERROR(fmt,...)
    ulib.syslog(3,string.format(fmt or "ERROR",...))
end

function has_module(m)
	if utils.file_exists(__ROOT__..'/'..m) then
		if m:find("%.ls$") then
			return true, true,  __ROOT__..'/'..m
		else
			return true, false, m:gsub(".lua$","")
		end
	elseif utils.file_exists(__ROOT__..'/'..string.gsub(m,'%.','/')..'.lua') then
		return true, false, m
	elseif utils.file_exists(__ROOT__..'/'..string.gsub(m,'%.','/')..'.ls') then
		return true, true, __ROOT__..'/'..string.gsub(m,'%.','/')..'.ls'
	end
	return false, false, nil
end

function echo(m)
    if m then std.t(m) else std.t("Undefined value") end
end

function loadscript(file, args)
    local f = io.open(file, "rb")
    local content = ""
    if f then
        local html = ""
        local pro = "local fn = function(...)"
        local s,e, mt
        local mtbegin = true -- find begin of scrit, 0 end of scrit
        local i = 1
        if args then
            pro = "local fn = function("..table.concat( args, ",")..")"
        end
        for line in io.lines(file) do
            line = std.trim(line, " ")
            if(line ~= "") then
                if(mtbegin) then
                    mt = "^%s*<%?lua"
                else
                    mt = "%?>%s*$"
                end
                s,e = line:find(mt)
                if(s) then
                    if mtbegin then
                        if html ~= "" then
                            pro= pro.."echo(\""..utils.escape(html).."\")\n"
                            html = ""
                        end
                        local b,f  = line:find("%?>%s*$")
                        if b then
                            pro = pro..line:sub(e+1,b-1).."\n"
                        else
                            pro = pro..line:sub(e+1).."\n"
                            mtbegin = not mtbegin
                        end
                    else
                        pro = pro..line:sub(0,s-1).."\n"
                        mtbegin = not mtbegin
                    end
                else -- no match
                    if mtbegin then
                        -- detect if we have inline lua with format <?=..?>
                        local b,f = line:find("<%?=")
                        if b then
                            local tmp = line
                            pro= pro.."echo("
                            while(b) do
                                -- find the close
                                local x,y = tmp:find("%?>")
                                if x then
                                    pro = pro.."\""..utils.escape(html..tmp:sub(0,b-1):gsub("%%","%%%%")).."\".."
                                    pro = pro..tmp:sub(f+1,x-1)..".."
                                    html = ""
                                    tmp = tmp:sub(y+1)
                                    b,f = tmp:find("<%?=")
                                else
                                    error("Syntax error near line "..i)
                                end
                            end
                            pro = pro.."\""..utils.escape(tmp:gsub("%%","%%%%")).."\")\n"
                        else
                            html = html..std.trim(line," "):gsub("%%","%%%%").."\n"
                        end
                    else
                        if line ~= "" then pro = pro..line.."\n" end
                    end
                end
            end
            i = i+ 1
        end
        f:close()
        if(html ~= "") then
            pro = pro.."echo(\""..utils.escape(html).."\")\n"
        end
        pro  = pro.."\nend \n return fn"
        local r,e = load(pro)
        if r then return r(), e else return nil,e end
    end
end

-- decode post data if any
local decode_request_data = function()
    if (not REQUEST.method)
        or (REQUEST.method ~= "POST"
            and REQUEST.method ~= "PUT"
            and REQUEST.method ~= "PATCH")
        or (not REQUEST.HAS_RAW_BODY) then
            return 0
    end
    local ctype = HEADER['Content-Type']
    local clen = HEADER['Content-Length'] or -1
    if clen then
        clen = tonumber(clen)
    end
    if not ctype or clen == -1 then
        LOG_ERROR("Invalid content type %s or content length %d", ctype, clen)
        return 400, "Bad Request, missing content description"
    end
    local raw_data, len = std.antd_recv(HTTP_REQUEST.id, clen)
    if len ~= clen then
        LOG_ERROR("Unable to read all data: read %d expected %d", len, clen)
        return 400, "Bad Request, missing content data"
    end
    if ctype:find("application/json") then
        REQUEST.json = bytes.__tostring(raw_data)
    else
        REQUEST[ctype] = raw_data
    end
    REQUEST.HAS_RAW_BODY = nil
    return 0
end

-- set compression level
local accept_encoding = HEADER["Accept-Encoding"]
if accept_encoding then
    if accept_encoding:find("gzip") then
        std.antd_set_zlevel(HTTP_REQUEST.id, "gzip")
    elseif accept_encoding:find("deflate") then
        std.antd_set_zlevel(HTTP_REQUEST.id, "deflate")
    end
end

local code, error = decode_request_data()

if code ~= 0 then
    LOG_ERROR(error)
    std.error(code, error)
    return
end

-- LOG_INFO(JSON.encode(REQUEST))

-- OOP support
--require("OOP")
-- load sqlite helper
--require("sqlite")
-- enable extra mime

-- run the file


local m, s, p  = has_module(HTTP_REQUEST.request.REQUEST_URI)
if m then
    -- run the correct module
    if s then
		local r,e = loadscript(p)
		if r then r() else unknow(e) end
    else
        LOG_INFO("RUNNING MODULE %s", p)
        require(p)
    end
else
	unknow("Resource not found for request "..HTTP_REQUEST.request.REQUEST_URI)
end


--require('router')