Software / code / prosody
Annotate
plugins/mod_console.lua @ 1065:3806173670f2
mod_*: Fix many unnecessary global accesses in modules (already committed to main repo)
| author | Matthew Wild <mwild1@gmail.com> |
|---|---|
| date | Wed, 29 Apr 2009 02:08:12 +0100 |
| parent | 896:2c0b9e3c11c3 |
| child | 1085:1ac11fb753ca |
| rev | line source |
|---|---|
| 896 | 1 -- Prosody IM v0.4 |
|
760
90ce865eebd8
Update copyright notices for 2009
Matthew Wild <mwild1@gmail.com>
parents:
759
diff
changeset
|
2 -- Copyright (C) 2008-2009 Matthew Wild |
|
90ce865eebd8
Update copyright notices for 2009
Matthew Wild <mwild1@gmail.com>
parents:
759
diff
changeset
|
3 -- Copyright (C) 2008-2009 Waqas Hussain |
|
519
cccd610a0ef9
Insert copyright/license headers
Matthew Wild <mwild1@gmail.com>
parents:
461
diff
changeset
|
4 -- |
| 758 | 5 -- This project is MIT/X11 licensed. Please see the |
| 6 -- COPYING file in the source package for more information. | |
|
519
cccd610a0ef9
Insert copyright/license headers
Matthew Wild <mwild1@gmail.com>
parents:
461
diff
changeset
|
7 -- |
|
cccd610a0ef9
Insert copyright/license headers
Matthew Wild <mwild1@gmail.com>
parents:
461
diff
changeset
|
8 |
| 736 | 9 module.host = "*"; |
|
519
cccd610a0ef9
Insert copyright/license headers
Matthew Wild <mwild1@gmail.com>
parents:
461
diff
changeset
|
10 |
|
1065
3806173670f2
mod_*: Fix many unnecessary global accesses in modules (already committed to main repo)
Matthew Wild <mwild1@gmail.com>
parents:
896
diff
changeset
|
11 local hosts = _G.hosts; |
| 736 | 12 local connlisteners_register = require "net.connlisteners".register; |
| 13 | |
| 14 local console_listener = { default_port = 5582; default_mode = "*l"; }; | |
| 15 | |
| 16 local commands = {}; | |
| 17 local def_env = {}; | |
| 18 local default_env_mt = { __index = def_env }; | |
| 19 | |
| 20 console = {}; | |
| 21 | |
| 22 function console:new_session(conn) | |
| 23 local w = function(s) conn.write(s:gsub("\n", "\r\n")); end; | |
| 24 local session = { conn = conn; | |
| 25 send = function (t) w(tostring(t)); end; | |
| 26 print = function (t) w("| "..tostring(t).."\n"); end; | |
| 27 disconnect = function () conn.close(); end; | |
| 28 }; | |
| 29 session.env = setmetatable({}, default_env_mt); | |
| 30 | |
| 31 -- Load up environment with helper objects | |
| 32 for name, t in pairs(def_env) do | |
| 33 if type(t) == "table" then | |
| 34 session.env[name] = setmetatable({ session = session }, { __index = t }); | |
| 35 end | |
| 36 end | |
| 37 | |
| 38 return session; | |
| 39 end | |
| 40 | |
| 41 local sessions = {}; | |
| 42 | |
| 43 function console_listener.listener(conn, data) | |
| 44 local session = sessions[conn]; | |
| 45 | |
| 46 if not session then | |
| 47 -- Handle new connection | |
| 48 session = console:new_session(conn); | |
| 49 sessions[conn] = session; | |
| 50 printbanner(session); | |
| 51 end | |
| 52 if data then | |
| 53 -- Handle data | |
| 54 (function(session, data) | |
| 55 if data:match("[!.]$") then | |
| 56 local command = data:lower(); | |
| 57 command = data:match("^%w+") or data:match("%p"); | |
| 58 if commands[command] then | |
| 59 commands[command](session, data); | |
| 60 return; | |
| 61 end | |
| 62 end | |
| 63 | |
| 64 session.env._ = data; | |
| 65 | |
| 66 local chunk, err = loadstring("return "..data); | |
| 67 if not chunk then | |
| 68 chunk, err = loadstring(data); | |
| 69 if not chunk then | |
| 70 err = err:gsub("^%[string .-%]:%d+: ", ""); | |
| 71 err = err:gsub("^:%d+: ", ""); | |
| 72 err = err:gsub("'<eof>'", "the end of the line"); | |
| 73 session.print("Sorry, I couldn't understand that... "..err); | |
| 74 return; | |
| 75 end | |
| 76 end | |
| 77 | |
| 78 setfenv(chunk, session.env); | |
| 79 local ranok, taskok, message = pcall(chunk); | |
| 80 | |
| 81 if not ranok then | |
| 82 session.print("Fatal error while running command, it did not complete"); | |
| 83 session.print("Error: "..taskok); | |
| 84 return; | |
| 85 end | |
| 86 | |
| 87 if not message then | |
| 88 session.print("Result: "..tostring(taskok)); | |
| 89 return; | |
| 90 elseif (not taskok) and message then | |
| 91 session.print("Command completed with a problem"); | |
| 92 session.print("Message: "..tostring(message)); | |
| 93 return; | |
| 94 end | |
| 95 | |
|
669
9255abbb3068
mod_console: replace all \n with \r\n in the output, and send \0 as a marker character after every response
Waqas Hussain <waqas20@gmail.com>
parents:
615
diff
changeset
|
96 session.print("OK: "..tostring(message)); |
|
9255abbb3068
mod_console: replace all \n with \r\n in the output, and send \0 as a marker character after every response
Waqas Hussain <waqas20@gmail.com>
parents:
615
diff
changeset
|
97 end)(session, data); |
| 736 | 98 end |
|
669
9255abbb3068
mod_console: replace all \n with \r\n in the output, and send \0 as a marker character after every response
Waqas Hussain <waqas20@gmail.com>
parents:
615
diff
changeset
|
99 session.send(string.char(0)); |
| 736 | 100 end |
| 101 | |
| 102 function console_listener.disconnect(conn, err) | |
| 103 | |
| 104 end | |
| 105 | |
| 106 connlisteners_register('console', console_listener); | |
| 107 | |
| 108 -- Console commands -- | |
| 109 -- These are simple commands, not valid standalone in Lua | |
| 110 | |
| 111 function commands.bye(session) | |
| 112 session.print("See you! :)"); | |
| 113 session.disconnect(); | |
| 114 end | |
| 115 | |
| 116 commands["!"] = function (session, data) | |
| 117 if data:match("^!!") then | |
| 118 session.print("!> "..session.env._); | |
| 119 return console_listener.listener(session.conn, session.env._); | |
| 120 end | |
| 121 local old, new = data:match("^!(.-[^\\])!(.-)!$"); | |
| 122 if old and new then | |
| 123 local ok, res = pcall(string.gsub, session.env._, old, new); | |
| 124 if not ok then | |
| 125 session.print(res) | |
| 126 return; | |
| 127 end | |
| 128 session.print("!> "..res); | |
| 129 return console_listener.listener(session.conn, res); | |
| 130 end | |
| 131 session.print("Sorry, not sure what you want"); | |
| 132 end | |
| 133 | |
| 134 -- Session environment -- | |
| 135 -- Anything in def_env will be accessible within the session as a global variable | |
| 136 | |
| 137 def_env.server = {}; | |
| 138 function def_env.server:reload() | |
| 139 dofile "prosody" | |
| 140 return true, "Server reloaded"; | |
| 141 end | |
| 142 | |
| 143 def_env.module = {}; | |
| 144 function def_env.module:load(name, host, config) | |
| 145 local mm = require "modulemanager"; | |
| 146 local ok, err = mm.load(host or self.env.host, name, config); | |
| 147 if not ok then | |
| 148 return false, err or "Unknown error loading module"; | |
| 149 end | |
| 150 return true, "Module loaded"; | |
| 151 end | |
| 152 | |
|
712
56410c0cd846
mod_console: Added module:reload
Waqas Hussain <waqas20@gmail.com>
parents:
669
diff
changeset
|
153 function def_env.module:unload(name, host) |
|
56410c0cd846
mod_console: Added module:reload
Waqas Hussain <waqas20@gmail.com>
parents:
669
diff
changeset
|
154 local mm = require "modulemanager"; |
|
56410c0cd846
mod_console: Added module:reload
Waqas Hussain <waqas20@gmail.com>
parents:
669
diff
changeset
|
155 local ok, err = mm.unload(host or self.env.host, name); |
|
56410c0cd846
mod_console: Added module:reload
Waqas Hussain <waqas20@gmail.com>
parents:
669
diff
changeset
|
156 if not ok then |
|
56410c0cd846
mod_console: Added module:reload
Waqas Hussain <waqas20@gmail.com>
parents:
669
diff
changeset
|
157 return false, err or "Unknown error unloading module"; |
|
56410c0cd846
mod_console: Added module:reload
Waqas Hussain <waqas20@gmail.com>
parents:
669
diff
changeset
|
158 end |
|
56410c0cd846
mod_console: Added module:reload
Waqas Hussain <waqas20@gmail.com>
parents:
669
diff
changeset
|
159 return true, "Module unloaded"; |
|
56410c0cd846
mod_console: Added module:reload
Waqas Hussain <waqas20@gmail.com>
parents:
669
diff
changeset
|
160 end |
|
56410c0cd846
mod_console: Added module:reload
Waqas Hussain <waqas20@gmail.com>
parents:
669
diff
changeset
|
161 |
|
56410c0cd846
mod_console: Added module:reload
Waqas Hussain <waqas20@gmail.com>
parents:
669
diff
changeset
|
162 function def_env.module:reload(name, host) |
|
56410c0cd846
mod_console: Added module:reload
Waqas Hussain <waqas20@gmail.com>
parents:
669
diff
changeset
|
163 local mm = require "modulemanager"; |
|
56410c0cd846
mod_console: Added module:reload
Waqas Hussain <waqas20@gmail.com>
parents:
669
diff
changeset
|
164 local ok, err = mm.reload(host or self.env.host, name); |
|
56410c0cd846
mod_console: Added module:reload
Waqas Hussain <waqas20@gmail.com>
parents:
669
diff
changeset
|
165 if not ok then |
|
56410c0cd846
mod_console: Added module:reload
Waqas Hussain <waqas20@gmail.com>
parents:
669
diff
changeset
|
166 return false, err or "Unknown error reloading module"; |
|
56410c0cd846
mod_console: Added module:reload
Waqas Hussain <waqas20@gmail.com>
parents:
669
diff
changeset
|
167 end |
|
56410c0cd846
mod_console: Added module:reload
Waqas Hussain <waqas20@gmail.com>
parents:
669
diff
changeset
|
168 return true, "Module reloaded"; |
|
56410c0cd846
mod_console: Added module:reload
Waqas Hussain <waqas20@gmail.com>
parents:
669
diff
changeset
|
169 end |
|
56410c0cd846
mod_console: Added module:reload
Waqas Hussain <waqas20@gmail.com>
parents:
669
diff
changeset
|
170 |
| 736 | 171 def_env.config = {}; |
| 172 function def_env.config:load(filename, format) | |
| 173 local config_load = require "core.configmanager".load; | |
| 174 local ok, err = config_load(filename, format); | |
| 175 if not ok then | |
| 176 return false, err or "Unknown error loading config"; | |
| 177 end | |
| 178 return true, "Config loaded"; | |
| 179 end | |
| 180 | |
| 181 function def_env.config:get(host, section, key) | |
| 182 local config_get = require "core.configmanager".get | |
| 183 return true, tostring(config_get(host, section, key)); | |
| 184 end | |
| 185 | |
| 186 def_env.hosts = {}; | |
| 187 function def_env.hosts:list() | |
| 188 for host, host_session in pairs(hosts) do | |
| 189 self.session.print(host); | |
| 190 end | |
| 191 return true, "Done"; | |
| 192 end | |
| 193 | |
| 194 function def_env.hosts:add(name) | |
| 195 end | |
| 196 | |
| 197 ------------- | |
| 198 | |
| 199 function printbanner(session) | |
| 200 session.print [[ | |
| 201 ____ \ / _ | |
| 202 | _ \ _ __ ___ ___ _-_ __| |_ _ | |
| 203 | |_) | '__/ _ \/ __|/ _ \ / _` | | | | | |
| 204 | __/| | | (_) \__ \ |_| | (_| | |_| | | |
| 205 |_| |_| \___/|___/\___/ \__,_|\__, | | |
| 206 A study in simplicity |___/ | |
| 207 | |
| 208 ]] | |
| 209 session.print("Welcome to the Prosody administration console. For a list of commands, type: help"); | |
| 210 session.print("You may find more help on using this console in our online documentation at "); | |
| 211 session.print("http://prosody.im/doc/console\n"); | |
| 212 end |
