/*
 * Kazoo module interface
 *
 * Copyright (C) 2010-2014 2600Hz
 *
 * This file is part of Kamailio, a free SIP server.
 *
 * Kamailio is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version
 *
 * Kamailio is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 */

#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include "../../core/mem/mem.h"
#include "../../core/timer_proc.h"
#include "../../core/sr_module.h"
#include "../presence/bind_presence.h"
#include "../../core/pvar.h"

#include "defs.h"
#include "const.h"
#include "kz_json.h"

#include "kz_pua.h"

extern int dbk_include_entity;
extern int dbk_pua_mode;

extern db1_con_t *kz_pa_db;
extern db_func_t kz_pa_dbf;
extern str kz_presentity_table;

extern db_locking_t kz_pua_lock_type;

int kz_pua_update_presentity(str *event, str *realm, str *user, str *etag,
		str *sender, str *body, int expires, int reset)
{
	db_key_t query_cols[13];
	db_op_t query_ops[13];
	db_val_t query_vals[13];
	int n_query_cols = 0;
	int ret = -1;
	int use_replace = 1;

	query_cols[n_query_cols] = &str_event_col;
	query_ops[n_query_cols] = OP_EQ;
	query_vals[n_query_cols].type = DB1_STR;
	query_vals[n_query_cols].nul = 0;
	query_vals[n_query_cols].val.str_val = *event;
	n_query_cols++;

	query_cols[n_query_cols] = &str_domain_col;
	query_ops[n_query_cols] = OP_EQ;
	query_vals[n_query_cols].type = DB1_STR;
	query_vals[n_query_cols].nul = 0;
	query_vals[n_query_cols].val.str_val = *realm;
	n_query_cols++;

	query_cols[n_query_cols] = &str_username_col;
	query_ops[n_query_cols] = OP_EQ;
	query_vals[n_query_cols].type = DB1_STR;
	query_vals[n_query_cols].nul = 0;
	query_vals[n_query_cols].val.str_val = *user;
	n_query_cols++;

	query_cols[n_query_cols] = &str_etag_col;
	query_ops[n_query_cols] = OP_EQ;
	query_vals[n_query_cols].type = DB1_STR;
	query_vals[n_query_cols].nul = 0;
	query_vals[n_query_cols].val.str_val = *etag;
	n_query_cols++;

	query_cols[n_query_cols] = &str_sender_col;
	query_vals[n_query_cols].type = DB1_STR;
	query_vals[n_query_cols].nul = 0;
	query_vals[n_query_cols].val.str_val = *sender;
	n_query_cols++;

	query_cols[n_query_cols] = &str_body_col;
	query_vals[n_query_cols].type = DB1_BLOB;
	query_vals[n_query_cols].nul = 0;
	query_vals[n_query_cols].val.str_val = *body;
	n_query_cols++;

	query_cols[n_query_cols] = &str_received_time_col;
	query_vals[n_query_cols].type = DB1_INT;
	query_vals[n_query_cols].nul = 0;
	query_vals[n_query_cols].val.int_val = (int)time(NULL);
	n_query_cols++;

	query_cols[n_query_cols] = &str_expires_col;
	query_vals[n_query_cols].type = DB1_INT;
	query_vals[n_query_cols].nul = 0;
	query_vals[n_query_cols].val.int_val = expires;
	n_query_cols++;

	query_cols[n_query_cols] = &str_priority_col;
	query_vals[n_query_cols].type = DB1_INT;
	query_vals[n_query_cols].nul = 0;
	query_vals[n_query_cols].val.int_val = 0;
	n_query_cols++;

	if(kz_pa_dbf.use_table(kz_pa_db, &kz_presentity_table) < 0) {
		LM_ERR("unsuccessful use_table\n");
		goto error;
	}

	if(kz_pa_dbf.replace == NULL || reset > 0) {
		use_replace = 0;
		LM_DBG("using delete/insert instead of replace\n");
	}

	if(kz_pa_dbf.start_transaction) {
		if(kz_pa_dbf.start_transaction(kz_pa_db, kz_pua_lock_type) < 0) {
			LM_ERR("in start_transaction\n");
			goto error;
		}
	}

	if(use_replace) {
		if(kz_pa_dbf.replace(
				   kz_pa_db, query_cols, query_vals, n_query_cols, 4, 0)
				< 0) {
			LM_ERR("replacing record in database\n");
			if(kz_pa_dbf.abort_transaction) {
				if(kz_pa_dbf.abort_transaction(kz_pa_db) < 0)
					LM_ERR("in abort_transaction\n");
			}
			goto error;
		}
	} else {
		if(kz_pa_dbf.delete(
				   kz_pa_db, query_cols, query_ops, query_vals, 4 - reset)
				< 0) {
			LM_ERR("deleting record in database\n");
			if(kz_pa_dbf.abort_transaction) {
				if(kz_pa_dbf.abort_transaction(kz_pa_db) < 0)
					LM_ERR("in abort_transaction\n");
			}
			goto error;
		}
		if(kz_pa_dbf.insert(kz_pa_db, query_cols, query_vals, n_query_cols)
				< 0) {
			LM_ERR("replacing record in database\n");
			if(kz_pa_dbf.abort_transaction) {
				if(kz_pa_dbf.abort_transaction(kz_pa_db) < 0)
					LM_ERR("in abort_transaction\n");
			}
			goto error;
		}
	}

	if(kz_pa_dbf.end_transaction) {
		if(kz_pa_dbf.end_transaction(kz_pa_db) < 0) {
			LM_ERR("in end_transaction\n");
			goto error;
		}
	}

error:

	return ret;
}


int kz_pua_publish_presence_to_presentity(struct json_object *json_obj)
{
	int ret = 1;
	str from = {0, 0}, to = {0, 0};
	str from_user = {0, 0}, to_user = {0, 0};
	str from_realm = {0, 0}, to_realm = {0, 0};
	str callid = {0, 0}, fromtag = {0, 0}, totag = {0, 0};
	str state = {0, 0};
	str direction = {0, 0};
	str event = str_init("presence");
	str presence_body = {0, 0};
	str activity = str_init("");
	str note = str_init("Available");
	str status = str_presence_status_online;
	int expires = 0;
	str sender = {0, 0}, etag = {0, 0};

	char *body = (char *)pkg_malloc(PRESENCE_BODY_BUFFER_SIZE);
	if(body == NULL) {
		LM_ERR("Error allocating buffer for publish\n");
		ret = -1;
		goto error;
	}

	json_extract_field(BLF_JSON_FROM, from);
	json_extract_field(BLF_JSON_FROM_USER, from_user);
	json_extract_field(BLF_JSON_FROM_REALM, from_realm);
	json_extract_field(BLF_JSON_TO, to);
	json_extract_field(BLF_JSON_TO_USER, to_user);
	json_extract_field(BLF_JSON_TO_REALM, to_realm);
	json_extract_field(BLF_JSON_CALLID, callid);
	json_extract_field(BLF_JSON_FROMTAG, fromtag);
	json_extract_field(BLF_JSON_TOTAG, totag);
	json_extract_field(BLF_JSON_DIRECTION, direction);
	json_extract_field(BLF_JSON_STATE, state);

	json_extract_field(BLF_JSON_ETAG, etag);
	json_extract_field(BLF_JSON_SENDER, sender);

	if(sender.len == 0) {
		json_extract_field(BLF_JSON_SWITCH_URI, sender);
	}

	struct json_object *ExpiresObj =
			kz_json_get_object(json_obj, BLF_JSON_EXPIRES);
	if(ExpiresObj != NULL) {
		expires = json_object_get_int(ExpiresObj);
		if(expires > 0)
			expires += (int)time(NULL);
	}

	if(!from_user.len || !to_user.len || !state.len) {
		LM_ERR("missing one of From / To / State\n");
		goto error;
	}

	if(!strcmp(state.s, "early")) {
		note = str_presence_note_busy;
		activity = str_presence_act_busy;

	} else if(!strcmp(state.s, "confirmed")) {
		note = str_presence_note_otp;
		activity = str_presence_act_otp;

	} else if(!strcmp(state.s, "offline")) {
		note = str_presence_note_offline;
		status = str_presence_status_offline;
	};

	sprintf(body, PRESENCE_BODY, from_user.s, callid.s, status.s, note.s,
			activity.s, note.s);

	presence_body.s = body;
	presence_body.len = strlen(body);

	if(sender.len == 0) {
		sender = from;
	}

	if(etag.len == 0) {
		etag = callid;
	}

	kz_pua_update_presentity(&event, &from_realm, &from_user, &etag, &sender,
			&presence_body, expires, 1);

error:

	if(body)
		pkg_free(body);

	return ret;
}

int kz_pua_publish_mwi_to_presentity(struct json_object *json_obj)
{
	int ret = 1;
	str event = str_init("message-summary");
	str from = {0, 0}, to = {0, 0};
	str from_user = {0, 0}, to_user = {0, 0};
	str from_realm = {0, 0}, to_realm = {0, 0};
	str callid = {0, 0}, fromtag = {0, 0}, totag = {0, 0};
	str mwi_user = {0, 0}, mwi_waiting = {0, 0}, mwi_new = {0, 0},
		mwi_saved = {0, 0}, mwi_urgent = {0, 0}, mwi_urgent_saved = {0, 0},
		mwi_account = {0, 0}, mwi_body = {0, 0};
	int expires = 0;
	str sender = {0, 0}, etag = {0, 0};

	char *body = (char *)pkg_malloc(MWI_BODY_BUFFER_SIZE);
	if(body == NULL) {
		LM_ERR("Error allocating buffer for publish\n");
		ret = -1;
		goto error;
	}

	json_extract_field(BLF_JSON_FROM, from);
	json_extract_field(BLF_JSON_FROM_USER, from_user);
	json_extract_field(BLF_JSON_FROM_REALM, from_realm);
	json_extract_field(BLF_JSON_TO, to);
	json_extract_field(BLF_JSON_TO_USER, to_user);
	json_extract_field(BLF_JSON_TO_REALM, to_realm);
	json_extract_field(BLF_JSON_CALLID, callid);
	json_extract_field(BLF_JSON_FROMTAG, fromtag);
	json_extract_field(BLF_JSON_TOTAG, totag);

	json_extract_field(MWI_JSON_TO, mwi_user);
	json_extract_field(MWI_JSON_WAITING, mwi_waiting);
	json_extract_field(MWI_JSON_NEW, mwi_new);
	json_extract_field(MWI_JSON_SAVED, mwi_saved);
	json_extract_field(MWI_JSON_URGENT, mwi_urgent);
	json_extract_field(MWI_JSON_URGENT_SAVED, mwi_urgent_saved);
	json_extract_field(MWI_JSON_ACCOUNT, mwi_account);

	json_extract_field(BLF_JSON_ETAG, etag);
	json_extract_field(BLF_JSON_SENDER, sender);

	struct json_object *ExpiresObj =
			kz_json_get_object(json_obj, BLF_JSON_EXPIRES);
	if(ExpiresObj != NULL) {
		expires = json_object_get_int(ExpiresObj);
		if(expires > 0)
			expires += (int)time(NULL);
	}

	sprintf(body, MWI_BODY, mwi_waiting.len, mwi_waiting.s, mwi_account.len,
			mwi_account.s, mwi_new.len, mwi_new.s, mwi_saved.len, mwi_saved.s,
			mwi_urgent.len, mwi_urgent.s, mwi_urgent_saved.len,
			mwi_urgent_saved.s);

	mwi_body.s = body;
	mwi_body.len = strlen(body);

	if(sender.len == 0) {
		sender = from;
	}

	if(etag.len == 0) {
		etag = callid;
	}

	kz_pua_update_presentity(&event, &from_realm, &from_user, &etag, &from,
			&mwi_body, expires, 1);

error:

	if(body)
		pkg_free(body);

	return ret;
}

int kz_pua_publish_dialoginfo_to_presentity(struct json_object *json_obj)
{
	int ret = 1;
	str from = {0, 0}, to = {0, 0}, pres = {0, 0};
	str from_user = {0, 0}, to_user = {0, 0}, pres_user = {0, 0};
	str from_realm = {0, 0}, to_realm = {0, 0}, pres_realm = {0, 0};
	str from_uri = {0, 0}, to_uri = {0, 0};
	str callid = {0, 0}, dialogid = {0, 0};
	str fromtag = {0, 0}, totag = {0, 0};
	str state = {0, 0};
	str direction = {0, 0};
	str dialoginfo_body = {0, 0};
	int expires = 0;
	str event = str_init("dialog");
	int reset = 0;
	char to_tag_buffer[100];
	char from_tag_buffer[100];
	char sender_buf[1024];
	str sender = {0, 0}, etag = {0, 0};

	char *body = (char *)pkg_malloc(DIALOGINFO_BODY_BUFFER_SIZE);
	if(body == NULL) {
		LM_ERR("Error allocating buffer for publish\n");
		ret = -1;
		goto error;
	}

	json_extract_field(BLF_JSON_PRES, pres);
	json_extract_field(BLF_JSON_PRES_USER, pres_user);
	json_extract_field(BLF_JSON_PRES_REALM, pres_realm);
	json_extract_field(BLF_JSON_FROM, from);
	json_extract_field(BLF_JSON_FROM_USER, from_user);
	json_extract_field(BLF_JSON_FROM_REALM, from_realm);
	json_extract_field(BLF_JSON_FROM_URI, from_uri);
	json_extract_field(BLF_JSON_TO, to);
	json_extract_field(BLF_JSON_TO_USER, to_user);
	json_extract_field(BLF_JSON_TO_REALM, to_realm);
	json_extract_field(BLF_JSON_TO_URI, to_uri);
	json_extract_field(BLF_JSON_CALLID, callid);
	json_extract_field(BLF_JSON_DIALOGID, dialogid);
	json_extract_field(BLF_JSON_FROMTAG, fromtag);
	json_extract_field(BLF_JSON_TOTAG, totag);
	json_extract_field(BLF_JSON_DIRECTION, direction);
	json_extract_field(BLF_JSON_STATE, state);

	json_extract_field(BLF_JSON_ETAG, etag);
	json_extract_field(BLF_JSON_SENDER, sender);

	if(sender.len == 0) {
		json_extract_field(BLF_JSON_SWITCH_URI, sender);
	}

	struct json_object *ExpiresObj =
			kz_json_get_object(json_obj, BLF_JSON_EXPIRES);
	if(ExpiresObj != NULL) {
		expires = json_object_get_int(ExpiresObj);
		if(expires > 0)
			expires += (int)time(NULL);
	}

	ExpiresObj = kz_json_get_object(json_obj, "Flush-Level");
	if(ExpiresObj != NULL) {
		reset = json_object_get_int(ExpiresObj);
	}

	if(!from.len || !to.len || !state.len) {
		LM_ERR("missing one of From / To / State\n");
		goto error;
	}

	if(!pres.len || !pres_user.len || !pres_realm.len) {
		pres = from;
		pres_user = from_user;
		pres_realm = from_realm;
	}

	if(!from_uri.len)
		from_uri = from;

	if(!to_uri.len)
		to_uri = to;

	if(fromtag.len > 0) {
		fromtag.len =
				sprintf(from_tag_buffer, LOCAL_TAG, fromtag.len, fromtag.s);
		fromtag.s = from_tag_buffer;
	}

	if(totag.len > 0) {
		totag.len = sprintf(to_tag_buffer, REMOTE_TAG, totag.len, totag.s);
		totag.s = to_tag_buffer;
	}

	if(dialogid.len == 0) {
		dialogid = callid;
	}

	if(callid.len) {

		if(dbk_include_entity) {
			sprintf(body, DIALOGINFO_BODY, pres.len, pres.s, dialogid.len,
					dialogid.s, callid.len, callid.s, fromtag.len, fromtag.s,
					totag.len, totag.s, direction.len, direction.s, state.len,
					state.s, from_user.len, from_user.s, from.len, from.s,
					from_uri.len, from_uri.s, to_user.len, to_user.s, to.len,
					to.s, to_uri.len, to_uri.s);
		} else {

			sprintf(body, DIALOGINFO_BODY_2, pres.len, pres.s, dialogid.len,
					dialogid.s, callid.len, callid.s, fromtag.len, fromtag.s,
					totag.len, totag.s, direction.len, direction.s, state.len,
					state.s, from_user.len, from_user.s, from.len, from.s,
					to_user.len, to_user.s, to.len, to.s);
		}

	} else {
		sprintf(body, DIALOGINFO_EMPTY_BODY, pres.len, pres.s);
	}

	if(sender.len == 0) {
		sprintf(sender_buf, "sip:%s", callid.s);
		sender.s = sender_buf;
		sender.len = strlen(sender_buf);
	}

	if(etag.len == 0) {
		etag = callid;
	}

	dialoginfo_body.s = body;
	dialoginfo_body.len = strlen(body);

	kz_pua_update_presentity(&event, &pres_realm, &pres_user, &etag, &sender,
			&dialoginfo_body, expires, reset);

error:

	if(body)
		pkg_free(body);

	return ret;
}

int kz_pua_publish(struct sip_msg *msg, char *json)
{
	str event_name = {0, 0}, event_package = {0, 0};
	struct json_object *json_obj = NULL;
	int ret = 1;

	if(dbk_pua_mode != 1) {
		LM_ERR("pua_mode must be 1 to publish\n");
		ret = -1;
		goto error;
	}

	/* extract info from json and construct xml */
	json_obj = kz_json_parse(json);
	if(json_obj == NULL) {
		ret = -1;
		goto error;
	}

	json_extract_field(BLF_JSON_EVENT_NAME, event_name);

	if(event_name.len == 6 && strncmp(event_name.s, "update", 6) == 0) {
		json_extract_field(BLF_JSON_EVENT_PKG, event_package);
		if(event_package.len == str_event_dialog.len
				&& strncmp(event_package.s, str_event_dialog.s,
						   event_package.len)
						   == 0) {
			ret = kz_pua_publish_dialoginfo_to_presentity(json_obj);
		} else if(event_package.len == str_event_message_summary.len
				  && strncmp(event_package.s, str_event_message_summary.s,
							 event_package.len)
							 == 0) {
			ret = kz_pua_publish_mwi_to_presentity(json_obj);
		} else if(event_package.len == str_event_presence.len
				  && strncmp(event_package.s, str_event_presence.s,
							 event_package.len)
							 == 0) {
			ret = kz_pua_publish_presence_to_presentity(json_obj);
		}
	}

error:
	if(json_obj)
		json_object_put(json_obj);

	return ret;
}

int kz_pua_publish_mwi(struct sip_msg *msg, char *json)
{
	struct json_object *json_obj = NULL;
	int ret = 1;

	if(dbk_pua_mode != 1) {
		LM_ERR("pua_mode must be 1 to publish\n");
		ret = -1;
		goto error;
	}

	/* extract info from json and construct xml */
	json_obj = kz_json_parse(json);
	if(json_obj == NULL) {
		ret = -1;
		goto error;
	}

	ret = kz_pua_publish_mwi_to_presentity(json_obj);

error:
	if(json_obj)
		json_object_put(json_obj);

	return ret;
}

int kz_pua_publish_presence(struct sip_msg *msg, char *json)
{
	struct json_object *json_obj = NULL;
	int ret = 1;

	if(dbk_pua_mode != 1) {
		LM_ERR("pua_mode must be 1 to publish\n");
		ret = -1;
		goto error;
	}

	/* extract info from json and construct xml */
	json_obj = kz_json_parse(json);
	if(json_obj == NULL) {
		ret = -1;
		goto error;
	}

	ret = kz_pua_publish_presence_to_presentity(json_obj);

error:
	if(json_obj)
		json_object_put(json_obj);

	return ret;
}

int kz_pua_publish_dialoginfo(struct sip_msg *msg, char *json)
{
	struct json_object *json_obj = NULL;
	int ret = 1;

	if(dbk_pua_mode != 1) {
		LM_ERR("pua_mode must be 1 to publish\n");
		ret = -1;
		goto error;
	}

	/* extract info from json and construct xml */
	json_obj = kz_json_parse(json);
	if(json_obj == NULL) {
		ret = -1;
		goto error;
	}

	ret = kz_pua_publish_dialoginfo_to_presentity(json_obj);

error:
	if(json_obj)
		json_object_put(json_obj);

	return ret;
}
