NyisBotCPP/main.cpp at master · SmallJoker/NyisBotCPP · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
#include "args_parser.h"
#include "logger.h"
#include "client/client_discord.h"
#include "client/client_irc.h"
#include "client/client_telegram.h"
#include "client/client_tui.h"
#include "connection.h"
#include "module.h"
#include "settings.h"
#include "tests/test.h"
#ifdef __unix__
	#include <signal.h>
#endif

static IClient *s_cli = nullptr;
static Settings *settings_ro = nullptr;
static Settings *settings_rw = nullptr;
static bool keep_running = true;

void sigint_handler(int signal)
{
	LOG("Received signal " << signal);
	keep_running = false;
}

void exit_main()
{
	LOG("Shutting down...");
	delete s_cli;
	delete settings_rw;
	delete settings_ro;
	delete g_logger;
	SLEEP_MS(100);
}


extern "C" {

DLL_EXPORT int main(int argc, char *argv[])
{
#ifdef __unix__
	{
		struct sigaction act;
		act.sa_handler = sigint_handler;
		sigaction(SIGINT, &act, NULL);
		sigaction(SIGTERM, &act, NULL);
	}
#endif
	atexit(exit_main);
	g_logger = new Logger();

	bool run_client = true;
	bool run_unittest = false;
	bool no_modules = false;
	bool no_random = false;
	bool verbose = false;
	std::string logfile("debug.txt");
	std::string config_type("user");

	CLIArg("config",     &config_type);
	CLIArg("no-modules", &no_modules);
	CLIArg("no-random",  &no_random);
	CLIArg("logfile",    &logfile);
	CLIArg("quicktest",  &run_client);
	CLIArg("unittest",   &run_unittest);
	CLIArg("verbose",    &verbose);
	CLIArg::parseArgs(argc, argv);

	g_logger->setupFile(logfile.c_str());
	g_logger->setLogLevels(verbose ? LL_VERBOSE : LL_NORMAL, LL_NORMAL);

	LOG("Startup");

	if (run_unittest) {
		Unittest test;
		if (!test.runTests())
			exit(EXIT_FAILURE);
		exit(EXIT_SUCCESS);
	}

	settings_ro = new Settings("config/main.defaults.conf", nullptr);
	settings_rw = new Settings("config/main." + config_type + ".conf", settings_ro);
	settings_ro->syncFileContents();
	settings_ro->set("_internal.type", config_type);
	if (!settings_rw->syncFileContents())
		exit(EXIT_FAILURE);


	if (!no_random)
		std::srand(std::time(nullptr));

	cstr_t &client_type = settings_rw->get("client.type");
	if (client_type == "irc")
		s_cli = new ClientIRC(settings_rw);
	else if (client_type == "tui")
		s_cli = new ClientTUI(settings_rw);
	else if (client_type == "discord")
		s_cli = new ClientDiscord(settings_rw);
	else if (client_type == "telegram")
		s_cli = new ClientTelegram(settings_rw);

	ASSERT(s_cli, "client.type setting value is not supported.");

	if (!no_modules) {
		if (!s_cli->getModuleMgr()->loadModules())
			exit(EXIT_FAILURE);
	}

	if (run_client) {
		s_cli->initialize();

		while (keep_running && s_cli->run()) {
			s_cli->processRequests();
			SLEEP_MS(20);
		}
	}
	return 0;
}

}