prosody/plugins/mod_sipwise_pushd.lua

--
-- Copyright (C) 2014 Sipwise GmbH <development@sipwise.com>
--
-- This project is MIT/X11 licensed. Please see the
-- COPYING file in the source package for more information.
--

module:depends("sipwise_vcard_cusax");

local datamanager = require "util.datamanager";
local mod_sql = module:require("sql");
local format = string.format;
local jid_split = require "util.jid".split;
local jid_bare = require "util.jid".bare;
local hosts = prosody.hosts;

local pushd_config = {
	url = "https://127.0.0.1:8080/push",
	gcm = true,
	apns = true,
	call_sound = 'incoming_call.caf',
	msg_sound  = 'incoming_message.caf'
};
local sql_config = {
	driver = "MySQL",
	database = "provisioning",
	username = "prosody",
	password = "PW_PROSODY",
	host = "localhost"
};

local push_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 vs ON vs.id = vup.subscriber_id
  LEFT JOIN provisioning.voip_domains vd ON vd.id = vs.domain_id
WHERE vp.attribute = 'mobile_push_enable'
  AND vs.username = ?
  AND vd.domain = ?;
]];

local push_dom_query = [[
SELECT vp.attribute, vup.value FROM provisioning.voip_preferences vp
  LEFT JOIN provisioning.voip_dom_preferences vup ON vup.attribute_id = vp.id
  LEFT JOIN provisioning.voip_domains vd ON vd.id = vup.domain_id
WHERE vp.attribute = 'mobile_push_enable'
  AND vd.domain = ?;
]];
local engine;

local function process_response(response, code, request)
	if code >= 200 and code < 299 then
		module:log("debug", "pushd response OK[%s] %s",
			tostring(code), tostring(response));
	else
		module:log("error", "pushd response KO[%s] %s",
			tostring(code), tostring(response));
	end
end

local function push_enable(username, domain)
	local row
	-- Reconnect to DB if necessary
	if not engine.conn:ping() then
		engine.conn = nil;
		engine:connect();
	end
	for row in engine:select(push_dom_query, domain) do
		if row[2] == "1" then
			module:log("debug", "domain mobile_push_enable pref set");
			return true;
		end
	end
	for row in engine:select(push_usr_query, username, domain) do
		if row[2] == "1" then
			module:log("debug", "usr mobile_push_enable pref set");
			return true;
		end
	end
	return false;
end

local function get_caller_info(jid)
	local node, host = jid_split(jid);
	local vcard = module:shared(format("/%s/sipwise_vcard_cusax/vcard", host));
	if vcard then
		local info = vcard.get_subscriber_info(node, host);
		return info;
	end
end

local function get_callee_badge(jid)
	local node, host = jid_split(jid);
	local result = datamanager.list_load(node, host, "offline");
	if result then
		module:log("debug", "%d offline messages for %s", #result, jid);
		return #result
	end
	module:log("debug", "0 offline messages for %s", jid);
	return 0
end

local function handle_offline(event)
	module:log("debug", "handle_offline");
	local origin, stanza = event.origin, event.stanza;
	local to = stanza.attr.to;
	local from = stanza.attr.from;
	local node, host;
	local caller = { username = 'unknow', host = 'unknown.local' };
	local http_options = {
		method = "POST",
		body = "",
	}
	local message;
	local function get_message()
		local body = stanza:get_child('body');
		if body then
			return body:get_text();
		end
	end
	local function build_push_common_query(caller_jid, type, message)
		local muc = stanza:get_child('x', 'jabber:x:conference');
		local query_muc = '';
		if muc then
			local muc_jid = muc.attr.jid;
			local muc_name, muc_domain = jid_split(muc_jid);
			local room = hosts[muc_domain].modules.muc.rooms[muc_jid];
			query_muc = format("data_room_jid=%s&data_room_description=%s",
				muc_jid, room:get_description() or 'Prosody chatroom');
		end
		local query = format("callee=%s&domain=%s", node, host);
		query = query .. '&' .. format("data_sender_jid=%s", caller_jid);
		local caller_info = get_caller_info(caller_jid) or {display_name = ''};
		query = query .. '&' .. format(
			"data_sender_name=%s&data_type=%s&data_message=%s",
			caller_info.display_name , type, message);

		if muc then
			return query .. '&' .. query_muc;
		end
		return query;
	end
	local function build_push_apns_query(type, message)
		local badge = get_callee_badge(to);
		local query_apns = format(
			"apns_sound=%s&apns_badge=%s&apns_alert=%s",
			pushd_config.msg_sound or '', badge, message);
		return http_options.body..'&'..query_apns;
	end
	local function build_push_query(message)
		local type = 'message';
		local invite = stanza:get_child('x',
			'http://jabber.org/protocol/muc#user');
		local caller_jid = format("%s@%s", origin.username or caller.username,
			origin.host or caller.host);
		if invite then
			type = 'invite';
			caller_jid = jid_bare(invite:get_child('invite').attr.from) or
				caller_jid;
		end

		http_options.body = build_push_common_query(caller_jid, type, message);
		if pushd_config.apns then
			http_options.body = build_push_apns_query(type, message);
		end
	end

	module:log("debug", "stanza[%s]", tostring(stanza));

	if from then
		caller.username, caller.host = jid_split(from);
	end

	if to then
		node, host = jid_split(to);
		message = get_message();
		if message then
			module:log("debug", "message OK");
			if push_enable(node, host) then
				build_push_query(message);
				if http_options.body then
					module:log("debug", "Sending http pushd request: %s data: %s",
						pushd_config.url, http_options.body);
					http.request(pushd_config.url, http_options, process_response);
				end
			else
				module:log("debug", "no mobile_push_enable pref set for %s", to);
			end
		else
			module:log("debug", "no message body");
		end
	end
end

module:hook("message/offline/handle", handle_offline, 20);

function module.load()
	pushd_config = module:get_option("pushd_config", pushd_config);
	sql_config   = module:get_option("auth_sql", sql_config);
	engine = mod_sql:create_engine(sql_config);
	engine:execute("SET NAMES 'utf8' COLLATE 'utf8_bin';");
	module:log("info", "load OK");
end