| rev | 
   line source | 
| 
jbe@203
 | 
     1 #!/usr/bin/env moonbridge
 | 
| 
jbe/bsw@0
 | 
     2 
 | 
| 
jbe@278
 | 
     3 WEBMCP_VERSION = "2.0.0"
 | 
| 
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 
 |