| rev | line source | 
| jbe@203 | 1 #!/usr/bin/env moonbridge | 
| jbe/bsw@0 | 2 | 
| jbe@203 | 3 WEBMCP_VERSION = "2.0.0_devel" | 
| jbe@64 | 4 | 
| jbe@231 | 5 -- allow control of global environment | 
| jbe@231 | 6 local _G = _G | 
| jbe@237 | 7 local allowed_globals = {} | 
| jbe@231 | 8 local global_metatable = { | 
| jbe@231 | 9   __index = _G, | 
| jbe@231 | 10   __newindex = function(self, key, value) | 
| jbe@231 | 11     _G[key] = value | 
| jbe@231 | 12   end | 
| jbe@231 | 13 } | 
| jbe@231 | 14 _ENV = setmetatable({}, global_metatable) | 
| jbe@231 | 15 | 
| jbe@203 | 16 -- check if interactive mode | 
| jbe@203 | 17 if listen then  -- defined by moonbridge | 
| jbe@203 | 18   WEBMCP_MODE = "listen" | 
| jbe@203 | 19 else | 
| jbe@203 | 20   WEBMCP_MODE = "interactive" | 
| jbe@64 | 21 end | 
| jbe@203 | 22 | 
| jbe@206 | 23 -- configuration names are provided as 4th, 5th, etc. argument | 
| jbe@206 | 24 WEBMCP_CONFIG_NAMES = {select(4, ...)} | 
| jbe@203 | 25 | 
| jbe@203 | 26 -- determine framework and bath path from command line arguments | 
| jbe@203 | 27 -- or print usage synopsis (if applicable) | 
| jbe@68 | 28 do | 
| jbe@206 | 29   local arg1, arg2, arg3 = ... | 
| jbe@203 | 30   local helpout | 
| jbe@203 | 31   if | 
| jbe@203 | 32     arg1 == "-h" or arg1 == "--help" or | 
| jbe@206 | 33     arg2 == "-h" or arg2 == "--help"  -- if first arg is provided by wrapper | 
| jbe@203 | 34   then | 
| jbe@203 | 35     helpout = io.stdout | 
| jbe@203 | 36   elseif | 
| jbe@217 | 37     #WEBMCP_CONFIG_NAMES < 1 or | 
| jbe@206 | 38     (WEBMCP_MODE == "interactive") ~= (arg3 == "INTERACTIVE") | 
| jbe@203 | 39   then | 
| jbe@203 | 40     helpout = io.stderr | 
| jbe@203 | 41   end | 
| jbe@203 | 42   if helpout then | 
| jbe@217 | 43     helpout:write("Usage: moonbridge -- <framework path>/bin/mcp.lua <framework path> <app base path> <app name> <config name> [<config name> ...]\n") | 
| jbe@217 | 44     helpout:write("   or: lua -i <framework path>/bin/mcp.lua <framework path> <app base path> INTERACTIVE <config name> [<config name> ...]\n") | 
| jbe@203 | 45     if helpout == io.stderr then | 
| jbe@203 | 46       return 1 | 
| jbe@68 | 47     else | 
| jbe@203 | 48       return 0 | 
| jbe@68 | 49     end | 
| jbe@68 | 50   end | 
| jbe@203 | 51   local function append_trailing_slash(str) | 
| jbe@217 | 52     return string.gsub(str, "([^/])$", function(last) return last .. "/" end) | 
| jbe@203 | 53   end | 
| jbe@203 | 54   WEBMCP_FRAMEWORK_PATH = append_trailing_slash(arg1) | 
| jbe@203 | 55   WEBMCP_BASE_PATH      = append_trailing_slash(arg2) | 
| jbe@206 | 56   if WEBMCP_MODE == "listen" then | 
| jbe@206 | 57     WEBMCP_APP_NAME = arg3 | 
| jbe@206 | 58   end | 
| jbe@68 | 59 end | 
| jbe@1 | 60 | 
| jbe@203 | 61 -- setup search paths for libraries | 
| jbe/bsw@0 | 62 do | 
| jbe@217 | 63   if string.match(package.path, "^[^;]") then | 
| jbe@217 | 64     package.path = ";" .. package.path | 
| jbe@217 | 65   end | 
| jbe@217 | 66   package.path = WEBMCP_FRAMEWORK_PATH .. "lib/?.lua" .. package.path | 
| jbe/bsw@0 | 67   -- find out which file name extension shared libraries have | 
| jbe/bsw@0 | 68   local slib_exts = {} | 
| jbe/bsw@0 | 69   for ext in string.gmatch(package.cpath, "%?%.([A-Za-z0-9_-]+)") do | 
| jbe@217 | 70     if not slib_exts[ext] then | 
| jbe@217 | 71       slib_exts[#slib_exts+1] = ext | 
| jbe@217 | 72       slib_exts[ext] = true | 
| jbe@217 | 73     end | 
| jbe/bsw@0 | 74   end | 
| jbe/bsw@0 | 75   local paths = {} | 
| jbe@217 | 76   for i, ext in ipairs(slib_exts) do | 
| jbe@203 | 77     paths[#paths+1] = WEBMCP_FRAMEWORK_PATH .. "accelerator/?." .. ext | 
| jbe/bsw@0 | 78   end | 
| jbe@217 | 79   for i, ext in ipairs(slib_exts) do | 
| jbe@203 | 80     paths[#paths+1] = WEBMCP_FRAMEWORK_PATH .. "lib/?." .. ext | 
| jbe/bsw@0 | 81   end | 
| jbe/bsw@0 | 82   paths[#paths+1] = package.cpath | 
| jbe/bsw@0 | 83   package.cpath = table.concat(paths, ";") | 
| jbe/bsw@0 | 84 end | 
| jbe/bsw@0 | 85 | 
| jbe@203 | 86 -- autoloader system for WebMCP environment "$WEBMCP_FRAMEWORK_PATH/env/", | 
| jbe@203 | 87 -- application environment extensions "$WEBMCP_BASE_PATH/env/" | 
| jbe@203 | 88 -- and models "$WEBMCP_BASE_PATH/model/" | 
| jbe/bsw@0 | 89 do | 
| jbe/bsw@0 | 90   local weakkey_mt = { __mode = "k" } | 
| jbe/bsw@0 | 91   local autoloader_category = setmetatable({}, weakkey_mt) | 
| jbe/bsw@0 | 92   local autoloader_path     = setmetatable({}, weakkey_mt) | 
| jbe/bsw@0 | 93   local autoloader_mt       = {} | 
| jbe@219 | 94   local function install_autoloader(self, category, path_fragment) | 
| jbe/bsw@0 | 95     autoloader_category[self] = category | 
| jbe@219 | 96     autoloader_path[self]     = path_fragment | 
| jbe/bsw@0 | 97     setmetatable(self, autoloader_mt) | 
| jbe/bsw@0 | 98   end | 
| jbe/bsw@0 | 99   local function try_exec(filename) | 
| jbe/bsw@0 | 100     local file = io.open(filename, "r") | 
| jbe/bsw@0 | 101     if file then | 
| jbe/bsw@0 | 102       local filedata = file:read("*a") | 
| jbe/bsw@0 | 103       io.close(file) | 
| jbe@232 | 104       local func, errmsg = load(filedata, "=" .. filename, nil, _ENV) | 
| jbe/bsw@0 | 105       if func then | 
| jbe/bsw@0 | 106         func() | 
| jbe/bsw@0 | 107         return true | 
| jbe/bsw@0 | 108       else | 
| jbe/bsw@0 | 109         error(errmsg, 0) | 
| jbe/bsw@0 | 110       end | 
| jbe/bsw@0 | 111     else | 
| jbe/bsw@0 | 112       return false | 
| jbe/bsw@0 | 113     end | 
| jbe/bsw@0 | 114   end | 
| jbe/bsw@0 | 115   function autoloader_mt.__index(self, key) | 
| jbe/bsw@0 | 116     local category, base_path, merge_base_path, file_key | 
| jbe/bsw@0 | 117     local merge = false | 
| jbe/bsw@0 | 118     if | 
| jbe/bsw@0 | 119       string.find(key, "^[a-z_][A-Za-z0-9_]*$") and | 
| jbe/bsw@0 | 120       not string.find(key, "^__") | 
| jbe/bsw@0 | 121     then | 
| jbe/bsw@0 | 122       category        = "env" | 
| jbe@203 | 123       base_path       = WEBMCP_FRAMEWORK_PATH .. "env/" | 
| jbe/bsw@0 | 124       merge           = true | 
| jbe@203 | 125       merge_base_path = WEBMCP_BASE_PATH .. "env/" | 
| jbe/bsw@0 | 126       file_key        = key | 
| jbe/bsw@0 | 127     elseif string.find(key, "^[A-Z][A-Za-z0-9]*$") then | 
| jbe/bsw@0 | 128       category        = "model" | 
| jbe@203 | 129       base_path       = WEBMCP_BASE_PATH .. "model/" | 
| jbe/bsw@0 | 130       local first = true | 
| jbe/bsw@0 | 131       file_key = string.gsub(key, "[A-Z]", | 
| jbe/bsw@0 | 132         function(c) | 
| jbe/bsw@0 | 133           if first then | 
| jbe/bsw@0 | 134             first = false | 
| jbe/bsw@0 | 135             return string.lower(c) | 
| jbe/bsw@0 | 136           else | 
| jbe/bsw@0 | 137             return "_" .. string.lower(c) | 
| jbe/bsw@0 | 138           end | 
| jbe/bsw@0 | 139         end | 
| jbe/bsw@0 | 140       ) | 
| jbe/bsw@0 | 141     else | 
| jbe/bsw@0 | 142       return | 
| jbe/bsw@0 | 143     end | 
| jbe/bsw@0 | 144     local required_category = autoloader_category[self] | 
| jbe/bsw@0 | 145     if required_category and required_category ~= category then return end | 
| jbe@219 | 146     local path_fragment = autoloader_path[self] | 
| jbe@219 | 147     local path = base_path .. path_fragment .. file_key | 
| jbe@219 | 148     local merge_path | 
| jbe/bsw@0 | 149     if merge then | 
| jbe@219 | 150       merge_path = merge_base_path .. path_fragment .. file_key | 
| jbe/bsw@0 | 151     end | 
| jbe/bsw@0 | 152     local function try_dir(dirname) | 
| jbe/bsw@0 | 153       local dir = io.open(dirname) | 
| jbe/bsw@0 | 154       if dir then | 
| jbe/bsw@0 | 155         io.close(dir) | 
| jbe/bsw@0 | 156         local obj = {} | 
| jbe@219 | 157         install_autoloader(obj, category, path_fragment .. file_key .. "/") | 
| jbe/bsw@0 | 158         rawset(self, key, obj) | 
| jbe@219 | 159         try_exec(path .. "/__init.lua") | 
| jbe@219 | 160         if merge then try_exec(merge_path .. "/__init.lua") end | 
| jbe/bsw@0 | 161         return true | 
| jbe/bsw@0 | 162       else | 
| jbe/bsw@0 | 163         return false | 
| jbe/bsw@0 | 164       end | 
| jbe/bsw@0 | 165     end | 
| jbe@238 | 166     if self == _G then | 
| jbe@237 | 167       allowed_globals[key] = true | 
| jbe@233 | 168     end | 
| jbe@219 | 169     if merge and try_exec(merge_path .. ".lua") then | 
| jbe@219 | 170     elseif merge and try_dir(merge_path .. "/") then | 
| jbe@219 | 171     elseif try_exec(path .. ".lua") then | 
| jbe@219 | 172     elseif try_dir(path .. "/") then | 
| jbe/bsw@0 | 173     else end | 
| jbe@238 | 174     if self == _G then | 
| jbe@237 | 175       allowed_globals[key] = nil | 
| jbe@233 | 176     end | 
| jbe@237 | 177     return rawget(self, key) | 
| jbe/bsw@0 | 178   end | 
| jbe@219 | 179   install_autoloader(_G, nil, "") | 
| jbe@203 | 180   try_exec(WEBMCP_FRAMEWORK_PATH .. "env/__init.lua") | 
| jbe@203 | 181   try_exec(WEBMCP_BASE_PATH .. "env/__init.lua") | 
| jbe/bsw@0 | 182 end | 
| jbe/bsw@0 | 183 | 
| jbe@214 | 184 -- replace Moonbridge listen function | 
| jbe@214 | 185 local moonbridge_listen = listen | 
| jbe@225 | 186 local listeners = {} | 
| jbe@214 | 187 function listen(args) | 
| jbe@214 | 188   listeners[#listeners+1] = args | 
| jbe@214 | 189 end | 
| jbe@214 | 190 | 
| jbe@203 | 191 -- prohibit (unintended) definition of new global variables | 
| jbe@237 | 192 function global_metatable.__newindex(self, key, value) | 
| jbe@237 | 193   if not allowed_globals[key] then | 
| jbe@237 | 194     error("Setting of global variable prohibited", 2) | 
| jbe@237 | 195   end | 
| jbe@239 | 196   _G[key] = value | 
| jbe@231 | 197 end | 
| jbe@203 | 198 | 
| jbe@220 | 199 -- execute configurations and pre-fork initializers | 
| jbe@206 | 200 for i, config_name in ipairs(WEBMCP_CONFIG_NAMES) do | 
| jbe@206 | 201   execute.config(config_name) | 
| jbe@206 | 202 end | 
| jbe@220 | 203 execute.prefork_initializers() | 
| jbe@206 | 204 | 
| jbe/bsw@0 | 205 -- interactive console mode | 
| jbe@203 | 206 if WEBMCP_MODE == "interactive" then | 
| jbe@229 | 207   _G.multirand = require "multirand"  -- TODO: cleaner solution | 
| jbe@215 | 208   execute.postfork_initializers() | 
| jbe@203 | 209   trace.disable()  -- avoids memory leakage (TODO: needs general solution for moonbridge?) | 
| jbe/bsw@0 | 210 end | 
| jbe/bsw@0 | 211 | 
| jbe@204 | 212 -- invoke moonbridge | 
| jbe@206 | 213 if WEBMCP_MODE == "listen" then | 
| jbe@264 | 214   local http_options = request.get_http_options() | 
| jbe@264 | 215   local min_requests_per_connect = http_options.min_requests_per_connect or 90 | 
| jbe@264 | 216   local max_requests_per_connect = http_options.max_requests_per_connect or 100 | 
| jbe@207 | 217   local http = require("moonbridge_http") | 
| jbe@211 | 218   for i, listener in ipairs(listeners) do | 
| jbe@264 | 219     local request_count = 0 | 
| jbe@266 | 220     local function inner_handler(http_request) | 
| jbe@264 | 221       request_count = request_count + 1 | 
| jbe@266 | 222       request.handler(http_request, request_count >= max_requests_per_connect) | 
| jbe@264 | 223     end | 
| jbe@264 | 224     local outer_handler = http.generate_handler(inner_handler, http_options) | 
| jbe@229 | 225     --listener.prepare = execute.postfork_initializers | 
| jbe@229 | 226     listener.prepare = function() | 
| jbe@229 | 227       _G.multirand = require "multirand" | 
| jbe@229 | 228       execute.postfork_initializers() | 
| jbe@229 | 229     end | 
| jbe@264 | 230     listener.connect = function(socket) | 
| jbe@264 | 231       outer_handler(socket) | 
| jbe@264 | 232       return request_count < min_requests_per_connect | 
| jbe@264 | 233     end | 
| jbe@211 | 234     listener.finish = execute.finalizers | 
| jbe@211 | 235     moonbridge_listen(listener) | 
| jbe@204 | 236   end | 
| jbe@204 | 237 end | 
| jbe@204 | 238 |