yo-moss-ai 0.0.9 → 0.0.11

package/bin/bootstrap-deps.mjs

@@ -0,0 +1,255 @@
+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 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)}`)
+}
+
+export function cdnUrls(version, relativePath) {
+  const encoded = relativePath.split('/').map((part) => encodeURIComponent(part)).join('/')
+  return [
+    `https://cdn.jsdelivr.net/npm/yo-moss-ai-media@${version}/${encoded}`,
+    `https://unpkg.com/yo-moss-ai-media@${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
+  }
+}
+
+function runNpmInstall(cwd) {
+  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'], {
+      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}`))
+    })
+  })
+}
+
+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(),
+    downloadFile = followDownload,
+    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) {
+    let doneBytes = 0
+    renderBar('下载资源', 0)
+
+    for (const file of mediaFiles) {
+      const dest = path.join(cacheDir, file.path)
+      if (fileReady(dest, file.size)) {
+        doneBytes += file.size || 0
+        renderBar('下载资源', totalBytes ? doneBytes / totalBytes : 1)
+        continue
+      }
+
+      const fileSize = file.size || 0
+      let fileDone = 0
+      let lastError
+      for (const url of cdnUrls(mediaVersion, file.path)) {
+        try {
+          await downloadFile(url, dest, (chunkLen) => {
+            fileDone += chunkLen
+            const current = doneBytes + Math.min(fileDone, fileSize || fileDone)
+            renderBar('下载资源', totalBytes ? current / totalBytes : 1)
+          })
+          lastError = null
+          break
+        } catch (err) {
+          lastError = err
+          fileDone = 0
+        }
+      }
+      if (lastError) throw new Error(`无法下载 ${file.path}: ${lastError.message}`)
+
+      doneBytes += fileSize || fs.statSync(dest).size
+      renderBar('下载资源', totalBytes ? doneBytes / totalBytes : 1)
+    }
+    process.stdout.write('\n')
+  }
+
+  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.11",
+  "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,15 @@
 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 = ''
+try {
+  pkgVersion = JSON.parse(fs.readFileSync(path.join(root, 'package.json'), 'utf8')).version
+} catch {
+  // ignore
+}
 
 const major = Number(process.versions.node.split('.')[0])
 if (major < 20) {
@@ -10,7 +19,6 @@ if (major < 20) {
   process.exit(1)
 }
 
-const root = path.join(path.dirname(fileURLToPath(import.meta.url)), '..')
 const pkg = JSON.parse(fs.readFileSync(path.join(root, 'package.json'), 'utf8'))
 const server = path.join(root, 'dist-server', 'index.mjs')
 
@@ -19,6 +27,15 @@ 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)
+  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