-- (C) 2011, Marco Cirillo (LW.Org) -- Display server stats in readable XML format module:set_global() local ports = module:get_option_array("xml_status_http_ports", {{ port = 5280 }}) local show_hosts = module:get_option_array("xml_status_show_hosts", nil) local show_comps = module:get_option_array("xml_status_show_comps", nil) local httpserver = require "net.httpserver" -- code begin if not prosody.stanza_counter and not show_hosts and not show_comps then module:log ("error", "mod_xml_status requires at least one of the following things:") module:log ("error", "mod_stanza_counter loaded, or either xml_status_show_hosts or xml_status_show_comps configuration values set.") module:log ("error", "check the module wiki at: http://code.google.com/p/prosody-modules/wiki/mod_xml_status") return false end local r_err = "\n\n\nProsody's XML Status - Error %s\n\n\n\n\n

%s

\n\n\n\n" local response_table = {} response_table.header = '' response_table.doc_header = '' response_table.doc_closure = '' response_table.stanzas = { elem_header = ' ', elem_closure = ' ', incoming = ' ', outgoing = ' ' } response_table.hosts = { elem_header = ' ', elem_closure = ' ', status = ' ' } response_table.comps = { elem_header = ' ', elem_closure = ' ', status = ' ' } local function forge_response() local hosts_s = {}; local components = {}; local stats = {}; local hosts_stats = {}; local comps_stats = {} local function t_builder(t,r) for _, bstring in ipairs(t) do r[#r+1] = bstring end end if show_hosts then t_builder(show_hosts, hosts_s) end if show_comps then t_builder(show_comps, components) end -- build stanza stats if there if prosody.stanza_counter then stats[1] = response_table.stanzas.elem_header stats[2] = response_table.stanzas.incoming:format(prosody.stanza_counter.iq["incoming"], prosody.stanza_counter.message["incoming"], prosody.stanza_counter.presence["incoming"]) stats[3] = response_table.stanzas.outgoing:format(prosody.stanza_counter.iq["outgoing"], prosody.stanza_counter.message["outgoing"], prosody.stanza_counter.presence["outgoing"]) stats[4] = response_table.stanzas.elem_closure end -- build hosts stats if there if hosts_s[1] then hosts_stats[1] = response_table.hosts.elem_header for _, name in ipairs(hosts_s) do hosts_stats[#hosts_stats+1] = response_table.hosts.status:format( name, hosts[name] and "online" or "offline") end hosts_stats[#hosts_stats+1] = response_table.hosts.elem_closure end -- build components stats if there if components[1] then comps_stats[1] = response_table.comps.elem_header for _, name in ipairs(components) do comps_stats[#comps_stats+1] = response_table.comps.status:format( name, hosts[name].modules.component and hosts[name].modules.component.connected and "online" or hosts[name] and hosts[name].modules.component == nil and "online" or "offline") end comps_stats[#comps_stats+1] = response_table.comps.elem_closure end -- build xml document local result = {} result[#result+1] = response_table.header; result[#result+1] = response_table.doc_header -- start t_builder(stats, result); t_builder(hosts_stats, result); t_builder(comps_stats, result) result[#result+1] = response_table.doc_closure -- end return table.concat(result, "\n") end -- http handlers local function response(code, r, h) local response = { status = code; body = r; } if h then response.headers = h; end return response end local function request(method, body, request) if method == "GET" then return response(200, forge_response()) else local err405 = r_err:format("405", "Only GET is supported") return response(405, err405, {["Allow"] = "GET"}) end end -- initialization. -- init http interface local function setup() httpserver.new_from_config(ports, request, { base = "xml-status" }) end if prosody.start_time then setup(); else module:hook("server-started", setup); end