-- (C) 2011, Marco Cirillo (LW.Org)
-- Exposes stats on HTTP for the stanza counter module.
module:set_global()
local ports = module:get_option_array("stanza_counter_http_ports", {{ port = 5280 }})
local httpserver = require "net.httpserver"
-- http handlers
local r_200 = "\n\n\nProsody's Stanza Counter\n\n\n\n\nIncoming and Outgoing stanzas divided per type
\nIncoming IQs: %d
\nOutgoing IQs: %d
\nIncoming Messages: %d
\nOutgoing Messages: %d
\nIncoming Presences: %d
\nOutgoing Presences: %d
\n\n\n\n"
local r_err = "\n\n
\nProsody's Stanza Counter - Error %s\n\n\n\n\n%s
\n\n\n\n"
local function res(code, r, h)
local response = {
status = code,
body = r
}
if h then response.headers = h end
return response
end
local function req(method, body, request)
if not prosody.stanza_counter then
local err500 = r_err:format("500", "Stats not found, is the counter module loaded?")
return res(500, err500) end
if method == "GET" then
local forge_res = r_200:format(prosody.stanza_counter.iq["incoming"],
prosody.stanza_counter.iq["outgoing"],
prosody.stanza_counter.message["incoming"],
prosody.stanza_counter.message["outgoing"],
prosody.stanza_counter.presence["incoming"],
prosody.stanza_counter.presence["outgoing"])
return res(200, forge_res)
else
local err405 = r_err:format("405", "Only GET is supported")
return res(405, err405, {["Allow"] = "GET"})
end
end
-- initialization.
-- init http and cleanup interface
function cleanup() -- recycled from mod_register_json, it's handy
module:log("debug", "Cleaning up handlers and stuff as module is being unloaded.")
for _, options in ipairs(ports) do
if options.port then
httpserver.new.http_servers[options.port].handlers[options.path or "stanza-counter"] = nil
end
end
-- if there are no handlers left clean and close the socket, doesn't work with server_event
local event = module:get_option_boolen("use_libevent", false)
if not event then
for _, options in ipairs(ports) do
if options.port and not next(httpserver.new.http_servers[options.port].handlers) then
httpserver.new.http_servers[options.port] = nil
if options.interface then
for _, value in ipairs(options.interface) do
if server.getserver(value, options.port) then server.removeserver(value, options.port) end
end
else if server.getserver("*", options.port) then server.removeserver("*", options.port) end end
end
end
end
prosody.events.remove_handler("module-unloaded", cleanup)
end
local function setup()
httpserver.new_from_config(ports, req, { base = "stanza-counter" })
prosody.events.add_handler("module-unloaded", cleanup)
end
-- set it
if prosody.start_time then setup() else module:hook("server-started", setup) end