/*
 * Copyright (C) 2015-2018 "IoT.bzh"
 * Author José Bollo <jose.bollo@iot.bzh>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#define _GNU_SOURCE         /* See feature_test_macros(7) */
#include <stdio.h>
#include <string.h>
#include <assert.h>

#include <json-c/json.h>

#define AFB_BINDING_VERSION 2
#include <afb/afb-binding.h>

#include "verbose.h"
#include "utils-systemd.h"
#include "afm-udb.h"
#include "afm-urun.h"
#include "wgt-info.h"
#include "wgtpkg-install.h"
#include "wgtpkg-uninstall.h"
#include "wrap-json.h"

/*
 * constant strings
 */
static const char _added_[]     = "added";
static const char _a_l_c_[]     = "application-list-changed";
static const char _bad_request_[] = "bad-request";
static const char _cannot_start_[] = "cannot-start";
static const char _detail_[]    = "detail";
static const char _id_[]        = "id";
static const char _install_[]   = "install";
static const char _lang_[]      = "lang";
static const char _not_found_[] = "not-found";
static const char _once_[]      = "once";
static const char _pause_[]     = "pause";
static const char _resume_[]    = "resume";
static const char _runid_[]     = "runid";
static const char _runnables_[] = "runnables";
static const char _runners_[]   = "runners";
static const char _start_[]     = "start";
static const char _state_[]     = "state";
static const char _terminate_[] = "terminate";
static const char _uninstall_[] = "uninstall";

/*
 * the permissions
 */
static const struct afb_auth
	auth_install = {
		.type = afb_auth_Permission,
		.text = "urn:AGL:permission:afm:system:widget:install"
	},
	auth_uninstall = {
		.type = afb_auth_Permission,
		.text = "urn:AGL:permission:afm:system:widget:uninstall"
	},
	auth_preinstall = {
		.type = afb_auth_Permission,
		.text = "urn:AGL:permission:afm:system:widget:preinstall"
	},
	auth_detail = {
		.type = afb_auth_Permission,
		.text = "urn:AGL:permission:afm:system:widget:detail"
	},
	auth_start = {
		.type = afb_auth_Permission,
		.text = "urn:AGL:permission:afm:system:widget:start"
	},
	auth_view_all = {
		.type = afb_auth_Permission,
		.text = "urn:AGL:permission:afm:system:widget:view-all"
	},
	auth_state = {
		.type = afb_auth_Permission,
		.text = "urn:AGL:permission:afm:system:runner:state"
	},
	auth_kill = {
		.type = afb_auth_Permission,
		.text = "urn:AGL:permission:afm:system:runner:kill"
	}
;

/*
 * default root
 */
static const char *rootdir = FWK_APP_DIR;

/*
 * the internal application database
 */
static struct afm_udb *afudb;

/*
 * the event signalling that application list changed
 */
static struct afb_event applist_changed_event;

/*
 * the preallocated true json_object
 */
static struct json_object *json_true;

/* enforce daemon reload */
static void do_reloads()
{
	systemd_daemon_reload(0);
	systemd_unit_restart_name(0, "sockets.target");
}

/* common bad request reply */
static void bad_request(struct afb_req req)
{
	afb_req_fail(req, _bad_request_, NULL);
}

/* common not found reply */
static void not_found(struct afb_req req)
{
	afb_req_fail(req, _not_found_, NULL);
}

/* common can't start reply */
static void cant_start(struct afb_req req)
{
	afb_req_fail(req, _cannot_start_, NULL);
}

/*
 * Broadcast the event "application-list-changed".
 * This event is sent was the event "changed" is received from dbus.
 */
static void application_list_changed(const char *operation, const char *data)
{
	struct json_object *e = NULL;
	wrap_json_pack(&e, "{ss ss}", "operation", operation, "data", data);
	afb_event_broadcast(applist_changed_event, e);
}

/*
 * Retrieve the required language from 'req'.
 */
static const char *get_lang(struct afb_req req)
{
	const char *lang;

	/* get the optional language */
	lang = afb_req_value(req, _lang_);

	/* TODO use the req to get the lang of the session (if any) */

	return lang;
}


/*
 * retrieves the 'appid' in parameters received with the
 * request 'req' for the 'method'.
 *
 * Returns 1 in case of success.
 * Otherwise, if the 'appid' can't be retrieved, an error stating
 * the bad request is replied for 'req' and 0 is returned.
 */
static int onappid(struct afb_req req, const char *method, const char **appid)
{
	struct json_object *json;

	/* get the paramaters of the request */
	json = afb_req_json(req);

	/* get the appid if any */
	if (!wrap_json_unpack(json, "s", appid)
	 || !wrap_json_unpack(json, "{si}", _id_, appid)) {
		/* found */
		INFO("method %s called for %s", method, *appid);
		return 1;
	}

	/* nothing appropriate */
	INFO("bad request method %s: %s", method,
					json_object_to_json_string(json));
	bad_request(req);
	return 0;
}

/*
 * retrieves the 'runid' in parameters received with the
 * request 'req' for the 'method'.
 *
 * Returns 1 in case of success.
 * Otherwise, if the 'runid' can't be retrieved, an error stating
 * the bad request is replied for 'req' and 0 is returned.
 */
static int onrunid(struct afb_req req, const char *method, int *runid)
{
	struct json_object *json;
	const char *appid;

	/* get the paramaters of the request */
	json = afb_req_json(req);

	/* get the runid if any */
	if (!wrap_json_unpack(json, "i", runid)
	 || !wrap_json_unpack(json, "{si}", _runid_, runid)) {
		INFO("method %s called for %d", method, *runid);
		return 1;
	}

	/* get the appid if any */
	if (!onappid(req, method, &appid))
		return 0;

	/* search the runid of the appid */
	*runid = afm_urun_search_runid(afudb, appid, afb_req_get_uid(req));
	if (*runid < 0) {
		/* nothing appropriate */
		INFO("method %s can't get runid for %s: %m", method,
							appid);
		not_found(req);
		return 0;
	}

	/* found */
	INFO("method %s called for %s -> %d", method, appid, *runid);
	return 1;
}

/*
 * Sends the reply 'resp' to the request 'req' if 'resp' is not NULLzero.
 * Otherwise, when 'resp' is NULL replies the error string 'errstr'.
 */
static void reply(struct afb_req req, struct json_object *resp, const char *errstr)
{
	if (!resp)
		afb_req_fail(req, errstr, NULL);
	else
		afb_req_success(req, resp, NULL);
}

/*
 * Sends the reply "true" to the request 'req' if 'status' is zero.
 * Otherwise, when 'status' is not zero replies the error string 'errstr'.
 */
static void reply_status(struct afb_req req, int status, const char *errstr)
{
	reply(req, status ? NULL : json_object_get(json_true), errstr);
}

/*
 * On query "runnables"
 */
static void runnables(struct afb_req req)
{
	const char *lang;
	struct json_object *resp;

	/* get the language */
	lang = get_lang(req);

	/* get the details */
	resp = afm_udb_applications_public(afudb, afb_req_get_uid(req), lang);
	afb_req_success(req, resp, NULL);
}

/*
 * On query "detail"
 */
static void detail(struct afb_req req)
{
	const char *lang;
	const char *appid;
	struct json_object *resp;

	/* scan the request */
	if (!onappid(req, _detail_, &appid))
		return;

	/* get the language */
	lang = get_lang(req);

	/* wants details for appid */
	resp = afm_udb_get_application_public(afudb, appid, afb_req_get_uid(req), lang);
	if (resp)
		afb_req_success(req, resp, NULL);
	else
		not_found(req);
}

/*
 * On query "start"
 */
static void start(struct afb_req req)
{
	const char *appid;
	struct json_object *appli, *resp;
	int runid;

	/* scan the request */
	if (!onappid(req, _start_, &appid))
		return;

	/* get the application */
	appli = afm_udb_get_application_private(afudb, appid, afb_req_get_uid(req));
	if (appli == NULL) {
		not_found(req);
		return;
	}

	/* launch the application */
	runid = afm_urun_start(appli, afb_req_get_uid(req));
	if (runid <= 0) {
		cant_start(req);
		return;
	}

	/* returns */
	resp = NULL;
#if 0
	wrap_json_pack(&resp, "{si}", _runid_, runid);
#else
	wrap_json_pack(&resp, "i", runid);
#endif
	afb_req_success(req, resp, NULL);
}

/*
 * On query "once"
 */
static void once(struct afb_req req)
{
	const char *appid;
	struct json_object *appli, *resp;
	int runid;

	/* scan the request */
	if (!onappid(req, _once_, &appid))
		return;

	/* get the application */
	appli = afm_udb_get_application_private(afudb, appid, afb_req_get_uid(req));
	if (appli == NULL) {
		not_found(req);
		return;
	}

	/* launch the application */
	runid = afm_urun_once(appli, afb_req_get_uid(req));
	if (runid <= 0) {
		cant_start(req);
		return;
	}

	/* returns the state */
	resp = afm_urun_state(afudb, runid, afb_req_get_uid(req));
	afb_req_success(req, resp, NULL);
}

/*
 * On query "pause"
 */
static void pause(struct afb_req req)
{
	int runid, status;
	if (onrunid(req, "pause", &runid)) {
		status = afm_urun_pause(runid, afb_req_get_uid(req));
		reply_status(req, status, _not_found_);
	}
}

/*
 * On query "resume" from 'smsg' with parameters of 'obj'.
 */
static void resume(struct afb_req req)
{
	int runid, status;
	if (onrunid(req, "resume", &runid)) {
		status = afm_urun_resume(runid, afb_req_get_uid(req));
		reply_status(req, status, _not_found_);
	}
}

/*
 * On query "terminate"
 */
static void terminate(struct afb_req req)
{
	int runid, status;
	if (onrunid(req, "terminate", &runid)) {
		status = afm_urun_terminate(runid, afb_req_get_uid(req));
		reply_status(req, status, _not_found_);
	}
}

/*
 * On query "runners"
 */
static void runners(struct afb_req req)
{
	struct json_object *resp;
	resp = afm_urun_list(afudb, afb_req_get_uid(req));
	afb_req_success(req, resp, NULL);
}

/*
 * On query "state"
 */
static void state(struct afb_req req)
{
	int runid;
	struct json_object *resp;
	if (onrunid(req, "state", &runid)) {
		resp = afm_urun_state(afudb, runid, afb_req_get_uid(req));
		reply(req, resp, _not_found_);
	}
}

/*
 * On querying installation of widget(s)
 */
static void install(struct afb_req req)
{
	const char *wgtfile;
	const char *root;
	int force;
	int reload;
	struct wgt_info *ifo;
	struct json_object *json;
	struct json_object *resp;

	/* default settings */
	root = rootdir;
	force = 0;
	reload = 1;

	/* scan the request */
	json = afb_req_json(req);
	if (wrap_json_unpack(json, "s", &wgtfile)
		&& wrap_json_unpack(json, "{ss s?s s?b s?b}",
				"wgt", &wgtfile,
				"root", &root,
				"force", &force,
				"reload", &reload)) {
		return bad_request(req);
	}

	/* install the widget */
	ifo = install_widget(wgtfile, root, force);
	if (ifo == NULL)
		afb_req_fail_f(req, "failed", "installation failed: %m");
	else {
		afm_udb_update(afudb);
		/* reload if needed */
		if (reload)
			do_reloads();

		/* build the response */
		wrap_json_pack(&resp, "{ss}", _added_, wgt_info_desc(ifo)->idaver);
		afb_req_success(req, resp, NULL);
		application_list_changed(_install_, wgt_info_desc(ifo)->idaver);

		/* clean-up */
		wgt_info_unref(ifo);
	}
}

/*
 * On querying uninstallation of widget(s)
 */
static void uninstall(struct afb_req req)
{
	const char *idaver;
	const char *root;
	struct json_object *json;
	int rc;

	/* default settings */
	root = rootdir;

	/* scan the request */
	json = afb_req_json(req);
	if (wrap_json_unpack(json, "s", &idaver)
		&& wrap_json_unpack(json, "{ss s?s}",
				_id_, &idaver,
				"root", &root)) {
		return bad_request(req);
	}

	/* install the widget */
	rc = uninstall_widget(idaver, root);
	if (rc)
		afb_req_fail_f(req, "failed", "uninstallation failed: %m");
	else {
		afm_udb_update(afudb);
		afb_req_success(req, NULL, NULL);
		application_list_changed(_uninstall_, idaver);
	}
}

static int init()
{
	/* init database */
	afudb = afm_udb_create(1, 0, "afm-appli-");
	if (!afudb) {
		ERROR("afm_udb_create failed");
		return -1;
	}

	/* create TRUE */
	json_true = json_object_new_boolean(1);

	/* create the event */
	applist_changed_event = afb_daemon_make_event(_a_l_c_);
	return -!afb_event_is_valid(applist_changed_event);
}

static const struct afb_verb_v2 verbs[] =
{
	{_runnables_, runnables, &auth_detail, "Get list of runnable applications",          AFB_SESSION_CHECK_V2 },
	{_detail_   , detail,    &auth_detail, "Get the details for one application",        AFB_SESSION_CHECK_V2 },
	{_start_    , start,     &auth_start, "Start an application",                       AFB_SESSION_CHECK_V2 },
	{_once_     , once,      &auth_start, "Start once an application",                  AFB_SESSION_CHECK_V2 },
	{_terminate_, terminate, &auth_kill, "Terminate a running application",            AFB_SESSION_CHECK_V2 },
	{_pause_    , pause,     &auth_kill, "Pause a running application",                AFB_SESSION_CHECK_V2 },
	{_resume_   , resume,    &auth_kill, "Resume a paused application",                AFB_SESSION_CHECK_V2 },
	{_runners_  , runners,   &auth_state, "Get the list of running applications",       AFB_SESSION_CHECK_V2 },
	{_state_    , state,     &auth_state, "Get the state of a running application",     AFB_SESSION_CHECK_V2 },
	{_install_  , install,   &auth_install, "Install an application using a widget file", AFB_SESSION_CHECK_V2 },
	{_uninstall_, uninstall, &auth_uninstall, "Uninstall an application",                   AFB_SESSION_CHECK_V2 },
	{ NULL, NULL, NULL, NULL, 0 }
};

const struct afb_binding_v2 afbBindingV2 = {
	.api = "afm-main",
	.specification = NULL,
	.info = "Application Framework Master Service",
	.verbs = verbs,
	.preinit = NULL,
	.init = init,
	.onevent = NULL,
	.noconcurrency = 1 /* relies on binder for serialisation of requests */
};