Sun, 23 Nov 2008 05:43:17 +0000
Backed out changeset 79bd7a3e906c
382 | 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 |