discord-html-transcripts-fix 1.0.0 → 1.2.0

package/dist/types.js

@@ -13,5 +13,6 @@ var ExportReturnType;
     ExportReturnType["Buffer"] = "buffer";
     ExportReturnType["String"] = "string";
     ExportReturnType["Attachment"] = "attachment";
+    ExportReturnType["Stream"] = "stream";
 })(ExportReturnType || (exports.ExportReturnType = ExportReturnType = {}));
 //# sourceMappingURL=types.js.map

package/dist/utils/buildProfiles.js

@@ -1,42 +1,154 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.buildProfiles = buildProfiles;
+exports.buildExtendedContext = buildExtendedContext;
+exports.buildAllContext = buildAllContext;
 const discord_js_1 = require("discord.js");
+
+/**
+ * Backwards compatible: returns the profile dict used by <discord-mention> web components.
+ */
 async function buildProfiles(messages) {
+    const { profiles } = await buildAllContext(messages, null);
+    return profiles;
+}
+
+/**
+ * Backwards compatible: returns extended users/roles/channels.
+ */
+async function buildExtendedContext(messages, channel) {
+    const { users, roles, channels } = await buildAllContext(messages, channel);
+    return { users, roles, channels };
+}
+
+/**
+ * Single-pass collector — builds the legacy `profiles` dict AND the extended
+ * users/roles/channels dicts in one go (saves an O(n) pass for big tickets).
+ */
+async function buildAllContext(messages, channel) {
     const profiles = {};
-    // loop through messages
+    const users = {};
+    const roles = {};
+    const channels = {};
+
+    // FIX: previous version was `!channel.isDMBased && !channel.isDMBased()`
+    // which is always false (isDMBased is a truthy function reference).
+    const guild = channel && typeof channel.isDMBased === 'function' && !channel.isDMBased()
+        ? channel.guild
+        : null;
+
+    const seenUserIds = new Set();
+    const seenRoleIds = new Set();
+    const seenChannelIds = new Set();
+
+    const includeRoleFromGuild = (roleId) => {
+        if (!guild || !roleId || seenRoleIds.has(roleId)) return;
+        const role = guild.roles.cache.get(roleId);
+        if (!role) return;
+        seenRoleIds.add(roleId);
+        roles[roleId] = {
+            id: role.id,
+            name: role.name,
+            color: role.hexColor && role.hexColor !== '#000000' ? role.hexColor : null,
+            position: role.position,
+            hoist: role.hoist,
+            mentionable: role.mentionable,
+            memberCount: typeof role.members?.size === 'number' ? role.members.size : null,
+            managed: role.managed,
+        };
+    };
+
+    const includeChannelFromGuild = (channelId) => {
+        if (!guild || !channelId || seenChannelIds.has(channelId)) return;
+        const ch = guild.channels.cache.get(channelId);
+        if (!ch) return;
+        seenChannelIds.add(channelId);
+        channels[channelId] = {
+            id: ch.id,
+            name: ch.name,
+            type: ch.type,
+            parent: ch.parent ? ch.parent.name : null,
+            topic: 'topic' in ch && typeof ch.topic === 'string' ? ch.topic : null,
+            nsfw: 'nsfw' in ch ? !!ch.nsfw : false,
+        };
+    };
+
+    const includeUser = (member, author) => {
+        if (!author) return;
+        const id = author.id;
+        if (seenUserIds.has(id)) return;
+        seenUserIds.add(id);
+
+        // legacy profile (used by <discord-mention>)
+        profiles[id] = {
+            author: member?.nickname ?? author.displayName ?? author.username,
+            avatar: member?.displayAvatarURL?.({ size: 64 }) ?? author.displayAvatarURL?.({ size: 64 }),
+            roleColor: member?.displayHexColor,
+            roleIcon: member?.roles?.icon?.iconURL?.() ?? undefined,
+            roleName: member?.roles?.hoist?.name ?? undefined,
+            bot: !!author.bot,
+            verified: author.flags?.has?.(discord_js_1.UserFlags.VerifiedBot) ?? false,
+        };
+
+        // extended record (used by popup)
+        const memberRoles = member && member.roles && member.roles.cache
+            ? Array.from(member.roles.cache.values())
+                .filter((r) => r.id !== guild?.id)
+                .sort((a, b) => b.position - a.position)
+            : [];
+        for (const r of memberRoles) includeRoleFromGuild(r.id);
+
+        users[id] = {
+            id,
+            username: author.username,
+            displayName: member?.nickname || author.displayName || author.username,
+            globalName: author.globalName || null,
+            avatar: member?.displayAvatarURL?.({ size: 128 }) || author.displayAvatarURL?.({ size: 128 }) || null,
+            bannerColor: member?.displayHexColor && member.displayHexColor !== '#000000' ? member.displayHexColor : null,
+            bot: !!author.bot,
+            system: !!author.system,
+            joinedAt: member?.joinedAt ? member.joinedAt.toISOString() : null,
+            createdAt: author.createdAt ? author.createdAt.toISOString() : null,
+            roles: memberRoles.map((r) => ({
+                id: r.id,
+                name: r.name,
+                color: r.hexColor && r.hexColor !== '#000000' ? r.hexColor : null,
+                position: r.position,
+            })),
+            highestRole: memberRoles[0]
+                ? { name: memberRoles[0].name, color: memberRoles[0].hexColor }
+                : null,
+        };
+    };
+
     for (const message of messages) {
-        // add all users
-        const author = message.author;
-        if (!profiles[author.id]) {
-            // add profile
-            profiles[author.id] = buildProfile(message.member, author);
+        if (message.author) includeUser(message.member, message.author);
+        if (message.interaction?.user) includeUser(null, message.interaction.user);
+        if (message.thread?.lastMessage?.author) {
+            includeUser(message.thread.lastMessage.member, message.thread.lastMessage.author);
         }
-        // add interaction users
-        if (message.interaction) {
-            const user = message.interaction.user;
-            if (!profiles[user.id]) {
-                profiles[user.id] = buildProfile(null, user);
+        if (message.mentions) {
+            if (message.mentions.users) {
+                for (const [uid, u] of message.mentions.users) {
+                    const member = message.mentions.members?.get?.(uid) || null;
+                    includeUser(member, u);
+                }
+            }
+            if (message.mentions.roles) {
+                for (const [rid] of message.mentions.roles) includeRoleFromGuild(rid);
+            }
+            if (message.mentions.channels) {
+                for (const [cid] of message.mentions.channels) includeChannelFromGuild(cid);
             }
         }
-        // threads
-        if (message.thread && message.thread.lastMessage) {
-            profiles[message.thread.lastMessage.author.id] = buildProfile(message.thread.lastMessage.member, message.thread.lastMessage.author);
+        // also include authors of forwarded snapshots
+        if (Array.isArray(message.messageSnapshots)) {
+            for (const snap of message.messageSnapshots) {
+                if (snap?.author) includeUser(null, snap.author);
+            }
         }
     }
-    // return as a JSON
-    return profiles;
-}
-function buildProfile(member, author) {
-    var _a, _b, _c, _d, _e, _f, _g, _h;
-    return {
-        author: (_b = (_a = member === null || member === void 0 ? void 0 : member.nickname) !== null && _a !== void 0 ? _a : author.displayName) !== null && _b !== void 0 ? _b : author.username,
-        avatar: (_c = member === null || member === void 0 ? void 0 : member.displayAvatarURL({ size: 64 })) !== null && _c !== void 0 ? _c : author.displayAvatarURL({ size: 64 }),
-        roleColor: member === null || member === void 0 ? void 0 : member.displayHexColor,
-        roleIcon: (_e = (_d = member === null || member === void 0 ? void 0 : member.roles.icon) === null || _d === void 0 ? void 0 : _d.iconURL()) !== null && _e !== void 0 ? _e : undefined,
-        roleName: (_g = (_f = member === null || member === void 0 ? void 0 : member.roles.hoist) === null || _f === void 0 ? void 0 : _f.name) !== null && _g !== void 0 ? _g : undefined,
-        bot: author.bot,
-        verified: (_h = author.flags) === null || _h === void 0 ? void 0 : _h.has(discord_js_1.UserFlags.VerifiedBot),
-    };
+
+    return { profiles, users, roles, channels };
 }
-//# sourceMappingURL=buildProfiles.js.map
+//# sourceMappingURL=buildProfiles.js.map

package/dist/utils/utils.js

@@ -7,38 +7,49 @@ exports.isDefined = isDefined;
 exports.formatBytes = formatBytes;
 exports.parseDiscordEmoji = parseDiscordEmoji;
 exports.streamToString = streamToString;
+exports.safeJsonForScript = safeJsonForScript;
+exports.safeHref = safeHref;
+exports.safeColor = safeColor;
+exports.safeImageMime = safeImageMime;
+exports.escapeHtml = escapeHtml;
 const twemoji_1 = __importDefault(require("twemoji"));
+
 function isDefined(value) {
     return value !== undefined && value !== null;
 }
-/**
- * Returns a tuple of [bytes (i.e. 1024), bytesUnit (i.e. KB)]
- * @param bytes
- * @param decimals
- * @returns
- */
+
 function formatBytes(bytes, decimals = 2) {
-    if (bytes === 0)
-        return [0, 'Bytes'];
+    if (!bytes || !Number.isFinite(bytes)) return [0, 'Bytes'];
     const k = 1024;
     const dm = decimals < 0 ? 0 : decimals;
     const sizes = ['Bytes', 'KB', 'MB', 'GB', 'TB', 'PB', 'EB', 'ZB', 'YB'];
-    const i = Math.floor(Math.log(bytes) / Math.log(k));
+    const i = Math.min(sizes.length - 1, Math.max(0, Math.floor(Math.log(bytes) / Math.log(k))));
     return [parseFloat((bytes / Math.pow(k, i)).toFixed(dm)), sizes[i]];
 }
+
+const emojiCache = new Map();
 function parseDiscordEmoji(emoji) {
+    if (!emoji) return '';
+    const key = emoji.id || emoji.name;
+    if (!key) return '';
+    if (emojiCache.has(key)) return emojiCache.get(key);
+    let url = '';
     if (emoji.id) {
-        return `https://cdn.discordapp.com/emojis/${emoji.id}.${emoji.animated ? 'gif' : 'png'}`;
+        url = `https://cdn.discordapp.com/emojis/${emoji.id}.${emoji.animated ? 'gif' : 'png'}`;
+    } else if (typeof emoji.name === 'string' && emoji.name) {
+        try {
+            const codepoints = twemoji_1.default.convert
+                .toCodePoint(emoji.name.indexOf(String.fromCharCode(0x200d)) < 0 ? emoji.name.replace(/️/g, '') : emoji.name)
+                .toLowerCase();
+            url = `https://cdnjs.cloudflare.com/ajax/libs/twemoji/14.0.2/svg/${codepoints}.svg`;
+        } catch (_e) {
+            url = '';
+        }
     }
-    const codepoints = twemoji_1.default.convert
-        .toCodePoint(emoji.name.indexOf(String.fromCharCode(0x200d)) < 0 ? emoji.name.replace(/\uFE0F/g, '') : emoji.name)
-        .toLowerCase();
-    return `https://cdnjs.cloudflare.com/ajax/libs/twemoji/14.0.2/svg/${codepoints}.svg`;
+    emojiCache.set(key, url);
+    return url;
 }
-/**
- * Converts a stream to a string
- * @param stream - The stream to convert
- */
+
 function streamToString(stream) {
     const chunks = [];
     return new Promise((resolve, reject) => {
@@ -47,4 +58,48 @@ function streamToString(stream) {
         stream.on('end', () => resolve(Buffer.concat(chunks).toString('utf8')));
     });
 }
-//# sourceMappingURL=utils.js.map
+
+/**
+ * JSON.stringify safe to embed in <script>. Escapes <, >, &, U+2028, U+2029.
+ */
+function safeJsonForScript(obj) {
+    return JSON.stringify(obj)
+        .replace(/</g, '\\u003c')
+        .replace(/>/g, '\\u003e')
+        .replace(/&/g, '\\u0026')
+        .replace(/\u2028/g, '\\u2028')
+        .replace(/\u2029/g, '\\u2029');
+}
+
+function safeHref(url) {
+    if (typeof url !== 'string' || !url) return '#';
+    try {
+        const u = new URL(url, 'https://example.invalid');
+        const proto = u.protocol.toLowerCase();
+        if (proto === 'http:' || proto === 'https:' || proto === 'mailto:' || proto === 'discord:') return url;
+        return '#';
+    } catch (_e) {
+        return '#';
+    }
+}
+
+function safeColor(c, fallback = '#5865F2') {
+    if (typeof c !== 'string') return fallback;
+    return /^#[0-9a-fA-F]{3,8}$/.test(c) ? c : fallback;
+}
+
+const ALLOWED_IMAGE_MIME = new Set([
+    'image/png', 'image/jpeg', 'image/jpg', 'image/gif', 'image/webp', 'image/avif', 'image/svg+xml',
+]);
+function safeImageMime(mime, fallback = 'image/png') {
+    if (typeof mime !== 'string') return fallback;
+    const m = mime.split(';')[0].trim().toLowerCase();
+    return ALLOWED_IMAGE_MIME.has(m) ? m : fallback;
+}
+
+function escapeHtml(str) {
+    return String(str == null ? '' : str).replace(/[&<>"']/g, (c) => ({
+        '&': '&amp;', '<': '&lt;', '>': '&gt;', '"': '&quot;', "'": '&#39;'
+    }[c]));
+}
+//# sourceMappingURL=utils.js.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "discord-html-transcripts-fix",
-  "version": "1.0.0",
+  "version": "1.2.0",
   "description": "A nicely formatted html transcript generator for discord.js. Bugfix fork with support for the latest discord.js and Components v2.",
   "main": "dist/index.js",
   "types": "./dist/index.d.ts",
@@ -19,7 +19,6 @@
     "@types/react-dom": "^19.2.3",
     "@typescript-eslint/eslint-plugin": "^8.50.0",
     "@typescript-eslint/parser": "^8.50.0",
-    "debug": "^4.4.3",
     "discord.js": "^14.25.1",
     "dotenv": "^17.2.3",
     "eslint": "^9.39.2",
@@ -34,16 +33,17 @@
   "dependencies": {
     "@lit-labs/ssr": "^3.3.1",
     "@skyra/discord-components-core": "^4.0.2",
+    "debug": "^4.4.3",
     "discord-markdown-parser": "~1.3.0",
     "highlight.js": "^11.11.1",
+    "react": "^19.2.3",
+    "react-dom": "^19.2.3",
     "simple-markdown": "^0.7.3",
     "twemoji": "^14.0.2",
     "undici": "^7.16.0"
   },
   "peerDependencies": {
-    "discord.js": "^14.0.0 || ^15.0.0",
-    "react": "^19.2.3",
-    "react-dom": "^19.2.3"
+    "discord.js": "^14.0.0 || ^15.0.0"
   },
   "scripts": {
     "test": "echo \"Error: no test specified\" && exit 1",