Files
pi-agent-integrated/pi-web/app/api/project-trust/route.ts

67 lines
2.5 KiB
TypeScript

import { stat } from "fs/promises";
import { resolve } from "path";
import { NextResponse } from "next/server";
import { getAgentDir } from "@earendil-works/pi-coding-agent";
import { getAllowedFileRoots, isExistingFilePathAllowed } from "@/lib/file-access";
import { invalidateModelsCache } from "@/lib/models-cache";
import { getProjectTrustStatus, trustProject } from "@/lib/project-trust";
import { destroyRpcSessionsForCwd, hasBusyRpcSessionForCwd } from "@/lib/rpc-manager";
export const dynamic = "force-dynamic";
async function validateCwd(value: unknown): Promise<
{ cwd: string } | { response: NextResponse }
> {
if (typeof value !== "string" || !value.trim()) {
return { response: NextResponse.json({ error: "cwd required" }, { status: 400 }) };
}
const cwd = resolve(value);
try {
if (!(await stat(cwd)).isDirectory()) {
return { response: NextResponse.json({ error: "cwd must be a directory" }, { status: 400 }) };
}
} catch {
return { response: NextResponse.json({ error: "Directory does not exist" }, { status: 400 }) };
}
const allowedRoots = await getAllowedFileRoots();
if (!isExistingFilePathAllowed(cwd, allowedRoots)) {
return { response: NextResponse.json({ error: "Access denied" }, { status: 403 }) };
}
return { cwd };
}
export async function GET(req: Request) {
const result = await validateCwd(new URL(req.url).searchParams.get("cwd"));
if ("response" in result) return result.response;
return NextResponse.json(getProjectTrustStatus(result.cwd, getAgentDir()));
}
export async function POST(req: Request) {
try {
const body = await req.json() as { cwd?: unknown };
const result = await validateCwd(body.cwd);
if ("response" in result) return result.response;
const agentDir = getAgentDir();
const current = getProjectTrustStatus(result.cwd, agentDir);
if (!current.requiresTrust) {
return NextResponse.json({ error: "This project has no resources that require trust" }, { status: 409 });
}
if (hasBusyRpcSessionForCwd(result.cwd)) {
return NextResponse.json({ error: "Wait for the active session to finish before trusting this project" }, { status: 409 });
}
const status = trustProject(result.cwd, agentDir);
invalidateModelsCache();
destroyRpcSessionsForCwd(result.cwd);
return NextResponse.json(status);
} catch (error) {
return NextResponse.json(
{ error: error instanceof Error ? error.message : String(error) },
{ status: 500 },
);
}
}