雷电芽衣
bc6983a62e
Split the 657-line operation service into eor, submit, and verify modules, add the AGENT_REVOKED and KEY_RETIRED codes the Backend already emits, and condense every multi-line comment. Mirror Node, Python, and Go at 2.1.0 with the deep-health 503, non-idempotent retry, rotation grace, and path-encoding fixes. Document which routes and response fields the self-hosted server does not provide. Claude-Session: https://claude.ai/code/session_01Agoc9bRVE9KL7mmWtey9Vq
116 lines
3.8 KiB
JavaScript
116 lines
3.8 KiB
JavaScript
import assert from 'node:assert/strict';
|
|
import { mkdir, readFile, writeFile } from 'node:fs/promises';
|
|
import path from 'node:path';
|
|
import { parse as parseToml } from '@decimalturn/toml-patch';
|
|
import {
|
|
createFixtureRoot,
|
|
distUrl,
|
|
homeEnvironment,
|
|
installConfig as baseInstallConfig,
|
|
optionalEnvironment,
|
|
runNode as baseNode,
|
|
runPlugin as basePlugin,
|
|
runShell,
|
|
} from './harness.mjs';
|
|
|
|
export { VALID_PRIVATE_KEY, cliPath, registryModuleUrl, startApiServer } from './harness.mjs';
|
|
|
|
export const pluginModuleUrl = distUrl('plugins/kimi.js');
|
|
export const contractModuleUrl = distUrl('plugins/kimi-contract.js');
|
|
export const ioModuleUrl = distUrl('plugins/kimi-io.js');
|
|
export const installationModuleUrl = distUrl('plugins/kimi-installation.js');
|
|
const PLUGIN = { exportName: 'kimiPlugin', moduleUrl: pluginModuleUrl };
|
|
|
|
export function runNode(args, env, cwd, input = '', unset = []) {
|
|
return baseNode(args, env, cwd, input, { unset });
|
|
}
|
|
|
|
export function runKimiHook(command, input, fixture, environment = {}) {
|
|
return runShell(command, {
|
|
...homeEnvironment(fixture),
|
|
KIMI_CODE_HOME: fixture.kimiHomeOverride ?? '',
|
|
...environment,
|
|
}, fixture.projectDir, input);
|
|
}
|
|
|
|
export function installConfig(fixture, overrides = {}) {
|
|
return baseInstallConfig('kimi', fixture, overrides);
|
|
}
|
|
|
|
export function runPlugin(fixture, method, argument, environment = {}) {
|
|
const home = optionalEnvironment('KIMI_CODE_HOME', fixture.kimiHomeOverride);
|
|
return basePlugin(PLUGIN, fixture, method, argument, {
|
|
env: { ...environment, ...home.env },
|
|
unset: home.unset,
|
|
});
|
|
}
|
|
|
|
async function writeOptional(filePath, contents) {
|
|
if (contents === undefined) return;
|
|
await mkdir(path.dirname(filePath), { recursive: true });
|
|
await writeFile(filePath, contents, { encoding: 'utf-8', mode: 0o600 });
|
|
}
|
|
|
|
export async function createFixture({
|
|
baseUrl = 'http://127.0.0.1:9',
|
|
stableConfig,
|
|
legacyConfig,
|
|
stableDetected = true,
|
|
legacyDetected = true,
|
|
explicitKimiHome = true,
|
|
} = {}) {
|
|
const base = await createFixtureRoot(
|
|
'elydora-kimi-',
|
|
"home with spaces and 'quote %ELYDORA_HOOK_PATH%",
|
|
);
|
|
const kimiHome = explicitKimiHome
|
|
? path.join(base.homeDir, 'custom kimi-code')
|
|
: path.join(base.homeDir, '.kimi-code');
|
|
const stablePath = path.join(kimiHome, 'config.toml');
|
|
const legacyHome = path.join(base.homeDir, '.kimi');
|
|
const legacyPath = path.join(legacyHome, 'config.toml');
|
|
if (stableDetected && !explicitKimiHome) await mkdir(kimiHome, { recursive: true });
|
|
if (legacyDetected) await mkdir(legacyHome, { recursive: true });
|
|
await writeOptional(stablePath, stableConfig);
|
|
await writeOptional(legacyPath, legacyConfig);
|
|
return {
|
|
...base,
|
|
baseUrl,
|
|
kimiHome,
|
|
kimiHomeOverride: explicitKimiHome ? kimiHome : undefined,
|
|
legacyHome,
|
|
legacyPath,
|
|
stablePath,
|
|
install(overrides = {}) {
|
|
return runPlugin(this, 'install', installConfig(this, overrides));
|
|
},
|
|
};
|
|
}
|
|
|
|
export async function readKimiConfig(filePath) {
|
|
const raw = await readFile(filePath, 'utf-8');
|
|
return { raw, config: parseToml(raw) };
|
|
}
|
|
|
|
export function managedHook(config, event) {
|
|
return config.hooks?.findLast((hook) => hook.event === event && hook.timeout === 10);
|
|
}
|
|
|
|
export function assertManagedHook(hook, event) {
|
|
assert(hook);
|
|
assert.deepEqual(Object.keys(hook).sort(), ['command', 'event', 'timeout']);
|
|
assert.equal(hook.event, event);
|
|
assert.equal(hook.timeout, 10);
|
|
if (process.platform === 'win32') {
|
|
assert.match(hook.command, /^"[^"\r\n]+powershell\.exe" .* -EncodedCommand /i);
|
|
} else {
|
|
assert.match(hook.command, /^'[^']*node[^']*' /i);
|
|
}
|
|
}
|
|
|
|
export function legacyCommand(scriptPath) {
|
|
if (process.platform === 'win32') return `"${process.execPath}" "${scriptPath}"`;
|
|
const quote = (value) => `'${value.replaceAll("'", `'"'"'`)}'`;
|
|
return `${quote(process.execPath)} ${quote(scriptPath)}`;
|
|
}
|