sudobot/trunk/setup.js

#!/bin/node

/**
* setup.js -- the installer script
* 
* 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/>.
*/

const path = require('path');
const fs = require('fs/promises');
const readline = require('readline');

const CONFIG_DIR = path.resolve(__dirname, 'config');
const { version } = require('./package.json');
const { existsSync } = require('fs');
const SAMPLE_CONFIG_PATH = path.resolve(CONFIG_DIR, 'sample-config.json');
const CONFIG_PATH = path.resolve(CONFIG_DIR, 'config.json');

const isSnowflake = text => /^\d+$/.test(text);

const rl = readline.createInterface({ input: process.stdin, output: process.stdout });

const prompt = text => new Promise(resolve => {
    rl.question(text, resolve);
});

const promptDefault = async (text, defaultValue) => {
    const input = await prompt(text);

    if (input.toString().trim() === '') {
        return defaultValue;
    }

    return input;
};

const promptLoop = async (text, validator, defaultValue) => {
    let fn = promptDefault;

    if (defaultValue === undefined) {
        fn = prompt;
    }

    const input = await fn(text, defaultValue);

    if (!(await validator(input ?? ''))) {
        return promptLoop(text, validator, defaultValue);
    }

    return input ?? '';
};

const snowflakeValidator = input => {
    if (!isSnowflake(input)) {
        console.log(`That's not a valid snowflake! Please enter a valid ID.`);
        return false;
    }

    return true;
};

(async () => {
    console.log(`SudoBot version ${version}`);
    console.log(`Copyright (C) OSN Inc 2022`);
    console.log(`Thanks for using SudoBot! We'll much appreciate if you star the repository on GitHub.\n`);

    let prefix = '-', homeGuild = '', owners = [];
    let config = Object.entries(JSON.parse((await fs.readFile(SAMPLE_CONFIG_PATH)).toString()));
    
    config[1][1].prefix = (await promptLoop(`What will be the bot prefix? [${prefix}]: `, input => {
        if (input.trim().includes(' ')) {
            console.log(`Prefixes must not contain spaces!`);
            return false;
        }

        return true;
    }, "-")).trim();

    homeGuild = await promptLoop(`What will be the Home/Support Guild ID?: `, snowflakeValidator);
    config[1][0] = homeGuild;

    config = Object.fromEntries(config);

    config.global.id = homeGuild;
    config.global.owners = (await promptLoop(`Who will be the owner? Specify the owner user IDs separated with comma (,): `, input => {
        const splitted = input.split(',');

        for (const snowflake of splitted) {
            if (!snowflakeValidator(snowflake)) {
                console.log(`Invalid snowflake given! Make sure that the IDs are correctly given!`);
                return false;
            }
        }

        return true;
    })).split(',').map(s => s.trim());

    // config[1][0] = homeGuild;
    // config[1][1].prefix = prefix;

    // config.global.owners = owners;

    const guildConfig = {...config[homeGuild]};

    guildConfig.mod_role = await promptLoop(`What will be the moderator role ID?: `, snowflakeValidator);
    guildConfig.admin = await promptLoop(`What will be the safe role ID?: `, snowflakeValidator);
    guildConfig.mute_role = await promptLoop(`What will be the muted role ID?: `, snowflakeValidator);
    guildConfig.gen_role = await promptLoop(`What will be the general role ID? [${homeGuild}]: `, snowflakeValidator, homeGuild);
    guildConfig.logging_channel = await promptLoop(`What will be the main logging channel ID?: `, snowflakeValidator);
    guildConfig.logging_channel_join_leave = await promptLoop(`What will be the join/leave logging channel ID?: `, snowflakeValidator);

    config[homeGuild] = guildConfig;

    console.log(config);

    if (existsSync(CONFIG_PATH)) {
        const input = await promptDefault("The config file (config/config.json) already exists. Do you want to overwrite the file? [y/N]: ", "n");

        if (input.trim().toLowerCase() !== "y" && input.trim().toLowerCase() !== "yes") {
            console.log("Aborting setup.");
            rl.close();
            process.exit(1);
        }
    }

    rl.close();

    if (existsSync(CONFIG_PATH))
        await fs.rename(CONFIG_PATH, path.join(CONFIG_DIR, 'config-old-' + Math.round(Math.random() * 100000) + '.json'));

    await fs.writeFile(CONFIG_PATH, JSON.stringify(config, undefined, ' '));

    console.log("Setup complete!");
    console.table([
        {
            prefix,
            homeGuild
        }
    ]);
})().catch(console.error);
1	rakin	442	#!/bin/node
2
3			/**
4			* setup.js -- the installer script
5			*
6			* This file is part of SudoBot.
7			*
8			* Copyright (C) 2021-2022 OSN Inc.
9			*
10			* SudoBot is free software; you can redistribute it and/or modify it
11			* under the terms of the GNU Affero General Public License as published by
12			* the Free Software Foundation, either version 3 of the License, or
13			* (at your option) any later version.
14			*
15			* SudoBot is distributed in the hope that it will be useful, but
16			* WITHOUT ANY WARRANTY; without even the implied warranty of
17			* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18			* GNU Affero General Public License for more details.
19			*
20			* You should have received a copy of the GNU Affero General Public License
21			* along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
22			*/
23
24			const path = require('path');
25			const fs = require('fs/promises');
26			const readline = require('readline');
27
28			const CONFIG_DIR = path.resolve(__dirname, 'config');
29			const { version } = require('./package.json');
30	rakin	443	const { existsSync } = require('fs');
31	rakin	442	const SAMPLE_CONFIG_PATH = path.resolve(CONFIG_DIR, 'sample-config.json');
32	rakin	443	const CONFIG_PATH = path.resolve(CONFIG_DIR, 'config.json');
33	rakin	442
34			const isSnowflake = text => /^\d+$/.test(text);
35
36			const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
37
38			const prompt = text => new Promise(resolve => {
39			rl.question(text, resolve);
40			});
41
42			const promptDefault = async (text, defaultValue) => {
43			const input = await prompt(text);
44
45			if (input.toString().trim() === '') {
46			return defaultValue;
47			}
48
49			return input;
50			};
51
52			const promptLoop = async (text, validator, defaultValue) => {
53			let fn = promptDefault;
54
55			if (defaultValue === undefined) {
56			fn = prompt;
57			}
58
59	rakin	443	const input = await fn(text, defaultValue);
60	rakin	442
61			if (!(await validator(input ?? ''))) {
62			return promptLoop(text, validator, defaultValue);
63			}
64
65			return input ?? '';
66			};
67
68			const snowflakeValidator = input => {
69			if (!isSnowflake(input)) {
70			console.log(`That's not a valid snowflake! Please enter a valid ID.`);
71			return false;
72			}
73
74			return true;
75			};
76
77			(async () => {
78			console.log(`SudoBot version ${version}`);
79			console.log(`Copyright (C) OSN Inc 2022`);
80			console.log(`Thanks for using SudoBot! We'll much appreciate if you star the repository on GitHub.\n`);
81	rakin	455
82			let prefix = '-', homeGuild = '', owners = [];
83			let config = Object.entries(JSON.parse((await fs.readFile(SAMPLE_CONFIG_PATH)).toString()));
84	rakin	442
85	rakin	455	config[1][1].prefix = (await promptLoop(`What will be the bot prefix? [${prefix}]: `, input => {
86	rakin	442	if (input.trim().includes(' ')) {
87			console.log(`Prefixes must not contain spaces!`);
88			return false;
89			}
90
91			return true;
92	rakin	455	}, "-")).trim();
93	rakin	443
94	rakin	442	homeGuild = await promptLoop(`What will be the Home/Support Guild ID?: `, snowflakeValidator);
95	rakin	455	config[1][0] = homeGuild;
96
97			config = Object.fromEntries(config);
98
99			config.global.id = homeGuild;
100			config.global.owners = (await promptLoop(`Who will be the owner? Specify the owner user IDs separated with comma (,): `, input => {
101	rakin	443	const splitted = input.split(',');
102	rakin	442
103	rakin	443	for (const snowflake of splitted) {
104			if (!snowflakeValidator(snowflake)) {
105			console.log(`Invalid snowflake given! Make sure that the IDs are correctly given!`);
106			return false;
107			}
108			}
109
110			return true;
111			})).split(',').map(s => s.trim());
112
113	rakin	455	// config[1][0] = homeGuild;
114			// config[1][1].prefix = prefix;
115	rakin	443
116	rakin	455	// config.global.owners = owners;
117	rakin	443
118	rakin	455	const guildConfig = {...config[homeGuild]};
119
120			guildConfig.mod_role = await promptLoop(`What will be the moderator role ID?: `, snowflakeValidator);
121			guildConfig.admin = await promptLoop(`What will be the safe role ID?: `, snowflakeValidator);
122			guildConfig.mute_role = await promptLoop(`What will be the muted role ID?: `, snowflakeValidator);
123			guildConfig.gen_role = await promptLoop(`What will be the general role ID? [${homeGuild}]: `, snowflakeValidator, homeGuild);
124			guildConfig.logging_channel = await promptLoop(`What will be the main logging channel ID?: `, snowflakeValidator);
125			guildConfig.logging_channel_join_leave = await promptLoop(`What will be the join/leave logging channel ID?: `, snowflakeValidator);
126
127			config[homeGuild] = guildConfig;
128
129	rakin	443	console.log(config);
130
131			if (existsSync(CONFIG_PATH)) {
132			const input = await promptDefault("The config file (config/config.json) already exists. Do you want to overwrite the file? [y/N]: ", "n");
133
134			if (input.trim().toLowerCase() !== "y" && input.trim().toLowerCase() !== "yes") {
135			console.log("Aborting setup.");
136			rl.close();
137			process.exit(1);
138			}
139			}
140
141	rakin	442	rl.close();
142
143	rakin	455	if (existsSync(CONFIG_PATH))
144			await fs.rename(CONFIG_PATH, path.join(CONFIG_DIR, 'config-old-' + Math.round(Math.random() * 100000) + '.json'));
145
146			await fs.writeFile(CONFIG_PATH, JSON.stringify(config, undefined, ' '));
147
148	rakin	442	console.log("Setup complete!");
149			console.table([
150			{
151			prefix,
152			homeGuild
153			}
154			]);
155			})().catch(console.error);