#include <errno.h>
#include <stdio.h>
#include <systemd/sd-bus.h>

#include "npassd/common.h"
#include "npassd/service.h"
#include "util.h"

#define SERVICE_IFACE "org.freedesktop.Secret.Service"

static int handle_open_session(sd_bus_message *msg, void *data,
			       sd_bus_error *ret_error);
static int handle_search_items(sd_bus_message *msg, void *data,
			       sd_bus_error *ret_error);

static const sd_bus_vtable service_vtable[] = {
	SD_BUS_VTABLE_START(0),
	SD_BUS_METHOD("OpenSession", "sv", "vo", handle_open_session,
		      SD_BUS_VTABLE_UNPRIVILEGED),
	SD_BUS_METHOD("SearchItems", "a{ss}", "aoao", handle_search_items,
		      SD_BUS_VTABLE_UNPRIVILEGED),
	SD_BUS_VTABLE_END,
};

static int handle_search_items(__attribute__((unused)) sd_bus_message *msg,
			       __attribute__((unused)) void *data,
			       __attribute__((unused)) sd_bus_error *ret_error)
{
	return sd_bus_reply_method_return(msg, "aoao", 0, 0);
}

static int handle_open_session(sd_bus_message *msg, void *data,
			       sd_bus_error *ret_error)
{
	struct service *service = data;
	int ret, session_slot_no;
	const char *sender;
	char *algo;

	ret = sd_bus_message_read(msg, "s", &algo);
	if (ret < 0)
		return ret;

	ret = strcmp(algo, "plain");
	if (ret) {
		sd_bus_error_set_const(ret_error, SD_BUS_ERROR_NOT_SUPPORTED,
				       "Unsupported encryption algorithm");
		return -ENOTSUP;
	}

	session_slot_no = session_slot_available(service->session, MAX_SESSION);
	if (session_slot_no < 0) {
		print_err("No free slot available: %s",
			  strerror(-session_slot_no));
		return ret;
	}

	sender = sd_bus_message_get_sender(msg);
	ret = session_new(service->bus, &service->session[session_slot_no],
			  session_slot_no, sender);
	if (ret < 0)
		return ret;

	return sd_bus_reply_method_return(
		msg, "vo", "s", NULL, service->session[session_slot_no].path);
}

void service_free(struct service *service)
{
	for (int i = 0; i < MAX_SESSION; i++)
		session_free(&service->session[i]);

	sd_bus_slot_unref(service->slot);
}

int service_init(sd_bus *bus, struct service *service)
{
	int ret;

	service->bus = bus;
	session_init(service->session, MAX_SESSION);

	ret = sd_bus_add_object_vtable(service->bus, &service->slot,
				       DBUS_OBJECT_PATH, SERVICE_IFACE,
				       service_vtable, service);
	if (ret < 0)
		print_err("Failed to connect to bus: %s", strerror(-ret));

	return ret;
}