@mulingai-npm/redis 2.1.2 → 2.2.1

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

@@ -59,14 +59,7 @@ export declare class MulingstreamChunkManager {
      * or null if the room doesn't exist in Redis.
      */
     getMulingstreamChunksByRoom(roomId: string): Promise<MulingstreamChunkData[] | null>;
-    /**
-     * Returns the room array "as is" for debugging
-     * (same as getMulingstreamChunksByRoom in this example).
-     */
     getRoomById(roomId: string): Promise<MulingstreamChunkData[] | null>;
-    /**
-     * Adds a new Mulingstream chunk to the array stored at [${roomId}].
-     */
     addMulingstreamChunk(params: {
         roomId: string;
         chunkNumber: number;
@@ -107,4 +100,8 @@ export declare class MulingstreamChunkManager {
      * Discards a specific language in both translation and tts for a chunk.
      */
     discardLanguage(roomId: string, chunkNumber: number, language: string): Promise<boolean>;
+    updateTranslation(roomId: string, chunkNumber: number, language: string, options: {
+        translation?: string;
+        status?: StepStatus;
+    }): Promise<boolean>;
 }

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

@@ -47,25 +47,12 @@ class MulingstreamChunkManager {
         }
         return chunks;
     }
-    /**
-     * Returns the room array "as is" for debugging
-     * (same as getMulingstreamChunksByRoom in this example).
-     */
     async getRoomById(roomId) {
         return this.getMulingstreamChunksByRoom(roomId);
     }
-    /**
-     * Adds a new Mulingstream chunk to the array stored at [${roomId}].
-     */
     async addMulingstreamChunk(params) {
+        var _a;
         const { roomId, chunkNumber, language, start, end, duration, isFirst, isLast, theme, sttProviders, translationTargetLanguages, ttsTargetLanguages } = params;
-        // ensure we only keep the last 5 elements
-        const currentLength = await this.redisClient.jsonArrLen(`[${roomId}]`, '.');
-        if ((currentLength !== null && currentLength !== void 0 ? currentLength : 0) >= ROOM_ARRAY_LENGTH) {
-            // remove the oldest (front of the array)
-            await this.redisClient.jsonArrPop(`[${roomId}]`, '.', 0);
-        }
-        // Build the chunk
         const audioChunk = {
             start,
             end,
@@ -75,25 +62,23 @@ class MulingstreamChunkManager {
             theme,
             processingStart: Date.now()
         };
-        // Build stt object
         const stt = {};
-        for (const svc of sttProviders) {
-            stt[svc] = {
+        for (const sttProvider of sttProviders) {
+            stt[sttProvider] = {
                 transcription: '',
                 status: 'INIT'
             };
         }
-        // Build translation and tts objects
         const translation = {};
         const tts = {};
-        for (const language of translationTargetLanguages) {
-            translation[language] = {
+        for (const lang of translationTargetLanguages) {
+            translation[lang] = {
                 translation: '',
                 status: 'INIT'
             };
         }
-        for (const language of ttsTargetLanguages) {
-            tts[language] = {
+        for (const lang of ttsTargetLanguages) {
+            tts[lang] = {
                 ttsAudioPath: '',
                 status: 'INIT',
                 isEmitted: false
@@ -113,9 +98,19 @@ class MulingstreamChunkManager {
             translation,
             tts
         };
-        // Append it to the array
-        await this.redisClient.jsonArrAppend(`[${roomId}]`, '.', // the root of the JSON document (the array)
-        newChunk);
+        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
@@ -220,5 +215,30 @@ class MulingstreamChunkManager {
         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/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@mulingai-npm/redis",
-    "version": "2.1.2",
+    "version": "2.2.1",
     "main": "dist/index.js",
     "types": "dist/index.d.ts",
     "repository": {