3.x/src/cli.ts

#!/usr/bin/ts-node

/**
* This file is part of SudoBot.
* 
* Copyright (C) 2021-2022 OSN Inc.
*
* SudoBot 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.
* 
* SudoBot 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 SudoBot. If not, see <https://www.gnu.org/licenses/>.
*/

import { registerCLICommands } from './utils/registry';
import DiscordClient from './client/Client';
import { Intents } from 'discord.js';
import { config } from 'dotenv';
import { existsSync } from 'fs';
import path from 'path';
import { yellow } from './utils/util';

if (existsSync(path.join(__dirname, '../.env'))) {
    config();
}
else {
    process.env.ENV = 'prod';
}

if (process.argv.includes('--prod')) {
    console.warn(yellow('WARNING: Forcing production mode (--prod option passed)'));
    process.env.ENV = 'prod';
}

if (process.argv.includes('--dev')) {
    console.warn(yellow('WARNING: Forcing development mode (--dev option passed)'));
    process.env.ENV = 'dev';
}

const client = new DiscordClient({
    partials: ["CHANNEL"],
    intents: [
        Intents.FLAGS.GUILDS,
        Intents.FLAGS.GUILD_MESSAGES,
        Intents.FLAGS.DIRECT_MESSAGES, 
        Intents.FLAGS.DIRECT_MESSAGE_TYPING,
        Intents.FLAGS.GUILD_PRESENCES,
        Intents.FLAGS.GUILD_MEMBERS,
        Intents.FLAGS.GUILD_BANS,
        Intents.FLAGS.GUILD_MESSAGE_REACTIONS,
        Intents.FLAGS.GUILD_EMOJIS_AND_STICKERS,
    ]
}, path.resolve(__dirname, '..'));

client.on('ready', async () => {
    console.log('The system has logged into discord.');
    await registerCLICommands(client, '../cli-commands');

    const { argv, exit } = process;

    argv.shift();
    argv.shift();

    if (!argv[0]) {
        console.log('No command provided.');
        exit(-1);
    }

    const commandName = argv.shift();
    const command = client.cliCommands.get(commandName!);

    if (!command) {
        console.log(`${commandName}: command not found`);
        exit(-1);
    }

    const options = argv.filter(a => a[0] === '-');
    const args = argv.filter(a => a[0] !== '-');

    if (command!.requiredArgs > args.length) {
        console.log(`${commandName}: expected at least ${command!.requiredArgs} arguments`);
        exit(-1);
    }

    if (command!.requiredOptions > options.length) {
        console.log(`${commandName}: expected at least ${command!.requiredOptions} options`);
        exit(-1);
    }

    await command!.run(client, argv, args, options);
});

client.login(process.env.TOKEN);
1	#!/usr/bin/ts-node
2
3	/**
4	* This file is part of SudoBot.
5	*
6	* Copyright (C) 2021-2022 OSN Inc.
7	*
8	* SudoBot is free software; you can redistribute it and/or modify it
9	* under the terms of the GNU Affero General Public License as published by
10	* the Free Software Foundation, either version 3 of the License, or
11	* (at your option) any later version.
12	*
13	* SudoBot is distributed in the hope that it will be useful, but
14	* WITHOUT ANY WARRANTY; without even the implied warranty of
15	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16	* GNU Affero General Public License for more details.
17	*
18	* You should have received a copy of the GNU Affero General Public License
19	* along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
20	*/
21
22	import { registerCLICommands } from './utils/registry';
23	import DiscordClient from './client/Client';
24	import { Intents } from 'discord.js';
25	import { config } from 'dotenv';
26	import { existsSync } from 'fs';
27	import path from 'path';
28	import { yellow } from './utils/util';
29
30	if (existsSync(path.join(__dirname, '../.env'))) {
31	config();
32	}
33	else {
34	process.env.ENV = 'prod';
35	}
36
37	if (process.argv.includes('--prod')) {
38	console.warn(yellow('WARNING: Forcing production mode (--prod option passed)'));
39	process.env.ENV = 'prod';
40	}
41
42	if (process.argv.includes('--dev')) {
43	console.warn(yellow('WARNING: Forcing development mode (--dev option passed)'));
44	process.env.ENV = 'dev';
45	}
46
47	const client = new DiscordClient({
48	partials: ["CHANNEL"],
49	intents: [
50	Intents.FLAGS.GUILDS,
51	Intents.FLAGS.GUILD_MESSAGES,
52	Intents.FLAGS.DIRECT_MESSAGES,
53	Intents.FLAGS.DIRECT_MESSAGE_TYPING,
54	Intents.FLAGS.GUILD_PRESENCES,
55	Intents.FLAGS.GUILD_MEMBERS,
56	Intents.FLAGS.GUILD_BANS,
57	Intents.FLAGS.GUILD_MESSAGE_REACTIONS,
58	Intents.FLAGS.GUILD_EMOJIS_AND_STICKERS,
59	]
60	}, path.resolve(__dirname, '..'));
61
62	client.on('ready', async () => {
63	console.log('The system has logged into discord.');
64	await registerCLICommands(client, '../cli-commands');
65
66	const { argv, exit } = process;
67
68	argv.shift();
69	argv.shift();
70
71	if (!argv[0]) {
72	console.log('No command provided.');
73	exit(-1);
74	}
75
76	const commandName = argv.shift();
77	const command = client.cliCommands.get(commandName!);
78
79	if (!command) {
80	console.log(`${commandName}: command not found`);
81	exit(-1);
82	}
83
84	const options = argv.filter(a => a[0] === '-');
85	const args = argv.filter(a => a[0] !== '-');
86
87	if (command!.requiredArgs > args.length) {
88	console.log(`${commandName}: expected at least ${command!.requiredArgs} arguments`);
89	exit(-1);
90	}
91
92	if (command!.requiredOptions > options.length) {
93	console.log(`${commandName}: expected at least ${command!.requiredOptions} options`);
94	exit(-1);
95	}
96
97	await command!.run(client, argv, args, options);
98	});
99
100	client.login(process.env.TOKEN);