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')