src/utils/util.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 fs from 'fs';
import DiscordClient from '../client/Client';
import { GuildMember, Message, CommandInteraction, MessageEmbed, ContextMenuInteraction, Interaction } from 'discord.js';
import Axios, { AxiosRequestHeaders } from 'axios';
import { formatDistanceToNowStrict, formatDuration, intervalToDuration } from 'date-fns';

export function parseEmbedsInString(content: string) {
    const embedExpressions = content.matchAll(/embed\:(\{[^\n]+\})/g);
    const newContent = content.replace(/embed\:(\{[^\n]+\})/g, '');
    let embeds: MessageEmbed[] = [];

    for (const expr of [...embedExpressions]) {
        const parsed = JSON.parse(expr[1]);

        try {
            embeds.push(new MessageEmbed(parsed).setColor(parsed.color));     
        }
        catch (e) {
            console.log(e);
        }
    }

    return { embeds, content: newContent };
}

export function splitMessage(message: string, limit: number = 1000, maxIterationCount: number = 100) {
    const splitted: string[] = [];
    let content = message;
    let { length } = content;

    if (length >= limit) {
        let i = 0;

        while (length !== 0 && content !== '') {
            splitted.push(content.substring(0, limit));
            content = content.substring(limit);
            length = content.length;
            i++;

            if (i >= maxIterationCount) {
                console.log('Break loop');
                break;
            }
        }
    }
    else {
        splitted.push(message);
    }

    return splitted;
}

export function getHomeGuild(client: DiscordClient) {
    return client.guilds.cache.get(client.config.props.global.id);
}

export function shouldNotModerate(client: DiscordClient, member: GuildMember) {
    if (!client.config.props[member.guild.id].admin) {
        return false;
    }

    const role = client.config.props[member.guild.id].admin;

    return role && role.trim() !== '' && member.roles.cache.has(role);
}

export async function hasPermission(client: DiscordClient, member: GuildMember, msg: Message | CommandInteraction | ContextMenuInteraction, mod: GuildMember | null, error: string = "You don't have permission to moderate this user") {
    let m = mod;
    
    if (!m) {
        m = msg.member! as GuildMember;
    }
    
    if (member.id !== m.id && member.roles.highest?.position >= m.roles.highest?.position) {
        if (msg instanceof Interaction && msg.deferred) {
            await msg.editReply({
                embeds: [
                    new MessageEmbed()
                    .setColor('#f14a60')
                    .setDescription(`:x: ${error}`)
                ]
            });

            return false;
        }

        await msg.reply({
            embeds: [
                new MessageEmbed()
                .setColor('#f14a60')
                .setDescription(`:x: ${error}`)
            ]
        });

        return false;
    }

    return true;
}

export function timeProcess(seconds: number) {
    return formatDuration(intervalToDuration({ start: 0, end: seconds * 1000 }));
}


/**
 * @deprecated
 */
export function timeProcessOld(seconds: number) {      
    let interval = seconds / (60 * 60 * 24 * 30 * 365);

    if (interval >= 1) {
        return Math.floor(interval) + " year" + (Math.floor(interval) === 1 ? '' : 's');
    }

    interval = seconds / (60 * 60 * 24 * 30);

    if (interval >= 1) {
        return Math.floor(interval) + " month" + (Math.floor(interval) === 1 ? '' : 's');
    }

    interval = seconds / (60 * 60 * 24);

    if (interval >= 1) {
        return Math.floor(interval) + " day" + (Math.floor(interval) === 1 ? '' : 's');
    }

    interval = seconds / (60 * 60);

    if (interval >= 1) {
        return Math.floor(interval) + " hour" + (Math.floor(interval) === 1 ? '' : 's');
    }

    interval = seconds / 60;

    if (interval >= 1) {
        return Math.floor(interval) + " minute" + (Math.floor(interval) === 1 ? '' : 's');
    }

    interval = seconds;

    return Math.floor(interval) + " second" + (Math.floor(interval) === 1 ? '' : 's');
}

export function escapeRegex(string: string) {
    return string.replace(/[-\/\\^$*+?.()|[\]{}]/g, '\\$&');
}

export function timeSince(date: number) {
    // const seconds = Math.floor((Date.now() - date) / 1000);
    // return timeProcess(seconds) + ' ago';
    return formatDistanceToNowStrict(new Date(date), { addSuffix: true });
}

export async function download(url: string, path: string, headers?: AxiosRequestHeaders) {  
    const writer = fs.createWriteStream(path);
    
    const response = await Axios({
        url,
        method: 'GET',
        responseType: 'stream',
        headers
    });
    
    response.data.pipe(writer);
    
    return new Promise((resolve, reject) => {
        if (response.status !== 200) {
            reject();
        }

        writer.on('finish', resolve);
        writer.on('error', reject);
    });
}

export async function deleteFile(path: string) {
    fs.rm(path, (err: any) => {
        if (err) {
            throw new Error(err);
        }
    });
}

export function random(arr: Array <any>) {
    let index = Math.floor(Math.random() * arr.length);
    return arr[index];
}

export function fill(length: number, string: string, token: string = ' ') {
    let safe = 0;

    if (length < string.length)
        return string;

    const diff = length - string.length;

    for (let i = 1; i <= diff; i++, safe++) {
        if (safe >= 500)
            break;        

        string += ' ';
    }   

    return string;
}

export function green(string: string) {
    return '\u001b[1;32m' + string + '\u001b[0m';
}

export function yellow(string: string) {
    return '\u001b[1;33m' + string + '\u001b[0m';
}

export function red(string: string) {
    return '\u001b[1;31m' + string + '\u001b[0m';
}
1	rakinar2	577	/**
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 fs from 'fs';
21			import DiscordClient from '../client/Client';
22			import { GuildMember, Message, CommandInteraction, MessageEmbed, ContextMenuInteraction, Interaction } from 'discord.js';
23			import Axios, { AxiosRequestHeaders } from 'axios';
24			import { formatDistanceToNowStrict, formatDuration, intervalToDuration } from 'date-fns';
25
26			export function parseEmbedsInString(content: string) {
27			const embedExpressions = content.matchAll(/embed\:(\{[^\n]+\})/g);
28			const newContent = content.replace(/embed\:(\{[^\n]+\})/g, '');
29			let embeds: MessageEmbed[] = [];
30
31			for (const expr of [...embedExpressions]) {
32			const parsed = JSON.parse(expr[1]);
33
34			try {
35			embeds.push(new MessageEmbed(parsed).setColor(parsed.color));
36			}
37			catch (e) {
38			console.log(e);
39			}
40			}
41
42			return { embeds, content: newContent };
43			}
44
45			export function splitMessage(message: string, limit: number = 1000, maxIterationCount: number = 100) {
46			const splitted: string[] = [];
47			let content = message;
48			let { length } = content;
49
50			if (length >= limit) {
51			let i = 0;
52
53			while (length !== 0 && content !== '') {
54			splitted.push(content.substring(0, limit));
55			content = content.substring(limit);
56			length = content.length;
57			i++;
58
59			if (i >= maxIterationCount) {
60			console.log('Break loop');
61			break;
62			}
63			}
64			}
65			else {
66			splitted.push(message);
67			}
68
69			return splitted;
70			}
71
72			export function getHomeGuild(client: DiscordClient) {
73			return client.guilds.cache.get(client.config.props.global.id);
74			}
75
76			export function shouldNotModerate(client: DiscordClient, member: GuildMember) {
77			if (!client.config.props[member.guild.id].admin) {
78			return false;
79			}
80
81			const role = client.config.props[member.guild.id].admin;
82
83			return role && role.trim() !== '' && member.roles.cache.has(role);
84			}
85
86			export async function hasPermission(client: DiscordClient, member: GuildMember, msg: Message \| CommandInteraction \| ContextMenuInteraction, mod: GuildMember \| null, error: string = "You don't have permission to moderate this user") {
87			let m = mod;
88
89			if (!m) {
90			m = msg.member! as GuildMember;
91			}
92
93			if (member.id !== m.id && member.roles.highest?.position >= m.roles.highest?.position) {
94			if (msg instanceof Interaction && msg.deferred) {
95			await msg.editReply({
96			embeds: [
97			new MessageEmbed()
98			.setColor('#f14a60')
99			.setDescription(`:x: ${error}`)
100			]
101			});
102
103			return false;
104			}
105
106			await msg.reply({
107			embeds: [
108			new MessageEmbed()
109			.setColor('#f14a60')
110			.setDescription(`:x: ${error}`)
111			]
112			});
113
114			return false;
115			}
116
117			return true;
118			}
119
120			export function timeProcess(seconds: number) {
121			return formatDuration(intervalToDuration({ start: 0, end: seconds * 1000 }));
122			}
123
124
125			/**
126			* @deprecated
127			*/
128			export function timeProcessOld(seconds: number) {
129			let interval = seconds / (60 * 60 * 24 * 30 * 365);
130
131			if (interval >= 1) {
132			return Math.floor(interval) + " year" + (Math.floor(interval) === 1 ? '' : 's');
133			}
134
135			interval = seconds / (60 * 60 * 24 * 30);
136
137			if (interval >= 1) {
138			return Math.floor(interval) + " month" + (Math.floor(interval) === 1 ? '' : 's');
139			}
140
141			interval = seconds / (60 * 60 * 24);
142
143			if (interval >= 1) {
144			return Math.floor(interval) + " day" + (Math.floor(interval) === 1 ? '' : 's');
145			}
146
147			interval = seconds / (60 * 60);
148
149			if (interval >= 1) {
150			return Math.floor(interval) + " hour" + (Math.floor(interval) === 1 ? '' : 's');
151			}
152
153			interval = seconds / 60;
154
155			if (interval >= 1) {
156			return Math.floor(interval) + " minute" + (Math.floor(interval) === 1 ? '' : 's');
157			}
158
159			interval = seconds;
160
161			return Math.floor(interval) + " second" + (Math.floor(interval) === 1 ? '' : 's');
162			}
163
164			export function escapeRegex(string: string) {
165			return string.replace(/[-\/\\^$*+?.()\|[\]{}]/g, '\\$&');
166			}
167
168			export function timeSince(date: number) {
169			// const seconds = Math.floor((Date.now() - date) / 1000);
170			// return timeProcess(seconds) + ' ago';
171			return formatDistanceToNowStrict(new Date(date), { addSuffix: true });
172			}
173
174			export async function download(url: string, path: string, headers?: AxiosRequestHeaders) {
175			const writer = fs.createWriteStream(path);
176
177			const response = await Axios({
178			url,
179			method: 'GET',
180			responseType: 'stream',
181			headers
182			});
183
184			response.data.pipe(writer);
185
186			return new Promise((resolve, reject) => {
187			if (response.status !== 200) {
188			reject();
189			}
190
191			writer.on('finish', resolve);
192			writer.on('error', reject);
193			});
194			}
195
196			export async function deleteFile(path: string) {
197			fs.rm(path, (err: any) => {
198			if (err) {
199			throw new Error(err);
200			}
201			});
202			}
203
204			export function random(arr: Array <any>) {
205			let index = Math.floor(Math.random() * arr.length);
206			return arr[index];
207			}
208
209			export function fill(length: number, string: string, token: string = ' ') {
210			let safe = 0;
211
212			if (length < string.length)
213			return string;
214
215			const diff = length - string.length;
216
217			for (let i = 1; i <= diff; i++, safe++) {
218			if (safe >= 500)
219			break;
220
221			string += ' ';
222			}
223
224			return string;
225			}
226
227			export function green(string: string) {
228			return '\u001b[1;32m' + string + '\u001b[0m';
229			}
230
231			export function yellow(string: string) {
232			return '\u001b[1;33m' + string + '\u001b[0m';
233			}
234
235			export function red(string: string) {
236			return '\u001b[1;31m' + string + '\u001b[0m';
237			}