mail/postfix/cgi/lib.inc

   1 -- library for cgi
   2 if _lib_inc_init ~= nil then return end
   3 _lib_inc_init = 1
   4 io.stderr = io.stdout
   5 dofile("config.cnf");
   6 dofile("lang_" .. CONF_LANG .. ".inc");
   7
   8 err_msg = nil   -- show err_msg
   9 -- archss_header output
  10 -- title: outout the web title
  11 -- fun_cb: callback function for private data
  12 function as_head(fun_cb, title)
  13         assert(type(title) == "string");
  14         assert(type(fun_cb) == "function");
  15
  16         print('Content-type: text/html\n\n' ..
  17                 '<HTML><HEAD><TITLE>' .. title .. '</TITLE>\n' ..
  18                 '<META HTTP-EQUIV="Content-Type" CONTENT="text/html; CHARSET="UTF-8"/>\n' ..
  19                 '<LINK REL=STYLESHEET HREF="styles.css" TYPE="TEXT/CSS">');
  20         fun_cb();
  21         print('</HEAD>');
  22 end
  23
  24 -- onload: web init with javascript
  25 function as_body(fun_cb, onload)
  26         assert(type(onload) == "string");
  27         assert(type(fun_cb) == "function");
  28
  29         print('<BODY ONLOAD="' .. onload .. '">');
  30         fun_cb();
  31         print('</BODY></HTML>');
  32 end
  33
  34 -- archss_form output
  35 -- target: post this form to which cgi
  36 -- datatype: used for binary or normal data
  37 function as_form(fun_cb, target, datatype)
  38         assert(type(fun_cb) == "function")
  39         assert(type(target) == "string")
  40
  41         local dt = "";
  42         if datatype ~= nil then dt = datatype end;
  43         print('<form action="' .. target .. CONF_EXT .. '" method="post"' .. dt .. '>');
  44         fun_cb();
  45         print('</form>');
  46 end
  47
  48 function as_table(fun_cb, width, class)
  49         assert(type(fun_cb) == "function")
  50         assert(type(width) == "string")
  51
  52         print('<table width="' .. width .. '" style="' .. class .. '">');
  53         fun_cb();
  54         print('</table>\n')
  55 end
  56
  57 -- parse STDIN and query_string
  58 local _as_stdin_init;   -- parsed flag
  59 local _as_query_data = {};      -- query data store
  60 --local _as_kv_token = "(%w+)=([%w_-%%]+)"
  61 local _as_kv_token = "([^&=]+)=([^&=]+)"
  62 -- FIXME this MUST work with multi-byte encoded data
  63 function QUERY()
  64         if _as_stdin_init ~= nil then return _as_query_data end
  65         local query_string = os.getenv("QUERY_STRING");
  66         local k, v
  67
  68         if query_string == nil then return _as_query_data end
  69         for k, v in string.gfind(query_string, _as_kv_token) do
  70                 _as_query_data[unescape(k)] = unescape(v);
  71         end
  72         _as_stdin_init = 1;
  73         return _as_query_data;
  74 end
  75
  76 local _as_post_init;
  77 local _as_post_data = {};       -- post data store
  78 function POST()
  79         if _as_post_init ~= nil then return _as_post_data end
  80         local read = io.read();
  81         local k, v
  82
  83         if read == nil then return _as_post_data end
  84         for k, v in string.gfind(read, _as_kv_token) do
  85                 _as_post_data[unescape(k)] = unescape(v);
  86         end
  87         _as_post_init = 1;
  88         return _as_post_data;
  89 end
  90
  91 --[[ all key/value in one
  92    FIXME how to do it
  93 function QP_ENV()
  94         return QUERY() .. POST();
  95 end
  96 ]]
  97
  98 function show_errmsg()
  99         if err_msg ~= nil then print('<div class="err_msg">'..err_msg..'</div>') end
 100 end
 101
 102 -- database deal with
 103 env, con = nil, nil;
 104 function db_open()
 105         require "luasql.mysql";
 106         env = assert (luasql.mysql())
 107         con = assert (env:connect(CONF_DB.DBN, CONF_DB.USR, CONF_DB.PWD, CONF_DB.SRV), "mysql connect bad")
 108 end
 109
 110 function rows(connection, sql_statement)
 111         local cursor = assert(connection:execute(sql_statement))
 112         return function()
 113                 return cursor:fetch()
 114         end
 115 end
 116
 117 -- form deal with
 118 function unescape (s)
 119         if s == nil then return s end
 120         s = string.gsub(s, "+", " ")
 121         s = string.gsub(s, "%%(%x%x)", function (h)
 122                   return string.char(tonumber(h, 16))
 123                 end)
 124         return s
 125 end
 126
 127 function escape (s)
 128         s = string.gsub(s, "([&=+%c])", function (c)
 129                   return string.format("%%%02X", string.byte(c))
 130                 end)
 131         s = string.gsub(s, " ", "+")
 132         return s
 133 end
 134
 135 function encode (t)
 136         local s = ""
 137         local k, v
 138         for k,v in pairs(t) do
 139                 s = s .. "&" .. escape(k) .. "=" .. escape(v)
 140         end
 141         return string.sub(s, 2)     -- remove first `&'
 142 end
 143
 144
 145 -- private msg store with "XXX_ARCHSS", we need config it when install
 146 -- vim:ft=lua