@mulingai-npm/redis 2.9.1 → 2.11.0

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

@@ -10,6 +10,7 @@ export type MulingstreamChunkData = {
     shortCodeTargetLanguages: string[];
     finalTranscription: string;
     sttStatus: StepStatus;
+    createdAt: number;
     audioChunk: {
         start: number;
         end: number;
@@ -123,4 +124,5 @@ export declare class MulingstreamChunkManager {
     }): Promise<MulingstreamChunkData | null>;
     increaseTotalCheck(roomId: string, chunkNumber: number, language: string): Promise<MulingstreamChunkData | null>;
     areTranslationsProcessed(roomId: string, chunkNumber: number): Promise<boolean>;
+    getAllReadyTts(roomId: string, language: string, timeoutMs?: number): Promise<MulingstreamChunkData[]>;
 }

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

@@ -53,6 +53,7 @@ class MulingstreamChunkManager {
     async addMulingstreamChunk(params) {
         var _a;
         const { roomId, chunkNumber, language, start, end, duration, isFirst, isLast, theme, sttProviders, targetLanguages, shortCodeTargetLanguages } = params;
+        /* ---------- build chunk skeleton ---------- */
         const audioChunk = {
             start,
             end,
@@ -63,25 +64,14 @@ class MulingstreamChunkManager {
             processingStart: Date.now()
         };
         const stt = {};
-        for (const sttProvider of sttProviders) {
-            stt[sttProvider] = {
-                transcription: '',
-                status: 'INIT'
-            };
+        for (const p of sttProviders) {
+            stt[p] = { transcription: '', status: 'INIT' };
         }
         const translation = {};
         const tts = {};
         for (const lang of shortCodeTargetLanguages) {
-            translation[lang] = {
-                translation: '',
-                status: 'INIT'
-            };
-            tts[lang] = {
-                ttsAudioPath: '',
-                status: 'INIT',
-                isEmitted: false,
-                totalCheck: 0
-            };
+            translation[lang] = { translation: '', status: 'INIT' };
+            tts[lang] = { ttsAudioPath: '', status: 'INIT', isEmitted: false, totalCheck: 0 };
         }
         const newChunk = {
             roomId,
@@ -92,22 +82,29 @@ class MulingstreamChunkManager {
             shortCodeTargetLanguages,
             finalTranscription: '',
             sttStatus: 'INIT',
+            createdAt: Date.now(),
             audioChunk,
             stt,
             translation,
             tts
         };
-        const chunks = (_a = (await this.redisClient.jsonGet(`[${roomId}]`, '.'))) !== null && _a !== void 0 ? _a : [];
+        /* ---------- pull existing room array ---------- */
+        let chunks = (_a = (await this.redisClient.jsonGet(`[${roomId}]`, '.'))) !== null && _a !== void 0 ? _a : [];
+        /* ---------- reset the room if this is the first chunk ---------- */
+        if (chunkNumber === 1) {
+            chunks = []; // start fresh
+        }
+        /* ---------- insert / replace ---------- */
         const idx = chunks.findIndex((c) => c.chunkNumber === chunkNumber);
         if (idx !== -1) {
             chunks[idx] = newChunk;
         }
         else {
             chunks.push(newChunk);
-            if (chunks.length > ROOM_ARRAY_LENGTH) {
+            if (chunks.length > ROOM_ARRAY_LENGTH)
                 chunks.shift();
-            }
         }
+        /* ---------- persist ---------- */
         await this.redisClient.jsonSet(`[${roomId}]`, '.', chunks);
         await this.redisClient.expire(`[${roomId}]`, EXPIRATION);
     }
@@ -398,5 +395,65 @@ class MulingstreamChunkManager {
         }
         return Object.values(chunk.translation).every((t) => t.status !== 'INIT');
     }
+    async getAllReadyTts(roomId, language, timeoutMs = 15000) {
+        const chunks = await this.getMulingstreamChunksByRoom(roomId);
+        if (!(chunks === null || chunks === void 0 ? void 0 : chunks.length))
+            return [];
+        chunks.sort((a, b) => a.chunkNumber - b.chunkNumber);
+        /* ---- isolate the tail where the language exists ---- */
+        let lastLangIdx = -1;
+        for (let i = chunks.length - 1; i >= 0; i--) {
+            if (chunks[i].tts[language]) {
+                lastLangIdx = i;
+                break;
+            }
+        }
+        if (lastLangIdx === -1)
+            return [];
+        let firstLangIdx = lastLangIdx;
+        for (let i = lastLangIdx - 1; i >= 0; i--) {
+            if (!chunks[i].tts[language])
+                break;
+            firstLangIdx = i;
+        }
+        const window = chunks.slice(firstLangIdx, lastLangIdx + 1);
+        /* ---- find last USED / DISCARDED in that window ---- */
+        let lastDone = -1;
+        for (let i = window.length - 1; i >= 0; i--) {
+            const st = window[i].tts[language].status;
+            if (st === 'USED' || st === 'DISCARDED') {
+                lastDone = i;
+                break;
+            }
+        }
+        const ready = [];
+        let blocked = false;
+        const now = Date.now();
+        for (let i = lastDone + 1; i < window.length; i++) {
+            const entry = window[i].tts[language];
+            if (entry.status === 'INIT') {
+                console.log(window[i].createdAt, ' --- ', window[i].chunkNumber);
+                if (now - window[i].createdAt > timeoutMs) {
+                    entry.status = 'DISCARDED';
+                }
+                else {
+                    blocked = true; // still waiting ⇒ cannot emit READYs
+                }
+                continue;
+            }
+            if (entry.status === 'READY') {
+                if (blocked)
+                    break; // earlier INIT still pending
+                ready.push(window[i]);
+                continue;
+            }
+            break; // unexpected USED / DISCARDED
+        }
+        // persist any DISCARDED changes we made
+        if (ready.length === 0 || blocked) {
+            await this.redisClient.jsonSet(`[${roomId}]`, '.', chunks);
+        }
+        return ready;
+    }
 }
 exports.MulingstreamChunkManager = MulingstreamChunkManager;

package/package.json

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