feat(source): 外部音源迁 src/ 并新增 source-simple 进程内直载

- apis/source(.ivm).ts → src/source(.ivm).ts（音源属服务端职责，不入 SDK） - 新增 src/source-simple.ts：sources/*.mjs 进程内直载（getMusicUrl 极简契约），与 *.js 的 IVM 沙箱（globalThis.lx 契约）按扩展名混排候选链，失败降权冷却 - server 挂载决策：SOURCE_ENABLED 非 0/false 启用，/url 取流分流到外部音源；音源模式下登录态不参与取流，自动续期不启动（手动 /login/refresh 仍可用） - 新增 src/env.ts（.env 最先加载）与 test/source-simple-test.ts（并入 pnpm test）
2026-10-01 16:35:08 +08:00
1 parent 66d06ef48a
commit 96f6a7fe91
@@ -0,0 +1,164 @@
+// 简单 API 音源脚本（.mjs）加载器测试 —— 离线运行，不依赖外部平台
+// 运行：node test/source-simple-test.ts（已并入 pnpm test）
+//
+// 覆盖：基本取流（字符串 / {type,url} 返回）、sources 导出限定平台、signal 取消、
+//       超时中断、脚本抛错传播、缺少 getMusicUrl 导出报错、候选链失败切换（ivm→simple）
+
+import fs from 'node:fs'
+import path from 'node:path'
+import { fileURLToPath } from 'node:url'
+import assert from 'node:assert/strict'
+import { createSimpleSource } from '../src/source-simple.ts'
+
+const root = path.join(path.dirname(fileURLToPath(import.meta.url)), '..')
+const tmpDir = path.join(root, 'tmp')
+const sourcesDir = path.join(root, 'sources')
+fs.mkdirSync(tmpDir, { recursive: true })
+
+const results: { name: string; ok: boolean; detail?: any }[] = []
+const test = async(name: string, fn: () => Promise<any>) => {
+  try {
+    const result = await fn()
+    results.push({ name, ok: true, detail: result })
+    console.log(`✅ ${name}`)
+  } catch (err: any) {
+    results.push({ name, ok: false, detail: err.message })
+    console.log(`❌ ${name}: ${err.message}`)
+  }
+}
+
+const songInfo = { songmid: '228908', hash: '228908', copyrightId: '228908' }
+
+// 生成临时脚本并返回路径；测试结束后清理
+const writeScript = (dir: string, name: string, code: string) => {
+  const p = path.join(dir, name)
+  fs.writeFileSync(p, code)
+  return p
+}
+const cleanup = (p: string) => { try { fs.unlinkSync(p) } catch {} }
+
+await test('基本取流（返回字符串）', async() => {
+  const p = writeScript(tmpDir, 'simple-test-basic.mjs', `
+    export async function getMusicUrl({ source, id, quality }) {
+      return 'https://example.test/' + source + '/' + id + '/' + quality + '.mp3'
+    }
+  `)
+  try {
+    const api = await createSimpleSource({ scriptPath: p })
+    const r = await api.kw.getMusicUrl(songInfo, '320k').promise
+    assert.equal(r.type, '320k')
+    assert.equal(r.url, 'https://example.test/kw/228908/320k.mp3')
+    return r.url
+  } finally { cleanup(p) }
+})
+
+await test('返回 { type, url } 对象', async() => {
+  const p = writeScript(tmpDir, 'simple-test-obj.mjs', `
+    export async function getMusicUrl({ quality }) {
+      return { type: quality, url: 'https://example.test/obj.mp3' }
+    }
+  `)
+  try {
+    const api = await createSimpleSource({ scriptPath: p })
+    const r = await api.tx.getMusicUrl(songInfo, 'flac').promise
+    assert.deepEqual(r, { type: 'flac', url: 'https://example.test/obj.mp3' })
+    return r.url
+  } finally { cleanup(p) }
+})
+
+await test('sources 导出限定平台', async() => {
+  const p = writeScript(tmpDir, 'simple-test-sources.mjs', `
+    export const sources = ['mg']
+    export async function getMusicUrl() { return 'https://example.test/x.mp3' }
+  `)
+  try {
+    const api = await createSimpleSource({ scriptPath: p })
+    assert.ok(api.mg, 'mg 应存在')
+    assert.ok(!api.kw, 'kw 不应存在')
+    const r = await api.mg.getMusicUrl({}, '128k').promise
+    assert.equal(r.url, 'https://example.test/x.mp3')
+    return '仅 mg'
+  } finally { cleanup(p) }
+})
+
+await test('cancelHttp 触发 signal', async() => {
+  const p = writeScript(tmpDir, 'simple-test-cancel.mjs', `
+    export async function getMusicUrl({ signal }) {
+      await new Promise((ok, err) => {
+        const t = setTimeout(() => ok('late'), 5000)
+        signal.addEventListener('abort', () => { clearTimeout(t); err(new Error('已取消')) })
+      })
+    }
+  `)
+  try {
+    const api = await createSimpleSource({ scriptPath: p })
+    const req = api.kw.getMusicUrl(songInfo, '128k')
+    setTimeout(() => req.cancelHttp(), 50)
+    await assert.rejects(req.promise, /已取消/)
+    return 'abort 生效'
+  } finally { cleanup(p) }
+})
+
+await test('脚本超时中断', async() => {
+  const p = writeScript(tmpDir, 'simple-test-timeout.mjs', `
+    export const timeout = 50
+    export async function getMusicUrl() {
+      await new Promise(ok => setTimeout(ok, 300))
+      return 'https://example.test/never.mp3'
+    }
+  `)
+  try {
+    const api = await createSimpleSource({ scriptPath: p })
+    await assert.rejects(api.kw.getMusicUrl(songInfo, '128k').promise, /超时/)
+    return 'timeout 生效'
+  } finally { cleanup(p) }
+})
+
+await test('脚本抛错传播', async() => {
+  const p = writeScript(tmpDir, 'simple-test-error.mjs', `
+    export async function getMusicUrl() { throw new Error('上游 403') }
+  `)
+  try {
+    const api = await createSimpleSource({ scriptPath: p })
+    await assert.rejects(api.kw.getMusicUrl(songInfo, '128k').promise, /上游 403/)
+    return '错误传播正常'
+  } finally { cleanup(p) }
+})
+
+await test('缺少 getMusicUrl 导出报错', async() => {
+  const p = writeScript(tmpDir, 'simple-test-bad.mjs', 'export const foo = 1')
+  try {
+    await assert.rejects(createSimpleSource({ scriptPath: p }), /getMusicUrl/)
+    return '校验生效'
+  } finally { cleanup(p) }
+})
+
+await test('候选链：ivm 脚本失败自动切换到 .mjs', async() => {
+  // SOURCE_SCRIPT_PATH 指向不存在的 .js 脚本（加载失败），sources/ 下的临时 .mjs 接管
+  cleanup(path.join(sourcesDir, 'zz-test-simple.mjs')) // 清理上次异常退出的残留
+  const p = writeScript(sourcesDir, 'zz-test-simple.mjs', `
+    export async function getMusicUrl({ source }) {
+      return 'https://example.test/chain-' + source + '.mp3'
+    }
+  `)
+  process.env.SOURCE_SCRIPT_PATH = path.join(root, 'nonexistent-source-script.js')
+  try {
+    const { createApi } = await import('../src/source.ts')
+    const r = await createApi('kw').getMusicUrl(songInfo, '128k').promise
+    assert.equal(r.url, 'https://example.test/chain-kw.mp3')
+    return '切换成功'
+  } finally {
+    cleanup(p)
+    delete process.env.SOURCE_SCRIPT_PATH
+  }
+})
+
+// 汇总
+const passed = results.filter(r => r.ok).length
+const failed = results.filter(r => !r.ok)
+console.log(`\n===== 结果: ${passed}/${results.length} 通过 =====`)
+if (failed.length) {
+  console.log('失败项:')
+  failed.forEach(f => console.log(`  - ${f.name}: ${f.detail}`))
+}
+if (failed.length) process.exit(1)