@atomic-chat/core 0.5.1 → 0.13.2

package/dist/core/model-setup/service.js

@@ -0,0 +1,329 @@
+/**
+ * The model-setup runner: `queued → installing_engine → downloading_model → downloading_projector →
+ * verifying → registering → ready`, or `failed | cancelled`; a setup a stopped core left mid-way is
+ * `interrupted` at the next start and continues on `resume`, from the files already on disk.
+ *
+ * The PrismML pack is shared: two setups that need the same pack wait on one install, and cancelling
+ * one of them stops its wait, never the install the other still needs. Credentials are never stored:
+ * the Hugging Face token goes into the download headers of the run that uses it.
+ */
+import { stat } from 'node:fs/promises';
+import { join, relative, sep } from 'node:path';
+import { AtomicCoreError, MODEL_SETUP_FINAL_STAGES } from '../../contracts/index.js';
+import { modelDirFromId } from '../../config/index.js';
+import { hfResolveUrl } from '../../models/index.js';
+/** A download task id the progress events can carry: no characters a path or event name chokes on. */
+export function setupTaskId(raw) {
+    return raw.replace(/[^A-Za-z0-9_/:.-]/g, '_');
+}
+/** The engine install's task id, one per pack, shared by every setup that needs it. */
+export function engineTaskId(engine) {
+    return setupTaskId(`atomic-prism-backend-${engine.version}/${engine.backend}`);
+}
+const isFinal = (stage) => MODEL_SETUP_FINAL_STAGES.includes(stage);
+const RESUMABLE = ['interrupted', 'failed', 'cancelled'];
+/** The error the wire carries for a stopped setup. */
+export function setupErrorBody(error) {
+    if (error instanceof AtomicCoreError)
+        return error.toJSON();
+    return { code: 'IO_ERROR', message: error instanceof Error ? error.message : String(error) };
+}
+/** The refusal a blocked plan answers `POST /model-setups` with. */
+export function blockedPlanError(plan) {
+    const blocker = plan.blockers[0];
+    if (!blocker)
+        return null;
+    const details = JSON.stringify({ blockers: plan.blockers });
+    switch (blocker.code) {
+        case 'legacy_artifact':
+            return new AtomicCoreError('MODEL_FORMAT_LEGACY', blocker.message, details);
+        case 'insufficient_disk_space':
+            return new AtomicCoreError('DISK_FULL', `[disk_full] ${blocker.message}`, details);
+        default:
+            return new AtomicCoreError('MODEL_ENGINE_INCOMPATIBLE', blocker.message, details);
+    }
+}
+const cancelled = () => new AtomicCoreError('MODEL_LOAD_CANCELLED', 'The model setup was cancelled.');
+function raceAbort(work, signal) {
+    if (signal.aborted)
+        return Promise.reject(cancelled());
+    return new Promise((resolve, reject) => {
+        const onAbort = () => reject(cancelled());
+        signal.addEventListener('abort', onAbort, { once: true });
+        work.then(resolve, reject).finally(() => signal.removeEventListener('abort', onAbort));
+    });
+}
+export class ModelSetupService {
+    deps;
+    records = new Map();
+    runs = new Map();
+    engines = new Map();
+    constructor(deps) {
+        this.deps = deps;
+    }
+    /** Load the records and mark what a stopped core left mid-way `interrupted`. Call once at start. */
+    async recover() {
+        for (const record of await this.deps.store.list()) {
+            this.records.set(record.setup_id, record);
+            if (isFinal(record.stage) || record.stage === 'interrupted')
+                continue;
+            await this.update(record.setup_id, { stage: 'interrupted', stopped_at: record.stage });
+        }
+    }
+    /** In-memory view for the control snapshot. */
+    snapshot() {
+        return [...this.records.values()].sort((a, b) => a.created_at - b.created_at);
+    }
+    plan(request) {
+        return this.deps.plan(request);
+    }
+    async list() {
+        return this.snapshot();
+    }
+    async get(setupId) {
+        const record = this.records.get(setupId) ?? (await this.deps.store.read(setupId));
+        if (!record)
+            throw new AtomicCoreError('MODEL_SETUP_NOT_FOUND', 'No such model setup.', setupId);
+        return record;
+    }
+    /**
+     * Start a setup from the plan the user saw. A retry with the same `request_id` returns the setup
+     * it started; a plan that changed since is `MODEL_SETUP_PLAN_STALE`; a model already being set up
+     * returns that setup.
+     */
+    async start(input) {
+        if (typeof input.request_id !== 'string' ||
+            input.request_id === '' ||
+            typeof input.plan_digest !== 'string') {
+            throw new AtomicCoreError('INVALID_ARGUMENT', 'A model setup needs request_id and plan_digest.');
+        }
+        const { request_id: requestId, plan_digest: planDigest, proxy, ...request } = input;
+        const same = this.snapshot().find((r) => r.request_id === requestId);
+        if (same) {
+            if (JSON.stringify(same.request) !== JSON.stringify(request)) {
+                throw new AtomicCoreError('INVALID_ARGUMENT', 'That request id was already used for another setup.', same.setup_id);
+            }
+            return same;
+        }
+        const plan = await this.deps.plan(input);
+        if (plan.digest !== planDigest) {
+            throw new AtomicCoreError('MODEL_SETUP_PLAN_STALE', 'What this setup would do has changed since it was planned; review the new plan.', JSON.stringify(plan));
+        }
+        const blocked = blockedPlanError(plan);
+        if (blocked)
+            throw blocked;
+        const active = this.snapshot().find((r) => r.plan.model_id === plan.model_id && !isFinal(r.stage) && r.stage !== 'interrupted');
+        if (active)
+            return active;
+        const setupId = this.deps.newId();
+        const now = this.now();
+        const record = {
+            setup_id: setupId,
+            request_id: requestId,
+            revision: 0,
+            stage: 'queued',
+            request,
+            plan,
+            task_ids: this.taskIds(setupId, plan),
+            created_at: now,
+            updated_at: now,
+        };
+        await this.deps.store.create(record);
+        this.remember(record);
+        this.launch(setupId, proxy ?? null);
+        return record;
+    }
+    /** Stop a setup. Files downloaded so far stay, so `resume` continues from them. */
+    async cancel(setupId) {
+        const record = await this.get(setupId);
+        const run = this.runs.get(setupId);
+        if (run) {
+            run.controller.abort();
+            this.deps.cancelDownload(record.task_ids.model);
+            if (record.task_ids.projector)
+                this.deps.cancelDownload(record.task_ids.projector);
+            await run.done;
+            return this.get(setupId);
+        }
+        if (isFinal(record.stage))
+            return record;
+        return this.update(setupId, { stage: 'cancelled', stopped_at: record.stage });
+    }
+    /** Run a stopped setup again, planned afresh from its request. */
+    async resume(setupId, options = {}) {
+        const record = await this.get(setupId);
+        if (this.runs.has(setupId) || !RESUMABLE.includes(record.stage))
+            return record;
+        const plan = await this.deps.plan(record.request);
+        const blocked = blockedPlanError(plan);
+        if (blocked) {
+            return this.update(setupId, {
+                stage: 'failed',
+                stopped_at: record.stage,
+                error: blocked.toJSON(),
+                plan,
+            });
+        }
+        const next = await this.update(setupId, { stage: 'queued', plan, task_ids: this.taskIds(setupId, plan) }, ['error', 'stopped_at']);
+        this.launch(setupId, options.proxy ?? null);
+        return next;
+    }
+    /** Wait for a run in flight (tests and shutdown). */
+    async settled(setupId) {
+        await this.runs.get(setupId)?.done;
+    }
+    taskIds(setupId, plan) {
+        return {
+            ...(plan.engine && !plan.engine.installed ? { engine: engineTaskId(plan.engine) } : {}),
+            model: setupTaskId(`model-setup-${setupId}-model`),
+            ...(plan.projector ? { projector: setupTaskId(`model-setup-${setupId}-projector`) } : {}),
+        };
+    }
+    launch(setupId, proxy) {
+        const controller = new AbortController();
+        const done = this.run(setupId, proxy, controller.signal).finally(() => {
+            if (this.runs.get(setupId)?.controller === controller)
+                this.runs.delete(setupId);
+        });
+        this.runs.set(setupId, { controller, done });
+    }
+    async run(setupId, proxy, signal) {
+        let stage = 'queued';
+        const enter = async (next) => {
+            if (signal.aborted)
+                throw cancelled();
+            stage = next;
+            return this.update(setupId, { stage: next });
+        };
+        try {
+            let record = await this.get(setupId);
+            const { plan } = record;
+            const paths = this.pathsOf(plan);
+            if (plan.engine) {
+                if (!plan.engine.installed) {
+                    record = await enter('installing_engine');
+                    await raceAbort(this.sharedInstall(plan.engine, proxy), signal);
+                }
+                await this.deps.selectEngine(plan.engine);
+            }
+            record = await enter('downloading_model');
+            await raceAbort(this.deps.download(record.task_ids.model, [this.item(plan.model, paths.model, plan.model_id, proxy)]), signal);
+            const projectorTask = record.task_ids.projector;
+            if (plan.projector && paths.projector && projectorTask) {
+                await enter('downloading_projector');
+                await raceAbort(this.deps.download(projectorTask, [
+                    this.item(plan.projector, paths.projector, plan.model_id, proxy),
+                ]), signal);
+            }
+            await enter('verifying');
+            this.check(plan, await this.deps.verify(join(this.deps.layout.root, paths.model)));
+            await enter('registering');
+            await this.registerModel(plan, paths);
+            await enter('ready');
+        }
+        catch (error) {
+            const wasCancelled = signal.aborted || (error instanceof Error && error.message.includes('Download cancelled'));
+            await this.update(setupId, wasCancelled
+                ? { stage: 'cancelled', stopped_at: stage }
+                : { stage: 'failed', stopped_at: stage, error: setupErrorBody(error) }).catch((e) => this.deps.log?.('warn', `[model-setup] could not record the end of ${setupId}: ${String(e)}`));
+        }
+    }
+    sharedInstall(engine, proxy) {
+        const key = `${engine.version}/${engine.backend}`;
+        let install = this.engines.get(key);
+        if (!install) {
+            install = this.deps
+                .installEngine(engine, engineTaskId(engine), proxy)
+                .finally(() => this.engines.delete(key));
+            this.engines.set(key, install);
+        }
+        return install;
+    }
+    /** Data-folder-relative paths, `/`-separated, as `model.yml` stores them. */
+    pathsOf(plan) {
+        const dir = modelDirFromId(this.deps.layout.provider('llamacpp-upstream').modelsDir, plan.model_id);
+        const rel = (file) => relative(this.deps.layout.root, join(dir, file.split('/').pop() ?? file))
+            .split(sep)
+            .join('/');
+        return { model: rel(plan.model.file), ...(plan.projector ? { projector: rel(plan.projector.file) } : {}) };
+    }
+    item(artifact, savePath, modelId, proxy) {
+        return {
+            url: hfResolveUrl(artifact.repo, artifact.file, artifact.revision, this.deps.hfEndpoint),
+            save_path: savePath,
+            ...(artifact.sha256 ? { sha256: artifact.sha256 } : {}),
+            ...(artifact.size > 0 ? { size: artifact.size } : {}),
+            model_id: modelId,
+            ...(proxy ? { proxy } : {}),
+        };
+    }
+    /** The header is the last word: a file that turned out to need another engine is not registered. */
+    check(plan, verdict) {
+        if (verdict.outcome === 'legacy_artifact')
+            throw new AtomicCoreError('MODEL_FORMAT_LEGACY', verdict.reason, JSON.stringify(verdict));
+        if (verdict.outcome === 'unsupported') {
+            throw new AtomicCoreError('MODEL_ENGINE_INCOMPATIBLE', verdict.reason, JSON.stringify(verdict));
+        }
+        if (verdict.provider !== null && verdict.provider !== plan.provider) {
+            throw new AtomicCoreError('MODEL_SETUP_PLAN_STALE', `The downloaded file needs ${verdict.provider}, not ${plan.provider}; plan the setup again.`, JSON.stringify(verdict));
+        }
+    }
+    async registerModel(plan, paths) {
+        const size = async (rel) => (await stat(join(this.deps.layout.root, rel))).size;
+        const modelSize = await size(paths.model);
+        const projectorSize = paths.projector ? await size(paths.projector) : 0;
+        const { verdict } = plan;
+        await this.deps.register(plan.model_id, {
+            model_path: paths.model,
+            ...(paths.projector ? { mmproj_path: paths.projector } : {}),
+            name: (plan.model.file.split('/').pop() ?? plan.model.file).replace(/\.gguf$/i, ''),
+            size_bytes: modelSize + projectorSize,
+            ...(plan.model.sha256 ? { model_sha256: plan.model.sha256 } : {}),
+            model_size_bytes: modelSize,
+            ...(plan.projector?.sha256 ? { mmproj_sha256: plan.projector.sha256 } : {}),
+            ...(paths.projector ? { mmproj_size_bytes: projectorSize, projector_vision: true } : {}),
+            embedding: false,
+            ...(plan.provider === 'atomic-prism'
+                ? {
+                    atomic_runtime: {
+                        provider: 'atomic-prism',
+                        requires: verdict.requires,
+                        ...(verdict.min_prism_build !== undefined ? { min_build: verdict.min_prism_build } : {}),
+                        ...(verdict.family ? { family: verdict.family } : {}),
+                    },
+                }
+                : {}),
+        });
+        this.deps.imported?.({
+            provider: plan.provider,
+            modelId: plan.model_id,
+            modelPath: paths.model,
+            ...(paths.projector ? { mmprojPath: paths.projector } : {}),
+        });
+    }
+    /** Commit a patch on top of the newest record, retrying if a concurrent write moved it. */
+    async update(setupId, patch, clear = []) {
+        for (let attempt = 0; attempt < 3; attempt++) {
+            const current = await this.deps.store.read(setupId);
+            if (!current)
+                throw new AtomicCoreError('MODEL_SETUP_NOT_FOUND', 'No such model setup.', setupId);
+            const next = { ...current, ...patch, updated_at: this.now() };
+            for (const key of clear)
+                delete next[key];
+            const stored = await this.deps.store.commit(next, current.revision);
+            if (stored) {
+                this.remember(stored);
+                return stored;
+            }
+        }
+        throw new AtomicCoreError('INTERNAL_ERROR', 'The model setup kept changing under this write.', setupId);
+    }
+    remember(record) {
+        this.records.set(record.setup_id, record);
+        this.deps.emit(record);
+    }
+    now() {
+        return (this.deps.now ?? Date.now)();
+    }
+}
+//# sourceMappingURL=service.js.map

package/dist/core/model-setup/service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"service.js","sourceRoot":"","sources":["../../../src/core/model-setup/service.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAA;AACvC,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,EAAE,MAAM,WAAW,CAAA;AAC/C,OAAO,EAAE,eAAe,EAAE,wBAAwB,EAAE,MAAM,0BAA0B,CAAA;AAapF,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAA;AAEtD,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAA;AAgCpD,sGAAsG;AACtG,MAAM,UAAU,WAAW,CAAC,GAAW;IACrC,OAAO,GAAG,CAAC,OAAO,CAAC,oBAAoB,EAAE,GAAG,CAAC,CAAA;AAC/C,CAAC;AAED,uFAAuF;AACvF,MAAM,UAAU,YAAY,CAAC,MAAqD;IAChF,OAAO,WAAW,CAAC,wBAAwB,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC,CAAA;AAChF,CAAC;AAED,MAAM,OAAO,GAAG,CAAC,KAAsB,EAAE,EAAE,CAAC,wBAAwB,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAA;AACpF,MAAM,SAAS,GAA+B,CAAC,aAAa,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAA;AAEpF,sDAAsD;AACtD,MAAM,UAAU,cAAc,CAAC,KAAc;IAC3C,IAAI,KAAK,YAAY,eAAe;QAAE,OAAO,KAAK,CAAC,MAAM,EAAE,CAAA;IAC3D,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAA;AAC9F,CAAC;AAED,oEAAoE;AACpE,MAAM,UAAU,gBAAgB,CAAC,IAAoB;IACnD,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAA;IAChC,IAAI,CAAC,OAAO;QAAE,OAAO,IAAI,CAAA;IACzB,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAA;IAC3D,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;QACrB,KAAK,iBAAiB;YACpB,OAAO,IAAI,eAAe,CAAC,qBAAqB,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAA;QAC7E,KAAK,yBAAyB;YAC5B,OAAO,IAAI,eAAe,CAAC,WAAW,EAAE,eAAe,OAAO,CAAC,OAAO,EAAE,EAAE,OAAO,CAAC,CAAA;QACpF;YACE,OAAO,IAAI,eAAe,CAAC,2BAA2B,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAA;IACrF,CAAC;AACH,CAAC;AAED,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC,IAAI,eAAe,CAAC,sBAAsB,EAAE,gCAAgC,CAAC,CAAA;AAErG,SAAS,SAAS,CAAI,IAAgB,EAAE,MAAmB;IACzD,IAAI,MAAM,CAAC,OAAO;QAAE,OAAO,OAAO,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAA;IACtD,OAAO,IAAI,OAAO,CAAI,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACxC,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAA;QACzC,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;QACzD,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAA;IACxF,CAAC,CAAC,CAAA;AACJ,CAAC;AAOD,MAAM,OAAO,iBAAiB;IAKC;IAJZ,OAAO,GAAG,IAAI,GAAG,EAAsB,CAAA;IACvC,IAAI,GAAG,IAAI,GAAG,EAAe,CAAA;IAC7B,OAAO,GAAG,IAAI,GAAG,EAAyB,CAAA;IAE3D,YAA6B,IAA2B;QAA3B,SAAI,GAAJ,IAAI,CAAuB;IAAG,CAAC;IAE5D,oGAAoG;IACpG,KAAK,CAAC,OAAO;QACX,KAAK,MAAM,MAAM,IAAI,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC;YAClD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAA;YACzC,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,KAAK,KAAK,aAAa;gBAAE,SAAQ;YACrE,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAA;QACxF,CAAC;IACH,CAAC;IAED,+CAA+C;IAC/C,QAAQ;QACN,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC,CAAA;IAC/E,CAAC;IAED,IAAI,CAAC,OAA8B;QACjC,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;IAChC,CAAC;IAED,KAAK,CAAC,IAAI;QACR,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAA;IACxB,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,OAAe;QACvB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAA;QACjF,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,eAAe,CAAC,uBAAuB,EAAE,sBAAsB,EAAE,OAAO,CAAC,CAAA;QAChG,OAAO,MAAM,CAAA;IACf,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,KAAK,CAAC,KAA6B;QACvC,IACE,OAAO,KAAK,CAAC,UAAU,KAAK,QAAQ;YACpC,KAAK,CAAC,UAAU,KAAK,EAAE;YACvB,OAAO,KAAK,CAAC,WAAW,KAAK,QAAQ,EACrC,CAAC;YACD,MAAM,IAAI,eAAe,CAAC,kBAAkB,EAAE,iDAAiD,CAAC,CAAA;QAClG,CAAC;QACD,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,WAAW,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,GAAG,KAAK,CAAA;QACnF,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,KAAK,SAAS,CAAC,CAAA;QACpE,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC7D,MAAM,IAAI,eAAe,CACvB,kBAAkB,EAClB,qDAAqD,EACrD,IAAI,CAAC,QAAQ,CACd,CAAA;YACH,CAAC;YACD,OAAO,IAAI,CAAA;QACb,CAAC;QACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACxC,IAAI,IAAI,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;YAC/B,MAAM,IAAI,eAAe,CACvB,wBAAwB,EACxB,iFAAiF,EACjF,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CACrB,CAAA;QACH,CAAC;QACD,MAAM,OAAO,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAA;QACtC,IAAI,OAAO;YAAE,MAAM,OAAO,CAAA;QAC1B,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC,IAAI,CACjC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,KAAK,IAAI,CAAC,QAAQ,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,KAAK,aAAa,CAC3F,CAAA;QACD,IAAI,MAAM;YAAE,OAAO,MAAM,CAAA;QAEzB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAA;QACjC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;QACtB,MAAM,MAAM,GAAe;YACzB,QAAQ,EAAE,OAAO;YACjB,UAAU,EAAE,SAAS;YACrB,QAAQ,EAAE,CAAC;YACX,KAAK,EAAE,QAAQ;YACf,OAAO;YACP,IAAI;YACJ,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC;YACrC,UAAU,EAAE,GAAG;YACf,UAAU,EAAE,GAAG;SAChB,CAAA;QACD,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;QACpC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAA;QACrB,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,KAAK,IAAI,IAAI,CAAC,CAAA;QACnC,OAAO,MAAM,CAAA;IACf,CAAC;IAED,mFAAmF;IACnF,KAAK,CAAC,MAAM,CAAC,OAAe;QAC1B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;QACtC,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;QAClC,IAAI,GAAG,EAAE,CAAC;YACR,GAAG,CAAC,UAAU,CAAC,KAAK,EAAE,CAAA;YACtB,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAA;YAC/C,IAAI,MAAM,CAAC,QAAQ,CAAC,SAAS;gBAAE,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAA;YAClF,MAAM,GAAG,CAAC,IAAI,CAAA;YACd,OAAO,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;QAC1B,CAAC;QACD,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;YAAE,OAAO,MAAM,CAAA;QACxC,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAA;IAC/E,CAAC;IAED,kEAAkE;IAClE,KAAK,CAAC,MAAM,CAAC,OAAe,EAAE,UAA0C,EAAE;QACxE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;QACtC,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC;YAAE,OAAO,MAAM,CAAA;QAC9E,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;QACjD,MAAM,OAAO,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAA;QACtC,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE;gBAC1B,KAAK,EAAE,QAAQ;gBACf,UAAU,EAAE,MAAM,CAAC,KAAK;gBACxB,KAAK,EAAE,OAAO,CAAC,MAAM,EAAE;gBACvB,IAAI;aACL,CAAC,CAAA;QACJ,CAAC;QACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAC5B,OAAO,EACP,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,EAAE,EAChE,CAAC,OAAO,EAAE,YAAY,CAAC,CACxB,CAAA;QACD,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,CAAA;QAC3C,OAAO,IAAI,CAAA;IACb,CAAC;IAED,qDAAqD;IACrD,KAAK,CAAC,OAAO,CAAC,OAAe;QAC3B,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,IAAI,CAAA;IACpC,CAAC;IAEO,OAAO,CAAC,OAAe,EAAE,IAAoB;QACnD,OAAO;YACL,GAAG,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACvF,KAAK,EAAE,WAAW,CAAC,eAAe,OAAO,QAAQ,CAAC;YAClD,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,WAAW,CAAC,eAAe,OAAO,YAAY,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC1F,CAAA;IACH,CAAC;IAEO,MAAM,CAAC,OAAe,EAAE,KAAyB;QACvD,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAA;QACxC,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE;YACpE,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,UAAU,KAAK,UAAU;gBAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;QAClF,CAAC,CAAC,CAAA;QACF,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAA;IAC9C,CAAC;IAEO,KAAK,CAAC,GAAG,CAAC,OAAe,EAAE,KAAyB,EAAE,MAAmB;QAC/E,IAAI,KAAK,GAAoB,QAAQ,CAAA;QACrC,MAAM,KAAK,GAAG,KAAK,EAAE,IAAqB,EAAE,EAAE;YAC5C,IAAI,MAAM,CAAC,OAAO;gBAAE,MAAM,SAAS,EAAE,CAAA;YACrC,KAAK,GAAG,IAAI,CAAA;YACZ,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAA;QAC9C,CAAC,CAAA;QACD,IAAI,CAAC;YACH,IAAI,MAAM,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;YACpC,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,CAAA;YACvB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;YAChC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAChB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;oBAC3B,MAAM,GAAG,MAAM,KAAK,CAAC,mBAAmB,CAAC,CAAA;oBACzC,MAAM,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,CAAC,CAAA;gBACjE,CAAC;gBACD,MAAM,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;YAC3C,CAAC;YACD,MAAM,GAAG,MAAM,KAAK,CAAC,mBAAmB,CAAC,CAAA;YACzC,MAAM,SAAS,CACb,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAC,EACrG,MAAM,CACP,CAAA;YACD,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAA;YAC/C,IAAI,IAAI,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,IAAI,aAAa,EAAE,CAAC;gBACvD,MAAM,KAAK,CAAC,uBAAuB,CAAC,CAAA;gBACpC,MAAM,SAAS,CACb,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,aAAa,EAAE;oBAChC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC;iBACjE,CAAC,EACF,MAAM,CACP,CAAA;YACH,CAAC;YACD,MAAM,KAAK,CAAC,WAAW,CAAC,CAAA;YACxB,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAA;YAClF,MAAM,KAAK,CAAC,aAAa,CAAC,CAAA;YAC1B,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;YACrC,MAAM,KAAK,CAAC,OAAO,CAAC,CAAA;QACtB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAChB,MAAM,CAAC,OAAO,IAAI,CAAC,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CAAC,CAAA;YAC5F,MAAM,IAAI,CAAC,MAAM,CACf,OAAO,EACP,YAAY;gBACV,CAAC,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,KAAK,EAAE;gBAC3C,CAAC,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,KAAK,EAAE,KAAK,EAAE,cAAc,CAAC,KAAK,CAAC,EAAE,CACzE,CAAC,KAAK,CAAC,CAAC,CAAU,EAAE,EAAE,CACrB,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,MAAM,EAAE,6CAA6C,OAAO,KAAK,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAC9F,CAAA;QACH,CAAC;IACH,CAAC;IAEO,aAAa,CAAC,MAAwB,EAAE,KAAyB;QACvE,MAAM,GAAG,GAAG,GAAG,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,OAAO,EAAE,CAAA;QACjD,IAAI,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;QACnC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,GAAG,IAAI,CAAC,IAAI;iBAChB,aAAa,CAAC,MAAM,EAAE,YAAY,CAAC,MAAM,CAAC,EAAE,KAAK,CAAC;iBAClD,OAAO,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAA;YAC1C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,OAAO,CAAC,CAAA;QAChC,CAAC;QACD,OAAO,OAAO,CAAA;IAChB,CAAC;IAED,6EAA6E;IACrE,OAAO,CAAC,IAAoB;QAClC,MAAM,GAAG,GAAG,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAA;QACnG,MAAM,GAAG,GAAG,CAAC,IAAY,EAAE,EAAE,CAC3B,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,CAAC;aACtE,KAAK,CAAC,GAAG,CAAC;aACV,IAAI,CAAC,GAAG,CAAC,CAAA;QACd,OAAO,EAAE,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAA;IAC5G,CAAC;IAEO,IAAI,CACV,QAAiC,EACjC,QAAgB,EAChB,OAAe,EACf,KAAyB;QAEzB,OAAO;YACL,GAAG,EAAE,YAAY,CAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC;YACxF,SAAS,EAAE,QAAQ;YACnB,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACvD,GAAG,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACrD,QAAQ,EAAE,OAAO;YACjB,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC5B,CAAA;IACH,CAAC;IAED,oGAAoG;IAC5F,KAAK,CAAC,IAAoB,EAAE,OAA6B;QAC/D,IAAI,OAAO,CAAC,OAAO,KAAK,iBAAiB;YACvC,MAAM,IAAI,eAAe,CAAC,qBAAqB,EAAE,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAA;QAC3F,IAAI,OAAO,CAAC,OAAO,KAAK,aAAa,EAAE,CAAC;YACtC,MAAM,IAAI,eAAe,CAAC,2BAA2B,EAAE,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAA;QACjG,CAAC;QACD,IAAI,OAAO,CAAC,QAAQ,KAAK,IAAI,IAAI,OAAO,CAAC,QAAQ,KAAK,IAAI,CAAC,QAAQ,EAAE,CAAC;YACpE,MAAM,IAAI,eAAe,CACvB,wBAAwB,EACxB,6BAA6B,OAAO,CAAC,QAAQ,SAAS,IAAI,CAAC,QAAQ,yBAAyB,EAC5F,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CACxB,CAAA;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,aAAa,CACzB,IAAoB,EACpB,KAA4C;QAE5C,MAAM,IAAI,GAAG,KAAK,EAAE,GAAW,EAAE,EAAE,CAAC,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAA;QACvF,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;QACzC,MAAM,aAAa,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;QACvE,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,CAAA;QACxB,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE;YACtC,UAAU,EAAE,KAAK,CAAC,KAAK;YACvB,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5D,IAAI,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC;YACnF,UAAU,EAAE,SAAS,GAAG,aAAa;YACrC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACjE,gBAAgB,EAAE,SAAS;YAC3B,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3E,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,aAAa,EAAE,gBAAgB,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxF,SAAS,EAAE,KAAK;YAChB,GAAG,CAAC,IAAI,CAAC,QAAQ,KAAK,cAAc;gBAClC,CAAC,CAAC;oBACE,cAAc,EAAE;wBACd,QAAQ,EAAE,cAAc;wBACxB,QAAQ,EAAE,OAAO,CAAC,QAAQ;wBAC1B,GAAG,CAAC,OAAO,CAAC,eAAe,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;wBACxF,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBACtD;iBACF;gBACH,CAAC,CAAC,EAAE,CAAC;SACR,CAAC,CAAA;QACF,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,OAAO,EAAE,IAAI,CAAC,QAAQ;YACtB,SAAS,EAAE,KAAK,CAAC,KAAK;YACtB,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC5D,CAAC,CAAA;IACJ,CAAC;IAED,2FAA2F;IACnF,KAAK,CAAC,MAAM,CAClB,OAAe,EACf,KAA0B,EAC1B,QAA6C,EAAE;QAE/C,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC;YAC7C,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YACnD,IAAI,CAAC,OAAO;gBAAE,MAAM,IAAI,eAAe,CAAC,uBAAuB,EAAE,sBAAsB,EAAE,OAAO,CAAC,CAAA;YACjG,MAAM,IAAI,GAAe,EAAE,GAAG,OAAO,EAAE,GAAG,KAAK,EAAE,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAA;YACzE,KAAK,MAAM,GAAG,IAAI,KAAK;gBAAE,OAAO,IAAI,CAAC,GAAG,CAAC,CAAA;YACzC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAA;YACnE,IAAI,MAAM,EAAE,CAAC;gBACX,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAA;gBACrB,OAAO,MAAM,CAAA;YACf,CAAC;QACH,CAAC;QACD,MAAM,IAAI,eAAe,CAAC,gBAAgB,EAAE,iDAAiD,EAAE,OAAO,CAAC,CAAA;IACzG,CAAC;IAEO,QAAQ,CAAC,MAAkB;QACjC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAA;QACzC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;IACxB,CAAC;IAEO,GAAG;QACT,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,EAAE,CAAA;IACtC,CAAC;CACF"}

package/dist/core/model-setup/store.d.ts

@@ -0,0 +1,29 @@
+/**
+ * `<data>/atomic-core/prism-setups/<setup id>.json`: one record per model setup. One core owns a
+ * data folder at a time (the instance lock), so the store serialises writers in memory rather than
+ * with a lock file. A write never overwrites the only good copy: the previous record is kept as
+ * `.bak` and read back when the newest is torn. `commit` applies only against the revision it was
+ * computed from, so a late writer cannot step on a newer state.
+ */
+import type { ModelSetup } from '../../contracts/index.js';
+/** Enough of a shape check to tell a record from a truncated write or a foreign file. */
+export declare function parseModelSetupRecord(text: string): ModelSetup | null;
+export declare class ModelSetupStore {
+    private readonly dir;
+    private tail;
+    constructor(dir: string);
+    /** Every record that can be read, oldest first. */
+    list(): Promise<ModelSetup[]>;
+    /** The record, its `.bak` when the newest write was torn, or `null`. */
+    read(setupId: string): Promise<ModelSetup | null>;
+    /** Write a new record; fails when one with that id exists. */
+    create(record: ModelSetup): Promise<ModelSetup>;
+    /**
+     * Replace the record when it is still at `expectedRevision`; the stored one gets
+     * `expectedRevision + 1`. Returns `null` without writing when another writer moved it first.
+     */
+    commit(next: ModelSetup, expectedRevision: number): Promise<ModelSetup | null>;
+    private path;
+    private write;
+    private serialise;
+}

package/dist/core/model-setup/store.js

@@ -0,0 +1,116 @@
+/**
+ * `<data>/atomic-core/prism-setups/<setup id>.json`: one record per model setup. One core owns a
+ * data folder at a time (the instance lock), so the store serialises writers in memory rather than
+ * with a lock file. A write never overwrites the only good copy: the previous record is kept as
+ * `.bak` and read back when the newest is torn. `commit` applies only against the revision it was
+ * computed from, so a late writer cannot step on a newer state.
+ */
+import { mkdir, readFile, readdir, rename, rm, writeFile } from 'node:fs/promises';
+import { join } from 'node:path';
+import { AtomicCoreError, MODEL_SETUP_STAGES } from '../../contracts/index.js';
+const SAFE_ID = /^[A-Za-z0-9_-]{1,128}$/;
+/** Enough of a shape check to tell a record from a truncated write or a foreign file. */
+export function parseModelSetupRecord(text) {
+    try {
+        const raw = JSON.parse(text);
+        if (typeof raw?.setup_id !== 'string' || !SAFE_ID.test(raw.setup_id))
+            return null;
+        if (typeof raw.request_id !== 'string')
+            return null;
+        if (!Number.isSafeInteger(raw.revision) || raw.revision < 0)
+            return null;
+        if (!MODEL_SETUP_STAGES.includes(raw.stage))
+            return null;
+        if (typeof raw.plan?.digest !== 'string' || typeof raw.plan.model_id !== 'string')
+            return null;
+        return raw;
+    }
+    catch {
+        return null;
+    }
+}
+export class ModelSetupStore {
+    dir;
+    tail = Promise.resolve();
+    constructor(dir) {
+        this.dir = dir;
+    }
+    /** Every record that can be read, oldest first. */
+    async list() {
+        const names = await readdir(this.dir).catch((err) => {
+            if (err.code === 'ENOENT')
+                return [];
+            throw err;
+        });
+        const ids = new Set(names
+            .filter((n) => n.endsWith('.json') || n.endsWith('.json.bak'))
+            .map((n) => n.replace(/\.json(\.bak)?$/, '')));
+        const records = [];
+        for (const id of ids) {
+            const record = SAFE_ID.test(id) ? await this.read(id) : null;
+            if (record)
+                records.push(record);
+        }
+        return records.sort((a, b) => a.created_at - b.created_at || a.setup_id.localeCompare(b.setup_id));
+    }
+    /** The record, its `.bak` when the newest write was torn, or `null`. */
+    async read(setupId) {
+        if (!SAFE_ID.test(setupId))
+            return null;
+        const path = this.path(setupId);
+        for (const candidate of [path, `${path}.bak`]) {
+            const text = await readFile(candidate, 'utf8').catch(() => undefined);
+            const record = text === undefined ? null : parseModelSetupRecord(text);
+            if (record)
+                return record;
+        }
+        return null;
+    }
+    /** Write a new record; fails when one with that id exists. */
+    async create(record) {
+        return this.serialise(async () => {
+            if (await this.read(record.setup_id)) {
+                throw new AtomicCoreError('INVALID_ARGUMENT', 'A model setup with that id already exists.', record.setup_id);
+            }
+            await this.write(record);
+            return record;
+        });
+    }
+    /**
+     * Replace the record when it is still at `expectedRevision`; the stored one gets
+     * `expectedRevision + 1`. Returns `null` without writing when another writer moved it first.
+     */
+    async commit(next, expectedRevision) {
+        return this.serialise(async () => {
+            const current = await this.read(next.setup_id);
+            if (!current || current.revision !== expectedRevision)
+                return null;
+            const stored = { ...next, revision: expectedRevision + 1 };
+            await this.write(stored);
+            return stored;
+        });
+    }
+    path(setupId) {
+        return join(this.dir, `${setupId}.json`);
+    }
+    async write(record) {
+        await mkdir(this.dir, { recursive: true });
+        const path = this.path(record.setup_id);
+        const temporary = `${path}.tmp-${process.pid}`;
+        await writeFile(temporary, `${JSON.stringify(record, null, 2)}\n`, 'utf8');
+        await rename(path, `${path}.bak`).catch((err) => {
+            if (err.code !== 'ENOENT')
+                throw err;
+        });
+        await rename(temporary, path).catch(async (err) => {
+            await rm(temporary, { force: true }).catch(() => { });
+            throw err;
+        });
+    }
+    serialise(operation) {
+        const result = this.tail.then(operation, operation);
+        this.tail = result.catch(() => { });
+        return result;
+    }
+}
+//# sourceMappingURL=store.js.map

package/dist/core/model-setup/store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"store.js","sourceRoot":"","sources":["../../../src/core/model-setup/store.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAA;AAClF,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,OAAO,EAAE,eAAe,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAA;AAG9E,MAAM,OAAO,GAAG,wBAAwB,CAAA;AAExC,yFAAyF;AACzF,MAAM,UAAU,qBAAqB,CAAC,IAAY;IAChD,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAe,CAAA;QAC1C,IAAI,OAAO,GAAG,EAAE,QAAQ,KAAK,QAAQ,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC;YAAE,OAAO,IAAI,CAAA;QACjF,IAAI,OAAO,GAAG,CAAC,UAAU,KAAK,QAAQ;YAAE,OAAO,IAAI,CAAA;QACnD,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,QAAQ,GAAG,CAAC;YAAE,OAAO,IAAI,CAAA;QACxE,IAAI,CAAE,kBAAwC,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC;YAAE,OAAO,IAAI,CAAA;QAC/E,IAAI,OAAO,GAAG,CAAC,IAAI,EAAE,MAAM,KAAK,QAAQ,IAAI,OAAO,GAAG,CAAC,IAAI,CAAC,QAAQ,KAAK,QAAQ;YAAE,OAAO,IAAI,CAAA;QAC9F,OAAO,GAAG,CAAA;IACZ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAA;IACb,CAAC;AACH,CAAC;AAED,MAAM,OAAO,eAAe;IAGG;IAFrB,IAAI,GAAqB,OAAO,CAAC,OAAO,EAAE,CAAA;IAElD,YAA6B,GAAW;QAAX,QAAG,GAAH,GAAG,CAAQ;IAAG,CAAC;IAE5C,mDAAmD;IACnD,KAAK,CAAC,IAAI;QACR,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,GAA0B,EAAE,EAAE;YACzE,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ;gBAAE,OAAO,EAAc,CAAA;YAChD,MAAM,GAAG,CAAA;QACX,CAAC,CAAC,CAAA;QACF,MAAM,GAAG,GAAG,IAAI,GAAG,CACjB,KAAK;aACF,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;aAC7D,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,CAAC,CAAC,CAChD,CAAA;QACD,MAAM,OAAO,GAAiB,EAAE,CAAA;QAChC,KAAK,MAAM,EAAE,IAAI,GAAG,EAAE,CAAC;YACrB,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAA;YAC5D,IAAI,MAAM;gBAAE,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;QAClC,CAAC;QACD,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,IAAI,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAA;IACpG,CAAC;IAED,wEAAwE;IACxE,KAAK,CAAC,IAAI,CAAC,OAAe;QACxB,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC;YAAE,OAAO,IAAI,CAAA;QACvC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAC/B,KAAK,MAAM,SAAS,IAAI,CAAC,IAAI,EAAE,GAAG,IAAI,MAAM,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAA;YACrE,MAAM,MAAM,GAAG,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAA;YACtE,IAAI,MAAM;gBAAE,OAAO,MAAM,CAAA;QAC3B,CAAC;QACD,OAAO,IAAI,CAAA;IACb,CAAC;IAED,8DAA8D;IAC9D,KAAK,CAAC,MAAM,CAAC,MAAkB;QAC7B,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,IAAI,EAAE;YAC/B,IAAI,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACrC,MAAM,IAAI,eAAe,CACvB,kBAAkB,EAClB,4CAA4C,EAC5C,MAAM,CAAC,QAAQ,CAChB,CAAA;YACH,CAAC;YACD,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;YACxB,OAAO,MAAM,CAAA;QACf,CAAC,CAAC,CAAA;IACJ,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,MAAM,CAAC,IAAgB,EAAE,gBAAwB;QACrD,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,IAAI,EAAE;YAC/B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;YAC9C,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,QAAQ,KAAK,gBAAgB;gBAAE,OAAO,IAAI,CAAA;YAClE,MAAM,MAAM,GAAG,EAAE,GAAG,IAAI,EAAE,QAAQ,EAAE,gBAAgB,GAAG,CAAC,EAAE,CAAA;YAC1D,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;YACxB,OAAO,MAAM,CAAA;QACf,CAAC,CAAC,CAAA;IACJ,CAAC;IAEO,IAAI,CAAC,OAAe;QAC1B,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,OAAO,CAAC,CAAA;IAC1C,CAAC;IAEO,KAAK,CAAC,KAAK,CAAC,MAAkB;QACpC,MAAM,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;QAC1C,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAA;QACvC,MAAM,SAAS,GAAG,GAAG,IAAI,QAAQ,OAAO,CAAC,GAAG,EAAE,CAAA;QAC9C,MAAM,SAAS,CAAC,SAAS,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,MAAM,CAAC,CAAA;QAC1E,MAAM,MAAM,CAAC,IAAI,EAAE,GAAG,IAAI,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,GAA0B,EAAE,EAAE;YACrE,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ;gBAAE,MAAM,GAAG,CAAA;QACtC,CAAC,CAAC,CAAA;QACF,MAAM,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,GAAY,EAAE,EAAE;YACzD,MAAM,EAAE,CAAC,SAAS,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;YACpD,MAAM,GAAG,CAAA;QACX,CAAC,CAAC,CAAA;IACJ,CAAC;IAEO,SAAS,CAAI,SAA2B;QAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,SAAS,CAAC,CAAA;QACnD,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;QAClC,OAAO,MAAM,CAAA;IACf,CAAC;CACF"}

package/dist/core/model-setup/wiring.d.ts

@@ -0,0 +1,59 @@
+/**
+ * Composition of the model setup for `create.ts`: the plan's facts (the verdict, the conf rule, the
+ * Hub's file facts, the PrismML manifest, this machine, the packs on disk, free space) gathered from
+ * the core's services, and the runner's I/O (engine install, downloads, header check, `model.yml`).
+ */
+import type { DataLayout } from '../../config/index.js';
+import type { CoreEvents, LocalProviderId, ModelCompatibilityRequest, ModelCompatibilityResponse, ModelSetupPlan, ModelSetupPlanRequest, ProxyConfig } from '../../contracts/index.js';
+import type { PrismCatalogService, PrismOfferOptions, RocmHostProbe } from '../../backend/index.js';
+import type { DownloadItem } from '../../downloads/index.js';
+import type { HardwareFacts } from '../../hardware/index.js';
+import type { ModelCompatibilityService, ModelYmlInput } from '../../models/index.js';
+import type { PrismPackRef } from './plan.js';
+import { ModelSetupService } from './service.js';
+export interface ModelSetupWiringDeps {
+    layout: DataLayout;
+    compatibility: Pick<ModelCompatibilityService, 'check' | 'ruleFor'>;
+    prismCatalog: Pick<PrismCatalogService, 'catalog'>;
+    hardware: () => Promise<HardwareFacts>;
+    offer: () => PrismOfferOptions;
+    currentPack: () => Promise<PrismPackRef | null>;
+    installEngine: (version: string, backend: string, options: {
+        taskId: string;
+        proxy: ProxyConfig | null;
+    }) => Promise<unknown>;
+    selectEngine: (versionBackend: string) => Promise<unknown>;
+    downloader: {
+        download: (taskId: string, items: DownloadItem[], options: {
+            headers?: Record<string, string>;
+            resume?: boolean;
+        }) => Promise<unknown>;
+        cancel: (taskId: string) => unknown;
+    };
+    register: (provider: LocalProviderId, modelId: string, yml: ModelYmlInput) => Promise<unknown>;
+    /** A registered model's file: absolute path and the sha256 its `model.yml` records. */
+    modelFile: (provider: LocalProviderId, modelId: string) => Promise<{
+        modelPath: string;
+        sha256?: string;
+    }>;
+    emit: <K extends keyof CoreEvents>(name: K, payload: CoreEvents[K]) => void;
+    freeBytes: () => Promise<number | null>;
+    /** The fetch for one request's proxy policy. */
+    fetchFor: (proxy?: ProxyConfig | null) => typeof fetch;
+    newId: () => string;
+    /** Test hook `ATOMIC_HF_ENDPOINT`: where model files are downloaded from. */
+    hfEndpoint?: string;
+    env?: NodeJS.ProcessEnv;
+    platform?: NodeJS.Platform;
+    rocmProbe?: () => Promise<RocmHostProbe>;
+    installedPacks?: () => Promise<PrismPackRef[]>;
+    log?: (level: 'info' | 'warn', message: string) => void;
+}
+/** The plan for one request, from what the core knows now. */
+export declare function planFor(deps: ModelSetupWiringDeps, request: ModelSetupPlanRequest): Promise<ModelSetupPlan>;
+/**
+ * `POST /models/compatibility`: a registered model by id (its file on disk), or a Hub file before
+ * download (the conf rule, then — when asked — its header over HTTP ranges).
+ */
+export declare function compatibilityFor(deps: Pick<ModelSetupWiringDeps, 'compatibility' | 'modelFile'>, request: ModelCompatibilityRequest): Promise<ModelCompatibilityResponse>;
+export declare function wireModelSetups(deps: ModelSetupWiringDeps): ModelSetupService;