-- Prosody IM
-- Copyright (C) 2008-2010 Matthew Wild
-- Copyright (C) 2008-2010 Waqas Hussain
-- Copyright (C) 2013-2015 Sipwise GmbH
--
-- This is a stripped down version of mod_vcard for returning
-- simply a vcard containing some info of
-- the requested user.
-- http://xmpp.org/extensions/xep-0054.html
--
-- This project is MIT/X11 licensed. Please see the
-- COPYING file in the source package for more information.
--

local st = require "util.stanza";
local jid_split = require "util.jid".split;
local vcard = module:shared("vcard");
local hosts = prosody.hosts;

local email_query = [[
SELECT bc.email, bccc.email FROM billing.voip_subscribers AS vs
  LEFT join billing.contacts AS bc ON vs.contact_id = bc.id
  LEFT join billing.contracts AS bcc ON vs.contract_id = bcc.id
  LEFT join billing.contacts AS bccc ON bcc.contact_id = bccc.id
  LEFT join billing.domains AS bd ON vs.domain_id = bd.id
WHERE vs.username = ? AND bd.domain = ?;
]]

local display_usr_query = [[
SELECT vp.attribute, vup.value FROM provisioning.voip_preferences vp
  LEFT JOIN provisioning.voip_usr_preferences vup ON vup.attribute_id = vp.id
  LEFT JOIN provisioning.voip_subscribers AS ps ON ps.id = vup.subscriber_id
  LEFT JOIN provisioning.voip_domains AS pd ON ps.domain_id = pd.id
WHERE vp.attribute = 'display_name'
  AND ps.username = ? AND pd.domain = ?
]];

local aliases_query = [[
SELECT pa.username, pa.is_primary FROM provisioning.voip_dbaliases AS pa
  LEFT JOIN provisioning.voip_subscribers AS ps ON ps.id = pa.subscriber_id
  LEFT JOIN provisioning.voip_domains AS pd ON ps.domain_id = pd.id
WHERE ps.username = ? AND pd.domain = ? ORDER BY pa.is_primary DESC;
]];

local um_user_exists = require "core.usermanager".user_exists;
local sql = require "util.sql";
local default_params = module:get_option("sql");
local engine;

module:add_feature("vcard-temp");

local function reconect_check()
	if not engine.conn:ping() then
		engine.conn = nil;
		engine:connect();
	end
end

function vcard.get_subscriber_info(user, host)
	if not hosts[host] or not hosts[host].users then
		return nil;
	end
	if not um_user_exists(user, host) then
		return nil;
	end
	local info = { user = user, domain = host, aliases = {} };
	reconect_check();

	for row in engine:select(display_usr_query, user, host) do
		info['display_name'] = row[2];
		module:log("debug", string.format("display_name:[%s]", row[2]));
	end

	for row in engine:select(aliases_query, user, host) do
		table.insert(info['aliases'], row[1]);
		module:log("debug", string.format("aliases:%s", row[1]));
	end

	for row in engine:select(email_query, user, host) do
		local email;
		-- NULL is not nil here, so checking length
		for i= 1, 2 do
			if row[i] and row[i]:len() > 0 then
				email = row[i];
				break;
			end
		end
		if email then
			info['email'] = email;
			module:log("debug", "email:", tostring(email));
		end
	end
	return info;
end

local function generate_vcard(info)
	local function add(t, name, value)
		local tmp = st.stanza( name,
			{ xmlns = "vcard-temp" });
		if value then tmp:text(value) end
		if t then
			t:add_child(tmp);
		else
			return tmp;
		end
	end
	local vCard = st.stanza("vCard",
		{
			xmlns = "vcard-temp",
			prodid = "-//HandGen//NONSGML vGen v1.0//EN",
			version = "2.0"
		}
	);
	local uri = info["user"] .. '@' .. info["domain"];

	add(vCard, "JABBERID", uri);
	local t = add(nil, "TEL")
	add(t, "VIDEO");
	add(t, "NUMBER",  "sip:" .. uri);
	vCard:add_child(t);
	for _,v in ipairs(info['aliases']) do
		t = add(nil, "TEL");
		add(t, "VOICE");
		add(t, "NUMBER", v);
		vCard:add_child(t);
	end
	if info['display_name'] then
		add(vCard, "FN", info['display_name']);
	end
	if info['email'] then
		t = add(nil, "EMAIL");
		add(t, "INTERNET");
		add(t, "PREF");
		add(t, "USERID", info['email']);
		vCard:add_child(t)
	end
	return vCard;
end

local function handle_vcard(event)
	module:log("debug", "handle_vcard");
	local session, stanza = event.origin, event.stanza;
	local to = stanza.attr.to;
	if stanza.attr.type == "get" then
		local user, host;
		if to then
			user, host = jid_split(to);
		else
			user = session.username;
			host = session.host;
		end
		if not (user and host) then
			module:log("error", "bad-request stanza: %s",
				tostring(stanza))
			session.send(st.error_reply(stanza, "modify", "jid-malformed"));
		end
		local info = vcard.get_subscriber_info(user, host);
		if info then
			local vCard = generate_vcard(info);
			local reply = st.reply(stanza):add_child(st.deserialize(vCard));
			session.send(reply);
		else
			module:log("error", "No info retrieve from stanza: %s",
				tostring(stanza))
			session.send(st.error_reply(stanza, "cancel", "item-not-found"));
		end
	else
		module:log("debug", "reject setting vcard");
		session.send(st.error_reply(stanza, "auth", "forbidden"));
	end
	return true;
end

module:hook("iq/bare/vcard-temp:vCard", handle_vcard);
module:hook("iq/host/vcard-temp:vCard", handle_vcard);

local function normalize_params(params)
	assert(params.driver and params.database,
		"Configuration error: Both the SQL driver and the database need to be specified");
	return params;
end

function module.load()
	if prosody.prosodyctl then return; end
	local engines = module:shared("/*/sql/connections");
	local params = normalize_params(module:get_option("auth_sql", default_params));
	engine = engines[sql.db2uri(params)];
	if not engine then
		module:log("debug", "Creating new engine");
		engine = sql:create_engine(params);
		engines[sql.db2uri(params)] = engine;
	end
	engine:connect();
	module:log("debug", "load OK");
end