85 lines
2.4 KiB
TypeScript
85 lines
2.4 KiB
TypeScript
import fs from "node:fs/promises";
|
|
import path from "node:path";
|
|
|
|
import JSON5 from "json5";
|
|
|
|
import {
|
|
DEFAULT_AGENT_WORKSPACE_DIR,
|
|
ensureAgentWorkspace,
|
|
} from "../agents/workspace.js";
|
|
import { type ClawdbotConfig, CONFIG_PATH_CLAWDBOT } from "../config/config.js";
|
|
import { applyModelDefaults } from "../config/defaults.js";
|
|
import { resolveSessionTranscriptsDir } from "../config/sessions.js";
|
|
import type { RuntimeEnv } from "../runtime.js";
|
|
import { defaultRuntime } from "../runtime.js";
|
|
|
|
async function readConfigFileRaw(): Promise<{
|
|
exists: boolean;
|
|
parsed: ClawdbotConfig;
|
|
}> {
|
|
try {
|
|
const raw = await fs.readFile(CONFIG_PATH_CLAWDBOT, "utf-8");
|
|
const parsed = JSON5.parse(raw);
|
|
if (parsed && typeof parsed === "object") {
|
|
return { exists: true, parsed: parsed as ClawdbotConfig };
|
|
}
|
|
return { exists: true, parsed: {} };
|
|
} catch {
|
|
return { exists: false, parsed: {} };
|
|
}
|
|
}
|
|
|
|
async function writeConfigFile(cfg: ClawdbotConfig) {
|
|
await fs.mkdir(path.dirname(CONFIG_PATH_CLAWDBOT), { recursive: true });
|
|
const json = JSON.stringify(applyModelDefaults(cfg), null, 2)
|
|
.trimEnd()
|
|
.concat("\n");
|
|
await fs.writeFile(CONFIG_PATH_CLAWDBOT, json, "utf-8");
|
|
}
|
|
|
|
export async function setupCommand(
|
|
opts?: { workspace?: string },
|
|
runtime: RuntimeEnv = defaultRuntime,
|
|
) {
|
|
const desiredWorkspace =
|
|
typeof opts?.workspace === "string" && opts.workspace.trim()
|
|
? opts.workspace.trim()
|
|
: undefined;
|
|
|
|
const existingRaw = await readConfigFileRaw();
|
|
const cfg = existingRaw.parsed;
|
|
const agent = cfg.agent ?? {};
|
|
|
|
const workspace =
|
|
desiredWorkspace ?? agent.workspace ?? DEFAULT_AGENT_WORKSPACE_DIR;
|
|
|
|
const next: ClawdbotConfig = {
|
|
...cfg,
|
|
agent: {
|
|
...agent,
|
|
workspace,
|
|
},
|
|
};
|
|
|
|
if (!existingRaw.exists || agent.workspace !== workspace) {
|
|
await writeConfigFile(next);
|
|
runtime.log(
|
|
!existingRaw.exists
|
|
? `Wrote ${CONFIG_PATH_CLAWDBOT}`
|
|
: `Updated ${CONFIG_PATH_CLAWDBOT} (set agent.workspace)`,
|
|
);
|
|
} else {
|
|
runtime.log(`Config OK: ${CONFIG_PATH_CLAWDBOT}`);
|
|
}
|
|
|
|
const ws = await ensureAgentWorkspace({
|
|
dir: workspace,
|
|
ensureBootstrapFiles: !next.agent?.skipBootstrap,
|
|
});
|
|
runtime.log(`Workspace OK: ${ws.dir}`);
|
|
|
|
const sessionsDir = resolveSessionTranscriptsDir();
|
|
await fs.mkdir(sessionsDir, { recursive: true });
|
|
runtime.log(`Sessions OK: ${sessionsDir}`);
|
|
}
|