55 lines
3.2 KiB
JavaScript
55 lines
3.2 KiB
JavaScript
// preset-inspect.cjs — 检查宿主预设花名册与默认预设指向的 provider
|
|
const http = require('http');
|
|
const { spawn } = require('child_process');
|
|
const WebSocket = require('ws');
|
|
const PORT = 9380;
|
|
const proc = spawn('C:\\Program Files\\Google\\Chrome\\Application\\chrome.exe', [
|
|
'--headless=new', '--disable-gpu', '--no-first-run', '--no-default-browser-check',
|
|
'--window-size=1440,900', '--force-device-scale-factor=1',
|
|
'--remote-debugging-port=' + PORT,
|
|
'--user-data-dir=C:\\Users\\SJL\\AppData\\Local\\Temp\\wt-chrome-preset1',
|
|
'about:blank',
|
|
], { stdio: 'ignore' });
|
|
const getJSON = (p) => new Promise((res, rej) => {
|
|
http.get({ host: '127.0.0.1', port: PORT, path: p }, (r) => { let d=''; r.on('data',c=>d+=c); r.on('end',()=>{try{res(JSON.parse(d))}catch(e){rej(e)}}) }).on('error', rej);
|
|
});
|
|
const sleep = (ms) => new Promise((r) => setTimeout(r, ms));
|
|
(async () => {
|
|
let list = null;
|
|
for (let i = 0; i < 40; i++) { try { list = await getJSON('/json/list'); if (list && list.length) break; } catch {} await sleep(500); }
|
|
const target = list.find((t) => t.type === 'page') || list[0];
|
|
const ws = new WebSocket(target.webSocketDebuggerUrl);
|
|
let id = 0; const pending = new Map();
|
|
const send = (method, params) => new Promise((res) => { const mid = ++id; pending.set(mid, res); ws.send(JSON.stringify({ id: mid, method, params })); });
|
|
ws.on('message', (raw) => { const m = JSON.parse(String(raw)); if (m.id && pending.has(m.id)) { pending.get(m.id)(m); pending.delete(m.id); } });
|
|
await new Promise((r) => ws.on('open', r));
|
|
await send('Runtime.enable');
|
|
await send('Page.enable');
|
|
await send('Page.navigate', { url: 'http://127.0.0.1:3080/' });
|
|
const evaluate = async (expr) => {
|
|
const r = await send('Runtime.evaluate', { expression: expr, returnByValue: true, awaitPromise: true });
|
|
if (r.result && r.result.exceptionDetails) return { __error: (r.result.exceptionDetails.exception || {}).description || r.result.exceptionDetails.text };
|
|
return r.result && r.result.result ? r.result.result.value : undefined;
|
|
};
|
|
let ready = null;
|
|
for (let i = 0; i < 30; i++) { await sleep(1000); ready = await evaluate("!!window.__dshPresetApi"); if (ready === true) break; }
|
|
await sleep(1200);
|
|
const out = await evaluate(`(async function(){
|
|
var out = {};
|
|
try {
|
|
var api = window.__dshPresetApi;
|
|
out.apiCaptured = !!api;
|
|
out.agentPresetsPresent = !!(api && api.agentPresets);
|
|
var lr = api && api.agentPresets && await api.agentPresets.list({});
|
|
out.listOk = lr && lr.result && lr.result.ok;
|
|
var presets = lr && lr.result && lr.result.ok ? (lr.result.value && lr.result.value.presets) : null;
|
|
out.count = Array.isArray(presets) ? presets.length : null;
|
|
out.rows = Array.isArray(presets) ? presets.map(function(p){ return { id: p.id, name: p.name, isDefault: !!p.isDefault, provider: p.provider || p.modelProvider || (p.composition && p.composition.provider) || null, keys: Object.keys(p).slice(0, 10) }; }) : null;
|
|
} catch (e) { out.err = String(e && e.stack || e); }
|
|
return out;
|
|
})()`);
|
|
console.log('RESULT ' + JSON.stringify(out));
|
|
try { ws.close(); } catch {}
|
|
proc.kill();
|
|
})().catch((e) => { console.error('PROBE FAIL', e); try { proc.kill(); } catch {} });
|