chore: import upstream snapshot with attribution

2026-07-13 13:01:18 +08:00
commit 979fb22d7c
613 changed files with 113494 additions and 0 deletions
@@ -0,0 +1,84 @@
+#!/usr/bin/env node
+//#region src/hooks/pre-tool-use.ts
+function isSdkChildContext(payload) {
+	if (process.env["AGENTMEMORY_SDK_CHILD"] === "1") return true;
+	if (!payload || typeof payload !== "object") return false;
+	return payload.entrypoint === "sdk-ts";
+}
+const INJECT_CONTEXT = process.env["AGENTMEMORY_INJECT_CONTEXT"] === "true";
+const REST_URL = process.env["AGENTMEMORY_URL"] || "http://localhost:3111";
+const SECRET = process.env["AGENTMEMORY_SECRET"] || "";
+function authHeaders() {
+	const h = { "Content-Type": "application/json" };
+	if (SECRET) h["Authorization"] = `Bearer ${SECRET}`;
+	return h;
+}
+async function main() {
+	if (!INJECT_CONTEXT) return;
+	let input = "";
+	for await (const chunk of process.stdin) input += chunk;
+	let data;
+	try {
+		data = JSON.parse(input);
+	} catch {
+		return;
+	}
+	if (isSdkChildContext(data)) return;
+	const toolName = typeof data.tool_name === "string" ? data.tool_name : typeof data.toolName === "string" ? data.toolName : void 0;
+	if (!toolName) return;
+	const normalizedToolName = toolName.toLowerCase();
+	if (![
+		"edit",
+		"write",
+		"create",
+		"read",
+		"view",
+		"glob",
+		"grep"
+	].includes(normalizedToolName)) return;
+	const rawToolInput = data.tool_input ?? data.toolArgs;
+	const toolInput = typeof rawToolInput === "object" && rawToolInput !== null && !Array.isArray(rawToolInput) ? rawToolInput : {};
+	const files = [];
+	const fileKeys = normalizedToolName === "grep" ? ["path", "file"] : [
+		"file_path",
+		"path",
+		"file",
+		"pattern"
+	];
+	for (const key of fileKeys) {
+		const val = toolInput[key];
+		if (typeof val === "string" && val.length > 0) files.push(val);
+	}
+	if (files.length === 0) return;
+	const terms = [];
+	if (normalizedToolName === "grep" || normalizedToolName === "glob") {
+		const pattern = toolInput["pattern"];
+		if (typeof pattern === "string" && pattern.length > 0) terms.push(pattern);
+	}
+	const rawSessionId = data.session_id || data.sessionId;
+	const sessionId = typeof rawSessionId === "string" && rawSessionId.length > 0 ? rawSessionId : "unknown";
+	const project = typeof data.project === "string" && data.project.trim().length > 0 ? data.project.trim() : void 0;
+	try {
+		const res = await fetch(`${REST_URL}/agentmemory/enrich`, {
+			method: "POST",
+			headers: authHeaders(),
+			body: JSON.stringify({
+				sessionId,
+				files,
+				terms,
+				toolName,
+				...project !== void 0 && { project }
+			}),
+			signal: AbortSignal.timeout(2e3)
+		});
+		if (res.ok) {
+			const result = await res.json();
+			if (result.context) process.stdout.write(result.context);
+		}
+	} catch {}
+}
+main();
+
+//#endregion
+export {  };
+//# sourceMappingURL=pre-tool-use.mjs.map