react-native-sherpa-onnx 0.3.6 → 0.3.8

package/lib/module/download/downloadTask.js

@@ -0,0 +1,438 @@
+"use strict";
+
+import { Platform } from 'react-native';
+import { createDownloadTask, completeHandler, getExistingDownloadTasks, setConfig } 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();
+let androidDownloaderNotificationConfigApplied = false;
+let didWarnConfigFailure = false;
+function warnBackgroundDownloaderConfigFailure(context, error) {
+  if (didWarnConfigFailure) return;
+  didWarnConfigFailure = true;
+  const reason = error instanceof Error ? error.message : String(error);
+  console.warn(`[Download] Background downloader config failed (${context}): ${reason}`);
+}
+/**
+ * Apply your own `@kesha-antonov/react-native-background-downloader` `setConfig` **before** the first
+ * model download. When called, the SDK will **not** overwrite it with built-in defaults on first download.
+ *
+ * Safe to call at app startup (e.g. `App.tsx`). Other `setConfig` options (e.g. headers) are forwarded
+ * where the native module supports them on each platform.
+ */
+export function configureModelDownloadBackgroundDownloader(options) {
+  try {
+    setConfig(options);
+    androidDownloaderNotificationConfigApplied = true;
+  } catch (error) {
+    // Keep fallback default config enabled if custom config fails.
+    warnBackgroundDownloaderConfigFailure('custom', error);
+  }
+}
+
+/**
+ * Library default is showNotificationsEnabled: false (silent empty FGS notification).
+ * Enable visible notifications unless the host app already called `configureModelDownloadBackgroundDownloader`.
+ */
+function ensureAndroidBackgroundDownloaderNotifications() {
+  if (androidDownloaderNotificationConfigApplied) return;
+  if (Platform.OS !== 'android') return;
+  try {
+    setConfig({
+      showNotificationsEnabled: true,
+      notificationsGrouping: {
+        enabled: false,
+        mode: 'individual',
+        texts: {
+          downloadTitle: 'Model download',
+          downloadStarting: 'Starting download…',
+          downloadProgress: 'Downloading… {progress}%'
+        }
+      }
+    });
+    androidDownloaderNotificationConfigApplied = true;
+  } catch (error) {
+    warnBackgroundDownloaderConfigFailure('default', error);
+  }
+}
+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}`);
+  }
+  ensureAndroidBackgroundDownloaderNotifications();
+  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) {
+  ensureAndroidBackgroundDownloaderNotifications();
+  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":["Platform","createDownloadTask","completeHandler","getExistingDownloadTasks","setConfig","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","androidDownloaderNotificationConfigApplied","didWarnConfigFailure","warnBackgroundDownloaderConfigFailure","context","error","reason","Error","message","String","console","warn","configureModelDownloadBackgroundDownloader","options","ensureAndroidBackgroundDownloaderNotifications","OS","showNotificationsEnabled","notificationsGrouping","enabled","mode","texts","downloadTitle","downloadStarting","downloadProgress","downloadModelByCategory","opts","isAborted","Boolean","signal","aborted","abortError","name","model","baseDir","downloadPath","archiveExt","isArchive","modelDir","sleep","ms","Promise","resolve","setTimeout","cleanupPartial","cleanupPartialWithRetry","attempt","diskSpaceCheck","bytes","success","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","errorCode","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,SAASA,QAAQ,QAAQ,cAAc;AAEvC,SACEC,kBAAkB,EAClBC,eAAe,EACfC,wBAAwB,EACxBC,SAAS,QACJ,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,IAAIC,0CAA0C,GAAG,KAAK;AACtD,IAAIC,oBAAoB,GAAG,KAAK;AAEhC,SAASC,qCAAqCA,CAC5CC,OAAe,EACfC,KAAc,EACd;EACA,IAAIH,oBAAoB,EAAE;EAC1BA,oBAAoB,GAAG,IAAI;EAC3B,MAAMI,MAAM,GAAGD,KAAK,YAAYE,KAAK,GAAGF,KAAK,CAACG,OAAO,GAAGC,MAAM,CAACJ,KAAK,CAAC;EACrEK,OAAO,CAACC,IAAI,CACV,mDAAmDP,OAAO,MAAME,MAAM,EACxE,CAAC;AACH;AAIA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,OAAO,SAASM,0CAA0CA,CACxDC,OAA6C,EACvC;EACN,IAAI;IACFtC,SAAS,CAACsC,OAAO,CAAC;IAClBZ,0CAA0C,GAAG,IAAI;EACnD,CAAC,CAAC,OAAOI,KAAK,EAAE;IACd;IACAF,qCAAqC,CAAC,QAAQ,EAAEE,KAAK,CAAC;EACxD;AACF;;AAEA;AACA;AACA;AACA;AACA,SAASS,8CAA8CA,CAAA,EAAG;EACxD,IAAIb,0CAA0C,EAAE;EAChD,IAAI9B,QAAQ,CAAC4C,EAAE,KAAK,SAAS,EAAE;EAC/B,IAAI;IACFxC,SAAS,CAAC;MACRyC,wBAAwB,EAAE,IAAI;MAC9BC,qBAAqB,EAAE;QACrBC,OAAO,EAAE,KAAK;QACdC,IAAI,EAAE,YAAY;QAClBC,KAAK,EAAE;UACLC,aAAa,EAAE,gBAAgB;UAC/BC,gBAAgB,EAAE,oBAAoB;UACtCC,gBAAgB,EAAE;QACpB;MACF;IACF,CAAC,CAAC;IACFtB,0CAA0C,GAAG,IAAI;EACnD,CAAC,CAAC,OAAOI,KAAK,EAAE;IACdF,qCAAqC,CAAC,SAAS,EAAEE,KAAK,CAAC;EACzD;AACF;AAEA,OAAO,eAAemB,uBAAuBA,CAC3C3B,QAAuB,EACvBC,EAAU,EACV2B,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,IAAIvB,KAAK,CAAC,kBAAkB,CAAC;IAChDuB,UAAU,CAACC,IAAI,GAAG,YAAY;IAC9B,MAAMD,UAAU;EAClB;EAEA,MAAME,KAAK,GAAG,MAAMtC,sBAAsB,CAAIG,QAAQ,EAAEC,EAAE,CAAC;EAC3D,IAAI,CAACkC,KAAK,EAAE;IACV,MAAM,IAAIzB,KAAK,CAAC,qBAAqBT,EAAE,EAAE,CAAC;EAC5C;EAEA,MAAMmC,OAAO,GAAGjD,gBAAgB,CAACa,QAAQ,CAAC;EAC1C,MAAMnB,KAAK,CAACuD,OAAO,CAAC;EAEpB,MAAMC,YAAY,GAAGhD,cAAc,CAACW,QAAQ,EAAEC,EAAE,EAAEkC,KAAK,CAACG,UAAU,CAAC;EACnE,MAAMC,SAAS,GAAGJ,KAAK,CAACG,UAAU,KAAK,SAAS;EAChD,MAAME,QAAQ,GAAGpD,WAAW,CAACY,QAAQ,EAAEC,EAAE,CAAC;EAE1C,MAAMwC,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,MAAM5D,MAAM,CAAC6D,QAAQ,CAAC,EAAE;MAC1B,MAAMxD,MAAM,CAACwD,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,MAAMnE,MAAM,CAAC6D,QAAQ,CAAC,CAAC,EAAE;MAC/B,MAAMC,KAAK,CAAC,GAAG,CAAC;IAClB;IACA,IAAI,MAAM9D,MAAM,CAAC6D,QAAQ,CAAC,EAAE;MAC1B3B,OAAO,CAACC,IAAI,CACV,wDAAwDd,QAAQ,IAAIC,EAAE,EACxE,CAAC;IACH;EACF,CAAC;EAED,MAAMgD,cAAc,GAAG,MAAMhE,cAAc,CAACkD,KAAK,CAACe,KAAK,CAAC;EACxD,IAAI,CAACD,cAAc,CAACE,OAAO,EAAE;IAC3B,MAAM,IAAIzC,KAAK,CAAC,4BAA4BuC,cAAc,CAACtC,OAAO,EAAE,CAAC;EACvE;EAEAM,8CAA8C,CAAC,CAAC;EAEhD,MAAMmC,SAAS,GAAG7D,oBAAoB,CAACS,QAAQ,EAAEC,EAAE,CAAC;EAEpD,IAAI2B,IAAI,EAAEyB,SAAS,EAAE;IACnB,IAAI,MAAM1E,MAAM,CAAC6D,QAAQ,CAAC,EAAE,MAAMxD,MAAM,CAACwD,QAAQ,CAAC;IAClD,IAAI,MAAM7D,MAAM,CAAC0D,YAAY,CAAC,EAAE,MAAMrD,MAAM,CAACqD,YAAY,CAAC;IAC1D,IAAI,MAAM1D,MAAM,CAACyE,SAAS,CAAC,EAAE,MAAMpE,MAAM,CAACoE,SAAS,CAAC;EACtD,CAAC,MAAM;IACL,MAAME,iBAAiB,GAAG,MAAM3E,MAAM,CAACW,kBAAkB,CAACU,QAAQ,EAAEC,EAAE,CAAC,CAAC;IACxE,IAAI,CAACqD,iBAAiB,IAAIf,SAAS,EAAE;MACnC,IAAI,MAAM5D,MAAM,CAAC6D,QAAQ,CAAC,EAAE,MAAMxD,MAAM,CAACwD,QAAQ,CAAC;IACpD;EACF;EAEA,IAAI;IACF,MAAMe,aAA4B,GAAG;MACnCC,OAAO,EAAEvD,EAAE;MACXD,QAAQ;MACRyD,KAAK,EAAE,aAAa;MACpBC,SAAS,EAAE,IAAIC,IAAI,CAAC,CAAC,CAACC,WAAW,CAAC,CAAC;MACnCC,WAAW,EAAExB,YAAY;MACzBF,KAAK;MACL2B,UAAU,EAAE3B,KAAK,CAACe;IACpB,CAAC;IACD,MAAMrE,KAAK,CAACM,gBAAgB,CAACa,QAAQ,CAAC,CAAC;IACvC,MAAMlB,SAAS,CAACsE,SAAS,EAAEW,IAAI,CAACC,SAAS,CAACT,aAAa,CAAC,EAAE,MAAM,CAAC;IAEjE,IAAI,CAAChB,SAAS,EAAE;MACd,MAAM1D,KAAK,CAAC2D,QAAQ,CAAC;IACvB;IAEA,MAAMyB,MAAM,GAAGlE,kBAAkB,CAACC,QAAQ,EAAEC,EAAE,CAAC;IAE/C,OAAO,IAAI0C,OAAO,CAAiB,CAACC,OAAO,EAAEsB,MAAM,KAAK;MACtD,IAAIC,YAAsC;MAE1C,MAAMC,OAAO,GAAGA,CAAA,KAAM;QACpB,IAAID,YAAY,IAAIvC,IAAI,EAAEG,MAAM,EAAE;UAChCH,IAAI,CAACG,MAAM,CAACsC,mBAAmB,CAAC,OAAO,EAAEF,YAAY,CAAC;UACtDA,YAAY,GAAGG,SAAS;QAC1B;QACApE,mBAAmB,CAACqE,MAAM,CAACN,MAAM,CAAC;MACpC,CAAC;MAED,MAAMO,IAAI,GAAGjG,kBAAkB,CAAC;QAC9B0B,EAAE,EAAEgE,MAAM;QACVQ,GAAG,EAAEtC,KAAK,CAACuC,WAAW;QACtBC,WAAW,EAAEtC,YAAY;QACzBuC,QAAQ,EAAE,CAAC;MACb,CAAC,CAAC,CACCC,QAAQ,CACP,CAAC;QACCC,eAAe;QACfC;MAIF,CAAC,KAAK;QACJ,IAAIlD,SAAS,CAAC,CAAC,EAAE;QACjB,MAAMmD,KAAK,GAAGD,UAAU,IAAI5C,KAAK,CAACe,KAAK,IAAI,CAAC;QAC5C,MAAM+B,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;QACD7B,IAAI,EAAEsD,UAAU,GAAGL,QAAQ,CAAC;QAC5BlF,oBAAoB,CAACK,QAAQ,EAAEC,EAAE,EAAE4E,QAAQ,CAAC;MAC9C,CACF,CAAC,CACAM,IAAI,CAAC,YAAY;QAChBf,OAAO,CAAC,CAAC;QACT,IAAI;UACF,MAAMgB,MAAM,GAAG,MAAMxF,yBAAyB,CAAC;YAC7CI,QAAQ;YACRC,EAAE;YACFkC,KAAK;YACLE,YAAY;YACZG,QAAQ;YACRD,SAAS;YACTa,SAAS;YACTrB,MAAM,EAAEH,IAAI,EAAEG,MAAM;YACpBsD,eAAe,EAAEzD,IAAI,EAAEyD,eAAe;YACtCC,yBAAyB,EAAE1D,IAAI,EAAE0D,yBAAyB;YAC1DJ,UAAU,EAAEtD,IAAI,EAAEsD,UAAU;YAC5BK,iBAAiB,EAAEA,CAAA,KACjBzF,8BAA8B,CAAgBE,QAAQ;UAC1D,CAAC,CAAC;UACFxB,eAAe,CAACyF,MAAM,CAAC;UACvBrB,OAAO,CAACwC,MAAM,CAAC;QACjB,CAAC,CAAC,OAAOI,CAAC,EAAE;UACVhH,eAAe,CAACyF,MAAM,CAAC;UACvBC,MAAM,CAACsB,CAAC,CAAC;QACX;MACF,CAAC,CAAC,CACDhF,KAAK,CACJ,CAAC;QAAEA,KAAK;QAAEiF;MAAkD,CAAC,KAAK;QAChErB,OAAO,CAAC,CAAC;QACT5F,eAAe,CAACyF,MAAM,CAAC;QACvB,CAAC,YAAY;UACX,IAAI;YACF,IAAI,MAAMtF,MAAM,CAACyE,SAAS,CAAC,EAAE,MAAMpE,MAAM,CAACoE,SAAS,CAAC;UACtD,CAAC,CAAC,MAAM;YACN;UAAA;QAEJ,CAAC,EAAE,CAAC;QACJc,MAAM,CACJ,IAAIxD,KAAK,CACP,OAAOF,KAAK,KAAK,QAAQ,GAAGA,KAAK,GAAGI,MAAM,CAAC6E,SAAS,IAAIjF,KAAK,CAC/D,CACF,CAAC;MACH,CACF,CAAC;MAEHN,mBAAmB,CAACwF,GAAG,CAACzB,MAAM,EAAEO,IAAI,CAAC;MACrC,IAAI5C,IAAI,EAAEG,MAAM,EAAE;QAChBoC,YAAY,GAAGA,CAAA,KAAM;UACnBK,IAAI,CAACmB,IAAI,CAAC,CAAC;UACXvB,OAAO,CAAC,CAAC;UACT,CAAC,YAAY;YACX,IAAI;cACF,IAAI,MAAMzF,MAAM,CAACyE,SAAS,CAAC,EAAE,MAAMpE,MAAM,CAACoE,SAAS,CAAC;YACtD,CAAC,CAAC,MAAM;cACN;YAAA;UAEJ,CAAC,EAAE,CAAC;UACJ,MAAMwC,GAAG,GAAG,IAAIlF,KAAK,CAAC,kBAAkB,CAAC;UACzCkF,GAAG,CAAC1D,IAAI,GAAG,YAAY;UACvBgC,MAAM,CAAC0B,GAAG,CAAC;QACb,CAAC;QACDhE,IAAI,CAACG,MAAM,CAAC8D,gBAAgB,CAAC,OAAO,EAAE1B,YAAY,CAAC;MACrD;MACAK,IAAI,CAACsB,KAAK,CAAC,CAAC;IACd,CAAC,CAAC;EACJ,CAAC,CAAC,OAAOF,GAAG,EAAE;IACZ,IAAKA,GAAG,YAAYlF,KAAK,IAAIkF,GAAG,CAAC1D,IAAI,KAAK,YAAY,IAAKL,SAAS,CAAC,CAAC,EAAE;MACtE,MAAMkB,uBAAuB,CAAC,CAAC;MAC/B,IAAI;QACF,IAAI,MAAMpE,MAAM,CAACyE,SAAS,CAAC,EAAE,MAAMpE,MAAM,CAACoE,SAAS,CAAC;MACtD,CAAC,CAAC,MAAM;QACN;MAAA;IAEJ;IACA,IAAIb,SAAS,IAAI,EAAEqD,GAAG,YAAYlF,KAAK,IAAIkF,GAAG,CAAC1D,IAAI,KAAK,YAAY,CAAC,EAAE;MACrE,IAAI;QACF,IAAI,MAAMvD,MAAM,CAAC0D,YAAY,CAAC,EAAE;UAC9B,MAAM0D,WAAW,GAAG,MAAMhH,IAAI,CAACsD,YAAY,CAAC;UAC5C,IAAIF,KAAK,CAACe,KAAK,GAAG,CAAC,IAAI6C,WAAW,CAACC,IAAI,GAAG7D,KAAK,CAACe,KAAK,EAAE;YACrDrC,OAAO,CAACC,IAAI,CACV,6CAA6Cd,QAAQ,IAAIC,EAAE,KAAK8F,WAAW,CAACC,IAAI,IAAI7D,KAAK,CAACe,KAAK,GACjG,CAAC;YACD,MAAMlE,MAAM,CAACqD,YAAY,CAAC;UAC5B;QACF;MACF,CAAC,CAAC,MAAM;QACN;MAAA;IAEJ;IACA,MAAMuD,GAAG;EACX;AACF;AAEA,OAAO,eAAeK,sBAAsBA,CAC1CjG,QAAuB,EACG;EAC1B,MAAMkG,MAAM,GAAGlG,QAAQ,GAAG,GAAG;EAC7B,MAAMmG,MAAuB,GAAG,EAAE;EAElC,MAAMC,aAAa,GAAG,MAAM3H,wBAAwB,CAAC,CAAC;EACtD,KAAK,MAAM+F,IAAI,IAAI4B,aAAa,EAAE;IAChC,IAAI,CAAC5B,IAAI,CAACvE,EAAE,IAAI,CAACuE,IAAI,CAACvE,EAAE,CAACoG,UAAU,CAACH,MAAM,CAAC,EAAE;IAC7C,MAAM1C,OAAO,GAAGgB,IAAI,CAACvE,EAAE,CAACqG,KAAK,CAACJ,MAAM,CAACK,MAAM,CAAC;IAC5C,MAAMC,SAAS,GAAGlH,kBAAkB,CAACU,QAAQ,EAAEwD,OAAO,CAAC;IACvD,IAAI,MAAM7E,MAAM,CAAC6H,SAAS,CAAC,EAAE;IAE7B,MAAMpD,SAAS,GAAG7D,oBAAoB,CAACS,QAAQ,EAAEwD,OAAO,CAAC;IACzD,IAAIrB,KAAgC;IACpC,IAAI2B,UAA8B;IAClC,IAAID,WAA+B;IACnC,IAAIH,SAA6B;IAEjC,IAAI,MAAM/E,MAAM,CAACyE,SAAS,CAAC,EAAE;MAC3B,IAAI;QACF,MAAMqD,GAAG,GAAG,MAAM7H,QAAQ,CAACwE,SAAS,EAAE,MAAM,CAAC;QAC7C,MAAMsD,QAAQ,GAAG3C,IAAI,CAAC4C,KAAK,CAACF,GAAG,CAAkB;QACjDtE,KAAK,GAAGuE,QAAQ,CAACvE,KAAK;QACtB2B,UAAU,GAAG4C,QAAQ,CAAC5C,UAAU,IAAI4C,QAAQ,CAACvE,KAAK,EAAEe,KAAK;QACzDW,WAAW,GAAG6C,QAAQ,CAAC7C,WAAW;QAClCH,SAAS,GAAGgD,QAAQ,CAAChD,SAAS;MAChC,CAAC,CAAC,MAAM;QACN;MAAA;IAEJ;IACA,IAAI,CAACvB,KAAK,EAAE;MACV,MAAMyE,IAAI,GAAG,MAAM/G,sBAAsB,CAACG,QAAQ,EAAEwD,OAAO,CAAC;MAC5D,IAAI,CAACoD,IAAI,EAAE;MACXzE,KAAK,GAAGyE,IAAqB;MAC7B9C,UAAU,GAAG3B,KAAK,CAACe,KAAK;MACxBW,WAAW,GAAGxE,cAAc,CAACW,QAAQ,EAAEwD,OAAO,EAAErB,KAAK,CAACG,UAAU,CAAC;IACnE;IAEA,IAAIwC,eAAmC;IACvC,IAAIjB,WAAW,EAAE;MACf,IAAI;QACF,MAAMgD,EAAE,GAAG,MAAM9H,IAAI,CAAC8E,WAAW,CAAC;QAClC,IAAIgD,EAAE,EAAEb,IAAI,IAAI,IAAI,IAAIa,EAAE,CAACb,IAAI,IAAI,CAAC,EAAElB,eAAe,GAAG+B,EAAE,CAACb,IAAI;MACjE,CAAC,CAAC,MAAM;QACN;MAAA;IAEJ;IAEAG,MAAM,CAACW,IAAI,CAAC;MACVtD,OAAO;MACPxD,QAAQ;MACRyD,KAAK,EAAE,aAAa;MACpBC,SAAS,EAAEA,SAAS,IAAI,IAAIC,IAAI,CAAC,CAAC,CAACC,WAAW,CAAC,CAAC;MAChDC,WAAW,EAAEA,WAAW,IAAI,EAAE;MAC9B1B,KAAK;MACL2C,eAAe;MACfhB,UAAU,EAAEA,UAAU,IAAI3B,KAAK,CAACe;IAClC,CAAC,CAAC;EACJ;EAEA,OAAOiD,MAAM;AACf;AAEA,OAAO,eAAeY,cAAcA,CAClC/G,QAAuB,EACvBC,EAAU,EACV2B,IAKC,EACwB;EACzBX,8CAA8C,CAAC,CAAC;EAChD,MAAMgD,MAAM,GAAGlE,kBAAkB,CAACC,QAAQ,EAAEC,EAAE,CAAC;EAC/C,MAAMmG,aAAa,GAAG,MAAM3H,wBAAwB,CAAC,CAAC;EACtD,MAAMuI,QAAQ,GAAGZ,aAAa,CAACa,IAAI,CAAEC,CAAC,IAAKA,CAAC,CAACjH,EAAE,KAAKgE,MAAM,CAAC;EAC3D,IAAI,CAAC+C,QAAQ,EAAE;IACb,OAAOrF,uBAAuB,CAAI3B,QAAQ,EAAEC,EAAE,EAAE;MAC9CiF,UAAU,EAAEtD,IAAI,EAAEsD,UAAU;MAC5BnD,MAAM,EAAEH,IAAI,EAAEG,MAAM;MACpBsD,eAAe,EAAEzD,IAAI,EAAEyD,eAAe;MACtCC,yBAAyB,EAAE1D,IAAI,EAAE0D;IACnC,CAAC,CAAC;EACJ;EAEA,MAAMnD,KAAK,GAAG,MAAMtC,sBAAsB,CAAIG,QAAQ,EAAEC,EAAE,CAAC;EAC3D,IAAI,CAACkC,KAAK,EAAE,MAAM,IAAIzB,KAAK,CAAC,qBAAqBT,EAAE,EAAE,CAAC;EACtD,MAAMoC,YAAY,GAAGhD,cAAc,CAACW,QAAQ,EAAEC,EAAE,EAAEkC,KAAK,CAACG,UAAU,CAAC;EACnE,MAAME,QAAQ,GAAGpD,WAAW,CAACY,QAAQ,EAAEC,EAAE,CAAC;EAC1C,MAAMsC,SAAS,GAAGJ,KAAK,CAACG,UAAU,KAAK,SAAS;EAChD,MAAMc,SAAS,GAAG7D,oBAAoB,CAACS,QAAQ,EAAEC,EAAE,CAAC;EACpD,MAAM4B,SAAS,GAAGA,CAAA,KAAMC,OAAO,CAACF,IAAI,EAAEG,MAAM,EAAEC,OAAO,CAAC;EAEtD,OAAO,IAAIW,OAAO,CAAiB,CAACC,OAAO,EAAEsB,MAAM,KAAK;IACtD,IAAIC,YAAsC;IAE1C,MAAMC,OAAO,GAAGA,CAAA,KAAM;MACpB,IAAID,YAAY,IAAIvC,IAAI,EAAEG,MAAM,EAAE;QAChCH,IAAI,CAACG,MAAM,CAACsC,mBAAmB,CAAC,OAAO,EAAEF,YAAY,CAAC;QACtDA,YAAY,GAAGG,SAAS;MAC1B;MACApE,mBAAmB,CAACqE,MAAM,CAACN,MAAM,CAAC;IACpC,CAAC;IAED+C,QAAQ,CACLnC,QAAQ,CACP,CAAC;MACCC,eAAe;MACfC;IAIF,CAAC,KAAK;MACJ,IAAIlD,SAAS,CAAC,CAAC,EAAE;MACjB,MAAMmD,KAAK,GAAGD,UAAU,IAAI5C,KAAK,CAACe,KAAK,IAAI,CAAC;MAC5C,MAAM+B,OAAO,GAAGD,KAAK,GAAG,CAAC,GAAIF,eAAe,GAAGE,KAAK,GAAI,GAAG,GAAG,CAAC;MAC/DpD,IAAI,EAAEsD,UAAU,GAAG;QACjBJ,eAAe;QACfhB,UAAU,EAAEkB,KAAK;QACjBC,OAAO;QACPxB,KAAK,EAAE;MACT,CAAC,CAAC;MACF9D,oBAAoB,CAACK,QAAQ,EAAEC,EAAE,EAAE;QACjC6E,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,MAAMxF,yBAAyB,CAAC;UAC7CI,QAAQ;UACRC,EAAE;UACFkC,KAAK;UACLE,YAAY;UACZG,QAAQ;UACRD,SAAS;UACTa,SAAS;UACTrB,MAAM,EAAEH,IAAI,EAAEG,MAAM;UACpBsD,eAAe,EAAEzD,IAAI,EAAEyD,eAAe;UACtCC,yBAAyB,EAAE1D,IAAI,EAAE0D,yBAAyB;UAC1DJ,UAAU,EAAEtD,IAAI,EAAEsD,UAAU;UAC5BK,iBAAiB,EAAEA,CAAA,KACjBzF,8BAA8B,CAAgBE,QAAQ;QAC1D,CAAC,CAAC;QACFxB,eAAe,CAACyF,MAAM,CAAC;QACvBrB,OAAO,CAACwC,MAAM,CAAC;MACjB,CAAC,CAAC,OAAOI,CAAC,EAAE;QACVhH,eAAe,CAACyF,MAAM,CAAC;QACvBC,MAAM,CAACsB,CAAC,CAAC;MACX;IACF,CAAC,CAAC,CACDhF,KAAK,CAAC,CAAC;MAAEA,KAAK;MAAEiF;IAAkD,CAAC,KAAK;MACvErB,OAAO,CAAC,CAAC;MACT5F,eAAe,CAACyF,MAAM,CAAC;MACvB,CAAC,YAAY;QACX,IAAI;UACF,IAAI,MAAMtF,MAAM,CAACyE,SAAS,CAAC,EAAE,MAAMpE,MAAM,CAACoE,SAAS,CAAC;QACtD,CAAC,CAAC,MAAM;UACN;QAAA;MAEJ,CAAC,EAAE,CAAC;MACJc,MAAM,CACJ,IAAIxD,KAAK,CACP,OAAOF,KAAK,KAAK,QAAQ,GAAGA,KAAK,GAAGI,MAAM,CAAC6E,SAAS,IAAIjF,KAAK,CAC/D,CACF,CAAC;IACH,CAAC,CAAC;IAEJN,mBAAmB,CAACwF,GAAG,CAACzB,MAAM,EAAE+C,QAAQ,CAAC;IACzC,IAAIpF,IAAI,EAAEG,MAAM,EAAE;MAChBoC,YAAY,GAAGA,CAAA,KAAM;QACnB6C,QAAQ,CAACrB,IAAI,CAAC,CAAC;QACfvB,OAAO,CAAC,CAAC;QACT,CAAC,YAAY;UACX,IAAI;YACF,IAAI,MAAMzF,MAAM,CAACyE,SAAS,CAAC,EAAE,MAAMpE,MAAM,CAACoE,SAAS,CAAC;UACtD,CAAC,CAAC,MAAM;YACN;UAAA;QAEJ,CAAC,EAAE,CAAC;QACJ,MAAMwC,GAAG,GAAG,IAAIlF,KAAK,CAAC,kBAAkB,CAAC;QACzCkF,GAAG,CAAC1D,IAAI,GAAG,YAAY;QACvBgC,MAAM,CAAC0B,GAAG,CAAC;MACb,CAAC;MACDhE,IAAI,CAACG,MAAM,CAAC8D,gBAAgB,CAAC,OAAO,EAAE1B,YAAY,CAAC;IACrD;IACA6C,QAAQ,CAACG,MAAM,CAAC,CAAC,CAACC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;EACnC,CAAC,CAAC;AACJ;AAEA,OAAO,eAAeC,wBAAwBA,CAC5CrH,QAAuB,EACvBC,EAAU,EACK;EACf,MAAMgE,MAAM,GAAGlE,kBAAkB,CAACC,QAAQ,EAAEC,EAAE,CAAC;EAC/C,MAAMmG,aAAa,GAAG,MAAM3H,wBAAwB,CAAC,CAAC;EACtD,MAAM+F,IAAI,GAAG4B,aAAa,CAACa,IAAI,CAAEC,CAAC,IAAKA,CAAC,CAACjH,EAAE,KAAKgE,MAAM,CAAC;EACvD,IAAIO,IAAI,EAAE;IACRA,IAAI,CAACmB,IAAI,CAAC,CAAC;IACXzF,mBAAmB,CAACqE,MAAM,CAACN,MAAM,CAAC;EACpC;EAEA,MAAMzB,QAAQ,GAAGpD,WAAW,CAACY,QAAQ,EAAEC,EAAE,CAAC;EAC1C,IAAI,MAAMtB,MAAM,CAAC6D,QAAQ,CAAC,EAAE;IAC1B,MAAMxD,MAAM,CAACwD,QAAQ,CAAC;EACxB;EACA,MAAM8E,OAAO,GAAG7H,iBAAiB,CAACO,QAAQ,EAAEC,EAAE,CAAC;EAC/C,MAAMsH,QAAQ,GAAG7H,WAAW,CAACM,QAAQ,EAAEC,EAAE,CAAC;EAC1C,IAAI,MAAMtB,MAAM,CAAC2I,OAAO,CAAC,EAAE;IACzB,MAAMtI,MAAM,CAACsI,OAAO,CAAC;EACvB;EACA,IAAI,MAAM3I,MAAM,CAAC4I,QAAQ,CAAC,EAAE;IAC1B,MAAMvI,MAAM,CAACuI,QAAQ,CAAC;EACxB;EACA,MAAMnE,SAAS,GAAG7D,oBAAoB,CAACS,QAAQ,EAAEC,EAAE,CAAC;EACpD,IAAI,MAAMtB,MAAM,CAACyE,SAAS,CAAC,EAAE;IAC3B,MAAMpE,MAAM,CAACoE,SAAS,CAAC;EACzB;EACA,MAAMlE,wBAAwB,CAACM,+BAA+B,CAACS,EAAE,CAAC,CAAC;AACrE;;AAEA;AACA,OAAO,SAASuH,yBAAyBA,CAAA,EAAa;EACpD,OAAO,CAAC,GAAGtH,mBAAmB,CAACuH,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, configureModelDownloadBackgroundDownloader } 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","configureModelDownloadBackgroundDownloader","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,EAC7BC,0CAA0C,QACrC,2BAAwB;AAkB/B,SACEC,gBAAgB,EAChBC,sBAAsB,EACtBC,cAAc,EACdC,qBAAqB,EACrBC,2BAA2B,EAC3BC,2BAA2B,EAC3BC,iBAAiB,EACjBC,qBAAqB,QAChB,iBAAc","ignoreList":[]}