src/utils/parseInput.ts

/**
* 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 DiscordClient from '../client/Client';
import { User, GuildMember, GuildChannel, Role, Guild } from 'discord.js';

export async function parseUser(client: DiscordClient, input: string): Promise <User | null> {
    if (!/^\<\@(\!)?(\d+)\>$/.test(input.trim())) {
        return null;
    }   

    let user: User | null = null;   

    try {
        user = await client.users.fetch(input.trim().substring(input.includes('!') ? 3 : 2, input.trim().length - 1));
    }
    catch (e) {
        return null;
    }

    return user;
}

export async function parseMember(guild: Guild, input: string): Promise <GuildMember | null> {
    if (!/^\<\@(\!)?(\d+)\>$/.test(input.trim())) {
        return null;
    }   

    let user: GuildMember | null = null;    

    try {
        user = await guild.members.fetch(input.trim().substring(input.includes('!') ? 3 : 2, input.trim().length - 1));
    }
    catch (e) {
        return null;
    }

    return user;
}

export async function parseChannel(guild: Guild, input: string): Promise <GuildChannel | null> {
    if (!/^\<\#(\d+)\>$/.test(input.trim())) {
        return null;
    }   

    let channel: GuildChannel | null = null;    

    try {
        channel = <GuildChannel | null> await guild.channels.fetch(input.trim().substring(2, input.trim().length - 1));
    }
    catch (e) {
        return null;
    }

    return channel;
}

export async function parseRole(guild: Guild, input: string): Promise <Role | null> {
    if (!/^\<\@\&(\d+)\>$/.test(input.trim())) {
        return null;
    }   

    let role: Role | null = null;   

    try {
        role = await guild.roles.fetch(input.trim().substring(3, input.trim().length - 1));
    }
    catch (e) {
        return null;
    }

    return role;
}
1	/**
2	* This file is part of SudoBot.
3	*
4	* Copyright (C) 2021-2022 OSN Inc.
5	*
6	* SudoBot is free software; you can redistribute it and/or modify it
7	* under the terms of the GNU Affero General Public License as published by
8	* the Free Software Foundation, either version 3 of the License, or
9	* (at your option) any later version.
10	*
11	* SudoBot is distributed in the hope that it will be useful, but
12	* WITHOUT ANY WARRANTY; without even the implied warranty of
13	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14	* GNU Affero General Public License for more details.
15	*
16	* You should have received a copy of the GNU Affero General Public License
17	* along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
18	*/
19
20	import DiscordClient from '../client/Client';
21	import { User, GuildMember, GuildChannel, Role, Guild } from 'discord.js';
22
23	export async function parseUser(client: DiscordClient, input: string): Promise <User \| null> {
24	if (!/^\<\@(\!)?(\d+)\>$/.test(input.trim())) {
25	return null;
26	}
27
28	let user: User \| null = null;
29
30	try {
31	user = await client.users.fetch(input.trim().substring(input.includes('!') ? 3 : 2, input.trim().length - 1));
32	}
33	catch (e) {
34	return null;
35	}
36
37	return user;
38	}
39
40	export async function parseMember(guild: Guild, input: string): Promise <GuildMember \| null> {
41	if (!/^\<\@(\!)?(\d+)\>$/.test(input.trim())) {
42	return null;
43	}
44
45	let user: GuildMember \| null = null;
46
47	try {
48	user = await guild.members.fetch(input.trim().substring(input.includes('!') ? 3 : 2, input.trim().length - 1));
49	}
50	catch (e) {
51	return null;
52	}
53
54	return user;
55	}
56
57	export async function parseChannel(guild: Guild, input: string): Promise <GuildChannel \| null> {
58	if (!/^\<\#(\d+)\>$/.test(input.trim())) {
59	return null;
60	}
61
62	let channel: GuildChannel \| null = null;
63
64	try {
65	channel = <GuildChannel \| null> await guild.channels.fetch(input.trim().substring(2, input.trim().length - 1));
66	}
67	catch (e) {
68	return null;
69	}
70
71	return channel;
72	}
73
74	export async function parseRole(guild: Guild, input: string): Promise <Role \| null> {
75	if (!/^\<\@\&(\d+)\>$/.test(input.trim())) {
76	return null;
77	}
78
79	let role: Role \| null = null;
80
81	try {
82	role = await guild.roles.fetch(input.trim().substring(3, input.trim().length - 1));
83	}
84	catch (e) {
85	return null;
86	}
87
88	return role;
89	}