/*
 * Copyright (C) 2017 Ortega Froysa, Nicolás <nortega@themusicinnoise.net>
 * Author: Ortega Froysa, Nicolás <nortega@themusicinnoise.net>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "neocomm.h"

#include "error.hpp"
#include "globals.hpp"

std::map<std::string, struct channel_info> channels;

int NeoComm_join_channel(const char *channel_name) {
	if(not node.isRunning())
	{
		add_error("NeoComm must be initialized.");
		return 0;
	}

	const std::string s_chan_name(channel_name);
	if(s_chan_name.empty())
	{
		add_error("Chan name is empty.");
		return 0;
	}

	// if it already exists then just exit
	if(channels.find(s_chan_name) == channels.end())
		return 1;

	dht::InfoHash chan_hash(s_chan_name);
	{
		static constexpr dht::InfoHash INVALID_ID {};
		if(chan_hash == INVALID_ID)
			chan_hash = dht::InfoHash::get(s_chan_name);
	}

	channels[s_chan_name] = {
		/*.hash =*/ chan_hash,
		/*.token =*/ node.listen<dht::ImMessage>(chan_hash,
				[=](dht::ImMessage &&msg) {
					channels[s_chan_name].msgs.push_back(msg);
					return true;
				}),
		/*.msgs = */ std::vector<dht::ImMessage>()
	};

	return 1;
}

void NeoComm_leave_channel(const char *channel_name) {
	if(not node.isRunning())
		return;
	const std::string s_chan_name(channel_name);
	if(channels.find(s_chan_name) == channels.end())
		return;

	struct channel_info *channel = &channels[s_chan_name];
	channel->token.wait();
	node.cancelListen(channel->hash, channel->token.get());
	channels.erase(s_chan_name);
}

struct message *NeoComm_get_next_message(const char *channel_name) {
	const std::string s_chan_name(channel_name);
	if(channels[s_chan_name].msgs.empty())
		return NULL;
	const dht::ImMessage *new_msg = &channels[s_chan_name].msgs.front();
	struct message *msg =
		static_cast<struct message*>(malloc(sizeof(struct message)));

	*msg = {
		/*.msg =*/ new_msg->msg.c_str(),
		/*.sent =*/ new_msg->date,
		// TODO: add the rest of these
	};

	return msg;
}

void NeoComm_free_message(struct message *msg) {
	free(msg);
}