feat: save avatars locally

src/lib/tickets/archiver.js

@@ -1,4 +1,5 @@
 const { reusable } = require('../threads');
+const { getAvatarData } = require('../users');
 
 
 /**
@@ -35,7 +36,8 @@ module.exports = class TicketArchiver {
 		const members = new Set(message.mentions.members.values());
 		const roles = new Set(message.mentions.roles.values());
 
-		const worker = await reusable('crypto');
+		const cryptoWorker = await reusable('crypto');
+		const usersWorker = await reusable('users');
 
 		try {
 			const queries = [];
@@ -71,13 +73,18 @@ module.exports = class TicketArchiver {
 			}
 
 			for (const member of members) {
+				const avatar = getAvatarData(member);
+
+				const savedAvatarFilename = usersWorker.saveAvatar(avatar);
+				if(!savedAvatarFilename) this.client.log.warn(`Couldn't save user avatar at: ${avatar.url}`);
+
 				const data = {
-					avatar: member.avatar || member.user.avatar, // TODO: save avatar in user/avatars/
+					avatar: avatar.hash,
 					bot: member.user.bot,
 					discriminator: member.user.discriminator,
-					displayName: member.displayName ? await worker.encrypt(member.displayName) : null,
+					displayName: member.displayName ? await cryptoWorker.encrypt(member.displayName) : null,
 					roleId: !!member && hoistedRole(member).id,
-					username: await worker.encrypt(member.user.username),
+					username: await cryptoWorker.encrypt(member.user.username),
 				};
 				queries.push(
 					this.client.prisma.archivedUser.upsert({
@@ -120,7 +127,7 @@ module.exports = class TicketArchiver {
 			}
 
 			const data = {
-				content: await worker.encrypt(
+				content: await cryptoWorker.encrypt(
 					JSON.stringify({
 						attachments: [...message.attachments.values()],
 						components: [...message.components.values()],
@@ -150,7 +157,8 @@ module.exports = class TicketArchiver {
 
 			return await this.client.prisma.$transaction(queries);
 		} finally {
-			await worker.terminate();
+			await cryptoWorker.terminate();
+			await usersWorker.terminate();
 		}
 	}
 };

src/lib/users.js

@@ -70,3 +70,37 @@ module.exports.getPrivilegeLevel = async member => {
 	else if (await this.isStaff(member.guild, member.id)) return 1;
 	else return 0;
 };
+
+/**
+ *
+ * @param {import("discord.js").GuildMember} member
+ * @returns {{hash: string, url: string} | false}
+ */
+module.exports.getAvatarData = member => {
+	if (!member?.user) return false;
+
+	const cdnBaseUrl = 'https://cdn.discordapp.com';
+
+	const {
+		id: userId,
+		avatar: userAvatar,
+	} = member.user;
+	const { id: guildId } = member.guild;
+	const memberAvatar = member.avatar;
+
+	const avatarHash = memberAvatar || userAvatar;
+	if (!avatarHash) return false;
+
+	const isAnimated = avatarHash.startsWith('a_');
+	const ext = isAnimated ? '.gif' : '.png';
+
+	const url = memberAvatar
+		? `${cdnBaseUrl}/guilds/${guildId}/users/${userId}/avatars/${avatarHash}${ext}`
+		: `${cdnBaseUrl}/avatars/${userId}/${avatarHash}${ext}`;
+
+	return {
+		hash: avatarHash,
+		isAnimated,
+		url,
+	};
+};

src/lib/workers/users.js

@@ -0,0 +1,38 @@
+const { expose } = require('threads/worker');
+const fs = require('fs').promises;
+const path = require('path');
+
+expose({
+	async saveAvatar(avatar) {
+		const avatarDir = path.join('user', 'avatars');
+
+		const ext = path.extname(avatar.url);
+		const filename = `${avatar.hash}${ext}`;
+		const filePath = path.join(avatarDir, filename);
+
+		try {
+			await fs.access(filePath);
+			return filename; // Avatar already saved
+		  } catch (err) {
+			if (err.code !== 'ENOENT') {
+			  return false;
+			}
+		  }
+
+		const res = await fetch(avatar.url);
+		if (!res.ok) {
+		  return false;
+		}
+
+		const arrayBuffer = await res.arrayBuffer();
+		const buffer = Buffer.from(arrayBuffer);
+
+		try {
+			await fs.writeFile(filePath, buffer);
+		} catch {
+			return false;
+		}
+
+		return filename;
+	},
+});