yo-moss-ai 0.0.10 → 0.0.12

package/bin/bootstrap-deps.mjs

@@ -0,0 +1,326 @@
+import fs from 'node:fs'
+import os from 'node:os'
+import path from 'node:path'
+import http from 'node:http'
+import https from 'node:https'
+import { spawn } from 'node:child_process'
+import { pipeline } from 'node:stream/promises'
+
+export const PROGRESS_BAR_WIDTH = 28
+export const MEDIA_PACKAGE_NAME = 'yo-moss-ai-media'
+
+export function mediaCacheDir() {
+  return process.env.MOSS_MEDIA_ROOT || path.join(os.homedir(), '.moss-desktop', 'media')
+}
+
+export function runtimeDepsRoot() {
+  return process.env.MOSS_RUNTIME_DEPS_ROOT || path.join(os.homedir(), '.moss-desktop', 'runtime-deps')
+}
+
+/** 纯函数：生成终端进度条文本（单行覆盖用） */
+export function formatProgressBar(label, ratio, width = PROGRESS_BAR_WIDTH) {
+  const pct = Math.min(100, Math.max(0, Math.round(ratio * 100)))
+  const filled = Math.round((pct / 100) * width)
+  const bar = '█'.repeat(filled) + '░'.repeat(width - filled)
+  return `${label} [${bar}] ${pct}%`
+}
+
+function renderBar(label, ratio) {
+  process.stdout.write(`\r\x1b[2K${formatProgressBar(label, ratio)}`)
+}
+
+/** @deprecated 单文件 CDN 对大体积 npm 包不可靠，保留仅供测试/兜底 */
+export function cdnUrls(version, relativePath) {
+  const encoded = relativePath.split('/').map((part) => encodeURIComponent(part)).join('/')
+  return [
+    `https://cdn.jsdelivr.net/npm/${MEDIA_PACKAGE_NAME}@${version}/${encoded}`,
+    `https://unpkg.com/${MEDIA_PACKAGE_NAME}@${version}/${encoded}`,
+  ]
+}
+
+export function followDownload(url, dest, onChunk, redirects = 0) {
+  if (redirects > 5) return Promise.reject(new Error('下载重定向过多'))
+  return new Promise((resolve, reject) => {
+    const client = url.startsWith('http:') ? http : https
+    const req = client.get(
+      url,
+      { headers: { 'User-Agent': 'yo-moss-ai' }, timeout: 120_000 },
+      (res) => {
+        const location = res.headers.location
+        if (res.statusCode >= 300 && res.statusCode < 400 && location) {
+          res.resume()
+          followDownload(new URL(location, url).href, dest, onChunk, redirects + 1).then(resolve, reject)
+          return
+        }
+        if (res.statusCode !== 200) {
+          res.resume()
+          reject(new Error(`HTTP ${res.statusCode}`))
+          return
+        }
+        fs.mkdirSync(path.dirname(dest), { recursive: true })
+        const tmp = `${dest}.download`
+        const out = fs.createWriteStream(tmp)
+        res.on('data', (chunk) => {
+          onChunk?.(chunk.length)
+        })
+        pipeline(res, out)
+          .then(() => {
+            fs.renameSync(tmp, dest)
+            resolve()
+          })
+          .catch((err) => {
+            try {
+              fs.unlinkSync(tmp)
+            } catch {
+              // ignore
+            }
+            reject(err)
+          })
+      },
+    )
+    req.on('timeout', () => req.destroy(new Error('下载超时')))
+    req.on('error', reject)
+  })
+}
+
+export function fileReady(filePath, expectedSize) {
+  try {
+    const stat = fs.statSync(filePath)
+    return stat.isFile() && (!expectedSize || stat.size === expectedSize)
+  } catch {
+    return false
+  }
+}
+
+export function dirByteSize(dir) {
+  if (!fs.existsSync(dir)) return 0
+  let total = 0
+  for (const name of fs.readdirSync(dir)) {
+    const full = path.join(dir, name)
+    const stat = fs.statSync(full)
+    if (stat.isDirectory()) total += dirByteSize(full)
+    else total += stat.size
+  }
+  return total
+}
+
+export function copyTree(src, dest) {
+  fs.mkdirSync(dest, { recursive: true })
+  for (const name of fs.readdirSync(src)) {
+    if (name.startsWith('.')) continue
+    const from = path.join(src, name)
+    const to = path.join(dest, name)
+    if (fs.statSync(from).isDirectory()) copyTree(from, to)
+    else fs.copyFileSync(from, to)
+  }
+}
+
+function runNpmInstall(cwd, extraArgs = []) {
+  return new Promise((resolve, reject) => {
+    const npmCmd = process.platform === 'win32' ? 'npm.cmd' : 'npm'
+    const child = spawn(
+      npmCmd,
+      ['install', '--omit=dev', '--no-audit', '--no-fund', '--loglevel=error', ...extraArgs],
+      {
+        cwd,
+        stdio: ['ignore', 'ignore', 'pipe'],
+        shell: process.platform === 'win32',
+      },
+    )
+    let errText = ''
+    child.stderr.on('data', (chunk) => {
+      errText += String(chunk)
+    })
+    child.on('exit', (code) => {
+      if (code === 0) resolve()
+      else reject(new Error(errText.trim() || `npm install 失败，退出码 ${code}`))
+    })
+  })
+}
+
+/**
+ * 通过 npm registry 安装 yo-moss-ai-media 整包（比 CDN 单文件拉取可靠）。
+ */
+export const NPM_REGISTRIES = [
+  'https://registry.npmjs.org',
+  'https://registry.npmmirror.com',
+]
+
+export async function installMediaBundleFromNpm(
+  version,
+  cacheDir,
+  { totalBytes = 0, onProgress, installImpl } = {},
+) {
+  const staging = path.join(cacheDir, '.npm-staging')
+  fs.rmSync(staging, { recursive: true, force: true })
+  fs.mkdirSync(staging, { recursive: true })
+  fs.writeFileSync(
+    path.join(staging, 'package.json'),
+    `${JSON.stringify({
+      name: 'yo-moss-ai-media-staging',
+      private: true,
+      dependencies: { [MEDIA_PACKAGE_NAME]: version },
+    }, null, 2)}\n`,
+  )
+
+  const expected = totalBytes > 0 ? totalBytes : 56_000_000
+  const poll = setInterval(() => {
+    const size = dirByteSize(path.join(staging, 'node_modules', MEDIA_PACKAGE_NAME))
+    onProgress?.(Math.min(0.92, size / expected))
+  }, 400)
+
+  let lastError
+  try {
+    for (const registry of NPM_REGISTRIES) {
+      try {
+        const installer = installImpl || ((cwd) => runNpmInstall(cwd, ['--registry', registry]))
+        await installer(staging)
+        lastError = null
+        break
+      } catch (err) {
+        lastError = err
+        fs.rmSync(path.join(staging, 'node_modules'), { recursive: true, force: true })
+      }
+    }
+    if (lastError) {
+      throw new Error(`运行资源自动安装失败（${MEDIA_PACKAGE_NAME}@${version}），请检查网络后重试 npx yo-moss-ai`)
+    }
+  } finally {
+    clearInterval(poll)
+  }
+
+  const pkgRoot = path.join(staging, 'node_modules', MEDIA_PACKAGE_NAME)
+  if (!fs.existsSync(pkgRoot)) {
+    throw new Error(`运行资源自动安装失败（${MEDIA_PACKAGE_NAME}@${version}），请检查网络后重试 npx yo-moss-ai`)
+  }
+
+  onProgress?.(0.95)
+  for (const dir of ['models', 'textures', 'stl']) {
+    const from = path.join(pkgRoot, dir)
+    if (!fs.existsSync(from)) continue
+    copyTree(from, path.join(cacheDir, dir))
+  }
+  fs.rmSync(staging, { recursive: true, force: true })
+  onProgress?.(1)
+}
+
+export async function ensureRuntimeNpmDeps(npmDeps, depsRoot = runtimeDepsRoot(), installImpl = runNpmInstall) {
+  fs.mkdirSync(depsRoot, { recursive: true })
+  const pkgPath = path.join(depsRoot, 'package.json')
+  const deps = Object.fromEntries(npmDeps.map((item) => [item.name, item.version]))
+  const pkg = {
+    name: 'yo-moss-ai-runtime-deps',
+    private: true,
+    dependencies: deps,
+    overrides: { phin: 'npm:phn@^0.2.8' },
+  }
+  fs.writeFileSync(pkgPath, `${JSON.stringify(pkg, null, 2)}\n`)
+
+  const marker = path.join(depsRoot, '.installed.json')
+  const nextMarker = JSON.stringify(deps)
+  if (fs.existsSync(marker) && fs.readFileSync(marker, 'utf8') === nextMarker) {
+    const nutEntry = path.join(depsRoot, 'node_modules', '@nut-tree-fork', 'nut-js', 'dist', 'index.js')
+    if (fs.existsSync(nutEntry)) return
+  }
+
+  await installImpl(depsRoot)
+  fs.writeFileSync(marker, nextMarker)
+}
+
+export function loadManifest(root) {
+  const manifestPath = path.join(root, 'bin', 'runtime-manifest.json')
+  if (!fs.existsSync(manifestPath)) return null
+  return JSON.parse(fs.readFileSync(manifestPath, 'utf8'))
+}
+
+export function hasBundledMedia(distDir) {
+  return fs.existsSync(path.join(distDir, 'models'))
+    || fs.existsSync(path.join(distDir, 'textures'))
+    || fs.existsSync(path.join(distDir, 'stl'))
+}
+
+export function runtimeDepsReady(npmDeps, depsRoot = runtimeDepsRoot()) {
+  if (!npmDeps.length) return true
+  const nutEntry = path.join(depsRoot, 'node_modules', '@nut-tree-fork', 'nut-js', 'dist', 'index.js')
+  return fs.existsSync(nutEntry)
+}
+
+export function mediaCacheReady(files, cacheDir = mediaCacheDir()) {
+  if (!files.length) return true
+  return files.every((file) => fileReady(path.join(cacheDir, file.path), file.size))
+}
+
+/**
+ * npx 安装精简包后，启动服务前下载媒体资源与扩展 npm 依赖。
+ */
+export async function bootstrapRuntimeDeps(opts) {
+  const {
+    root,
+    version,
+    cacheDir = mediaCacheDir(),
+    depsRoot = runtimeDepsRoot(),
+    installMediaBundle = installMediaBundleFromNpm,
+    installNpmDeps = (npmDeps) => ensureRuntimeNpmDeps(npmDeps, depsRoot),
+  } = opts
+
+  const distDir = path.join(root, 'dist')
+  if (process.env.MOSS_SKIP_BOOTSTRAP === '1' || hasBundledMedia(distDir)) {
+    return { skipped: true }
+  }
+
+  const manifest = loadManifest(root)
+  if (!manifest) {
+    console.warn('缺少 bin/runtime-manifest.json，跳过资源预下载')
+    return { skipped: true }
+  }
+
+  const mediaVersion = manifest.mediaVersion || version
+  const mediaFiles = manifest.mediaFiles || []
+  const npmDeps = manifest.npmDeps || []
+  const totalBytes = mediaFiles.reduce((sum, file) => sum + (file.size || 0), 0)
+  const needMedia = !mediaCacheReady(mediaFiles, cacheDir)
+  const needNpm = !runtimeDepsReady(npmDeps, depsRoot)
+
+  if (!needMedia && !needNpm) {
+    process.env.MOSS_MEDIA_ROOT = cacheDir
+    process.env.MOSS_RUNTIME_DEPS = path.join(depsRoot, 'node_modules')
+    return { skipped: false, cached: true }
+  }
+
+  console.log('')
+  if (needMedia) {
+    const mb = (totalBytes / 1024 / 1024).toFixed(0)
+    console.log(`正在下载运行资源（约 ${mb} MB），完成后自动启动服务…`)
+  } else if (needNpm) {
+    console.log('正在安装扩展依赖，完成后自动启动服务…')
+  }
+
+  if (needMedia) {
+    renderBar('下载资源', 0)
+    await installMediaBundle(mediaVersion, cacheDir, {
+      totalBytes,
+      onProgress: (ratio) => renderBar('下载资源', ratio),
+    })
+    process.stdout.write('\n')
+
+    if (!mediaCacheReady(mediaFiles, cacheDir)) {
+      throw new Error('媒体资源安装完成但校验失败，请删除 ~/.moss-desktop/media 后重试')
+    }
+  }
+
+  if (needNpm) {
+    renderBar('安装扩展依赖', 0)
+    try {
+      await installNpmDeps(npmDeps)
+      renderBar('安装扩展依赖', 1)
+      process.stdout.write('\n')
+    } catch (err) {
+      process.stdout.write('\n')
+      console.warn('扩展依赖安装失败，快捷键/媒体键可能不可用：', err.message)
+    }
+  }
+
+  process.env.MOSS_MEDIA_ROOT = cacheDir
+  process.env.MOSS_RUNTIME_DEPS = path.join(depsRoot, 'node_modules')
+  return { skipped: false, cached: false }
+}

package/bin/runtime-manifest.json

@@ -0,0 +1,67 @@
+{
+  "mediaVersion": "0.0.12",
+  "mediaFiles": [
+    {
+      "path": "models/mediapipe/blaze_face_full_range.tflite",
+      "size": 1083786
+    },
+    {
+      "path": "models/mediapipe/blaze_face_short_range.tflite",
+      "size": 229746
+    },
+    {
+      "path": "models/mediapipe/wasm/vision_wasm_internal.js",
+      "size": 323377
+    },
+    {
+      "path": "models/mediapipe/wasm/vision_wasm_internal.wasm",
+      "size": 11756954
+    },
+    {
+      "path": "models/mediapipe/wasm/vision_wasm_nosimd_internal.js",
+      "size": 323180
+    },
+    {
+      "path": "models/mediapipe/wasm/vision_wasm_nosimd_internal.wasm",
+      "size": 10960242
+    },
+    {
+      "path": "textures/earth-lights.png",
+      "size": 734910
+    },
+    {
+      "path": "textures/earth-night.jpg",
+      "size": 8277759
+    },
+    {
+      "path": "textures/earth_clouds.jpg",
+      "size": 12343415
+    },
+    {
+      "path": "textures/moon-base.jpg",
+      "size": 1591331
+    },
+    {
+      "path": "textures/moon-bump.jpg",
+      "size": 1021019
+    },
+    {
+      "path": "textures/wandering.png",
+      "size": 309032
+    },
+    {
+      "path": "stl/moss.stl",
+      "size": 1649184
+    },
+    {
+      "path": "stl/planetary_engine.stl",
+      "size": 5807034
+    }
+  ],
+  "npmDeps": [
+    {
+      "name": "@nut-tree-fork/nut-js",
+      "version": "^4.2.0"
+    }
+  ]
+}

package/bin/yo-moss-ai.mjs

@@ -2,6 +2,7 @@
 import fs from 'node:fs'
 import path from 'node:path'
 import { fileURLToPath, pathToFileURL } from 'node:url'
+import { bootstrapRuntimeDeps } from './bootstrap-deps.mjs'
 
 const root = path.join(path.dirname(fileURLToPath(import.meta.url)), '..')
 let pkgVersion = ''
@@ -10,7 +11,6 @@ try {
 } catch {
   // ignore
 }
-console.log(`yo-moss-ai${pkgVersion ? ` v${pkgVersion}` : ''} 正在启动…`)
 
 const major = Number(process.versions.node.split('.')[0])
 if (major < 20) {
@@ -27,6 +27,16 @@ if (!fs.existsSync(server)) {
   process.exit(1)
 }
 
+console.log(`yo-moss-ai${pkgVersion ? ` v${pkgVersion}` : ''}`)
+
+try {
+  await bootstrapRuntimeDeps({ root, version: pkg.version })
+} catch (err) {
+  console.error('\n启动准备失败：', err?.message || err)
+  console.error('请检查网络连接后重新运行：npx yo-moss-ai')
+  process.exit(1)
+}
+
 process.env.NODE_ENV = 'production'
 process.env.MOSS_OPEN_BROWSER = process.env.MOSS_OPEN_BROWSER || '1'
 process.env.MOSS_VERSION = pkg.version