Jupiter-Bot/Bot/Main.cpp

/**
 * Copyright (C) 2013-2017 Jessica James.
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
 * SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
 * OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
 * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 *
 * Written by Jessica James <jessica.aj@outlook.com>
 */

#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <exception>
#include <thread>
#include <mutex>
#include "Jupiter/Functions.h"
#include "Jupiter/INIConfig.h"
#include "Jupiter/Socket.h"
#include "Jupiter/Plugin.h"
#include "Jupiter/Timer.h"
#include "IRC_Bot.h"
#include "Console_Command.h"
#include "IRC_Command.h"

#if defined _WIN32
#include <Windows.h>
#endif // _WIN32

using namespace Jupiter::literals;

Jupiter::INIConfig o_config;
Jupiter::Config *Jupiter::g_config = &o_config;
std::chrono::steady_clock::time_point Jupiter::g_start_time = std::chrono::steady_clock::now();

#define INPUT_BUFFER_SIZE 2048

struct ConsoleInput
{
	Jupiter::String input;
	std::mutex input_mutex;
	bool awaiting_processing = false;

	ConsoleInput() : input(INPUT_BUFFER_SIZE) {}
} console_input;

void onTerminate()
{
	puts("Terminate signal received...");
}

void onExit()
{
	puts("Exit signal received; Cleaning up...");
	Jupiter::Socket::cleanup();
	puts("Clean-up complete. Closing...");
}

void inputLoop()
{
	char input[INPUT_BUFFER_SIZE];
	size_t input_length;
	while (ftell(stdin) != -1) // This can be expanded later to check for EBADF specifically.
	{
		fgets(input, sizeof(input), stdin);
		input_length = strcspn(input, "\r\n");

	check_input_processing:

		console_input.input_mutex.lock();
		if (console_input.awaiting_processing == false)
		{
			console_input.input.set(input, input_length);
			console_input.awaiting_processing = true;
			console_input.input_mutex.unlock();
		}
		else // User input received before previous input was processed.
		{
			std::this_thread::sleep_for((std::chrono::seconds(1)));
			goto check_input_processing;
		}
	}
}

int main(int argc, const char **args)
{
	atexit(onExit);
	std::set_terminate(onTerminate);
	std::thread inputThread(inputLoop);
	Jupiter::ReferenceString command, plugins_directory, configs_directory;
	size_t index;

#if defined _WIN32
	// Sets console to UTF-8
	SetConsoleCP(65001);
	SetConsoleOutputCP(65001);
#endif // _WIN32

	srand(static_cast<unsigned int>(std::chrono::system_clock::now().time_since_epoch().count()));
	puts(Jupiter::copyright);
	const char *configFileName = "Config.ini";

	for (int i = 1; i < argc; i++)
	{
		if ("-help"_jrs.equalsi(args[i]))
		{
			puts("Help coming soon, to a theatre near you!");
			return 0;
		}
		else if ("-config"_jrs.equalsi(args[i]) && ++i < argc)
			configFileName = args[i];
		else if ("-pluginsdir"_jrs.equalsi(args[i]) && ++i < argc)
			plugins_directory = args[i];
		else if ("-configsdir"_jrs.equals(args[i]) && ++i < argc)
			configs_directory = args[i];
		else if ("-configFormat"_jrs.equalsi(args[i]) && ++i < argc)
			puts("Feature not yet supported!");
		else
			printf("Warning: Unknown command line argument \"%s\" specified. Ignoring...", args[i]);
	}

	std::chrono::steady_clock::time_point load_start = std::chrono::steady_clock::now();

	puts("Loading config file...");
	if (!o_config.read(configFileName))
	{
		puts("Unable to read config file. Closing...");
		exit(0);
	}

	double time_taken = static_cast<double>(std::chrono::duration_cast<std::chrono::microseconds>(std::chrono::steady_clock::now() - load_start).count()) / 1000.0;

	printf("Config loaded (%fms)." ENDL, time_taken);
	
	if (plugins_directory.isEmpty())
		plugins_directory = o_config.get("PluginsDirectory"_jrs);

	if (configs_directory.isEmpty())
		configs_directory = o_config.get("ConfigsDirectory"_jrs);

	if (plugins_directory.isNotEmpty())
	{
		Jupiter::Plugin::setDirectory(plugins_directory);
		printf("Plugins will be loaded from \"%.*s\"." ENDL, plugins_directory.size(), plugins_directory.ptr());
	}

	if (configs_directory.isNotEmpty())
	{
		Jupiter::Plugin::setDirectory(configs_directory);
		printf("Plugin configs will be loaded from \"%.*s\"." ENDL, configs_directory.size(), configs_directory.ptr());
	}

	puts("Loading plugins...");
	const Jupiter::ReadableString &pluginList = o_config.get("Plugins"_jrs);
	if (pluginList.isEmpty())
		puts("No plugins to load!");
	else
	{
		// initialize plugins
		unsigned int nPlugins = pluginList.wordCount(WHITESPACE);
		printf("Attempting to load %u plugins..." ENDL, nPlugins);

		bool load_success;

		for (unsigned int i = 0; i < nPlugins; i++)
		{
			Jupiter::ReferenceString plugin = Jupiter::ReferenceString::getWord(pluginList, i, WHITESPACE);

			load_start = std::chrono::steady_clock::now();
			load_success = Jupiter::Plugin::load(plugin) != nullptr;
			time_taken = static_cast<double>(std::chrono::duration_cast<std::chrono::microseconds>(std::chrono::steady_clock::now() - load_start).count()) / 1000.0;
			
			if (load_success)
				printf("\"%.*s\" loaded successfully (%fms)." ENDL, plugin.size(), plugin.ptr(), time_taken);
			else
				fprintf(stderr, "WARNING: Failed to load plugin \"%.*s\" (%fms)!" ENDL, plugin.size(), plugin.ptr(), time_taken);
		}

		// OnPostInitialize
		for (index = 0; index != Jupiter::plugins->size(); ++index)
			Jupiter::plugins->get(index)->OnPostInitialize();
	}

	printf("Initialization completed in %f milliseconds." ENDL, static_cast<double>(std::chrono::duration_cast<std::chrono::microseconds>(std::chrono::steady_clock::now() - Jupiter::g_start_time).count()) / 1000.0 );

	if (consoleCommands->size() > 0)
		printf("%zu Console Commands have been initialized%s" ENDL, consoleCommands->size(), getConsoleCommand("help"_jrs) == nullptr ? "." : "; type \"help\" for more information.");
	if (IRCMasterCommandList->size() > 0)
		printf("%zu IRC Commands have been loaded into the master list." ENDL, IRCMasterCommandList->size());

	while (1)
	{
		index = 0;
		while (index < Jupiter::plugins->size())
			if (Jupiter::plugins->get(index)->shouldRemove() || Jupiter::plugins->get(index)->think() != 0)
				Jupiter::Plugin::free(index);
			else
				++index;
		Jupiter::Timer::check();
		
		if (console_input.input_mutex.try_lock())
		{
			if (console_input.awaiting_processing)
			{
				console_input.awaiting_processing = false;
				command = console_input.input.getWord(0, WHITESPACE);

				ConsoleCommand *cmd = getConsoleCommand(command);
				if (cmd != nullptr)
					cmd->trigger(console_input.input.gotoWord(1, WHITESPACE));
				else
					printf("Error: Command \"%.*s\" not found." ENDL, command.size(), command.ptr());
			}
			console_input.input_mutex.unlock();
		}
		std::this_thread::sleep_for((std::chrono::milliseconds(1)));
	}
	return 0;
}