/*
 * 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;

static std::mt19937_64 rand_dev { dht::crypto::random_device{}() };
static std::uniform_int_distribution<dht::Value::Id> rand_id;

static std::map<time_t, int> sent_messages;

static std::string channel_list;

int NeoComm_join_channel(const char *channel_name) {
	if(not node.isRunning())
	{
		add_error("NeoComm must be initialized.");
		return 0;
	}
	else if(channel_name[0] != '#')
	{
		add_error("Channel name must start with `#'.");
		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) not_eq 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;
}

int NeoComm_leave_channel(const char *channel_name) {
	if(not node.isRunning())
	{
		add_error("NeoComm must be initialized.");
		return 0;
	}
	else if(channel_name[0] != '#')
	{
		add_error("Channel name must start with `#'.");
		return 0;
	}

	const std::string s_chan_name(channel_name);
	if(channels.find(s_chan_name) == channels.end())
	{
		add_error("Channel does not exist.");
		return 0;
	}
	else if(s_chan_name.empty())
	{
		add_error("Chan name is empty.");
		return 0;
	}

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

const char *NeoComm_get_channel_list() {
	channel_list.clear();
	for(auto &i : channels)
		channel_list += i.first + " ";
	return channel_list.c_str();
}

unsigned int NeoComm_get_num_channels() {
	return channels.size();
}