@mulingai-npm/redis 2.3.1 → 2.4.0

package/dist/managers/mulingstream-chunk-manager.js

@@ -1,242 +1,242 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.MulingstreamChunkManager = void 0;
-const EXPIRATION = 12 * 60 * 60; // 12 hours in seconds
-const ROOM_ARRAY_LENGTH = 50; // keep only the last 5 elements
-class MulingstreamChunkManager {
-    constructor(redisClient) {
-        this.redisClient = redisClient;
-    }
-    /**
-     * Initializes a room in Redis as an empty JSON array.
-     * If the key [roomId] already exists, we do NOT overwrite it.
-     * Returns true if room was created, false if room already existed.
-     */
-    async initRoom(roomId) {
-        // Check if the key already exists (JSON.GET)
-        const isRoomExisting = await this.redisClient.jsonGet(`[${roomId}]`, '.');
-        if (isRoomExisting !== null) {
-            return false;
-        }
-        // Create an empty array at the root
-        await this.redisClient.jsonSet(`[${roomId}]`, '.', []);
-        await this.redisClient.expire(`[${roomId}]`, EXPIRATION);
-        return true;
-    }
-    /**
-     * Returns all rooms. This is naive if you store many other keys in Redis,
-     * because we search keys for pattern "[*]".
-     * Adjust as needed if you have a separate naming prefix.
-     */
-    async getRooms() {
-        // e.g., if we store everything as [foo], [bar], etc., we do:
-        const rooms = await this.redisClient.keys('\\[*\\]');
-        // This returns the raw keys, e.g. ['[myRoom]', '[anotherRoom]']
-        // We might want to strip off the brackets:
-        return rooms.map((k) => k.replace(/^\[|\]$/g, ''));
-    }
-    /**
-     * Returns the entire array of chunks for the given roomId,
-     * or null if the room doesn't exist in Redis.
-     */
-    async getMulingstreamChunksByRoom(roomId) {
-        // JSON.GET [roomId] .
-        const chunks = await this.redisClient.jsonGet(`[${roomId}]`, '.');
-        if (chunks === null) {
-            return null;
-        }
-        return chunks;
-    }
-    async getRoomById(roomId) {
-        return this.getMulingstreamChunksByRoom(roomId);
-    }
-    async addMulingstreamChunk(params) {
-        var _a;
-        const { roomId, chunkNumber, language, start, end, duration, isFirst, isLast, theme, sttProviders, targetLanguages, shortCodeTargetLanguages } = params;
-        const audioChunk = {
-            start,
-            end,
-            duration,
-            isFirst,
-            isLast,
-            theme,
-            processingStart: Date.now()
-        };
-        const stt = {};
-        for (const sttProvider of sttProviders) {
-            stt[sttProvider] = {
-                transcription: '',
-                status: 'INIT'
-            };
-        }
-        const translation = {};
-        const tts = {};
-        for (const lang of shortCodeTargetLanguages) {
-            translation[lang] = {
-                translation: '',
-                status: 'INIT'
-            };
-            tts[lang] = {
-                ttsAudioPath: '',
-                status: 'INIT',
-                isEmitted: false
-            };
-        }
-        const newChunk = {
-            roomId,
-            chunkNumber,
-            language,
-            sttProviders,
-            targetLanguages,
-            shortCodeTargetLanguages,
-            finalTranscription: '',
-            sttStatus: 'INIT',
-            audioChunk,
-            stt,
-            translation,
-            tts
-        };
-        const chunks = (_a = (await this.redisClient.jsonGet(`[${roomId}]`, '.'))) !== null && _a !== void 0 ? _a : [];
-        const idx = chunks.findIndex((c) => c.chunkNumber === chunkNumber);
-        if (idx !== -1) {
-            chunks[idx] = newChunk;
-        }
-        else {
-            chunks.push(newChunk);
-            if (chunks.length > ROOM_ARRAY_LENGTH) {
-                chunks.shift();
-            }
-        }
-        await this.redisClient.jsonSet(`[${roomId}]`, '.', chunks);
-        await this.redisClient.expire(`[${roomId}]`, EXPIRATION);
-    }
-    /**
-     * Given roomId and chunkNumber, return the single chunk from the array
-     * or null if not found.
-     */
-    async getMulingstreamChunkById(roomId, chunkNumber) {
-        // Retrieve the entire array
-        const chunks = await this.getMulingstreamChunksByRoom(roomId);
-        if (!chunks)
-            return null;
-        // Find by chunkNumber
-        const chunk = chunks.find((c) => c.chunkNumber === chunkNumber);
-        return chunk || null;
-    }
-    /**
-     * Update STT fields for a given chunk.
-     * If transcription or sttStatus is null, skip that field.
-     */
-    async updateStt(roomId, chunkNumber, sttProvider, options) {
-        const chunks = await this.getMulingstreamChunksByRoom(roomId);
-        if (!chunks)
-            return false;
-        // Find the chunk
-        const chunkIndex = chunks.findIndex((c) => c.chunkNumber === chunkNumber);
-        if (chunkIndex === -1)
-            return false;
-        const chunk = chunks[chunkIndex];
-        // If this stt provider doesn't exist, ignore
-        if (!chunk.stt[sttProvider])
-            return false;
-        // Update
-        if (options.transcription != null) {
-            chunk.stt[sttProvider].transcription = options.transcription;
-        }
-        if (options.sttStatus != null) {
-            chunk.stt[sttProvider].status = options.sttStatus;
-        }
-        // Write back
-        chunks[chunkIndex] = chunk;
-        await this.redisClient.jsonSet(`[${roomId}]`, '.', chunks);
-        return true;
-    }
-    /**
-     * Update the final transcription field for a chunk.
-     */
-    async updateFinalTranscription(roomId, chunkNumber, transcription, sttStatus) {
-        const chunks = await this.getMulingstreamChunksByRoom(roomId);
-        if (!chunks)
-            return false;
-        // Find the chunk
-        const chunkIndex = chunks.findIndex((c) => c.chunkNumber === chunkNumber);
-        if (chunkIndex === -1)
-            return false;
-        chunks[chunkIndex].finalTranscription = transcription;
-        chunks[chunkIndex].sttStatus = sttStatus;
-        // Write back
-        await this.redisClient.jsonSet(`[${roomId}]`, '.', chunks);
-        return true;
-    }
-    /**
-     * Discards all post-STT steps for a given chunk:
-     * sets all translation[].status & tts[].status to "DISCARDED".
-     */
-    async discardPostStt(roomId, chunkNumber) {
-        const chunks = await this.getMulingstreamChunksByRoom(roomId);
-        if (!chunks)
-            return false;
-        // Find chunk
-        const chunkIndex = chunks.findIndex((c) => c.chunkNumber === chunkNumber);
-        if (chunkIndex === -1)
-            return false;
-        // Discard translation & TTS statuses
-        const chunk = chunks[chunkIndex];
-        for (const lang of Object.keys(chunk.translation)) {
-            chunk.translation[lang].status = 'DISCARDED';
-        }
-        for (const lang of Object.keys(chunk.tts)) {
-            chunk.tts[lang].status = 'DISCARDED';
-        }
-        chunks[chunkIndex] = chunk;
-        await this.redisClient.jsonSet(`[${roomId}]`, '.', chunks);
-        return true;
-    }
-    /**
-     * Discards a specific language in both translation and tts for a chunk.
-     */
-    async discardLanguage(roomId, chunkNumber, language) {
-        const chunks = await this.getMulingstreamChunksByRoom(roomId);
-        if (!chunks)
-            return false;
-        const chunkIndex = chunks.findIndex((c) => c.chunkNumber === chunkNumber);
-        if (chunkIndex === -1)
-            return false;
-        const chunk = chunks[chunkIndex];
-        if (chunk.translation[language]) {
-            chunk.translation[language].status = 'DISCARDED';
-        }
-        if (chunk.tts[language]) {
-            chunk.tts[language].status = 'DISCARDED';
-        }
-        chunks[chunkIndex] = chunk;
-        await this.redisClient.jsonSet(`[${roomId}]`, '.', chunks);
-        return true;
-    }
-    async updateTranslation(roomId, chunkNumber, language, options) {
-        // Fetch all chunks for this room
-        const chunks = await this.getMulingstreamChunksByRoom(roomId);
-        if (!chunks)
-            return false;
-        // Locate the target chunk
-        const chunkIndex = chunks.findIndex((c) => c.chunkNumber === chunkNumber);
-        if (chunkIndex === -1)
-            return false;
-        const chunk = chunks[chunkIndex];
-        // Make sure the requested language exists in this chunk
-        if (!chunk.translation[language]) {
-            return false;
-        }
-        if (options.translation !== undefined) {
-            chunk.translation[language].translation = options.translation;
-        }
-        if (options.status !== undefined) {
-            chunk.translation[language].status = options.status;
-        }
-        // Persist the whole array back to Redis
-        chunks[chunkIndex] = chunk;
-        await this.redisClient.jsonSet(`[${roomId}]`, '.', chunks);
-        return true;
-    }
-}
-exports.MulingstreamChunkManager = MulingstreamChunkManager;
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MulingstreamChunkManager = void 0;
+const EXPIRATION = 12 * 60 * 60; // 12 hours in seconds
+const ROOM_ARRAY_LENGTH = 50; // keep only the last 5 elements
+class MulingstreamChunkManager {
+    constructor(redisClient) {
+        this.redisClient = redisClient;
+    }
+    /**
+     * Initializes a room in Redis as an empty JSON array.
+     * If the key [roomId] already exists, we do NOT overwrite it.
+     * Returns true if room was created, false if room already existed.
+     */
+    async initRoom(roomId) {
+        // Check if the key already exists (JSON.GET)
+        const isRoomExisting = await this.redisClient.jsonGet(`[${roomId}]`, '.');
+        if (isRoomExisting !== null) {
+            return false;
+        }
+        // Create an empty array at the root
+        await this.redisClient.jsonSet(`[${roomId}]`, '.', []);
+        await this.redisClient.expire(`[${roomId}]`, EXPIRATION);
+        return true;
+    }
+    /**
+     * Returns all rooms. This is naive if you store many other keys in Redis,
+     * because we search keys for pattern "[*]".
+     * Adjust as needed if you have a separate naming prefix.
+     */
+    async getRooms() {
+        // e.g., if we store everything as [foo], [bar], etc., we do:
+        const rooms = await this.redisClient.keys('\\[*\\]');
+        // This returns the raw keys, e.g. ['[myRoom]', '[anotherRoom]']
+        // We might want to strip off the brackets:
+        return rooms.map((k) => k.replace(/^\[|\]$/g, ''));
+    }
+    /**
+     * Returns the entire array of chunks for the given roomId,
+     * or null if the room doesn't exist in Redis.
+     */
+    async getMulingstreamChunksByRoom(roomId) {
+        // JSON.GET [roomId] .
+        const chunks = await this.redisClient.jsonGet(`[${roomId}]`, '.');
+        if (chunks === null) {
+            return null;
+        }
+        return chunks;
+    }
+    async getRoomById(roomId) {
+        return this.getMulingstreamChunksByRoom(roomId);
+    }
+    async addMulingstreamChunk(params) {
+        var _a;
+        const { roomId, chunkNumber, language, start, end, duration, isFirst, isLast, theme, sttProviders, targetLanguages, shortCodeTargetLanguages } = params;
+        const audioChunk = {
+            start,
+            end,
+            duration,
+            isFirst,
+            isLast,
+            theme,
+            processingStart: Date.now()
+        };
+        const stt = {};
+        for (const sttProvider of sttProviders) {
+            stt[sttProvider] = {
+                transcription: '',
+                status: 'INIT'
+            };
+        }
+        const translation = {};
+        const tts = {};
+        for (const lang of shortCodeTargetLanguages) {
+            translation[lang] = {
+                translation: '',
+                status: 'INIT'
+            };
+            tts[lang] = {
+                ttsAudioPath: '',
+                status: 'INIT',
+                isEmitted: false
+            };
+        }
+        const newChunk = {
+            roomId,
+            chunkNumber,
+            language,
+            sttProviders,
+            targetLanguages,
+            shortCodeTargetLanguages,
+            finalTranscription: '',
+            sttStatus: 'INIT',
+            audioChunk,
+            stt,
+            translation,
+            tts
+        };
+        const chunks = (_a = (await this.redisClient.jsonGet(`[${roomId}]`, '.'))) !== null && _a !== void 0 ? _a : [];
+        const idx = chunks.findIndex((c) => c.chunkNumber === chunkNumber);
+        if (idx !== -1) {
+            chunks[idx] = newChunk;
+        }
+        else {
+            chunks.push(newChunk);
+            if (chunks.length > ROOM_ARRAY_LENGTH) {
+                chunks.shift();
+            }
+        }
+        await this.redisClient.jsonSet(`[${roomId}]`, '.', chunks);
+        await this.redisClient.expire(`[${roomId}]`, EXPIRATION);
+    }
+    /**
+     * Given roomId and chunkNumber, return the single chunk from the array
+     * or null if not found.
+     */
+    async getMulingstreamChunkById(roomId, chunkNumber) {
+        // Retrieve the entire array
+        const chunks = await this.getMulingstreamChunksByRoom(roomId);
+        if (!chunks)
+            return null;
+        // Find by chunkNumber
+        const chunk = chunks.find((c) => c.chunkNumber === chunkNumber);
+        return chunk || null;
+    }
+    /**
+     * Update STT fields for a given chunk.
+     * If transcription or sttStatus is null, skip that field.
+     */
+    async updateStt(roomId, chunkNumber, sttProvider, options) {
+        const chunks = await this.getMulingstreamChunksByRoom(roomId);
+        if (!chunks)
+            return false;
+        // Find the chunk
+        const chunkIndex = chunks.findIndex((c) => c.chunkNumber === chunkNumber);
+        if (chunkIndex === -1)
+            return false;
+        const chunk = chunks[chunkIndex];
+        // If this stt provider doesn't exist, ignore
+        if (!chunk.stt[sttProvider])
+            return false;
+        // Update
+        if (options.transcription != null) {
+            chunk.stt[sttProvider].transcription = options.transcription;
+        }
+        if (options.sttStatus != null) {
+            chunk.stt[sttProvider].status = options.sttStatus;
+        }
+        // Write back
+        chunks[chunkIndex] = chunk;
+        await this.redisClient.jsonSet(`[${roomId}]`, '.', chunks);
+        return true;
+    }
+    /**
+     * Update the final transcription field for a chunk.
+     */
+    async updateFinalTranscription(roomId, chunkNumber, transcription, sttStatus) {
+        const chunks = await this.getMulingstreamChunksByRoom(roomId);
+        if (!chunks)
+            return false;
+        // Find the chunk
+        const chunkIndex = chunks.findIndex((c) => c.chunkNumber === chunkNumber);
+        if (chunkIndex === -1)
+            return false;
+        chunks[chunkIndex].finalTranscription = transcription;
+        chunks[chunkIndex].sttStatus = sttStatus;
+        // Write back
+        await this.redisClient.jsonSet(`[${roomId}]`, '.', chunks);
+        return true;
+    }
+    /**
+     * Discards all post-STT steps for a given chunk:
+     * sets all translation[].status & tts[].status to "DISCARDED".
+     */
+    async discardPostStt(roomId, chunkNumber) {
+        const chunks = await this.getMulingstreamChunksByRoom(roomId);
+        if (!chunks)
+            return false;
+        // Find chunk
+        const chunkIndex = chunks.findIndex((c) => c.chunkNumber === chunkNumber);
+        if (chunkIndex === -1)
+            return false;
+        // Discard translation & TTS statuses
+        const chunk = chunks[chunkIndex];
+        for (const lang of Object.keys(chunk.translation)) {
+            chunk.translation[lang].status = 'DISCARDED';
+        }
+        for (const lang of Object.keys(chunk.tts)) {
+            chunk.tts[lang].status = 'DISCARDED';
+        }
+        chunks[chunkIndex] = chunk;
+        await this.redisClient.jsonSet(`[${roomId}]`, '.', chunks);
+        return true;
+    }
+    /**
+     * Discards a specific language in both translation and tts for a chunk.
+     */
+    async discardLanguage(roomId, chunkNumber, language) {
+        const chunks = await this.getMulingstreamChunksByRoom(roomId);
+        if (!chunks)
+            return false;
+        const chunkIndex = chunks.findIndex((c) => c.chunkNumber === chunkNumber);
+        if (chunkIndex === -1)
+            return false;
+        const chunk = chunks[chunkIndex];
+        if (chunk.translation[language]) {
+            chunk.translation[language].status = 'DISCARDED';
+        }
+        if (chunk.tts[language]) {
+            chunk.tts[language].status = 'DISCARDED';
+        }
+        chunks[chunkIndex] = chunk;
+        await this.redisClient.jsonSet(`[${roomId}]`, '.', chunks);
+        return true;
+    }
+    async updateTranslation(roomId, chunkNumber, language, options) {
+        // Fetch all chunks for this room
+        const chunks = await this.getMulingstreamChunksByRoom(roomId);
+        if (!chunks)
+            return false;
+        // Locate the target chunk
+        const chunkIndex = chunks.findIndex((c) => c.chunkNumber === chunkNumber);
+        if (chunkIndex === -1)
+            return false;
+        const chunk = chunks[chunkIndex];
+        // Make sure the requested language exists in this chunk
+        if (!chunk.translation[language]) {
+            return false;
+        }
+        if (options.translation !== undefined) {
+            chunk.translation[language].translation = options.translation;
+        }
+        if (options.status !== undefined) {
+            chunk.translation[language].status = options.status;
+        }
+        // Persist the whole array back to Redis
+        chunks[chunkIndex] = chunk;
+        await this.redisClient.jsonSet(`[${roomId}]`, '.', chunks);
+        return true;
+    }
+}
+exports.MulingstreamChunkManager = MulingstreamChunkManager;

package/dist/managers/mulingstream-listener-manager.d.ts

@@ -1,29 +1,29 @@
-import { RedisClient } from '../redis-client';
-export type MulingstreamListenerData = {
-    listenerId: string;
-    roomId: string;
-    socketId?: string;
-    token: string;
-    name?: string;
-    firstJoined: number;
-    language?: string;
-    isActive?: boolean;
-};
-export declare class MulingstreamListenerManager {
-    private redisClient;
-    constructor(redisClient: RedisClient);
-    private parseHashData;
-    addListener(listenerData: Omit<MulingstreamListenerData, 'listenerId'>): Promise<string>;
-    getAllListeners(): Promise<MulingstreamListenerData[]>;
-    getListenersByRoom(roomId: string): Promise<MulingstreamListenerData[]>;
-    getListener(listenerIdOrToken: string): Promise<MulingstreamListenerData | null>;
-    removeListenerByToken(token: string): Promise<boolean>;
-    updateNameLanguage(listenerIdOrToken: string, name: string, language: string): Promise<boolean>;
-    updateSocketId(listenerIdOrToken: string, socketId: string): Promise<boolean>;
-    getTargetSocketIdsByRoomLanguage(roomId: string, language: string): Promise<string[]>;
-    /**
-     * Returns an array of unique languages for the given room, from highest frequency to lowest.
-     * If a listener has no language (empty string / undefined), it is ignored.
-     */
-    getUniqueLanguagesByRoom(roomId: string): Promise<string[]>;
-}
+import { RedisClient } from '../redis-client';
+export type MulingstreamListenerData = {
+    listenerId: string;
+    roomId: string;
+    socketId?: string;
+    token: string;
+    name?: string;
+    firstJoined: number;
+    language?: string;
+    isActive?: boolean;
+};
+export declare class MulingstreamListenerManager {
+    private redisClient;
+    constructor(redisClient: RedisClient);
+    private parseHashData;
+    addListener(listenerData: Omit<MulingstreamListenerData, 'listenerId'>): Promise<string>;
+    getAllListeners(): Promise<MulingstreamListenerData[]>;
+    getListenersByRoom(roomId: string): Promise<MulingstreamListenerData[]>;
+    getListener(listenerIdOrToken: string): Promise<MulingstreamListenerData | null>;
+    removeListenerByToken(token: string): Promise<boolean>;
+    updateNameLanguage(listenerIdOrToken: string, name: string, language: string): Promise<boolean>;
+    updateSocketId(listenerIdOrToken: string, socketId: string): Promise<boolean>;
+    getTargetSocketIdsByRoomLanguage(roomId: string, language: string): Promise<string[]>;
+    /**
+     * Returns an array of unique languages for the given room, from highest frequency to lowest.
+     * If a listener has no language (empty string / undefined), it is ignored.
+     */
+    getUniqueLanguagesByRoom(roomId: string): Promise<string[]>;
+}