@zihanw/pi-forge 0.3.0 → 0.3.2

package/src/web-editor/server.ts

@@ -78,6 +78,11 @@ async function handleRequest(host: WebEditorHost, token: string, req: IncomingMe
 		return;
 	}
 
+	if (req.method === "GET" && parts[1] === "resources" && parts.length === 2) {
+		sendJson(res, 200, host.listResources());
+		return;
+	}
+
 	if (req.method === "POST" && parts[1] === "stacks" && parts.length === 2) {
 		const body = await readJsonBody(req);
 		const parsed = readStackPayload(body);
@@ -88,7 +93,7 @@ async function handleRequest(host: WebEditorHost, token: string, req: IncomingMe
 		const options = isPlainObject(body)
 			? { activate: body.activate === true, overwrite: body.overwrite === true }
 			: {};
-		const result = host.createStack(parsed.stack, options);
+		const result = await host.createStack(parsed.stack, options);
 		if (result.ok && parsed.importFormat) {
 			sendJson(res, 200, { ...result, importFormat: parsed.importFormat, importReport: parsed.importReport });
 			return;
@@ -114,12 +119,12 @@ async function handleRequest(host: WebEditorHost, token: string, req: IncomingMe
 			sendJson(res, 400, { error: parsed.error });
 			return;
 		}
-		sendOperation(res, host.saveStack(parts[2]!, parsed.stack));
+		sendOperation(res, await host.saveStack(parts[2]!, parsed.stack));
 		return;
 	}
 
 	if (req.method === "DELETE" && parts[1] === "stacks" && parts.length === 3) {
-		sendOperation(res, host.deleteStack(parts[2]!));
+		sendOperation(res, await host.deleteStack(parts[2]!));
 		return;
 	}
 
@@ -173,7 +178,7 @@ async function handleRequest(host: WebEditorHost, token: string, req: IncomingMe
 	}
 
 	if (req.method === "POST" && parts[1] === "reload" && parts.length === 2) {
-		sendOperation(res, host.reloadStacks());
+		sendOperation(res, await host.reloadStacks());
 		return;
 	}
 

package/src/web-editor/types.ts

@@ -16,10 +16,11 @@ export interface WebEditorStackSummary {
 export interface WebEditorHost {
 	cwd: string;
 	listStacks(): WebEditorStackSummary[];
+	listResources(): WebEditorPolicyResources;
 	getStack(id: string): { stack: PromptStack; filePath: string; diagnostics: PromptStackDiagnostic[] } | undefined;
-	createStack(stack: PromptStack, options: WebEditorCreateStackOptions): WebEditorOperationResult<{ stack: WebEditorStackSummary; stacks: WebEditorStackSummary[] }>;
-	saveStack(id: string, stack: PromptStack): WebEditorOperationResult<{ stack: WebEditorStackSummary; stacks: WebEditorStackSummary[] }>;
-	deleteStack(id: string): WebEditorOperationResult<{ activeId?: string; stacks: WebEditorStackSummary[] }>;
+	createStack(stack: PromptStack, options: WebEditorCreateStackOptions): Promise<WebEditorOperationResult<{ stack: WebEditorStackSummary; stacks: WebEditorStackSummary[] }>>;
+	saveStack(id: string, stack: PromptStack): Promise<WebEditorOperationResult<{ stack: WebEditorStackSummary; stacks: WebEditorStackSummary[] }>>;
+	deleteStack(id: string): Promise<WebEditorOperationResult<{ activeId?: string; stacks: WebEditorStackSummary[] }>>;
 	validateStack(stack: PromptStack): PromptStackDiagnostic[];
 	previewStack(id: string, stack: PromptStack): WebEditorOperationResult<{ text: string; preview?: WebEditorPreview; diagnostics: PromptStackDiagnostic[] }>;
 	getPayload(): WebEditorOperationResult<WebEditorPayloadSnapshot>;
@@ -27,7 +28,7 @@ export interface WebEditorHost {
 	clearPayload(): WebEditorOperationResult<WebEditorPayloadSnapshot>;
 	activateStack(id: string): WebEditorOperationResult<{ activeId?: string; stacks: WebEditorStackSummary[] }>;
 	disableStacks(): WebEditorOperationResult<{ activeId?: string; stacks: WebEditorStackSummary[] }>;
-	reloadStacks(): WebEditorOperationResult<{ activeId?: string; stacks: WebEditorStackSummary[] }>;
+	reloadStacks(): Promise<WebEditorOperationResult<{ activeId?: string; stacks: WebEditorStackSummary[] }>>;
 }
 
 export interface WebEditorPreviewSection {
@@ -70,6 +71,19 @@ export interface WebEditorCreateStackOptions {
 	overwrite?: boolean;
 }
 
+export interface WebEditorPolicyResource {
+	name: string;
+	description?: string;
+	source?: string;
+	active?: boolean;
+	hidden?: boolean;
+}
+
+export interface WebEditorPolicyResources {
+	tools: WebEditorPolicyResource[];
+	skills: WebEditorPolicyResource[];
+}
+
 export type WebEditorOperationResult<T> = ({ ok: true } & T) | { ok: false; status?: number; error: string };
 
 export interface WebEditorServer {

package/src/web-host.ts

@@ -12,6 +12,7 @@ import type {
 	WebEditorHost,
 	WebEditorOperationResult,
 	WebEditorPayloadSnapshot,
+	WebEditorPolicyResources,
 	WebEditorPreview,
 	WebEditorStackSummary,
 } from "./web-editor/index.ts";
@@ -22,12 +23,13 @@ export interface WebHostRuntime {
 	getActiveId(): string | undefined;
 	getSelectedActiveId(): string | undefined;
 	setActive(id: string | undefined): boolean;
-	reloadStacks(preferredId?: string): void;
+	reloadStacks(preferredId?: string): Promise<void>;
 	buildPreview(target: LoadedPromptStack): {
 		text: string;
 		preview: WebEditorPreview;
 		diagnostics: PromptStackDiagnostic[];
 	};
+	getPolicyResources(): WebEditorPolicyResources;
 	getPayload(): WebEditorOperationResult<WebEditorPayloadSnapshot>;
 	armPayload(savePath?: string): WebEditorOperationResult<WebEditorPayloadSnapshot>;
 	clearPayload(): WebEditorOperationResult<WebEditorPayloadSnapshot>;
@@ -37,6 +39,7 @@ export function createWebEditorHost(ctx: ExtensionCommandContext, runtime: WebHo
 	return {
 		cwd: ctx.cwd,
 		listStacks: () => stackSummaries(runtime.getStacks(), runtime.getActive()),
+		listResources: () => runtime.getPolicyResources(),
 		getStack: (id) => {
 			const loaded = runtime.getStacks().find((candidate) => candidate.stack.id === id);
 			return loaded ? { stack: loaded.stack, filePath: loaded.filePath, diagnostics: loaded.diagnostics } : undefined;
@@ -63,8 +66,8 @@ export function createWebEditorHost(ctx: ExtensionCommandContext, runtime: WebHo
 			runtime.setActive("none");
 			return { ok: true, activeId: runtime.getActiveId(), stacks: stackSummaries(runtime.getStacks(), runtime.getActive()) };
 		},
-		reloadStacks: () => {
-			runtime.reloadStacks(runtime.getSelectedActiveId());
+		reloadStacks: async () => {
+			await runtime.reloadStacks(runtime.getSelectedActiveId());
 			return { ok: true, activeId: runtime.getActiveId(), stacks: stackSummaries(runtime.getStacks(), runtime.getActive()) };
 		},
 	};
@@ -128,12 +131,12 @@ export function loadWebEditorSettings(ctx: ExtensionCommandContext): { preferred
 	};
 }
 
-function saveStackFile(
+async function saveStackFile(
 	ctx: ExtensionCommandContext,
 	runtime: WebHostRuntime,
 	id: string,
 	stack: PromptStack,
-): WebEditorOperationResult<{ stack: WebEditorStackSummary; stacks: WebEditorStackSummary[] }> {
+): Promise<WebEditorOperationResult<{ stack: WebEditorStackSummary; stacks: WebEditorStackSummary[] }>> {
 	if (!ctx.isProjectTrusted()) {
 		return { ok: false, status: 403, error: "Project is not trusted; refusing to save prompt stacks." };
 	}
@@ -146,18 +149,18 @@ function saveStackFile(
 
 	writeFileSync(target.filePath, `${JSON.stringify(stack, null, 2)}\n`, "utf8");
 	const preferredId = runtime.getActive()?.stack.id === id ? stack.id : runtime.getSelectedActiveId();
-	runtime.reloadStacks(preferredId);
+	await runtime.reloadStacks(preferredId);
 	const saved = runtime.getStacks().find((candidate) => candidate.stack.id === stack.id) ?? runtime.getStacks().find((candidate) => candidate.filePath === target.filePath);
 	if (!saved) return { ok: false, status: 500, error: "Saved stack could not be reloaded." };
 	return { ok: true, stack: stackSummary(saved, runtime.getActive()), stacks: stackSummaries(runtime.getStacks(), runtime.getActive()) };
 }
 
-function createStackFile(
+async function createStackFile(
 	ctx: ExtensionCommandContext,
 	runtime: WebHostRuntime,
 	stack: PromptStack,
 	options: WebEditorCreateStackOptions,
-): WebEditorOperationResult<{ stack: WebEditorStackSummary; stacks: WebEditorStackSummary[] }> {
+): Promise<WebEditorOperationResult<{ stack: WebEditorStackSummary; stacks: WebEditorStackSummary[] }>> {
 	if (!ctx.isProjectTrusted()) {
 		return { ok: false, status: 403, error: "Project is not trusted; refusing to create prompt stacks." };
 	}
@@ -182,7 +185,7 @@ function createStackFile(
 	const previousSelection = runtime.getSelectedActiveId();
 	mkdirSync(dirname(targetPath), { recursive: true });
 	writeFileSync(targetPath, `${JSON.stringify(stack, null, 2)}\n`, "utf8");
-	runtime.reloadStacks(options.activate ? stack.id : (previousSelection ?? "none"));
+	await runtime.reloadStacks(options.activate ? stack.id : (previousSelection ?? "none"));
 	if (options.activate) runtime.setActive(stack.id);
 
 	const created = runtime.getStacks().find((candidate) => candidate.stack.id === stack.id);
@@ -190,11 +193,11 @@ function createStackFile(
 	return { ok: true, stack: stackSummary(created, runtime.getActive()), stacks: stackSummaries(runtime.getStacks(), runtime.getActive()) };
 }
 
-function deleteStackFile(
+async function deleteStackFile(
 	ctx: ExtensionCommandContext,
 	runtime: WebHostRuntime,
 	id: string,
-): WebEditorOperationResult<{ activeId?: string; stacks: WebEditorStackSummary[] }> {
+): Promise<WebEditorOperationResult<{ activeId?: string; stacks: WebEditorStackSummary[] }>> {
 	if (!ctx.isProjectTrusted()) {
 		return { ok: false, status: 403, error: "Project is not trusted; refusing to delete prompt stacks." };
 	}
@@ -209,9 +212,9 @@ function deleteStackFile(
 	unlinkSync(target.filePath);
 	if (wasActive) {
 		runtime.setActive("none");
-		runtime.reloadStacks("none");
+		await runtime.reloadStacks("none");
 	} else {
-		runtime.reloadStacks(runtime.getSelectedActiveId());
+		await runtime.reloadStacks(runtime.getSelectedActiveId());
 	}
 	return { ok: true, activeId: runtime.getActiveId(), stacks: stackSummaries(runtime.getStacks(), runtime.getActive()) };
 }