react-native-sherpa-onnx 0.3.6 → 0.3.7

package/lib/module/download/downloadTask.js

@@ -0,0 +1,385 @@
+"use strict";
+
+import { createDownloadTask, completeHandler, getExistingDownloadTasks } from '@kesha-antonov/react-native-background-downloader';
+import { exists, readFile, mkdir, writeFile, stat, unlink } from '@dr.pogodin/react-native-fs';
+import { checkDiskSpace, removeDirectoryRecursive } from "./validation.js";
+import { getModelsBaseDir, getModelDir, getArchivePath, getReadyMarkerPath, getDownloadStatePath, getNativeAssetExtractedModelDir, getTarArchivePath, getOnnxPath } from "./paths.js";
+import { emitDownloadProgress } from "./downloadEvents.js";
+import { runPostDownloadProcessing } from "./postDownloadProcessing.js";
+import { getModelByIdByCategory } from "./registry.js";
+import { listDownloadedModelsByCategory } from "./localModels.js";
+function makeDownloadTaskId(category, id) {
+  return `${category}:${id}`;
+}
+const activeDownloadTasks = new Map();
+export async function downloadModelByCategory(category, id, opts) {
+  const isAborted = () => Boolean(opts?.signal?.aborted);
+  if (opts?.signal?.aborted) {
+    const abortError = new Error('Download aborted');
+    abortError.name = 'AbortError';
+    throw abortError;
+  }
+  const model = await getModelByIdByCategory(category, id);
+  if (!model) {
+    throw new Error(`Unknown model id: ${id}`);
+  }
+  const baseDir = getModelsBaseDir(category);
+  await mkdir(baseDir);
+  const downloadPath = getArchivePath(category, id, model.archiveExt);
+  const isArchive = model.archiveExt === 'tar.bz2';
+  const modelDir = getModelDir(category, id);
+  const sleep = ms => new Promise(resolve => {
+    setTimeout(resolve, ms);
+  });
+  const cleanupPartial = async () => {
+    if (!isArchive) return;
+    if (await exists(modelDir)) {
+      await unlink(modelDir);
+    }
+  };
+  const cleanupPartialWithRetry = async () => {
+    for (let attempt = 0; attempt < 4; attempt += 1) {
+      await cleanupPartial();
+      if (!(await exists(modelDir))) return;
+      await sleep(400);
+    }
+    if (await exists(modelDir)) {
+      console.warn(`Model cleanup after abort did not fully complete for ${category}:${id}`);
+    }
+  };
+  const diskSpaceCheck = await checkDiskSpace(model.bytes);
+  if (!diskSpaceCheck.success) {
+    throw new Error(`Insufficient disk space: ${diskSpaceCheck.message}`);
+  }
+  const statePath = getDownloadStatePath(category, id);
+  if (opts?.overwrite) {
+    if (await exists(modelDir)) await unlink(modelDir);
+    if (await exists(downloadPath)) await unlink(downloadPath);
+    if (await exists(statePath)) await unlink(statePath);
+  } else {
+    const readyMarkerExists = await exists(getReadyMarkerPath(category, id));
+    if (!readyMarkerExists && isArchive) {
+      if (await exists(modelDir)) await unlink(modelDir);
+    }
+  }
+  try {
+    const downloadState = {
+      modelId: id,
+      category,
+      phase: 'downloading',
+      startedAt: new Date().toISOString(),
+      archivePath: downloadPath,
+      model,
+      totalBytes: model.bytes
+    };
+    await mkdir(getModelsBaseDir(category));
+    await writeFile(statePath, JSON.stringify(downloadState), 'utf8');
+    if (!isArchive) {
+      await mkdir(modelDir);
+    }
+    const taskId = makeDownloadTaskId(category, id);
+    return new Promise((resolve, reject) => {
+      let abortHandler;
+      const cleanup = () => {
+        if (abortHandler && opts?.signal) {
+          opts.signal.removeEventListener('abort', abortHandler);
+          abortHandler = undefined;
+        }
+        activeDownloadTasks.delete(taskId);
+      };
+      const task = createDownloadTask({
+        id: taskId,
+        url: model.downloadUrl,
+        destination: downloadPath,
+        metadata: {}
+      }).progress(({
+        bytesDownloaded,
+        bytesTotal
+      }) => {
+        if (isAborted()) return;
+        const total = bytesTotal ?? model.bytes ?? 0;
+        const percent = total > 0 ? bytesDownloaded / total * 100 : 0;
+        const progress = {
+          bytesDownloaded,
+          totalBytes: total,
+          percent,
+          phase: 'downloading'
+        };
+        opts?.onProgress?.(progress);
+        emitDownloadProgress(category, id, progress);
+      }).done(async () => {
+        cleanup();
+        try {
+          const result = await runPostDownloadProcessing({
+            category,
+            id,
+            model,
+            downloadPath,
+            modelDir,
+            isArchive,
+            statePath,
+            signal: opts?.signal,
+            onChecksumIssue: opts?.onChecksumIssue,
+            deleteArchiveAfterExtract: opts?.deleteArchiveAfterExtract,
+            onProgress: opts?.onProgress,
+            getDownloadedList: () => listDownloadedModelsByCategory(category)
+          });
+          completeHandler(taskId);
+          resolve(result);
+        } catch (e) {
+          completeHandler(taskId);
+          reject(e);
+        }
+      }).error(({
+        error,
+        errorCode
+      }) => {
+        cleanup();
+        completeHandler(taskId);
+        (async () => {
+          try {
+            if (await exists(statePath)) await unlink(statePath);
+          } catch {
+            // ignore
+          }
+        })();
+        reject(new Error(typeof error === 'string' ? error : String(errorCode ?? error)));
+      });
+      activeDownloadTasks.set(taskId, task);
+      if (opts?.signal) {
+        abortHandler = () => {
+          task.stop();
+          cleanup();
+          (async () => {
+            try {
+              if (await exists(statePath)) await unlink(statePath);
+            } catch {
+              // ignore
+            }
+          })();
+          const err = new Error('Download aborted');
+          err.name = 'AbortError';
+          reject(err);
+        };
+        opts.signal.addEventListener('abort', abortHandler);
+      }
+      task.start();
+    });
+  } catch (err) {
+    if (err instanceof Error && err.name === 'AbortError' || isAborted()) {
+      await cleanupPartialWithRetry();
+      try {
+        if (await exists(statePath)) await unlink(statePath);
+      } catch {
+        // ignore
+      }
+    }
+    if (isArchive && !(err instanceof Error && err.name === 'AbortError')) {
+      try {
+        if (await exists(downloadPath)) {
+          const archiveStat = await stat(downloadPath);
+          if (model.bytes > 0 && archiveStat.size < model.bytes) {
+            console.warn(`[Download] Deleting truncated archive for ${category}:${id} (${archiveStat.size}/${model.bytes})`);
+            await unlink(downloadPath);
+          }
+        }
+      } catch {
+        // ignore
+      }
+    }
+    throw err;
+  }
+}
+export async function getIncompleteDownloads(category) {
+  const prefix = category + ':';
+  const states = [];
+  const existingTasks = await getExistingDownloadTasks();
+  for (const task of existingTasks) {
+    if (!task.id || !task.id.startsWith(prefix)) continue;
+    const modelId = task.id.slice(prefix.length);
+    const readyPath = getReadyMarkerPath(category, modelId);
+    if (await exists(readyPath)) continue;
+    const statePath = getDownloadStatePath(category, modelId);
+    let model;
+    let totalBytes;
+    let archivePath;
+    let startedAt;
+    if (await exists(statePath)) {
+      try {
+        const raw = await readFile(statePath, 'utf8');
+        const fromFile = JSON.parse(raw);
+        model = fromFile.model;
+        totalBytes = fromFile.totalBytes ?? fromFile.model?.bytes;
+        archivePath = fromFile.archivePath;
+        startedAt = fromFile.startedAt;
+      } catch {
+        // ignore
+      }
+    }
+    if (!model) {
+      const meta = await getModelByIdByCategory(category, modelId);
+      if (!meta) continue;
+      model = meta;
+      totalBytes = model.bytes;
+      archivePath = getArchivePath(category, modelId, model.archiveExt);
+    }
+    let bytesDownloaded;
+    if (archivePath) {
+      try {
+        const st = await stat(archivePath);
+        if (st?.size != null && st.size >= 0) bytesDownloaded = st.size;
+      } catch {
+        // ignore
+      }
+    }
+    states.push({
+      modelId,
+      category,
+      phase: 'downloading',
+      startedAt: startedAt ?? new Date().toISOString(),
+      archivePath: archivePath ?? '',
+      model,
+      bytesDownloaded,
+      totalBytes: totalBytes ?? model.bytes
+    });
+  }
+  return states;
+}
+export async function resumeDownload(category, id, opts) {
+  const taskId = makeDownloadTaskId(category, id);
+  const existingTasks = await getExistingDownloadTasks();
+  const existing = existingTasks.find(t => t.id === taskId);
+  if (!existing) {
+    return downloadModelByCategory(category, id, {
+      onProgress: opts?.onProgress,
+      signal: opts?.signal,
+      onChecksumIssue: opts?.onChecksumIssue,
+      deleteArchiveAfterExtract: opts?.deleteArchiveAfterExtract
+    });
+  }
+  const model = await getModelByIdByCategory(category, id);
+  if (!model) throw new Error(`Unknown model id: ${id}`);
+  const downloadPath = getArchivePath(category, id, model.archiveExt);
+  const modelDir = getModelDir(category, id);
+  const isArchive = model.archiveExt === 'tar.bz2';
+  const statePath = getDownloadStatePath(category, id);
+  const isAborted = () => Boolean(opts?.signal?.aborted);
+  return new Promise((resolve, reject) => {
+    let abortHandler;
+    const cleanup = () => {
+      if (abortHandler && opts?.signal) {
+        opts.signal.removeEventListener('abort', abortHandler);
+        abortHandler = undefined;
+      }
+      activeDownloadTasks.delete(taskId);
+    };
+    existing.progress(({
+      bytesDownloaded,
+      bytesTotal
+    }) => {
+      if (isAborted()) return;
+      const total = bytesTotal ?? model.bytes ?? 0;
+      const percent = total > 0 ? bytesDownloaded / total * 100 : 0;
+      opts?.onProgress?.({
+        bytesDownloaded,
+        totalBytes: total,
+        percent,
+        phase: 'downloading'
+      });
+      emitDownloadProgress(category, id, {
+        bytesDownloaded,
+        totalBytes: total,
+        percent,
+        phase: 'downloading'
+      });
+    }).done(async () => {
+      cleanup();
+      try {
+        const result = await runPostDownloadProcessing({
+          category,
+          id,
+          model,
+          downloadPath,
+          modelDir,
+          isArchive,
+          statePath,
+          signal: opts?.signal,
+          onChecksumIssue: opts?.onChecksumIssue,
+          deleteArchiveAfterExtract: opts?.deleteArchiveAfterExtract,
+          onProgress: opts?.onProgress,
+          getDownloadedList: () => listDownloadedModelsByCategory(category)
+        });
+        completeHandler(taskId);
+        resolve(result);
+      } catch (e) {
+        completeHandler(taskId);
+        reject(e);
+      }
+    }).error(({
+      error,
+      errorCode
+    }) => {
+      cleanup();
+      completeHandler(taskId);
+      (async () => {
+        try {
+          if (await exists(statePath)) await unlink(statePath);
+        } catch {
+          // ignore
+        }
+      })();
+      reject(new Error(typeof error === 'string' ? error : String(errorCode ?? error)));
+    });
+    activeDownloadTasks.set(taskId, existing);
+    if (opts?.signal) {
+      abortHandler = () => {
+        existing.stop();
+        cleanup();
+        (async () => {
+          try {
+            if (await exists(statePath)) await unlink(statePath);
+          } catch {
+            // ignore
+          }
+        })();
+        const err = new Error('Download aborted');
+        err.name = 'AbortError';
+        reject(err);
+      };
+      opts.signal.addEventListener('abort', abortHandler);
+    }
+    existing.resume().catch(() => {});
+  });
+}
+export async function deleteIncompleteDownload(category, id) {
+  const taskId = makeDownloadTaskId(category, id);
+  const existingTasks = await getExistingDownloadTasks();
+  const task = existingTasks.find(t => t.id === taskId);
+  if (task) {
+    task.stop();
+    activeDownloadTasks.delete(taskId);
+  }
+  const modelDir = getModelDir(category, id);
+  if (await exists(modelDir)) {
+    await unlink(modelDir);
+  }
+  const tarPath = getTarArchivePath(category, id);
+  const onnxPath = getOnnxPath(category, id);
+  if (await exists(tarPath)) {
+    await unlink(tarPath);
+  }
+  if (await exists(onnxPath)) {
+    await unlink(onnxPath);
+  }
+  const statePath = getDownloadStatePath(category, id);
+  if (await exists(statePath)) {
+    await unlink(statePath);
+  }
+  await removeDirectoryRecursive(getNativeAssetExtractedModelDir(id));
+}
+
+/** Task ids in the form `category:modelId` for downloads currently tracked in JS (before post-processing). */
+export function getActiveDownloadTaskKeys() {
+  return [...activeDownloadTasks.keys()];
+}
+//# sourceMappingURL=downloadTask.js.map

package/lib/module/download/downloadTask.js.map

@@ -0,0 +1 @@
+{"version":3,"names":["createDownloadTask","completeHandler","getExistingDownloadTasks","exists","readFile","mkdir","writeFile","stat","unlink","checkDiskSpace","removeDirectoryRecursive","getModelsBaseDir","getModelDir","getArchivePath","getReadyMarkerPath","getDownloadStatePath","getNativeAssetExtractedModelDir","getTarArchivePath","getOnnxPath","emitDownloadProgress","runPostDownloadProcessing","getModelByIdByCategory","listDownloadedModelsByCategory","makeDownloadTaskId","category","id","activeDownloadTasks","Map","downloadModelByCategory","opts","isAborted","Boolean","signal","aborted","abortError","Error","name","model","baseDir","downloadPath","archiveExt","isArchive","modelDir","sleep","ms","Promise","resolve","setTimeout","cleanupPartial","cleanupPartialWithRetry","attempt","console","warn","diskSpaceCheck","bytes","success","message","statePath","overwrite","readyMarkerExists","downloadState","modelId","phase","startedAt","Date","toISOString","archivePath","totalBytes","JSON","stringify","taskId","reject","abortHandler","cleanup","removeEventListener","undefined","delete","task","url","downloadUrl","destination","metadata","progress","bytesDownloaded","bytesTotal","total","percent","onProgress","done","result","onChecksumIssue","deleteArchiveAfterExtract","getDownloadedList","e","error","errorCode","String","set","stop","err","addEventListener","start","archiveStat","size","getIncompleteDownloads","prefix","states","existingTasks","startsWith","slice","length","readyPath","raw","fromFile","parse","meta","st","push","resumeDownload","existing","find","t","resume","catch","deleteIncompleteDownload","tarPath","onnxPath","getActiveDownloadTaskKeys","keys"],"sourceRoot":"../../../src","sources":["download/downloadTask.ts"],"mappings":";;AAAA,SACEA,kBAAkB,EAClBC,eAAe,EACfC,wBAAwB,QACnB,mDAAmD;AAC1D,SACEC,MAAM,EACNC,QAAQ,EACRC,KAAK,EACLC,SAAS,EACTC,IAAI,EACJC,MAAM,QACD,6BAA6B;AACpC,SAASC,cAAc,EAAEC,wBAAwB,QAAQ,iBAAc;AASvE,SACEC,gBAAgB,EAChBC,WAAW,EACXC,cAAc,EACdC,kBAAkB,EAClBC,oBAAoB,EACpBC,+BAA+B,EAC/BC,iBAAiB,EACjBC,WAAW,QACN,YAAS;AAChB,SAASC,oBAAoB,QAAQ,qBAAkB;AACvD,SAASC,yBAAyB,QAAQ,6BAA0B;AACpE,SAASC,sBAAsB,QAAQ,eAAY;AACnD,SAASC,8BAA8B,QAAQ,kBAAe;AAE9D,SAASC,kBAAkBA,CAACC,QAAuB,EAAEC,EAAU,EAAU;EACvE,OAAO,GAAGD,QAAQ,IAAIC,EAAE,EAAE;AAC5B;AAEA,MAAMC,mBAAmB,GAAG,IAAIC,GAAG,CAA+B,CAAC;AAEnE,OAAO,eAAeC,uBAAuBA,CAC3CJ,QAAuB,EACvBC,EAAU,EACVI,IAOC,EACwB;EACzB,MAAMC,SAAS,GAAGA,CAAA,KAAMC,OAAO,CAACF,IAAI,EAAEG,MAAM,EAAEC,OAAO,CAAC;EAEtD,IAAIJ,IAAI,EAAEG,MAAM,EAAEC,OAAO,EAAE;IACzB,MAAMC,UAAU,GAAG,IAAIC,KAAK,CAAC,kBAAkB,CAAC;IAChDD,UAAU,CAACE,IAAI,GAAG,YAAY;IAC9B,MAAMF,UAAU;EAClB;EAEA,MAAMG,KAAK,GAAG,MAAMhB,sBAAsB,CAAIG,QAAQ,EAAEC,EAAE,CAAC;EAC3D,IAAI,CAACY,KAAK,EAAE;IACV,MAAM,IAAIF,KAAK,CAAC,qBAAqBV,EAAE,EAAE,CAAC;EAC5C;EAEA,MAAMa,OAAO,GAAG3B,gBAAgB,CAACa,QAAQ,CAAC;EAC1C,MAAMnB,KAAK,CAACiC,OAAO,CAAC;EAEpB,MAAMC,YAAY,GAAG1B,cAAc,CAACW,QAAQ,EAAEC,EAAE,EAAEY,KAAK,CAACG,UAAU,CAAC;EACnE,MAAMC,SAAS,GAAGJ,KAAK,CAACG,UAAU,KAAK,SAAS;EAChD,MAAME,QAAQ,GAAG9B,WAAW,CAACY,QAAQ,EAAEC,EAAE,CAAC;EAE1C,MAAMkB,KAAK,GAAIC,EAAU,IACvB,IAAIC,OAAO,CAAQC,OAAO,IAAK;IAC7BC,UAAU,CAACD,OAAO,EAAEF,EAAE,CAAC;EACzB,CAAC,CAAC;EAEJ,MAAMI,cAAc,GAAG,MAAAA,CAAA,KAAY;IACjC,IAAI,CAACP,SAAS,EAAE;IAChB,IAAI,MAAMtC,MAAM,CAACuC,QAAQ,CAAC,EAAE;MAC1B,MAAMlC,MAAM,CAACkC,QAAQ,CAAC;IACxB;EACF,CAAC;EAED,MAAMO,uBAAuB,GAAG,MAAAA,CAAA,KAAY;IAC1C,KAAK,IAAIC,OAAO,GAAG,CAAC,EAAEA,OAAO,GAAG,CAAC,EAAEA,OAAO,IAAI,CAAC,EAAE;MAC/C,MAAMF,cAAc,CAAC,CAAC;MACtB,IAAI,EAAE,MAAM7C,MAAM,CAACuC,QAAQ,CAAC,CAAC,EAAE;MAC/B,MAAMC,KAAK,CAAC,GAAG,CAAC;IAClB;IACA,IAAI,MAAMxC,MAAM,CAACuC,QAAQ,CAAC,EAAE;MAC1BS,OAAO,CAACC,IAAI,CACV,wDAAwD5B,QAAQ,IAAIC,EAAE,EACxE,CAAC;IACH;EACF,CAAC;EAED,MAAM4B,cAAc,GAAG,MAAM5C,cAAc,CAAC4B,KAAK,CAACiB,KAAK,CAAC;EACxD,IAAI,CAACD,cAAc,CAACE,OAAO,EAAE;IAC3B,MAAM,IAAIpB,KAAK,CAAC,4BAA4BkB,cAAc,CAACG,OAAO,EAAE,CAAC;EACvE;EAEA,MAAMC,SAAS,GAAG1C,oBAAoB,CAACS,QAAQ,EAAEC,EAAE,CAAC;EAEpD,IAAII,IAAI,EAAE6B,SAAS,EAAE;IACnB,IAAI,MAAMvD,MAAM,CAACuC,QAAQ,CAAC,EAAE,MAAMlC,MAAM,CAACkC,QAAQ,CAAC;IAClD,IAAI,MAAMvC,MAAM,CAACoC,YAAY,CAAC,EAAE,MAAM/B,MAAM,CAAC+B,YAAY,CAAC;IAC1D,IAAI,MAAMpC,MAAM,CAACsD,SAAS,CAAC,EAAE,MAAMjD,MAAM,CAACiD,SAAS,CAAC;EACtD,CAAC,MAAM;IACL,MAAME,iBAAiB,GAAG,MAAMxD,MAAM,CAACW,kBAAkB,CAACU,QAAQ,EAAEC,EAAE,CAAC,CAAC;IACxE,IAAI,CAACkC,iBAAiB,IAAIlB,SAAS,EAAE;MACnC,IAAI,MAAMtC,MAAM,CAACuC,QAAQ,CAAC,EAAE,MAAMlC,MAAM,CAACkC,QAAQ,CAAC;IACpD;EACF;EAEA,IAAI;IACF,MAAMkB,aAA4B,GAAG;MACnCC,OAAO,EAAEpC,EAAE;MACXD,QAAQ;MACRsC,KAAK,EAAE,aAAa;MACpBC,SAAS,EAAE,IAAIC,IAAI,CAAC,CAAC,CAACC,WAAW,CAAC,CAAC;MACnCC,WAAW,EAAE3B,YAAY;MACzBF,KAAK;MACL8B,UAAU,EAAE9B,KAAK,CAACiB;IACpB,CAAC;IACD,MAAMjD,KAAK,CAACM,gBAAgB,CAACa,QAAQ,CAAC,CAAC;IACvC,MAAMlB,SAAS,CAACmD,SAAS,EAAEW,IAAI,CAACC,SAAS,CAACT,aAAa,CAAC,EAAE,MAAM,CAAC;IAEjE,IAAI,CAACnB,SAAS,EAAE;MACd,MAAMpC,KAAK,CAACqC,QAAQ,CAAC;IACvB;IAEA,MAAM4B,MAAM,GAAG/C,kBAAkB,CAACC,QAAQ,EAAEC,EAAE,CAAC;IAE/C,OAAO,IAAIoB,OAAO,CAAiB,CAACC,OAAO,EAAEyB,MAAM,KAAK;MACtD,IAAIC,YAAsC;MAE1C,MAAMC,OAAO,GAAGA,CAAA,KAAM;QACpB,IAAID,YAAY,IAAI3C,IAAI,EAAEG,MAAM,EAAE;UAChCH,IAAI,CAACG,MAAM,CAAC0C,mBAAmB,CAAC,OAAO,EAAEF,YAAY,CAAC;UACtDA,YAAY,GAAGG,SAAS;QAC1B;QACAjD,mBAAmB,CAACkD,MAAM,CAACN,MAAM,CAAC;MACpC,CAAC;MAED,MAAMO,IAAI,GAAG7E,kBAAkB,CAAC;QAC9ByB,EAAE,EAAE6C,MAAM;QACVQ,GAAG,EAAEzC,KAAK,CAAC0C,WAAW;QACtBC,WAAW,EAAEzC,YAAY;QACzB0C,QAAQ,EAAE,CAAC;MACb,CAAC,CAAC,CACCC,QAAQ,CACP,CAAC;QACCC,eAAe;QACfC;MAIF,CAAC,KAAK;QACJ,IAAItD,SAAS,CAAC,CAAC,EAAE;QACjB,MAAMuD,KAAK,GAAGD,UAAU,IAAI/C,KAAK,CAACiB,KAAK,IAAI,CAAC;QAC5C,MAAMgC,OAAO,GAAGD,KAAK,GAAG,CAAC,GAAIF,eAAe,GAAGE,KAAK,GAAI,GAAG,GAAG,CAAC;QAC/D,MAAMH,QAA0B,GAAG;UACjCC,eAAe;UACfhB,UAAU,EAAEkB,KAAK;UACjBC,OAAO;UACPxB,KAAK,EAAE;QACT,CAAC;QACDjC,IAAI,EAAE0D,UAAU,GAAGL,QAAQ,CAAC;QAC5B/D,oBAAoB,CAACK,QAAQ,EAAEC,EAAE,EAAEyD,QAAQ,CAAC;MAC9C,CACF,CAAC,CACAM,IAAI,CAAC,YAAY;QAChBf,OAAO,CAAC,CAAC;QACT,IAAI;UACF,MAAMgB,MAAM,GAAG,MAAMrE,yBAAyB,CAAC;YAC7CI,QAAQ;YACRC,EAAE;YACFY,KAAK;YACLE,YAAY;YACZG,QAAQ;YACRD,SAAS;YACTgB,SAAS;YACTzB,MAAM,EAAEH,IAAI,EAAEG,MAAM;YACpB0D,eAAe,EAAE7D,IAAI,EAAE6D,eAAe;YACtCC,yBAAyB,EAAE9D,IAAI,EAAE8D,yBAAyB;YAC1DJ,UAAU,EAAE1D,IAAI,EAAE0D,UAAU;YAC5BK,iBAAiB,EAAEA,CAAA,KACjBtE,8BAA8B,CAAgBE,QAAQ;UAC1D,CAAC,CAAC;UACFvB,eAAe,CAACqE,MAAM,CAAC;UACvBxB,OAAO,CAAC2C,MAAM,CAAC;QACjB,CAAC,CAAC,OAAOI,CAAC,EAAE;UACV5F,eAAe,CAACqE,MAAM,CAAC;UACvBC,MAAM,CAACsB,CAAC,CAAC;QACX;MACF,CAAC,CAAC,CACDC,KAAK,CACJ,CAAC;QAAEA,KAAK;QAAEC;MAAkD,CAAC,KAAK;QAChEtB,OAAO,CAAC,CAAC;QACTxE,eAAe,CAACqE,MAAM,CAAC;QACvB,CAAC,YAAY;UACX,IAAI;YACF,IAAI,MAAMnE,MAAM,CAACsD,SAAS,CAAC,EAAE,MAAMjD,MAAM,CAACiD,SAAS,CAAC;UACtD,CAAC,CAAC,MAAM;YACN;UAAA;QAEJ,CAAC,EAAE,CAAC;QACJc,MAAM,CACJ,IAAIpC,KAAK,CACP,OAAO2D,KAAK,KAAK,QAAQ,GAAGA,KAAK,GAAGE,MAAM,CAACD,SAAS,IAAID,KAAK,CAC/D,CACF,CAAC;MACH,CACF,CAAC;MAEHpE,mBAAmB,CAACuE,GAAG,CAAC3B,MAAM,EAAEO,IAAI,CAAC;MACrC,IAAIhD,IAAI,EAAEG,MAAM,EAAE;QAChBwC,YAAY,GAAGA,CAAA,KAAM;UACnBK,IAAI,CAACqB,IAAI,CAAC,CAAC;UACXzB,OAAO,CAAC,CAAC;UACT,CAAC,YAAY;YACX,IAAI;cACF,IAAI,MAAMtE,MAAM,CAACsD,SAAS,CAAC,EAAE,MAAMjD,MAAM,CAACiD,SAAS,CAAC;YACtD,CAAC,CAAC,MAAM;cACN;YAAA;UAEJ,CAAC,EAAE,CAAC;UACJ,MAAM0C,GAAG,GAAG,IAAIhE,KAAK,CAAC,kBAAkB,CAAC;UACzCgE,GAAG,CAAC/D,IAAI,GAAG,YAAY;UACvBmC,MAAM,CAAC4B,GAAG,CAAC;QACb,CAAC;QACDtE,IAAI,CAACG,MAAM,CAACoE,gBAAgB,CAAC,OAAO,EAAE5B,YAAY,CAAC;MACrD;MACAK,IAAI,CAACwB,KAAK,CAAC,CAAC;IACd,CAAC,CAAC;EACJ,CAAC,CAAC,OAAOF,GAAG,EAAE;IACZ,IAAKA,GAAG,YAAYhE,KAAK,IAAIgE,GAAG,CAAC/D,IAAI,KAAK,YAAY,IAAKN,SAAS,CAAC,CAAC,EAAE;MACtE,MAAMmB,uBAAuB,CAAC,CAAC;MAC/B,IAAI;QACF,IAAI,MAAM9C,MAAM,CAACsD,SAAS,CAAC,EAAE,MAAMjD,MAAM,CAACiD,SAAS,CAAC;MACtD,CAAC,CAAC,MAAM;QACN;MAAA;IAEJ;IACA,IAAIhB,SAAS,IAAI,EAAE0D,GAAG,YAAYhE,KAAK,IAAIgE,GAAG,CAAC/D,IAAI,KAAK,YAAY,CAAC,EAAE;MACrE,IAAI;QACF,IAAI,MAAMjC,MAAM,CAACoC,YAAY,CAAC,EAAE;UAC9B,MAAM+D,WAAW,GAAG,MAAM/F,IAAI,CAACgC,YAAY,CAAC;UAC5C,IAAIF,KAAK,CAACiB,KAAK,GAAG,CAAC,IAAIgD,WAAW,CAACC,IAAI,GAAGlE,KAAK,CAACiB,KAAK,EAAE;YACrDH,OAAO,CAACC,IAAI,CACV,6CAA6C5B,QAAQ,IAAIC,EAAE,KAAK6E,WAAW,CAACC,IAAI,IAAIlE,KAAK,CAACiB,KAAK,GACjG,CAAC;YACD,MAAM9C,MAAM,CAAC+B,YAAY,CAAC;UAC5B;QACF;MACF,CAAC,CAAC,MAAM;QACN;MAAA;IAEJ;IACA,MAAM4D,GAAG;EACX;AACF;AAEA,OAAO,eAAeK,sBAAsBA,CAC1ChF,QAAuB,EACG;EAC1B,MAAMiF,MAAM,GAAGjF,QAAQ,GAAG,GAAG;EAC7B,MAAMkF,MAAuB,GAAG,EAAE;EAElC,MAAMC,aAAa,GAAG,MAAMzG,wBAAwB,CAAC,CAAC;EACtD,KAAK,MAAM2E,IAAI,IAAI8B,aAAa,EAAE;IAChC,IAAI,CAAC9B,IAAI,CAACpD,EAAE,IAAI,CAACoD,IAAI,CAACpD,EAAE,CAACmF,UAAU,CAACH,MAAM,CAAC,EAAE;IAC7C,MAAM5C,OAAO,GAAGgB,IAAI,CAACpD,EAAE,CAACoF,KAAK,CAACJ,MAAM,CAACK,MAAM,CAAC;IAC5C,MAAMC,SAAS,GAAGjG,kBAAkB,CAACU,QAAQ,EAAEqC,OAAO,CAAC;IACvD,IAAI,MAAM1D,MAAM,CAAC4G,SAAS,CAAC,EAAE;IAE7B,MAAMtD,SAAS,GAAG1C,oBAAoB,CAACS,QAAQ,EAAEqC,OAAO,CAAC;IACzD,IAAIxB,KAAgC;IACpC,IAAI8B,UAA8B;IAClC,IAAID,WAA+B;IACnC,IAAIH,SAA6B;IAEjC,IAAI,MAAM5D,MAAM,CAACsD,SAAS,CAAC,EAAE;MAC3B,IAAI;QACF,MAAMuD,GAAG,GAAG,MAAM5G,QAAQ,CAACqD,SAAS,EAAE,MAAM,CAAC;QAC7C,MAAMwD,QAAQ,GAAG7C,IAAI,CAAC8C,KAAK,CAACF,GAAG,CAAkB;QACjD3E,KAAK,GAAG4E,QAAQ,CAAC5E,KAAK;QACtB8B,UAAU,GAAG8C,QAAQ,CAAC9C,UAAU,IAAI8C,QAAQ,CAAC5E,KAAK,EAAEiB,KAAK;QACzDY,WAAW,GAAG+C,QAAQ,CAAC/C,WAAW;QAClCH,SAAS,GAAGkD,QAAQ,CAAClD,SAAS;MAChC,CAAC,CAAC,MAAM;QACN;MAAA;IAEJ;IACA,IAAI,CAAC1B,KAAK,EAAE;MACV,MAAM8E,IAAI,GAAG,MAAM9F,sBAAsB,CAACG,QAAQ,EAAEqC,OAAO,CAAC;MAC5D,IAAI,CAACsD,IAAI,EAAE;MACX9E,KAAK,GAAG8E,IAAqB;MAC7BhD,UAAU,GAAG9B,KAAK,CAACiB,KAAK;MACxBY,WAAW,GAAGrD,cAAc,CAACW,QAAQ,EAAEqC,OAAO,EAAExB,KAAK,CAACG,UAAU,CAAC;IACnE;IAEA,IAAI2C,eAAmC;IACvC,IAAIjB,WAAW,EAAE;MACf,IAAI;QACF,MAAMkD,EAAE,GAAG,MAAM7G,IAAI,CAAC2D,WAAW,CAAC;QAClC,IAAIkD,EAAE,EAAEb,IAAI,IAAI,IAAI,IAAIa,EAAE,CAACb,IAAI,IAAI,CAAC,EAAEpB,eAAe,GAAGiC,EAAE,CAACb,IAAI;MACjE,CAAC,CAAC,MAAM;QACN;MAAA;IAEJ;IAEAG,MAAM,CAACW,IAAI,CAAC;MACVxD,OAAO;MACPrC,QAAQ;MACRsC,KAAK,EAAE,aAAa;MACpBC,SAAS,EAAEA,SAAS,IAAI,IAAIC,IAAI,CAAC,CAAC,CAACC,WAAW,CAAC,CAAC;MAChDC,WAAW,EAAEA,WAAW,IAAI,EAAE;MAC9B7B,KAAK;MACL8C,eAAe;MACfhB,UAAU,EAAEA,UAAU,IAAI9B,KAAK,CAACiB;IAClC,CAAC,CAAC;EACJ;EAEA,OAAOoD,MAAM;AACf;AAEA,OAAO,eAAeY,cAAcA,CAClC9F,QAAuB,EACvBC,EAAU,EACVI,IAKC,EACwB;EACzB,MAAMyC,MAAM,GAAG/C,kBAAkB,CAACC,QAAQ,EAAEC,EAAE,CAAC;EAC/C,MAAMkF,aAAa,GAAG,MAAMzG,wBAAwB,CAAC,CAAC;EACtD,MAAMqH,QAAQ,GAAGZ,aAAa,CAACa,IAAI,CAAEC,CAAC,IAAKA,CAAC,CAAChG,EAAE,KAAK6C,MAAM,CAAC;EAC3D,IAAI,CAACiD,QAAQ,EAAE;IACb,OAAO3F,uBAAuB,CAAIJ,QAAQ,EAAEC,EAAE,EAAE;MAC9C8D,UAAU,EAAE1D,IAAI,EAAE0D,UAAU;MAC5BvD,MAAM,EAAEH,IAAI,EAAEG,MAAM;MACpB0D,eAAe,EAAE7D,IAAI,EAAE6D,eAAe;MACtCC,yBAAyB,EAAE9D,IAAI,EAAE8D;IACnC,CAAC,CAAC;EACJ;EAEA,MAAMtD,KAAK,GAAG,MAAMhB,sBAAsB,CAAIG,QAAQ,EAAEC,EAAE,CAAC;EAC3D,IAAI,CAACY,KAAK,EAAE,MAAM,IAAIF,KAAK,CAAC,qBAAqBV,EAAE,EAAE,CAAC;EACtD,MAAMc,YAAY,GAAG1B,cAAc,CAACW,QAAQ,EAAEC,EAAE,EAAEY,KAAK,CAACG,UAAU,CAAC;EACnE,MAAME,QAAQ,GAAG9B,WAAW,CAACY,QAAQ,EAAEC,EAAE,CAAC;EAC1C,MAAMgB,SAAS,GAAGJ,KAAK,CAACG,UAAU,KAAK,SAAS;EAChD,MAAMiB,SAAS,GAAG1C,oBAAoB,CAACS,QAAQ,EAAEC,EAAE,CAAC;EACpD,MAAMK,SAAS,GAAGA,CAAA,KAAMC,OAAO,CAACF,IAAI,EAAEG,MAAM,EAAEC,OAAO,CAAC;EAEtD,OAAO,IAAIY,OAAO,CAAiB,CAACC,OAAO,EAAEyB,MAAM,KAAK;IACtD,IAAIC,YAAsC;IAE1C,MAAMC,OAAO,GAAGA,CAAA,KAAM;MACpB,IAAID,YAAY,IAAI3C,IAAI,EAAEG,MAAM,EAAE;QAChCH,IAAI,CAACG,MAAM,CAAC0C,mBAAmB,CAAC,OAAO,EAAEF,YAAY,CAAC;QACtDA,YAAY,GAAGG,SAAS;MAC1B;MACAjD,mBAAmB,CAACkD,MAAM,CAACN,MAAM,CAAC;IACpC,CAAC;IAEDiD,QAAQ,CACLrC,QAAQ,CACP,CAAC;MACCC,eAAe;MACfC;IAIF,CAAC,KAAK;MACJ,IAAItD,SAAS,CAAC,CAAC,EAAE;MACjB,MAAMuD,KAAK,GAAGD,UAAU,IAAI/C,KAAK,CAACiB,KAAK,IAAI,CAAC;MAC5C,MAAMgC,OAAO,GAAGD,KAAK,GAAG,CAAC,GAAIF,eAAe,GAAGE,KAAK,GAAI,GAAG,GAAG,CAAC;MAC/DxD,IAAI,EAAE0D,UAAU,GAAG;QACjBJ,eAAe;QACfhB,UAAU,EAAEkB,KAAK;QACjBC,OAAO;QACPxB,KAAK,EAAE;MACT,CAAC,CAAC;MACF3C,oBAAoB,CAACK,QAAQ,EAAEC,EAAE,EAAE;QACjC0D,eAAe;QACfhB,UAAU,EAAEkB,KAAK;QACjBC,OAAO;QACPxB,KAAK,EAAE;MACT,CAAC,CAAC;IACJ,CACF,CAAC,CACA0B,IAAI,CAAC,YAAY;MAChBf,OAAO,CAAC,CAAC;MACT,IAAI;QACF,MAAMgB,MAAM,GAAG,MAAMrE,yBAAyB,CAAC;UAC7CI,QAAQ;UACRC,EAAE;UACFY,KAAK;UACLE,YAAY;UACZG,QAAQ;UACRD,SAAS;UACTgB,SAAS;UACTzB,MAAM,EAAEH,IAAI,EAAEG,MAAM;UACpB0D,eAAe,EAAE7D,IAAI,EAAE6D,eAAe;UACtCC,yBAAyB,EAAE9D,IAAI,EAAE8D,yBAAyB;UAC1DJ,UAAU,EAAE1D,IAAI,EAAE0D,UAAU;UAC5BK,iBAAiB,EAAEA,CAAA,KACjBtE,8BAA8B,CAAgBE,QAAQ;QAC1D,CAAC,CAAC;QACFvB,eAAe,CAACqE,MAAM,CAAC;QACvBxB,OAAO,CAAC2C,MAAM,CAAC;MACjB,CAAC,CAAC,OAAOI,CAAC,EAAE;QACV5F,eAAe,CAACqE,MAAM,CAAC;QACvBC,MAAM,CAACsB,CAAC,CAAC;MACX;IACF,CAAC,CAAC,CACDC,KAAK,CAAC,CAAC;MAAEA,KAAK;MAAEC;IAAkD,CAAC,KAAK;MACvEtB,OAAO,CAAC,CAAC;MACTxE,eAAe,CAACqE,MAAM,CAAC;MACvB,CAAC,YAAY;QACX,IAAI;UACF,IAAI,MAAMnE,MAAM,CAACsD,SAAS,CAAC,EAAE,MAAMjD,MAAM,CAACiD,SAAS,CAAC;QACtD,CAAC,CAAC,MAAM;UACN;QAAA;MAEJ,CAAC,EAAE,CAAC;MACJc,MAAM,CACJ,IAAIpC,KAAK,CACP,OAAO2D,KAAK,KAAK,QAAQ,GAAGA,KAAK,GAAGE,MAAM,CAACD,SAAS,IAAID,KAAK,CAC/D,CACF,CAAC;IACH,CAAC,CAAC;IAEJpE,mBAAmB,CAACuE,GAAG,CAAC3B,MAAM,EAAEiD,QAAQ,CAAC;IACzC,IAAI1F,IAAI,EAAEG,MAAM,EAAE;MAChBwC,YAAY,GAAGA,CAAA,KAAM;QACnB+C,QAAQ,CAACrB,IAAI,CAAC,CAAC;QACfzB,OAAO,CAAC,CAAC;QACT,CAAC,YAAY;UACX,IAAI;YACF,IAAI,MAAMtE,MAAM,CAACsD,SAAS,CAAC,EAAE,MAAMjD,MAAM,CAACiD,SAAS,CAAC;UACtD,CAAC,CAAC,MAAM;YACN;UAAA;QAEJ,CAAC,EAAE,CAAC;QACJ,MAAM0C,GAAG,GAAG,IAAIhE,KAAK,CAAC,kBAAkB,CAAC;QACzCgE,GAAG,CAAC/D,IAAI,GAAG,YAAY;QACvBmC,MAAM,CAAC4B,GAAG,CAAC;MACb,CAAC;MACDtE,IAAI,CAACG,MAAM,CAACoE,gBAAgB,CAAC,OAAO,EAAE5B,YAAY,CAAC;IACrD;IACA+C,QAAQ,CAACG,MAAM,CAAC,CAAC,CAACC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;EACnC,CAAC,CAAC;AACJ;AAEA,OAAO,eAAeC,wBAAwBA,CAC5CpG,QAAuB,EACvBC,EAAU,EACK;EACf,MAAM6C,MAAM,GAAG/C,kBAAkB,CAACC,QAAQ,EAAEC,EAAE,CAAC;EAC/C,MAAMkF,aAAa,GAAG,MAAMzG,wBAAwB,CAAC,CAAC;EACtD,MAAM2E,IAAI,GAAG8B,aAAa,CAACa,IAAI,CAAEC,CAAC,IAAKA,CAAC,CAAChG,EAAE,KAAK6C,MAAM,CAAC;EACvD,IAAIO,IAAI,EAAE;IACRA,IAAI,CAACqB,IAAI,CAAC,CAAC;IACXxE,mBAAmB,CAACkD,MAAM,CAACN,MAAM,CAAC;EACpC;EAEA,MAAM5B,QAAQ,GAAG9B,WAAW,CAACY,QAAQ,EAAEC,EAAE,CAAC;EAC1C,IAAI,MAAMtB,MAAM,CAACuC,QAAQ,CAAC,EAAE;IAC1B,MAAMlC,MAAM,CAACkC,QAAQ,CAAC;EACxB;EACA,MAAMmF,OAAO,GAAG5G,iBAAiB,CAACO,QAAQ,EAAEC,EAAE,CAAC;EAC/C,MAAMqG,QAAQ,GAAG5G,WAAW,CAACM,QAAQ,EAAEC,EAAE,CAAC;EAC1C,IAAI,MAAMtB,MAAM,CAAC0H,OAAO,CAAC,EAAE;IACzB,MAAMrH,MAAM,CAACqH,OAAO,CAAC;EACvB;EACA,IAAI,MAAM1H,MAAM,CAAC2H,QAAQ,CAAC,EAAE;IAC1B,MAAMtH,MAAM,CAACsH,QAAQ,CAAC;EACxB;EACA,MAAMrE,SAAS,GAAG1C,oBAAoB,CAACS,QAAQ,EAAEC,EAAE,CAAC;EACpD,IAAI,MAAMtB,MAAM,CAACsD,SAAS,CAAC,EAAE;IAC3B,MAAMjD,MAAM,CAACiD,SAAS,CAAC;EACzB;EACA,MAAM/C,wBAAwB,CAACM,+BAA+B,CAACS,EAAE,CAAC,CAAC;AACrE;;AAEA;AACA,OAAO,SAASsG,yBAAyBA,CAAA,EAAa;EACpD,OAAO,CAAC,GAAGrG,mBAAmB,CAACsG,IAAI,CAAC,CAAC,CAAC;AACxC","ignoreList":[]}

package/lib/module/download/ensureModel.js

@@ -0,0 +1,89 @@
+"use strict";
+
+import { exists, stat } from '@dr.pogodin/react-native-fs';
+import { getArchivePath } from "./paths.js";
+import { getModelByIdByCategory } from "./registry.js";
+import { isModelDownloadedByCategory, getLocalModelPathByCategory, deleteModelByCategory } from "./localModels.js";
+import { downloadModelByCategory } from "./downloadTask.js";
+import { getIncompleteDownloads, resumeDownload } from "./downloadTask.js";
+import { getIncompleteExtractions, resumeExtraction, extractModelByCategory, deleteIncompleteExtraction } from "./modelExtraction.js";
+import { deleteIncompleteDownload } from "./downloadTask.js";
+/**
+ * Single entry point to ensure a model is available locally: handles download, extraction,
+ * and all edge cases (already ready, incomplete download, incomplete extraction, archive
+ * already present). Call this with category, id, and optional opts; the function decides
+ * whether to return the existing path, resume an interrupted operation, or start download/extraction.
+ *
+ * Use this as the main API when you only need "make this model ready"; the lower-level
+ * APIs (downloadModelByCategory, resumeDownload, extractModelByCategory, getIncompleteExtractions,
+ * etc.) remain available for advanced flows.
+ */
+export async function ensureModelByCategory(category, id, opts) {
+  const model = await getModelByIdByCategory(category, id);
+  if (!model) {
+    throw new Error(`Unknown model id: ${id}`);
+  }
+  const isArchive = model.archiveExt === 'tar.bz2';
+  if (opts?.overwrite) {
+    await deleteModelByCategory(category, id);
+    await deleteIncompleteExtraction(category, id);
+    await deleteIncompleteDownload(category, id);
+  }
+  if (!opts?.overwrite && (await isModelDownloadedByCategory(category, id))) {
+    const localPath = await getLocalModelPathByCategory(category, id);
+    if (localPath) {
+      return {
+        modelId: id,
+        localPath
+      };
+    }
+  }
+  if (isArchive) {
+    const incompleteExtractions = await getIncompleteExtractions(category);
+    const extractionState = incompleteExtractions.find(s => s.modelId === id);
+    if (extractionState) {
+      return resumeExtraction(category, id, {
+        onProgress: opts?.onProgress,
+        signal: opts?.signal,
+        onChecksumIssue: opts?.onChecksumIssue,
+        deleteArchiveAfterExtract: opts?.deleteArchiveAfterExtract
+      });
+    }
+  }
+  const incompleteDownloads = await getIncompleteDownloads(category);
+  const downloadState = incompleteDownloads.find(s => s.modelId === id);
+  if (downloadState) {
+    return resumeDownload(category, id, {
+      onProgress: opts?.onProgress,
+      signal: opts?.signal,
+      onChecksumIssue: opts?.onChecksumIssue,
+      deleteArchiveAfterExtract: opts?.deleteArchiveAfterExtract
+    });
+  }
+  if (isArchive) {
+    const downloadPath = getArchivePath(category, id, model.archiveExt);
+    if (await exists(downloadPath)) {
+      try {
+        const st = await stat(downloadPath);
+        if (model.bytes <= 0 || st.size != null && st.size >= model.bytes) {
+          return extractModelByCategory(category, id, {
+            onProgress: opts?.onProgress,
+            signal: opts?.signal,
+            onChecksumIssue: opts?.onChecksumIssue,
+            deleteArchiveAfterExtract: opts?.deleteArchiveAfterExtract
+          });
+        }
+      } catch {
+        // fall through to download
+      }
+    }
+  }
+  return downloadModelByCategory(category, id, {
+    onProgress: opts?.onProgress,
+    overwrite: opts?.overwrite ?? false,
+    signal: opts?.signal,
+    onChecksumIssue: opts?.onChecksumIssue,
+    deleteArchiveAfterExtract: opts?.deleteArchiveAfterExtract
+  });
+}
+//# sourceMappingURL=ensureModel.js.map

package/lib/module/download/ensureModel.js.map

@@ -0,0 +1 @@
+{"version":3,"names":["exists","stat","getArchivePath","getModelByIdByCategory","isModelDownloadedByCategory","getLocalModelPathByCategory","deleteModelByCategory","downloadModelByCategory","getIncompleteDownloads","resumeDownload","getIncompleteExtractions","resumeExtraction","extractModelByCategory","deleteIncompleteExtraction","deleteIncompleteDownload","ensureModelByCategory","category","id","opts","model","Error","isArchive","archiveExt","overwrite","localPath","modelId","incompleteExtractions","extractionState","find","s","onProgress","signal","onChecksumIssue","deleteArchiveAfterExtract","incompleteDownloads","downloadState","downloadPath","st","bytes","size"],"sourceRoot":"../../../src","sources":["download/ensureModel.ts"],"mappings":";;AAAA,SAASA,MAAM,EAAEC,IAAI,QAAQ,6BAA6B;AAQ1D,SAASC,cAAc,QAAQ,YAAS;AACxC,SAASC,sBAAsB,QAAQ,eAAY;AACnD,SACEC,2BAA2B,EAC3BC,2BAA2B,EAC3BC,qBAAqB,QAChB,kBAAe;AACtB,SAASC,uBAAuB,QAAQ,mBAAgB;AACxD,SAASC,sBAAsB,EAAEC,cAAc,QAAQ,mBAAgB;AACvE,SACEC,wBAAwB,EACxBC,gBAAgB,EAChBC,sBAAsB,EACtBC,0BAA0B,QACrB,sBAAmB;AAC1B,SAASC,wBAAwB,QAAQ,mBAAgB;AAezD;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,OAAO,eAAeC,qBAAqBA,CACzCC,QAAuB,EACvBC,EAAU,EACVC,IAAyB,EACA;EACzB,MAAMC,KAAK,GAAG,MAAMhB,sBAAsB,CAAIa,QAAQ,EAAEC,EAAE,CAAC;EAC3D,IAAI,CAACE,KAAK,EAAE;IACV,MAAM,IAAIC,KAAK,CAAC,qBAAqBH,EAAE,EAAE,CAAC;EAC5C;EAEA,MAAMI,SAAS,GAAGF,KAAK,CAACG,UAAU,KAAK,SAAS;EAEhD,IAAIJ,IAAI,EAAEK,SAAS,EAAE;IACnB,MAAMjB,qBAAqB,CAACU,QAAQ,EAAEC,EAAE,CAAC;IACzC,MAAMJ,0BAA0B,CAACG,QAAQ,EAAEC,EAAE,CAAC;IAC9C,MAAMH,wBAAwB,CAACE,QAAQ,EAAEC,EAAE,CAAC;EAC9C;EAEA,IAAI,CAACC,IAAI,EAAEK,SAAS,KAAK,MAAMnB,2BAA2B,CAACY,QAAQ,EAAEC,EAAE,CAAC,CAAC,EAAE;IACzE,MAAMO,SAAS,GAAG,MAAMnB,2BAA2B,CAACW,QAAQ,EAAEC,EAAE,CAAC;IACjE,IAAIO,SAAS,EAAE;MACb,OAAO;QAAEC,OAAO,EAAER,EAAE;QAAEO;MAAU,CAAC;IACnC;EACF;EAEA,IAAIH,SAAS,EAAE;IACb,MAAMK,qBAAqB,GAAG,MAAMhB,wBAAwB,CAACM,QAAQ,CAAC;IACtE,MAAMW,eAAe,GAAGD,qBAAqB,CAACE,IAAI,CAAEC,CAAC,IAAKA,CAAC,CAACJ,OAAO,KAAKR,EAAE,CAAC;IAC3E,IAAIU,eAAe,EAAE;MACnB,OAAOhB,gBAAgB,CAAIK,QAAQ,EAAEC,EAAE,EAAE;QACvCa,UAAU,EAAEZ,IAAI,EAAEY,UAAU;QAC5BC,MAAM,EAAEb,IAAI,EAAEa,MAAM;QACpBC,eAAe,EAAEd,IAAI,EAAEc,eAAe;QACtCC,yBAAyB,EAAEf,IAAI,EAAEe;MACnC,CAAC,CAAC;IACJ;EACF;EAEA,MAAMC,mBAAmB,GAAG,MAAM1B,sBAAsB,CAACQ,QAAQ,CAAC;EAClE,MAAMmB,aAAa,GAAGD,mBAAmB,CAACN,IAAI,CAAEC,CAAC,IAAKA,CAAC,CAACJ,OAAO,KAAKR,EAAE,CAAC;EACvE,IAAIkB,aAAa,EAAE;IACjB,OAAO1B,cAAc,CAAIO,QAAQ,EAAEC,EAAE,EAAE;MACrCa,UAAU,EAAEZ,IAAI,EAAEY,UAAU;MAC5BC,MAAM,EAAEb,IAAI,EAAEa,MAAM;MACpBC,eAAe,EAAEd,IAAI,EAAEc,eAAe;MACtCC,yBAAyB,EAAEf,IAAI,EAAEe;IACnC,CAAC,CAAC;EACJ;EAEA,IAAIZ,SAAS,EAAE;IACb,MAAMe,YAAY,GAAGlC,cAAc,CAACc,QAAQ,EAAEC,EAAE,EAAEE,KAAK,CAACG,UAAU,CAAC;IACnE,IAAI,MAAMtB,MAAM,CAACoC,YAAY,CAAC,EAAE;MAC9B,IAAI;QACF,MAAMC,EAAE,GAAG,MAAMpC,IAAI,CAACmC,YAAY,CAAC;QACnC,IAAIjB,KAAK,CAACmB,KAAK,IAAI,CAAC,IAAKD,EAAE,CAACE,IAAI,IAAI,IAAI,IAAIF,EAAE,CAACE,IAAI,IAAIpB,KAAK,CAACmB,KAAM,EAAE;UACnE,OAAO1B,sBAAsB,CAAII,QAAQ,EAAEC,EAAE,EAAE;YAC7Ca,UAAU,EAAEZ,IAAI,EAAEY,UAAU;YAC5BC,MAAM,EAAEb,IAAI,EAAEa,MAAM;YACpBC,eAAe,EAAEd,IAAI,EAAEc,eAAe;YACtCC,yBAAyB,EAAEf,IAAI,EAAEe;UACnC,CAAC,CAAC;QACJ;MACF,CAAC,CAAC,MAAM;QACN;MAAA;IAEJ;EACF;EAEA,OAAO1B,uBAAuB,CAAIS,QAAQ,EAAEC,EAAE,EAAE;IAC9Ca,UAAU,EAAEZ,IAAI,EAAEY,UAAU;IAC5BP,SAAS,EAAEL,IAAI,EAAEK,SAAS,IAAI,KAAK;IACnCQ,MAAM,EAAEb,IAAI,EAAEa,MAAM;IACpBC,eAAe,EAAEd,IAAI,EAAEc,eAAe;IACtCC,yBAAyB,EAAEf,IAAI,EAAEe;EACnC,CAAC,CAAC;AACJ","ignoreList":[]}

package/lib/module/download/index.js

@@ -1,7 +1,7 @@
 "use strict";
 
-export { extractTarBz2 } from "./extractTarBz2.js";
-export { extractTarZst } from "./extractTarZst.js";
-export { listModelsByCategory, refreshModelsByCategory, getModelsCacheStatusByCategory, getModelByIdByCategory, listDownloadedModelsByCategory, isModelDownloadedByCategory, getLocalModelPathByCategory, downloadModelByCategory, deleteModelByCategory, clearModelCacheByCategory, getDownloadStorageBase, subscribeDownloadProgress, subscribeModelsListUpdated, updateModelLastUsed, listDownloadedModelsWithMetadata, cleanupLeastRecentlyUsed, ModelCategory } from "./ModelDownloadManager.js";
-export { validateChecksum, validateExtractedFiles, checkDiskSpace, setExpectedFilesForCategory, getExpectedFilesForCategory, parseChecksumFile, calculateFileChecksum } from "./validation.js";
+export { extractTarBz2 } from "../extraction/extractTarBz2.js";
+export { extractTarZst } from "../extraction/extractTarZst.js";
+export { listModelsByCategory, refreshModelsByCategory, getModelsCacheStatusByCategory, getModelByIdByCategory, listDownloadedModelsByCategory, isModelDownloadedByCategory, getLocalModelPathByCategory, downloadModelByCategory, deleteModelByCategory, clearModelCacheByCategory, getDownloadStorageBase, subscribeDownloadProgress, subscribeModelsListUpdated, updateModelLastUsed, listDownloadedModelsWithMetadata, cleanupLeastRecentlyUsed, getIncompleteDownloads, resumeDownload, deleteIncompleteDownload, extractModelByCategory, getIncompleteExtractions, resumeExtraction, deleteIncompleteExtraction, ensureModelByCategory, ModelCategory, getProtectedModelKeysForBulkDelete, purgeDownloadedModelArtifacts } from "./ModelDownloadManager.js";
+export { validateChecksum, validateExtractedFiles, checkDiskSpace, resolveActualModelDir, setExpectedFilesForCategory, getExpectedFilesForCategory, parseChecksumFile, calculateFileChecksum } from "./validation.js";
 //# sourceMappingURL=index.js.map

package/lib/module/download/index.js.map

@@ -1 +1 @@
-{"version":3,"names":["extractTarBz2","extractTarZst","listModelsByCategory","refreshModelsByCategory","getModelsCacheStatusByCategory","getModelByIdByCategory","listDownloadedModelsByCategory","isModelDownloadedByCategory","getLocalModelPathByCategory","downloadModelByCategory","deleteModelByCategory","clearModelCacheByCategory","getDownloadStorageBase","subscribeDownloadProgress","subscribeModelsListUpdated","updateModelLastUsed","listDownloadedModelsWithMetadata","cleanupLeastRecentlyUsed","ModelCategory","validateChecksum","validateExtractedFiles","checkDiskSpace","setExpectedFilesForCategory","getExpectedFilesForCategory","parseChecksumFile","calculateFileChecksum"],"sourceRoot":"../../../src","sources":["download/index.ts"],"mappings":";;AAAA,SAASA,aAAa,QAAQ,oBAAiB;AAE/C,SAASC,aAAa,QAAQ,oBAAiB;AAE/C,SACEC,oBAAoB,EACpBC,uBAAuB,EACvBC,8BAA8B,EAC9BC,sBAAsB,EACtBC,8BAA8B,EAC9BC,2BAA2B,EAC3BC,2BAA2B,EAC3BC,uBAAuB,EACvBC,qBAAqB,EACrBC,yBAAyB,EACzBC,sBAAsB,EACtBC,yBAAyB,EACzBC,0BAA0B,EAC1BC,mBAAmB,EACnBC,gCAAgC,EAChCC,wBAAwB,EACxBC,aAAa,QACR,2BAAwB;AAa/B,SACEC,gBAAgB,EAChBC,sBAAsB,EACtBC,cAAc,EACdC,2BAA2B,EAC3BC,2BAA2B,EAC3BC,iBAAiB,EACjBC,qBAAqB,QAChB,iBAAc","ignoreList":[]}
+{"version":3,"names":["extractTarBz2","extractTarZst","listModelsByCategory","refreshModelsByCategory","getModelsCacheStatusByCategory","getModelByIdByCategory","listDownloadedModelsByCategory","isModelDownloadedByCategory","getLocalModelPathByCategory","downloadModelByCategory","deleteModelByCategory","clearModelCacheByCategory","getDownloadStorageBase","subscribeDownloadProgress","subscribeModelsListUpdated","updateModelLastUsed","listDownloadedModelsWithMetadata","cleanupLeastRecentlyUsed","getIncompleteDownloads","resumeDownload","deleteIncompleteDownload","extractModelByCategory","getIncompleteExtractions","resumeExtraction","deleteIncompleteExtraction","ensureModelByCategory","ModelCategory","getProtectedModelKeysForBulkDelete","purgeDownloadedModelArtifacts","validateChecksum","validateExtractedFiles","checkDiskSpace","resolveActualModelDir","setExpectedFilesForCategory","getExpectedFilesForCategory","parseChecksumFile","calculateFileChecksum"],"sourceRoot":"../../../src","sources":["download/index.ts"],"mappings":";;AAAA,SAASA,aAAa,QAAQ,gCAA6B;AAE3D,SAASC,aAAa,QAAQ,gCAA6B;AAE3D,SACEC,oBAAoB,EACpBC,uBAAuB,EACvBC,8BAA8B,EAC9BC,sBAAsB,EACtBC,8BAA8B,EAC9BC,2BAA2B,EAC3BC,2BAA2B,EAC3BC,uBAAuB,EACvBC,qBAAqB,EACrBC,yBAAyB,EACzBC,sBAAsB,EACtBC,yBAAyB,EACzBC,0BAA0B,EAC1BC,mBAAmB,EACnBC,gCAAgC,EAChCC,wBAAwB,EACxBC,sBAAsB,EACtBC,cAAc,EACdC,wBAAwB,EACxBC,sBAAsB,EACtBC,wBAAwB,EACxBC,gBAAgB,EAChBC,0BAA0B,EAC1BC,qBAAqB,EACrBC,aAAa,EACbC,kCAAkC,EAClCC,6BAA6B,QACxB,2BAAwB;AAiB/B,SACEC,gBAAgB,EAChBC,sBAAsB,EACtBC,cAAc,EACdC,qBAAqB,EACrBC,2BAA2B,EAC3BC,2BAA2B,EAC3BC,iBAAiB,EACjBC,qBAAqB,QAChB,iBAAc","ignoreList":[]}