enigma-bot/index.js at master · Capaural/enigma-bot · 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
const discord = require("discord.js");
const admin = require("firebase-admin");
const serviceAccount = require("./private/firebase.json");
const { bot_token_dev, bot_token_prod } = require("./private/creds.json");

const bot_token = process.argv["2"] == "--prod" ? bot_token_prod : bot_token_dev;

admin.initializeApp({
	credential: admin.credential.cert(serviceAccount),
	databaseURL: "https://capaural-717e1.firebaseio.com",
});

const db = admin.database();
db.ref("/enigmas").on("value", (snapshot) => {
	const data = snapshot.val();
	if (data) {
		global.enigmas = [];
		for (var i in data.solutions) {
			if (data.solutions[i].unlocked) {
				global.enigmas.push(data.solutions[i]);
			}
		}
		// console.log(global.enigmas);

		global.teams = [];
		for (var i in data.teams) {
			global.teams.push(data.teams[i]);
		}
		global.teams.sort((a, b) => (a.score < b.score ? 1 : -1));
	}
});

// Import all the functions
const utils = require("./utils/utils");
const help = require("./commands/help");
const teams = require("./commands/teams");
const contact = require("./commands/contact");
const report = require("./commands/report");
const submit = require("./commands/submit");
const admin_commands = require("./commands/admin");
const config = require("./utils/config").config;
const rules = require("./commands/rules");
const hint = require("./commands/hint");

global.users = {};

const prefix = "-enigma ";
const client = new discord.Client();
client.once("ready", () => {
	console.log(utils.logPrefix() + " Enigma bot is online !");
	//utils.saveInLog(["BOT", "Enigma bot is online !"]);
});

const commands_not_in_dm = {
	contact: contact.triggerMessage,
	help: help.helpMessage,
	sendtemplate: admin_commands.sendTemplate,
	createenigma: admin_commands.createEnigma,
	leaderboard: teams.leaderboard,
	infos: teams.globalInfos,
	rules: rules.rulesMessage,
};

const commands_in_dm = {
	create: teams.createTeam,
	join: teams.joinTeam,
	infos: teams.infosTeam,
	report: report.report,
	submit: submit.submitAnswer,
	help: help.DMHelpMessage,
	rules: rules.rulesMessage,
	hint: hint.hintEnigma
};

client.on("message", (message) => {
	const isDM = message.channel.type == "dm";
	if ((!isDM && !message.content.startsWith(prefix)) || message.author.bot) {
		return;
	}

	let args = message.content.split(" ");
	if (args[0] + " " == prefix) {
		args.shift();
	}

	let command = "help";
	let parameters = "";
	if (args.length > 0) {
		// This is needed because when creating a new Enigma, the params start with a
		// line feed and not a space
		if (args[0].includes("\n")) {
			let first_args = args[0].split("\n").reverse();
			args.shift();
			for (const element of first_args) {
				args.unshift(element);
			}
		}

		command = args.shift().toLowerCase();
		parameters = args.join(" ").toLowerCase();
	}

	const function_params = {
		message: message,
		db: db,
		client: client,
		config: config,
		command_params: parameters,
	};

	const map = isDM ? commands_in_dm : commands_not_in_dm;
	const type = map[command] !== undefined ? command : "help";
	utils.saveInLog([client.user.username, "[+] " + command]);
	map[type](function_params);
});

client.login(bot_token);