File size: 3,393 Bytes
fc93158 | 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 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 | import fs from "node:fs/promises";
import path from "node:path";
import { describe, expect, it } from "vitest";
import type { OpenClawConfig } from "../../../config/config.js";
import { makeTempWorkspace, writeWorkspaceFile } from "../../../test-helpers/workspace.js";
import type { AgentBootstrapHookContext } from "../../hooks.js";
import { createHookEvent } from "../../hooks.js";
import handler from "./handler.js";
function createBootstrapExtraConfig(paths: string[]): OpenClawConfig {
return {
hooks: {
internal: {
entries: {
"bootstrap-extra-files": {
enabled: true,
paths,
},
},
},
},
};
}
async function createBootstrapContext(params: {
workspaceDir: string;
cfg: OpenClawConfig;
sessionKey: string;
rootFiles: Array<{ name: string; content: string }>;
}): Promise<AgentBootstrapHookContext> {
const bootstrapFiles = (await Promise.all(
params.rootFiles.map(async (file) => ({
name: file.name,
path: await writeWorkspaceFile({
dir: params.workspaceDir,
name: file.name,
content: file.content,
}),
content: file.content,
missing: false,
})),
)) as AgentBootstrapHookContext["bootstrapFiles"];
return {
workspaceDir: params.workspaceDir,
bootstrapFiles,
cfg: params.cfg,
sessionKey: params.sessionKey,
};
}
describe("bootstrap-extra-files hook", () => {
it("appends extra bootstrap files from configured patterns", async () => {
const tempDir = await makeTempWorkspace("openclaw-bootstrap-extra-");
const extraDir = path.join(tempDir, "packages", "core");
await fs.mkdir(extraDir, { recursive: true });
await fs.writeFile(path.join(extraDir, "AGENTS.md"), "extra agents", "utf-8");
const cfg = createBootstrapExtraConfig(["packages/*/AGENTS.md"]);
const context = await createBootstrapContext({
workspaceDir: tempDir,
cfg,
sessionKey: "agent:main:main",
rootFiles: [{ name: "AGENTS.md", content: "root agents" }],
});
const event = createHookEvent("agent", "bootstrap", "agent:main:main", context);
await handler(event);
const injected = context.bootstrapFiles.filter((f) => f.name === "AGENTS.md");
expect(injected).toHaveLength(2);
expect(injected.some((f) => f.path.endsWith(path.join("packages", "core", "AGENTS.md")))).toBe(
true,
);
});
it("re-applies subagent bootstrap allowlist after extras are added", async () => {
const tempDir = await makeTempWorkspace("openclaw-bootstrap-extra-subagent-");
const extraDir = path.join(tempDir, "packages", "persona");
await fs.mkdir(extraDir, { recursive: true });
await fs.writeFile(path.join(extraDir, "SOUL.md"), "evil", "utf-8");
const cfg = createBootstrapExtraConfig(["packages/*/SOUL.md"]);
const context = await createBootstrapContext({
workspaceDir: tempDir,
cfg,
sessionKey: "agent:main:subagent:abc",
rootFiles: [
{ name: "AGENTS.md", content: "root agents" },
{ name: "TOOLS.md", content: "root tools" },
],
});
const event = createHookEvent("agent", "bootstrap", "agent:main:subagent:abc", context);
await handler(event);
expect(context.bootstrapFiles.map((f) => f.name).toSorted()).toEqual([
"AGENTS.md",
"SOUL.md",
"TOOLS.md",
]);
});
});
|