w-dispatch-ai 1.0.0 → 1.0.2

package/test/unit-dispatchAiFallback.test.mjs

@@ -0,0 +1,422 @@
+import assert from 'assert'
+import dispatchAiFallback from '../src/dispatchAiFallback.mjs'
+import createFakeCli from './tools/fakeCliForTest.mjs'
+
+
+//assertKey, 由假CLI回聲之OPENCODE_AUTH_CONTENT解出本次注入之金鑰
+let getInjectedKey = (stdout) => {
+    let o = JSON.parse(stdout)
+    let auth = o.env.OPENCODE_AUTH_CONTENT
+    if (auth === '') {
+        return ''
+    }
+    let j = JSON.parse(auth)
+    return Object.values(j)[0].key
+}
+
+
+describe('dispatchAiFallback', function() {
+
+    let fake = null
+
+    before(function() {
+        fake = createFakeCli('fake-fallback')
+    })
+
+    after(function() {
+        if (fake) {
+            fake.clean()
+        }
+    })
+
+    it('prompt非有效字串時回傳錯誤結果物件且tried為空', async function() {
+        let t = await dispatchAiFallback('', { providers: [{ kind: 'claude' }] })
+        let r = [t.ok, t.error, t.tried]
+        let rr = [false, 'prompt must be a non-empty string', []]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('providers非有效陣列時回傳錯誤結果物件', async function() {
+        let r = []
+        for (let providers of [null, undefined, [], {}, 'abc', [null, 'x', 123]]) {
+            let t = await dispatchAiFallback('abc', { providers })
+            r.push([t.ok, t.error])
+        }
+        let rr = [
+            [false, 'providers must be a non-empty array'],
+            [false, 'providers must be a non-empty array'],
+            [false, 'providers must be a non-empty array'],
+            [false, 'providers must be a non-empty array'],
+            [false, 'providers must be a non-empty array'],
+            [false, 'providers must be a non-empty array'],
+        ]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('某把金鑰失敗自動換組內下一把並成功', async function() {
+        let t = await dispatchAiFallback('abc', {
+            providers: [
+                {
+                    id: 'g-rotate',
+                    kind: 'opencode',
+                    exe: fake.exe,
+                    provider: 'opencode',
+                    keys: ['sk-bad-aaa', 'sk-good-bbb'],
+                    extraArgs: ['--fake-fail-key=sk-bad'],
+                },
+            ],
+        })
+        let r = [
+            t.ok,
+            t.providerId,
+            t.keyIndex,
+            getInjectedKey(t.stdout),
+            t.tried.map((x) => [x.keyId, x.outcome]),
+        ]
+        let rr = [
+            true,
+            'g-rotate',
+            1,
+            'sk-good-bbb',
+            [['g-rotate#0', 'next-key'], ['g-rotate#1', 'ok']],
+        ]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('成功亦推進游標, 多把金鑰跨次呼叫輪替均攤', async function() {
+        let providers = [{
+            id: 'g-cursor',
+            kind: 'opencode',
+            exe: fake.exe,
+            provider: 'opencode',
+            keys: ['sk-k0', 'sk-k1', 'sk-k2'],
+        }]
+        let r = []
+        for (let i = 0; i < 4; i++) {
+            let t = await dispatchAiFallback('abc', { providers })
+            r.push(getInjectedKey(t.stdout))
+        }
+        let rr = ['sk-k0', 'sk-k1', 'sk-k2', 'sk-k0']
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('store注入時游標經store持久化, 各群組游標互不干擾', async function() {
+        let stored = { cursors: {} }
+        let store = {
+            get: () => stored,
+            set: (s) => {
+                stored = s
+            },
+        }
+        let providers = [{
+            id: 'g-store-a',
+            kind: 'opencode',
+            exe: fake.exe,
+            provider: 'opencode',
+            keys: ['sk-a0', 'sk-a1'],
+        }]
+        let t1 = await dispatchAiFallback('abc', { providers, store })
+        let t2 = await dispatchAiFallback('abc', { providers, store })
+        let r = [
+            getInjectedKey(t1.stdout),
+            getInjectedKey(t2.stdout),
+            stored.cursors['g-store-a'],
+            stored.cursors['g-store-b'],
+        ]
+        let rr = ['sk-a0', 'sk-a1', 0, undefined]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('整組金鑰用盡時遞補下一組', async function() {
+        let t = await dispatchAiFallback('abc', {
+            providers: [
+                {
+                    id: 'g-dead',
+                    kind: 'opencode',
+                    exe: fake.exe,
+                    provider: 'opencode',
+                    keys: ['sk-bad-1', 'sk-bad-2'],
+                    extraArgs: ['--fake-fail-key=sk-bad'],
+                },
+                { id: 'g-alive', kind: 'claude', exe: fake.exe },
+            ],
+        })
+        let o = JSON.parse(t.stdout)
+        let r = [
+            t.ok,
+            t.providerId,
+            t.keyIndex,
+            o.args[0], //claude轉接器之固定首旗標
+            t.tried.map((x) => [x.providerId, x.outcome]),
+        ]
+        let rr = [
+            true,
+            'g-alive',
+            null,
+            '-p',
+            [['g-dead', 'next-key'], ['g-dead', 'next-key'], ['g-alive', 'ok']],
+        ]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('逾時為與金鑰無關之失敗, 一把即整組跳過不逐把空耗', async function() {
+        let t = await dispatchAiFallback('abc', {
+            providers: [
+                {
+                    id: 'g-hang',
+                    kind: 'opencode',
+                    exe: fake.exe,
+                    provider: 'opencode',
+                    keys: ['sk-h0', 'sk-h1', 'sk-h2'],
+                    extraArgs: ['--fake-sleep=10000'],
+                    timeoutMs: 1000,
+                },
+                { id: 'g-next', kind: 'claude', exe: fake.exe },
+            ],
+        })
+        let r = [
+            t.ok,
+            t.providerId,
+            t.tried.length, //hang組僅1筆而非3筆
+            t.tried[0].outcome,
+            t.tried[0].error.indexOf('TIMEOUT') === 0,
+        ]
+        let rr = [true, 'g-next', 2, 'skip-group', true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('參數錯誤(code=2)整組跳過', async function() {
+        let t = await dispatchAiFallback('abc', {
+            providers: [
+                {
+                    id: 'g-arg',
+                    kind: 'opencode',
+                    exe: fake.exe,
+                    provider: 'opencode',
+                    keys: ['sk-x0', 'sk-x1'],
+                    extraArgs: ['--fake-exit=2'],
+                },
+                { id: 'g-ok', kind: 'codex', exe: fake.exe },
+            ],
+        })
+        let r = [t.ok, t.providerId, t.tried.map((x) => x.outcome)]
+        let rr = [true, 'g-ok', ['skip-group', 'ok']]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('輸出未過驗證整組跳過, 條目validate覆寫共用預設', async function() {
+        let t = await dispatchAiFallback('abc', {
+            providers: [
+                {
+                    id: 'g-badout',
+                    kind: 'opencode',
+                    exe: fake.exe,
+                    provider: 'opencode',
+                    keys: ['sk-v0', 'sk-v1'],
+                    validate: 'min:100000',
+                },
+                { id: 'g-anyout', kind: 'claude', exe: fake.exe },
+            ],
+        })
+        let r = [
+            t.ok,
+            t.providerId,
+            t.tried.map((x) => [x.outcome, x.error || null]),
+        ]
+        let rr = [
+            true,
+            'g-anyout',
+            [['skip-group', 'OUTPUT_VALIDATION_FAILED'], ['ok', null]],
+        ]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('執行檔不存在(ENOENT)整組跳過', async function() {
+        let t = await dispatchAiFallback('abc', {
+            providers: [
+                { id: 'g-noexe', kind: 'claude', exe: 'no-such-cli-for-test', keys: ['sk-n0', 'sk-n1'] },
+                { id: 'g-hasexe', kind: 'claude', exe: fake.exe },
+            ],
+        })
+        let r = [t.ok, t.providerId, t.tried.map((x) => x.outcome)]
+        let rr = [true, 'g-hasexe', ['skip-group', 'ok']]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('kind無效之條目整組跳過, 不中斷後續遞補', async function() {
+        let t = await dispatchAiFallback('abc', {
+            providers: [
+                { id: 'g-unknown', kind: 'gemini' },
+                { id: 'g-known', kind: 'claude', exe: fake.exe },
+            ],
+        })
+        let r = [
+            t.ok,
+            t.providerId,
+            t.tried[0].outcome,
+            t.tried[0].error.indexOf('unknown ai kind') === 0,
+        ]
+        let rr = [true, 'g-known', 'skip-group', true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('全數供應商失敗時回傳最後一筆失敗結果與完整歷程', async function() {
+        let t = await dispatchAiFallback('abc', {
+            providers: [
+                { id: 'g-f1', kind: 'claude', exe: fake.exe, extraArgs: ['--fake-exit=1'] },
+                { id: 'g-f2', kind: 'codex', exe: fake.exe, extraArgs: ['--fake-exit=1'] },
+            ],
+        })
+        let r = [
+            t.ok,
+            t.code,
+            t.error,
+            t.providerId,
+            t.tried.map((x) => [x.providerId, x.outcome]),
+        ]
+        let rr = [
+            false,
+            1,
+            'Exit code 1',
+            'g-f2',
+            [['g-f1', 'next-key'], ['g-f2', 'next-key']],
+        ]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('預算不足一次最低嘗試時停止並回報budget exhausted', async function() {
+        let t = await dispatchAiFallback('abc', {
+            providers: [{ id: 'g-nobudget', kind: 'claude', exe: fake.exe }],
+            budgetMs: 50,
+            minAttemptMs: 100,
+        })
+        let r = [t.ok, t.error, t.tried.map((x) => x.outcome)]
+        let rr = [false, 'budget exhausted', ['budget-out']]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('剩餘預算會壓進attempt之timeoutMs', async function() {
+        let t = await dispatchAiFallback('abc', {
+            providers: [{
+                id: 'g-clamp',
+                kind: 'claude',
+                exe: fake.exe,
+                extraArgs: ['--fake-sleep=10000'], //timeoutMs未給, 預設120000, 應被預算4000封頂
+            }],
+            budgetMs: 4000,
+            minAttemptMs: 100,
+        })
+        let r = [
+            t.ok,
+            t.tried[0].outcome,
+            t.error.indexOf('TIMEOUT') === 0,
+            t.durationMs < 9000, //遠小於fake-sleep之10000, 證明timeout被封頂
+        ]
+        let rr = [false, 'skip-group', true, true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('onEvent依序回報事件, 回調拋出例外不影響主流程', async function() {
+        let evs = []
+        let t = await dispatchAiFallback('abc', {
+            providers: [{
+                id: 'g-ev',
+                kind: 'opencode',
+                exe: fake.exe,
+                provider: 'opencode',
+                keys: ['sk-bad-e0', 'sk-good-e1'],
+                extraArgs: ['--fake-fail-key=sk-bad'],
+            }],
+            onEvent: (ev) => {
+                evs.push([ev.type, ev.providerId, ev.keyIndex])
+                throw new Error('callback error should not break the loop')
+            },
+        })
+        let r = [t.ok, evs]
+        let rr = [
+            true,
+            [
+                ['try', 'g-ev', 0],
+                ['next-key', 'g-ev', 0],
+                ['try', 'g-ev', 1],
+                ['ok', 'g-ev', 1],
+            ],
+        ]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('條目即opt, config與provider等鍵原樣透傳對應轉接器', async function() {
+        let config = { provider: { 'agnes-ai': { npm: '@ai-sdk/openai-compatible' } } }
+        let t = await dispatchAiFallback('abc', {
+            providers: [{
+                id: 'g-agnes',
+                kind: 'opencode',
+                exe: fake.exe,
+                model: 'agnes-ai/agnes-2.0-flash',
+                provider: 'agnes-ai',
+                keys: ['sk-agnes-0'],
+                config,
+            }],
+        })
+        let o = JSON.parse(t.stdout)
+        let r = [
+            t.ok,
+            JSON.parse(o.env.OPENCODE_CONFIG_CONTENT),
+            JSON.parse(o.env.OPENCODE_AUTH_CONTENT),
+            o.args.includes('agnes-ai/agnes-2.0-flash'),
+        ]
+        let rr = [
+            true,
+            config,
+            { 'agnes-ai': { type: 'api', key: 'sk-agnes-0' } },
+            true,
+        ]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('四層情境: 組內輪替與跨組遞補複合', async function() {
+        //deepseek兩把全敗(額度型) → agnes第1把敗換第2把成功
+        let t = await dispatchAiFallback('abc', {
+            providers: [
+                {
+                    id: 'g-ds',
+                    kind: 'opencode',
+                    exe: fake.exe,
+                    provider: 'opencode',
+                    keys: ['sk-dsbad-0', 'sk-dsbad-1'],
+                    extraArgs: ['--fake-fail-key=sk-dsbad'],
+                },
+                {
+                    id: 'g-ag',
+                    kind: 'opencode',
+                    exe: fake.exe,
+                    provider: 'agnes-ai',
+                    keys: ['sk-agbad-0', 'sk-aggood-1'],
+                    extraArgs: ['--fake-fail-key=sk-agbad'],
+                },
+                { id: 'g-cl', kind: 'claude', exe: fake.exe },
+            ],
+        })
+        let r = [
+            t.ok,
+            t.providerId,
+            t.keyIndex,
+            getInjectedKey(t.stdout),
+            t.tried.map((x) => [x.keyId, x.outcome]),
+        ]
+        let rr = [
+            true,
+            'g-ag',
+            1,
+            'sk-aggood-1',
+            [
+                ['g-ds#0', 'next-key'],
+                ['g-ds#1', 'next-key'],
+                ['g-ag#0', 'next-key'],
+                ['g-ag#1', 'ok'],
+            ],
+        ]
+        assert.strict.deepEqual(r, rr)
+    })
+
+})

package/test/unit-dispatchAntigravity.test.mjs

@@ -0,0 +1,173 @@
+import assert from 'assert'
+import dispatchAntigravity from '../src/dispatchAntigravity.mjs'
+import createFakeCli from './tools/fakeCliForTest.mjs'
+
+
+describe('dispatchAntigravity', function() {
+
+    let fake = null
+
+    before(function() {
+        fake = createFakeCli('fake-agy')
+    })
+
+    after(function() {
+        if (fake) {
+            fake.clean()
+        }
+    })
+
+    it('prompt非有效字串時回傳錯誤結果物件且不reject', async function() {
+        let r = []
+        for (let prompt of [null, undefined, '', 123, {}, []]) {
+            let t = await dispatchAntigravity(prompt)
+            r.push([t.ok, t.error, t.attempts])
+        }
+        let rr = [
+            [false, 'prompt must be a non-empty string', 0],
+            [false, 'prompt must be a non-empty string', 0],
+            [false, 'prompt must be a non-empty string', 0],
+            [false, 'prompt must be a non-empty string', 0],
+            [false, 'prompt must be a non-empty string', 0],
+            [false, 'prompt must be a non-empty string', 0],
+        ]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('prompt超過30000字元時回傳prompt too long且不spawn', async function() {
+        let t = await dispatchAntigravity('x'.repeat(30001), { exe: fake.exe })
+        let r = [t.ok, t.error.indexOf('prompt too long (30001 chars > 30000)') === 0, t.attempts]
+        let rr = [false, true, 0]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('prompt恰為30000字元時可正常spawn', async function() {
+        let prompt = 'x'.repeat(30000)
+        let t = await dispatchAntigravity(prompt, { exe: fake.exe })
+        let o = JSON.parse(t.stdout)
+        let r = [t.ok, o.args[o.args.length - 1] === prompt]
+        let rr = [true, true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('轉義膨脹突破命令列上限時由try/catch兜底回傳錯誤結果物件不reject', async function() {
+        //29000個引號長度過前置檢查, 但spawn轉義後約58000字元 → 同步拋ENAMETOOLONG
+        let t = await dispatchAntigravity('"'.repeat(29000), { exe: fake.exe })
+        let r = [t.ok, t.error.includes('ENAMETOOLONG'), t.attempts]
+        let rr = [false, true, 0]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('預設帶--dangerously-skip-permissions與--print-timeout, prompt為--print之值且位於最後', async function() {
+        let t = await dispatchAntigravity('abc', { exe: fake.exe })
+        let o = JSON.parse(t.stdout)
+        let r = [t.ok, o.args]
+        let rr = [true, ['--dangerously-skip-permissions', '--print-timeout', '270s', '--print', 'abc']]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('prompt不走stdin(agy由--print取得, 塞stdin會進互動模式)', async function() {
+        let t = await dispatchAntigravity('abc', { exe: fake.exe })
+        let o = JSON.parse(t.stdout)
+        let r = [o.stdin, o.args.includes('abc')]
+        let rr = ['', true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('有給model時帶--model旗標', async function() {
+        let t = await dispatchAntigravity('abc', { exe: fake.exe, model: 'gemini-3.6-flash-low' })
+        let o = JSON.parse(t.stdout)
+        let r = o.args
+        let rr = ['--dangerously-skip-permissions', '--print-timeout', '270s', '--model', 'gemini-3.6-flash-low', '--print', 'abc']
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('有給effort時帶--effort旗標', async function() {
+        let t = await dispatchAntigravity('abc', { exe: fake.exe, model: 'gemini-3.1-pro', effort: 'low' })
+        let o = JSON.parse(t.stdout)
+        let r = o.args
+        let rr = ['--dangerously-skip-permissions', '--print-timeout', '270s', '--model', 'gemini-3.1-pro', '--effort', 'low', '--print', 'abc']
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('skipPermissions為false時不帶--dangerously-skip-permissions旗標', async function() {
+        let t = await dispatchAntigravity('abc', { exe: fake.exe, skipPermissions: false })
+        let o = JSON.parse(t.stdout)
+        let r = [t.ok, o.args]
+        let rr = [true, ['--print-timeout', '270s', '--print', 'abc']]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('printTimeout由timeoutMs推導並預留30秒緩衝, 下限30秒', async function() {
+        let r = []
+        for (let timeoutMs of [600000, 120000, 5000, undefined]) {
+            let opt = { exe: fake.exe }
+            if (timeoutMs !== undefined) {
+                opt.timeoutMs = timeoutMs
+            }
+            let t = await dispatchAntigravity('abc', opt)
+            let o = JSON.parse(t.stdout)
+            let i = o.args.indexOf('--print-timeout')
+            r.push(o.args[i + 1])
+        }
+        let rr = ['570s', '90s', '30s', '270s'] //600-30, 120-30, max(30,5-30), 預設300-30
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('printTimeout明確給定時不推導', async function() {
+        let t = await dispatchAntigravity('abc', { exe: fake.exe, printTimeout: '10m', timeoutMs: 600000 })
+        let o = JSON.parse(t.stdout)
+        let i = o.args.indexOf('--print-timeout')
+        let r = o.args[i + 1]
+        let rr = '10m'
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('addDirs逐項展開為--add-dir並濾除非有效字串', async function() {
+        let t = await dispatchAntigravity('abc', { exe: fake.exe, addDirs: ['c:/a', null, 'c:/b', 123] })
+        let o = JSON.parse(t.stdout)
+        let r = o.args
+        let rr = ['--dangerously-skip-permissions', '--print-timeout', '270s', '--add-dir', 'c:/a', '--add-dir', 'c:/b', '--print', 'abc']
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('extraArgs接於固定旗標之後且於--print之前', async function() {
+        let t = await dispatchAntigravity('abc', { exe: fake.exe, extraArgs: ['--output-format', 'json'] })
+        let o = JSON.parse(t.stdout)
+        let r = o.args
+        let rr = ['--dangerously-skip-permissions', '--print-timeout', '270s', '--output-format', 'json', '--print', 'abc']
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('中文與多行prompt於--print值內完整保留', async function() {
+        let prompt = '第一行中文\n第二行 with "quote" & <sym>\n第三行'
+        let t = await dispatchAntigravity(prompt, { exe: fake.exe })
+        let o = JSON.parse(t.stdout)
+        let r = o.args[o.args.length - 1] === prompt
+        let rr = true
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('執行檔不存在時回傳ENOENT且不重試', async function() {
+        let t = await dispatchAntigravity('abc', { exe: 'agy-not-exist-for-test', maxRetries: 3 })
+        let r = [t.ok, t.error.includes('ENOENT'), t.attempts]
+        let rr = [false, true, 1]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('離開碼非0時回傳失敗(如model與effort衝突時agy回exit 1)', async function() {
+        let t = await dispatchAntigravity('abc', { exe: fake.exe, extraArgs: ['--fake-exit=1', '--fake-stderr=conflicts with --effort'] })
+        let r = [t.ok, t.code, t.error, t.stderr]
+        let rr = [false, 1, 'Exit code 1', 'conflicts with --effort']
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('cwd等execCli設定原樣轉傳', async function() {
+        let t = await dispatchAntigravity('abc', { exe: fake.exe, cwd: process.cwd() })
+        let o = JSON.parse(t.stdout)
+        let r = [t.ok, o.cwd === process.cwd()]
+        let rr = [true, true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+})