#!/usr/bin/env lua

package.path = package.path:gsub("([^;]*)(?[^;]*)", "%1prosody/%2;%1%2");
package.cpath = package.cpath:gsub("([^;]*)(?[^;]*)", "%1prosody/%2;%1%2");

local t_insert = table.insert;
local t_sort = table.sort;

local jid = require "util.jid";
local st = require "util.stanza";

local xs = require "util.xmppstream";

local function skeleton(s)
	local o = st.stanza(s.name, { xmlns = s.attr.xmlns });

	local children = {};
	for _, child in ipairs(s.tags) do t_insert(children, skeleton(child)) end
	t_sort(children, function(a, b)
		if a.attr.xmlns == b.attr.xmlns then return a.name < b.name; end
		return (a.attr.xmlns or "") < (b.attr.xmlns or "");
	end);
	for _, child in ipairs(children) do o:add_direct_child(child); end
	return o;
end

local function classify_jid(s)
	if not s then return "" end
	local u, h, r = jid.split(s);
	if r then
		return "full"
	elseif u then
		return "bare"
	elseif h then
		return "host"
	else
		return "invalid"
	end
end

local stream_session = { notopen = true };
local stream_callbacks = { stream_ns = "jabber:client"; default_ns = "jabber:client" };
function stream_callbacks:handlestanza(item)
	local clean = skeleton(item);

	-- Normalize top level attributes
	clean.attr.type = item.attr.type;
	if clean.attr.type == nil and clean.name == "message" then clean.attr.type = "normal"; end
	clean.attr.id = string.rep("x", math.floor(math.log(1 + #(item.attr.id or ""), 2)));
	clean.attr.from = classify_jid(item.attr.from);
	clean.attr.to = classify_jid(item.attr.to);
	print(clean);
end
local stream = xs.new(stream_session, stream_callbacks);
assert(stream:feed(st.stanza("stream", { xmlns = "jabber:client" }):top_tag()));
stream_session.notopen = nil;

local data = io.read(4096);
while data do
	stream:feed(data);
	data = io.read(4096);
end

assert(stream:feed("</stream>"));