dsh-drawai 0.4.1 → 0.4.3

package/lib/motion/asset-pack.js

@@ -16,11 +16,12 @@ import { spawn } from 'node:child_process'
 import { createHash } from 'node:crypto'
 import { createReadStream, createWriteStream, existsSync, mkdirSync, readFileSync, renameSync, rmSync, statSync, writeFileSync } from 'node:fs'
 import { homedir } from 'node:os'
-import { dirname, join } from 'node:path'
+import { dirname, join, resolve, sep } from 'node:path'
 import { fileURLToPath } from 'node:url'
 import { gunzipSync } from 'node:zlib'
 
 import { ASSETS_LOCK } from './asset-pack-lock.js'
+import { DEMO_ASSETS_LOCK } from './demo-assets-lock.js'
 
 export const ASSETS_VERSION = ASSETS_LOCK.version
 export const ASSETS_FILE = ASSETS_LOCK.file
@@ -30,6 +31,16 @@ export const ASSETS_TAG = `assets-${ASSETS_VERSION}`
 export const ASSETS_URL = process.env.DRAWAI_ASSETS_URL || `https://github.com/${ASSETS_REPO}/releases/download/${ASSETS_TAG}/${ASSETS_FILE}`
 export const ASSETS_PAGE = `https://github.com/${ASSETS_REPO}/releases/tag/${ASSETS_TAG}`
 
+/**
+ * 上游 demo 用的数据文件（现在只有 22-hud 的 land-50m.json）单独一个小包：
+ * 530KB，跟 200MB 的字体/素材库包分开，谁用到谁下载。
+ */
+export const DEMO_ASSETS_VERSION = DEMO_ASSETS_LOCK.version
+export const DEMO_ASSETS_FILE = DEMO_ASSETS_LOCK.file
+export const DEMO_ASSETS_TAG = `demo-assets-${DEMO_ASSETS_VERSION}`
+export const DEMO_ASSETS_URL = process.env.DRAWAI_DEMO_ASSETS_URL || `https://github.com/${ASSETS_REPO}/releases/download/${DEMO_ASSETS_TAG}/${DEMO_ASSETS_FILE}`
+export const DEMO_ASSETS_PAGE = `https://github.com/${ASSETS_REPO}/releases/tag/${DEMO_ASSETS_TAG}`
+
 const SENTINELS = ['fonts/font-faces.json', 'media-library']
 const READY_FILE = '.ready.json'
 
@@ -39,7 +50,7 @@ export class AssetPackError extends Error {
     super(message)
     this.name = 'AssetPackError'
     this.code = detail.code || 'assets-unavailable'
-    this.url = ASSETS_URL
+    this.url = detail.url || ASSETS_URL
     this.dir = detail.dir || cacheDir()
     this.cause = detail.cause
   }
@@ -86,6 +97,29 @@ export function assetsState() {
   return { ready: false, source: null, root: null, dir: cache, version: ASSETS_VERSION, url: ASSETS_URL }
 }
 
+/** demo 数据包没有「随包一份」的形态，永远只落在用户缓存目录。 */
+export function demoCacheDir() {
+  const custom = process.env.DRAWAI_DEMO_ASSETS_DIR
+  if (custom) return custom
+  return join(homedir(), '.dsh', 'drawai-assets', 'demo-' + DEMO_ASSETS_VERSION)
+}
+
+/** 已经落地的 demo 数据根目录；没有就返回 null。同步、很便宜。 */
+export function demoAssetsRoot() {
+  const dir = demoCacheDir()
+  return existsSync(join(dir, READY_FILE)) ? dir : null
+}
+
+/** 素材根目录下的某个文件（带越界校验）；没有就返回 null。 */
+export function demoAssetsFile(relativePath) {
+  const root = demoAssetsRoot()
+  if (!root) return null
+  const base = resolve(root)
+  const file = resolve(join(base, relativePath))
+  if (file !== base && !file.startsWith(base + sep)) return null
+  return existsSync(file) ? file : null
+}
+
 function readManifest(root) {
   const file = join(root, 'manifest.json')
   if (!existsSync(file)) return null
@@ -222,7 +256,58 @@ async function downloadTo(url, dest, onProgress) {
   }
 }
 
+/**
+ * 下载并安装一个素材包。两个包（字体素材库 / demo 数据）共用这一段：
+ * 落地、大小与 sha256 校验、逐文件校验、写 ready 标记，失败时清干净临时目录。
+ */
+async function installPack({ lock, url, dir, onProgress }) {
+  const tmp = `${dir}.tmp-${process.pid}-${Date.now()}`
+  const archive = `${tmp}.tar.gz`
+  try {
+    mkdirSync(dirname(dir), { recursive: true })
+    rmSync(tmp, { recursive: true, force: true })
+    mkdirSync(tmp, { recursive: true })
+    const { bytes, sha256 } = await downloadTo(url, archive, onProgress)
+    if (lock.bytes && bytes !== lock.bytes) {
+      throw new AssetPackError(`素材包大小不对（收到 ${bytes} 字节，应为 ${lock.bytes}），可能下载被截断了`, { code: 'size-mismatch', url, dir })
+    }
+    if (lock.sha256 && sha256 !== lock.sha256) {
+      throw new AssetPackError('素材包校验失败（sha256 不匹配），请重试；反复失败请手动下载安装', { code: 'checksum', url, dir })
+    }
+    const entries = readTar(gunzipSync(readFileSync(archive)))
+    const manifestEntry = entries.find((item) => item.path === 'manifest.json')
+    const manifest = manifestEntry ? JSON.parse(manifestEntry.data.toString('utf8')) : null
+    if (!manifest || !Array.isArray(manifest.files)) throw new AssetPackError('素材包里没有 manifest.json，无法校验', { code: 'no-manifest', url, dir })
+    const expected = new Map(manifest.files.map((item) => [item.path, item.sha256]))
+    let written = 0
+    for (const entry of entries) {
+      if (entry.path === 'manifest.json') continue
+      assertSafePath(entry.path)
+      const want = expected.get(entry.path)
+      const got = createHash('sha256').update(entry.data).digest('hex')
+      if (want && want !== got) throw new AssetPackError('素材包内容损坏：' + entry.path, { code: 'bad-file', url, dir })
+      const target = join(tmp, entry.path)
+      mkdirSync(dirname(target), { recursive: true })
+      writeFileSync(target, entry.data)
+      written += 1
+    }
+    writeFileSync(join(tmp, READY_FILE), JSON.stringify({
+      version: lock.version, files: written, bytes, sha256, at: new Date().toISOString(),
+    }, null, 2) + '\n', 'utf8')
+    rmSync(dir, { recursive: true, force: true })
+    renameSync(tmp, dir)
+    return { root: dir, source: 'download', downloaded: true, files: written }
+  } catch (err) {
+    rmSync(tmp, { recursive: true, force: true })
+    if (err instanceof AssetPackError) throw err
+    throw new AssetPackError('素材包安装失败：' + (err?.message || err), { code: 'install', cause: err, url, dir })
+  } finally {
+    rmSync(archive, { force: true })
+  }
+}
+
 let inflight = null
+let demoInflight = null
 
 /**
  * 保证素材可用。已经可用时是一次同步检查、立刻返回。
@@ -232,54 +317,28 @@ export async function ensureAssets(opts = {}) {
   const ready = assetsRoot()
   if (ready && !opts.force) return { root: ready, source: 'existing', downloaded: false }
   if (inflight) return inflight
-  inflight = (async () => {
-    const dir = cacheDir()
-    const tmp = `${dir}.tmp-${process.pid}-${Date.now()}`
-    const archive = `${tmp}.tar.gz`
-    try {
-      mkdirSync(dirname(dir), { recursive: true })
-      rmSync(tmp, { recursive: true, force: true })
-      mkdirSync(tmp, { recursive: true })
-      const { bytes, sha256 } = await downloadTo(ASSETS_URL, archive, opts.onProgress)
-      if (ASSETS_LOCK.bytes && bytes !== ASSETS_LOCK.bytes) {
-        throw new AssetPackError(`素材包大小不对（收到 ${bytes} 字节，应为 ${ASSETS_LOCK.bytes}），可能下载被截断了`, { code: 'size-mismatch' })
-      }
-      if (ASSETS_LOCK.sha256 && sha256 !== ASSETS_LOCK.sha256) {
-        throw new AssetPackError('素材包校验失败（sha256 不匹配），请重试；反复失败请手动下载安装', { code: 'checksum' })
-      }
-      const entries = readTar(gunzipSync(readFileSync(archive)))
-      const manifestEntry = entries.find((item) => item.path === 'manifest.json')
-      const manifest = manifestEntry ? JSON.parse(manifestEntry.data.toString('utf8')) : null
-      if (!manifest || !Array.isArray(manifest.files)) throw new AssetPackError('素材包里没有 manifest.json，无法校验', { code: 'no-manifest' })
-      const expected = new Map(manifest.files.map((item) => [item.path, item.sha256]))
-      let written = 0
-      for (const entry of entries) {
-        if (entry.path === 'manifest.json') continue
-        assertSafePath(entry.path)
-        const want = expected.get(entry.path)
-        const got = createHash('sha256').update(entry.data).digest('hex')
-        if (want && want !== got) throw new AssetPackError('素材包内容损坏：' + entry.path, { code: 'bad-file' })
-        const target = join(tmp, entry.path)
-        mkdirSync(dirname(target), { recursive: true })
-        writeFileSync(target, entry.data)
-        written += 1
-      }
-      writeFileSync(join(tmp, READY_FILE), JSON.stringify({
-        version: ASSETS_VERSION, files: written, bytes, sha256, at: new Date().toISOString(),
-      }, null, 2) + '\n', 'utf8')
-      rmSync(dir, { recursive: true, force: true })
-      renameSync(tmp, dir)
-      return { root: dir, source: 'download', downloaded: true, files: written }
-    } catch (err) {
-      rmSync(tmp, { recursive: true, force: true })
-      if (err instanceof AssetPackError) throw err
-      throw new AssetPackError('素材包安装失败：' + (err?.message || err), { code: 'install', cause: err })
-    } finally {
-      rmSync(archive, { force: true })
-      inflight = null
-    }
-  })()
-  return inflight
+  inflight = installPack({ lock: ASSETS_LOCK, url: ASSETS_URL, dir: cacheDir(), onProgress: opts.onProgress })
+  try {
+    return await inflight
+  } finally {
+    inflight = null
+  }
+}
+
+/**
+ * 保证 demo 数据包可用：上游 demo 第一次请求缺的数据文件时下载一次（约 530KB）。
+ * 已经落地时是同步检查。
+ */
+export async function ensureDemoAssets(opts = {}) {
+  const ready = demoAssetsRoot()
+  if (ready && !opts.force) return { root: ready, source: 'existing', downloaded: false }
+  if (demoInflight) return demoInflight
+  demoInflight = installPack({ lock: DEMO_ASSETS_LOCK, url: DEMO_ASSETS_URL, dir: demoCacheDir(), onProgress: opts.onProgress })
+  try {
+    return await demoInflight
+  } finally {
+    demoInflight = null
+  }
 }
 
 /** 工具层用：拿到素材根目录，或者抛一条人能照着做的错误。 */