avtc-pi-featyard 1.1.0

package/src/snippets/vendored/subscribe-to-parallel-work-guardrail.ts

@@ -0,0 +1,35 @@
+// SPDX-License-Identifier: MIT
+// SPDX-FileCopyrightText: 2026 avtc <tarasenkov@gmail.com>
+
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+
+export type GuardrailWhitelistCheck = (categoryId: string) => boolean;
+
+/** Pass as `addWhitelistCheck` when no whitelist check is needed. */
+export const NO_GUARDRAIL_WHITELIST_CHECK: GuardrailWhitelistCheck | null = null;
+
+/**
+ * Subscribe to pi-parallel-work-guardrail:ready and register hooks.
+ * Reload-safe: session_shutdown fires before reload, cleaning all listeners.
+ * Copy this file into your consumer's src/snippets/vendored/ directory verbatim — no changes needed.
+ */
+export function subscribeToGuardrail(pi: ExtensionAPI, addWhitelistCheck: GuardrailWhitelistCheck | null): void {
+  const unsubs: Array<() => void> = [];
+
+  // On session_shutdown (fires before reload): clean pi.events.on listeners
+  // Note: pi.on handlers are auto-cleaned on reload (new Extension objects)
+  pi.on("session_shutdown", () => {
+    for (const unsub of unsubs) unsub();
+    unsubs.length = 0;
+  });
+
+  // Register :ready listener
+  unsubs.push(
+    pi.events.on("pi-parallel-work-guardrail:ready", (data: unknown) => {
+      const api = data as {
+        addWhitelistCheck?: (check: GuardrailWhitelistCheck) => void;
+      };
+      if (addWhitelistCheck) api.addWhitelistCheck?.(addWhitelistCheck);
+    }),
+  );
+}

package/src/snippets/vendored/subscribe-to-subagent.ts

@@ -0,0 +1,67 @@
+// SPDX-License-Identifier: MIT
+// SPDX-FileCopyrightText: 2026 avtc <tarasenkov@gmail.com>
+
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+
+export type SubagentPromptTransformer = (
+  systemPrompt: string,
+  context: {
+    agentName: string;
+    task?: string;
+    isFork: boolean;
+  },
+) => string | Promise<string>;
+export type SubagentModelResolver = (args: {
+  agentName: string;
+  explicitModel: string | undefined;
+}) => string | undefined;
+
+/** Pass as `addPromptTransformer` when no prompt-transformer hook is needed. */
+export const NO_SUBAGENT_PROMPT_TRANSFORMER: SubagentPromptTransformer | null = null;
+/** Pass as `addModelResolver` when no model-resolver hook is needed. */
+export const NO_SUBAGENT_MODEL_RESOLVER: SubagentModelResolver | null = null;
+/** Pass as `addSkillPaths` when no extra skill paths are needed. */
+export const NO_SUBAGENT_SKILL_PATHS: string[] | null = null;
+/** Pass as `addAgentsPaths` when no extra agents paths are needed. */
+export const NO_SUBAGENT_AGENTS_PATHS: string[] | null = null;
+
+/** API object emitted on the `pi-subagent:ready` event, exposing registration hooks. */
+interface SubagentReadyApi {
+  addPromptTransformer: (transformer: SubagentPromptTransformer) => void;
+  addModelResolver: (resolver: SubagentModelResolver) => void;
+  addSkillPaths: (paths: string[]) => void;
+  addAgentsPaths: (paths: string[], extensionName: string) => void;
+}
+
+/**
+ * Subscribe to pi-subagent:ready and register hooks.
+ * Reload-safe: session_shutdown fires before reload, cleaning all listeners.
+ * Copy this file into your consumer's src/snippets/vendored/ directory verbatim — no changes needed.
+ */
+export function subscribeToSubagent(
+  pi: ExtensionAPI,
+  addPromptTransformer: SubagentPromptTransformer | null,
+  addModelResolver: SubagentModelResolver | null,
+  addSkillPaths: string[] | null,
+  addAgentsPaths: string[] | null,
+  extensionName: string,
+): void {
+  const unsubs: Array<() => void> = [];
+
+  // On session_shutdown (fires before reload): clean pi.events.on listeners
+  pi.on("session_shutdown", () => {
+    for (const unsub of unsubs) unsub();
+    unsubs.length = 0;
+  });
+
+  // Register :ready listener
+  unsubs.push(
+    pi.events.on("pi-subagent:ready", (data) => {
+      const api = data as SubagentReadyApi;
+      if (addPromptTransformer) api.addPromptTransformer(addPromptTransformer);
+      if (addModelResolver) api.addModelResolver(addModelResolver);
+      if (addSkillPaths) api.addSkillPaths(addSkillPaths);
+      if (addAgentsPaths) api.addAgentsPaths(addAgentsPaths, extensionName);
+    }),
+  );
+}

package/src/snippets/vendored/subscribe-to-todo.ts

@@ -0,0 +1,63 @@
+// SPDX-License-Identifier: MIT
+// SPDX-FileCopyrightText: 2026 avtc <tarasenkov@gmail.com>
+
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+
+/** API object emitted on the `pi-todo:ready` event. */
+export interface TodoReadyApi {
+  disableBuiltInFollowUp: () => void;
+  getCompletedItemId?: () => string | null;
+  getInProgressItem?: () => string | null;
+  areAllTodosDone?: () => boolean;
+}
+
+/**
+ * Subscribe to pi-todo:ready and register hooks.
+ * Reload-safe: session_shutdown fires before reload, cleaning all listeners.
+ * Copy this file into your consumer's src/snippets/vendored/ directory verbatim — no changes needed.
+ *
+ * Returns sync lazy proxy — getCompletedItemId, getInProgressItem and areAllTodosDone delegate to
+ * internal _api ref populated when :ready fires at session_start.
+ * disableBuiltInFollowUp is passed as a boolean flag, applied inside :ready handler.
+ * No pending queue needed — the flag is evaluated when _api is available.
+ */
+export function subscribeToTodo(
+  pi: ExtensionAPI,
+  disableBuiltInFollowUp: boolean,
+): {
+  getCompletedItemId(): string | null;
+  getInProgressItem(): string | null;
+  areAllTodosDone(): boolean;
+} {
+  const unsubs: Array<() => void> = [];
+
+  // Internal API ref — populated when :ready fires
+  let _api: TodoReadyApi | null = null;
+
+  // On session_shutdown (fires before reload): clean pi.events.on listeners
+  pi.on("session_shutdown", () => {
+    for (const unsub of unsubs) unsub();
+    unsubs.length = 0;
+  });
+
+  // Register :ready listener
+  unsubs.push(
+    pi.events.on("pi-todo:ready", (api: unknown) => {
+      _api = api as TodoReadyApi;
+      if (disableBuiltInFollowUp) _api.disableBuiltInFollowUp();
+    }),
+  );
+
+  // Return sync lazy proxy — methods delegate to _api when available
+  return {
+    getCompletedItemId(): string | null {
+      return _api?.getCompletedItemId?.() ?? null;
+    },
+    getInProgressItem(): string | null {
+      return _api?.getInProgressItem?.() ?? null;
+    },
+    areAllTodosDone(): boolean {
+      return _api?.areAllTodosDone?.() ?? true;
+    },
+  };
+}