feat(runtime): 支持共享 Demucs 运行时

Author: coderDJing

env.example

@@ -1,4 +1,7 @@
 CLOUD_SYNC_BASE_URL_DEV=http://localhost:3001
+
+# 可选：共享 Demucs runtime 根目录；未配置时使用本项目 vendor/demucs
+# FRKB_DEMUCS_ROOT=D:/playground/FRKB_Rapid-Audio-Organization-Tool/vendor/demucs
 # 生产地址（一般不需要填写，通过 server.json 发现）
 # CLOUD_SYNC_BASE_URL_PROD=
 

scripts/ensure-demucs-runtime.mjs

@@ -1,3 +1,4 @@
+import 'dotenv/config'
 import fs from 'node:fs'
 import path from 'node:path'
 import { spawnSync } from 'node:child_process'
@@ -56,7 +57,8 @@ const toShortText = (value, maxLen = 400) => {
   return text.slice(0, maxLen)
 }
 
-const runtimeRootArg = getArgValue('--runtime-root', 'vendor/demucs')
+const configuredDemucsRoot = String(process.env.FRKB_DEMUCS_ROOT || '').trim()
+const runtimeRootArg = getArgValue('--runtime-root', configuredDemucsRoot || 'vendor/demucs')
 const platformArg = getArgValue('--platform', platformDefault)
 const profileArg = getArgValue('--profiles', '')
 const modelsArg = getArgValue('--models', '')

scripts/master-tempo-ab-benchmark.mjs

@@ -1,3 +1,4 @@
+import 'dotenv/config'
 import { spawnSync } from 'node:child_process'
 import { createHash } from 'node:crypto'
 import { createRequire } from 'node:module'
@@ -10,6 +11,14 @@ const repoRoot = path.resolve(scriptDir, '..')
 const require = createRequire(import.meta.url)
 const RUNNER_VERSION = 'master-tempo-ab-benchmark-v2'
 
+const resolveManagedRuntimePython = () => {
+  const configuredDemucsRoot = String(process.env.FRKB_DEMUCS_ROOT || '').trim()
+  const demucsRoot = configuredDemucsRoot
+    ? path.resolve(configuredDemucsRoot)
+    : path.join(repoRoot, 'vendor', 'demucs')
+  return path.join(demucsRoot, 'win32-x64', 'runtime-cpu', 'python.exe')
+}
+
 const parseArgs = (argv) => {
   const args = new Map()
   for (let index = 0; index < argv.length; index += 2) {
@@ -82,10 +91,10 @@ const resolvePython = () => {
   const configured = String(process.env.FRKB_MASTER_TEMPO_BENCHMARK_PYTHON || '').trim()
   const candidate = configured
     ? path.resolve(configured)
-    : path.join(repoRoot, 'vendor', 'demucs', 'win32-x64', 'runtime-cpu', 'python.exe')
+    : resolveManagedRuntimePython()
   if (!fs.existsSync(candidate)) {
     throw new Error(
-      '找不到 benchmark Python；请设置 FRKB_MASTER_TEMPO_BENCHMARK_PYTHON 指向项目受管 Python'
+      '找不到 benchmark Python；请设置 FRKB_MASTER_TEMPO_BENCHMARK_PYTHON 或 FRKB_DEMUCS_ROOT'
     )
   }
   return candidate

scripts/package-demucs-runtime-assets.mjs

@@ -1,3 +1,4 @@
+import 'dotenv/config'
 import fs from 'node:fs'
 import path from 'node:path'
 import crypto from 'node:crypto'
@@ -36,7 +37,9 @@ const platformDefault = (() => {
   return process.arch === 'arm64' ? 'linux-arm64' : 'linux-x64'
 })()
 
-const runtimeRoot = path.resolve(getArgValue('--runtime-root', 'vendor/demucs'))
+const configuredDemucsRoot = String(process.env.FRKB_DEMUCS_ROOT || '').trim()
+const runtimeRootArg = getArgValue('--runtime-root', configuredDemucsRoot || 'vendor/demucs')
+const runtimeRoot = path.resolve(runtimeRootArg)
 const platformKey = getArgValue('--platform', platformDefault)
 const outputRoot = path.resolve(getArgValue('--output-root', 'dist/demucs-runtime-assets'))
 const releaseTag = getArgValue('--release-tag', 'demucs-runtime-assets')

scripts/prep_move_tracks_between_playlists.py

@@ -21,6 +21,7 @@
 
 import argparse
 import json
+import os
 import subprocess
 import sys
 import time
@@ -40,8 +41,40 @@
     pass
 
 REPO_ROOT = Path(__file__).resolve().parents[1]
+
+
+def _read_project_env_value(name: str) -> str:
+    configured = str(os.environ.get(name) or "").strip()
+    if configured:
+        return configured
+    try:
+        lines = (REPO_ROOT / ".env").read_text(encoding="utf-8").splitlines()
+    except OSError:
+        return ""
+    for raw_line in lines:
+        line = raw_line.strip()
+        if not line or line.startswith("#") or "=" not in line:
+            continue
+        key, raw_value = line.split("=", 1)
+        if key.strip() != name:
+            continue
+        value = raw_value.strip()
+        if len(value) >= 2 and value[0] == value[-1] and value[0] in {"'", '"'}:
+            value = value[1:-1]
+        return value
+    return ""
+
+
+def _resolve_demucs_root() -> Path:
+    configured = _read_project_env_value("FRKB_DEMUCS_ROOT")
+    if not configured:
+        return REPO_ROOT / "vendor" / "demucs"
+    candidate = Path(configured).expanduser()
+    return candidate if candidate.is_absolute() else REPO_ROOT / candidate
+
+
 DEFAULT_RUNTIME_PYTHON = (
-    REPO_ROOT / "vendor" / "demucs" / "win32-x64" / "runtime-cpu" / "python.exe"
+    _resolve_demucs_root() / "win32-x64" / "runtime-cpu" / "python.exe"
 )
 DEFAULT_SEALED_SCRIPT = REPO_ROOT / "scripts" / "rkb_sealed_batch.py"
 DEFAULT_TRIAGE_SCRIPT = REPO_ROOT / "scripts" / "move_rekordbox_playlist_grid_diffs.py"

scripts/prepare-demucs-runtimes.mjs

@@ -1,3 +1,4 @@
+import 'dotenv/config'
 import fs from 'node:fs'
 import path from 'node:path'
 import { spawnSync } from 'node:child_process'
@@ -30,7 +31,8 @@ const getArgValue = (flag, fallback = '') => {
 
 const hasFlag = (flag) => args.includes(flag)
 
-const runtimeRootArg = getArgValue('--runtime-root', 'vendor/demucs')
+const configuredDemucsRoot = String(process.env.FRKB_DEMUCS_ROOT || '').trim()
+const runtimeRootArg = getArgValue('--runtime-root', configuredDemucsRoot || 'vendor/demucs')
 const platformArg = getArgValue('--platform', platformDefault)
 const profileArg = getArgValue('--profiles', '')
 const pipExtraArg = getArgValue('--pip-extra', '')

scripts/test_prep_move_tracks_between_playlists.py

@@ -320,12 +320,10 @@ def test_main_passes_explicit_workflow_overrides(self) -> None:
         self.assertEqual(triage.call_args.kwargs["dataset_registry"], Path("registry.json"))
         self.assertTrue(triage.call_args.kwargs["pre_review"])
 
-    def test_default_workflow_python_is_vendored_runtime(self) -> None:
+    def test_default_workflow_python_uses_configured_demucs_root(self) -> None:
         self.assertEqual(
             prep.DEFAULT_RUNTIME_PYTHON,
-            prep.REPO_ROOT
-            / "vendor"
-            / "demucs"
+            prep._resolve_demucs_root()
             / "win32-x64"
             / "runtime-cpu"
             / "python.exe",