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