diff mod_discodot/mod_discodot.lua @ 4577:253df0798996

mod_discodot: Print a graph of service discovery
author Kim Alvefur <zash@zash.se>
date Thu, 27 May 2021 17:23:43 +0200
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/mod_discodot/mod_discodot.lua	Thu May 27 17:23:43 2021 +0200
@@ -0,0 +1,42 @@
+local cm = require("core.configmanager");
+
+local function format_host(host, conf)
+	if host == "*" then
+		return "Global"
+	end
+	local component_module = conf["component_module"];
+	if type(component_module) == "string" then
+		if component_module == "component" then
+			return string.format("Component %q", host)
+		else
+			return string.format("Component %q %q", host, component_module)
+		end
+	else
+		return string.format("VirtualHost %q", host)
+	end
+end
+
+function module.command(arg)
+
+	local config = cm.getconfig();
+
+	print("digraph \"prosody\" {")
+	for host, conf in pairs(config) do
+		print(string.format("%q [label=%q]", host, format_host(host, conf)));
+		local parent = host:match("%.(.*)");
+		if parent and rawget(config, parent) then
+			print(string.format("%q -> %q", parent, host));
+		end
+		local disco_items = conf["disco_items"]
+		if type(disco_items) == "table" then
+			for _, pair in ipairs(disco_items) do
+				print(string.format("%q -> %q", host, pair[1]));
+			end
+		end
+
+	end
+
+	print("}")
+
+	return 0
+end