File size: 2,396 Bytes
f59fbe2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 | import { readFile } from "node:fs/promises";
import { join } from "node:path";
import type { JsonObject, PermissionMode } from "@moonshot-ai/kimi-code-sdk";
export const LEGACY_APPROVAL_METADATA_KEY = "vscode_legacy_approval";
export interface LegacyApprovalFlags {
readonly yolo: boolean;
readonly afk: boolean;
}
export function readLegacyApprovalFlags(
metadata: Readonly<Record<string, unknown>> | undefined,
): LegacyApprovalFlags | undefined {
const value = metadata?.[LEGACY_APPROVAL_METADATA_KEY];
return parseLegacyApprovalFlags(value);
}
export async function readMigratedLegacyApprovalFlags(
metadata: Readonly<Record<string, unknown>> | undefined,
): Promise<LegacyApprovalFlags | undefined> {
const sourcePath = metadata?.["kimi_cli_source_path"];
if (typeof sourcePath !== "string" || sourcePath.length === 0) return undefined;
let text: string;
try {
text = await readFile(join(sourcePath, "state.json"), "utf8");
} catch (error) {
if ((error as NodeJS.ErrnoException).code === "ENOENT") return undefined;
throw error;
}
const state = JSON.parse(text) as { readonly approval?: unknown };
return parseLegacyApprovalFlags(state.approval);
}
function parseLegacyApprovalFlags(value: unknown): LegacyApprovalFlags | undefined {
if (typeof value !== "object" || value === null || Array.isArray(value)) return undefined;
const yolo = Reflect.get(value, "yolo");
const afk = Reflect.get(value, "afk");
if (typeof yolo !== "boolean" && typeof afk !== "boolean") return undefined;
return {
yolo: typeof yolo === "boolean" ? yolo : false,
afk: typeof afk === "boolean" ? afk : false,
};
}
export function legacyApprovalMetadata(flags: LegacyApprovalFlags): JsonObject {
return {
[LEGACY_APPROVAL_METADATA_KEY]: {
yolo: flags.yolo,
afk: flags.afk,
},
};
}
export function corePermissionForLegacyApproval(flags: LegacyApprovalFlags): PermissionMode {
if (flags.afk) return "auto";
return flags.yolo ? "yolo" : "manual";
}
/**
* The global `kimi.yoloMode` setting is authoritative whenever a session
* attaches to the runtime; afk stays per-session because it has no global
* setting counterpart.
*/
export function withGlobalYoloMode(
flags: LegacyApprovalFlags,
yoloMode: boolean,
): LegacyApprovalFlags {
return flags.yolo === yoloMode ? flags : { yolo: yoloMode, afk: flags.afk };
}
|