Mercurial > lxmppd-hg-xmllex / file comparison
comparison: plugins/mod_console.lua
plugins/mod_console.lua
- changeset 382
- 9daf1e3d278e
- child 411
- 64982773cc15
equal
deleted
inserted
replaced
| 381:e967f162501e | 382:9daf1e3d278e |
|---|---|
| 1 | |
| 2 local connlisteners_register = require "net.connlisteners".register; | |
| 3 | |
| 4 local console_listener = { default_port = 5582; default_mode = "*l"; }; | |
| 5 | |
| 6 local commands = {}; | |
| 7 local default_env = {}; | |
| 8 local default_env_mt = { __index = default_env }; | |
| 9 | |
| 10 console = {}; | |
| 11 | |
| 12 function console:new_session(conn) | |
| 13 local w = conn.write; | |
| 14 return { conn = conn; | |
| 15 send = function (t) w(tostring(t)); end; | |
| 16 print = function (t) w("| "..tostring(t).."\n"); end; | |
| 17 disconnect = function () conn.close(); end; | |
| 18 env = setmetatable({}, default_env_mt); | |
| 19 }; | |
| 20 end | |
| 21 | |
| 22 local sessions = {}; | |
| 23 | |
| 24 function console_listener.listener(conn, data) | |
| 25 local session = sessions[conn]; | |
| 26 | |
| 27 if not session then | |
| 28 -- Handle new connection | |
| 29 session = console:new_session(conn); | |
| 30 sessions[conn] = session; | |
| 31 session.print("Welcome to the lxmppd admin console!"); | |
| 32 end | |
| 33 if data then | |
| 34 -- Handle data | |
| 35 | |
| 36 if data:match("[!.]$") then | |
| 37 local command = data:lower(); | |
| 38 command = data:match("^%w+") or data:match("%p"); | |
| 39 if commands[command] then | |
| 40 commands[command](session, data); | |
| 41 return; | |
| 42 end | |
| 43 end | |
| 44 | |
| 45 session.env._ = data; | |
| 46 | |
| 47 local chunk, err = loadstring("return "..data); | |
| 48 if not chunk then | |
| 49 chunk, err = loadstring(data); | |
| 50 if not chunk then | |
| 51 err = err:gsub("^%[string .-%]:%d+: ", ""); | |
| 52 err = err:gsub("^:%d+: ", ""); | |
| 53 err = err:gsub("'<eof>'", "the end of the line"); | |
| 54 session.print("Sorry, I couldn't understand that... "..err); | |
| 55 return; | |
| 56 end | |
| 57 end | |
| 58 | |
| 59 setfenv(chunk, session.env); | |
| 60 local ranok, taskok, message = pcall(chunk); | |
| 61 | |
| 62 if not ranok then | |
| 63 session.print("Fatal error while running command, it did not complete"); | |
| 64 session.print("Error: "..taskok); | |
| 65 return; | |
| 66 end | |
| 67 | |
| 68 if not message then | |
| 69 session.print("Result: "..tostring(taskok)); | |
| 70 return; | |
| 71 elseif (not taskok) and message then | |
| 72 session.print("Command completed with a problem"); | |
| 73 session.print("Message: "..tostring(message)); | |
| 74 return; | |
| 75 end | |
| 76 | |
| 77 session.print("OK: "..tostring(message)); | |
| 78 end | |
| 79 end | |
| 80 | |
| 81 function console_listener.disconnect(conn, err) | |
| 82 | |
| 83 end | |
| 84 | |
| 85 connlisteners_register('console', console_listener); | |
| 86 | |
| 87 -- Console commands -- | |
| 88 -- These are simple commands, not valid standalone in Lua | |
| 89 | |
| 90 function commands.bye(session) | |
| 91 session.print("See you! :)"); | |
| 92 session.disconnect(); | |
| 93 end | |
| 94 | |
| 95 commands["!"] = function (session, data) | |
| 96 if data:match("^!!") then | |
| 97 session.print("!> "..session.env._); | |
| 98 return console_listener.listener(session.conn, session.env._); | |
| 99 end | |
| 100 local old, new = data:match("^!(.-[^\\])!(.-)!$"); | |
| 101 if old and new then | |
| 102 local ok, res = pcall(string.gsub, session.env._, old, new); | |
| 103 if not ok then | |
| 104 session.print(res) | |
| 105 return; | |
| 106 end | |
| 107 session.print("!> "..res); | |
| 108 return console_listener.listener(session.conn, res); | |
| 109 end | |
| 110 session.print("Sorry, not sure what you want"); | |
| 111 end | |
| 112 | |
| 113 -- Session environment -- | |
| 114 -- Anything in default_env will be accessible within the session as a global variable | |
| 115 | |
| 116 default_env.server = {}; | |
| 117 function default_env.server.reload() | |
| 118 dofile "main.lua" | |
| 119 return true, "Server reloaded"; | |
| 120 end | |
| 121 | |
| 122 default_env.module = {}; | |
| 123 function default_env.module.load(name) | |
| 124 local mm = require "modulemanager"; | |
| 125 local ok, err = mm.load(name); | |
| 126 if not ok then | |
| 127 return false, err or "Unknown error loading module"; | |
| 128 end | |
| 129 return true, "Module loaded"; | |
| 130 end | |
| 131 | |
| 132 default_env.config = {}; | |
| 133 function default_env.config.load(filename, format) | |
| 134 local cfgm_load = require "core.configmanager".load; | |
| 135 local ok, err = cfgm_load(filename, format); | |
| 136 if not ok then | |
| 137 return false, err or "Unknown error loading config"; | |
| 138 end | |
| 139 return true, "Config loaded"; | |
| 140 end |
