Files
harness-tools/extensions/subagents/docs/effect-v4-extension-guide.md
T

355 lines
14 KiB
Markdown

# Building a pi extension on the Effect v4 setup
> Practical, migration-oriented companion to [`effect-v4-notes.md`](./effect-v4-notes.md).
> That file is the Effect API cheat-sheet (v3→v4 renames, child processes, concurrency);
> **this** file is how to stand up _one pi extension_ on the same toolchain and where to
> draw the line between "wrap in Effect" and "leave as plain TS".
>
> **Verified against** `effect@4.0.0-beta.98`, `@effect/platform-node@4.0.0-beta.98`,
> `@effect/tsgo@0.19.0`, `typescript@7.0.2` (checked 2026-07-13, in `extensions/subagents`).
> `npm run check` there passes clean — use it as the reference implementation.
>
> Audience: the agents migrating `firecrawl-search`, `ask-user`, `model-info`,
> `git-info`, `ui-customization`, and `copy-all`.
---
## 0. The one rule: Effect is for the async core, not the whole file
A pi extension's public surface is **plain callbacks** the host calls:
`export default function (pi: ExtensionAPI)`, `pi.registerTool({ …, async execute() })`,
`pi.registerCommand`, `pi.on(...)`, renderers. None of that changes. Effect lives _inside_
those callbacks — you run an effect at the top of `execute` and return its value.
Reach for Effect only where you actually get something from it:
- **Yes:** async work that needs typed errors, cancellation via the tool `AbortSignal`,
timeouts, retries/polling, or a resource whose lifetime must outlive one call
(child process, subscription) → child processes (`git-info`, `copy-all`), the
Firecrawl SDK calls (`firecrawl-search`), git/gh polling (`git-info`).
- **No / barely:** pure TUI popups and rendering (`ask-user`, `ui-customization`),
cross-extension channel plumbing, cost/token bookkeeping (`model-info`). These are
synchronous or already-Promise UI code; wrapping them in Effect adds ceremony and no
safety. Migrate them by keeping the logic and only touching whatever genuinely async
part benefits (usually nothing).
If an extension has no async core worth typing, the "migration" may be just adopting the
toolchain (§1) and leaving the body plain. Don't invent an Effect layer to have one.
---
## 1. Per-extension toolchain (copy this exactly)
Each extension is its own npm package with its own `node_modules`. Replicate the pinned
setup — do **not** float the versions.
`package.json`:
```jsonc
{
"name": "<ext>",
"private": true,
"type": "module",
"scripts": {
"check": "tsc --noEmit -p .",
"prepare": "effect-tsgo patch", // patches the Effect LS into the tsgo binary
},
"dependencies": {
"effect": "4.0.0-beta.98", // EXACT pin, no ^
"@effect/platform-node": "4.0.0-beta.98", // only if you touch fs / child processes
},
"devDependencies": {
"@effect/tsgo": "^0.19.0",
"typescript": "^7.0.2",
},
}
```
`tsconfig.json` (extends the repo root, adds the Effect language-service plugin):
```jsonc
{
"extends": "../../tsconfig.json",
"compilerOptions": {
"plugins": [{ "name": "@effect/language-service" }],
},
"include": ["index.ts", "src/**/*.ts", "*.test.ts"],
}
```
The root `tsconfig.json` already sets `strict`, `module`/`moduleResolution: NodeNext`,
`verbatimModuleSyntax`, `allowImportingTsExtensions`, `target: ES2022`, `types: ["node"]`.
Keep local `.ts` imports written **with the `.ts` extension** (`./src/manager.ts`), matching
the house style.
### The LSP / language-service, precisely
- `typescript@7` is the **native (tsgo) TypeScript** — its `tsc` is a Go binary, and it's
what `npm run check` runs.
- `@effect/language-service` is **not an installed npm package** here (you won't find it in
`node_modules`). It's delivered by `effect-tsgo patch`, run automatically by the `prepare`
lifecycle script on `npm install`. The patch injects the Effect Language Service into the
tsgo binary; the tsconfig `plugins` entry then turns on Effect-aware editor diagnostics and
quickfixes (e.g. "yield missing services", floating effects).
- Practical sequence for a fresh/edited extension: `npm install` (runs the patch) →
`npm run check`. If editor diagnostics look stale, re-run `npx effect-tsgo patch`;
`npx effect-tsgo get-exe-path` prints the patched binary it resolved.
- The plugin drives the _editor_; it does not change `tsc` exit codes. `npm run check` is
still your ground-truth green/red signal.
---
## 2. The async boundary: one `ManagedRuntime` + a `runTool` helper
Every extension that uses Effect needs exactly this shape. Build **one** runtime lazily,
dispose it on shutdown, and funnel every tool handler through a single helper that turns
Effect exits into what pi expects (a value, or a thrown `Error`; interruption → a friendly
message). This is lifted verbatim from `src/runtime.ts` — reuse it.
```ts
// src/runtime.ts
import { Cause, Exit, Layer, ManagedRuntime, type Effect } from "effect";
import { NodeServices } from "@effect/platform-node"; // only if you need fs / processes
// Compose your services here (see §3). NodeServices.layer =
// ChildProcessSpawner | FileSystem | Path | Crypto | Stdio | Terminal.
const AppLayer = Layer.mergeAll(NodeServices.layer /*, MyServiceLive */);
export function createRuntime() {
return ManagedRuntime.make(AppLayer);
}
export type ExtRuntime = ReturnType<typeof createRuntime>;
/** Run an effect from an async handler: value on success, thrown Error otherwise. */
export async function runTool<A, E>(
runtime: ExtRuntime,
effect: Effect.Effect<A, E>,
options: { signal?: AbortSignal; interruptMessage?: string } = {},
) {
const exit = await runtime.runPromiseExit(
effect,
options.signal ? { signal: options.signal } : undefined,
);
if (Exit.isSuccess(exit)) return exit.value;
if (Cause.hasInterruptsOnly(exit.cause)) {
throw new Error(options.interruptMessage ?? "Operation was aborted.");
}
const [first] = Cause.prettyErrors(exit.cause);
throw new Error(first?.message ?? Cause.pretty(exit.cause));
}
```
Wire it into the extension lifecycle (lazy build, dispose on shutdown):
```ts
// index.ts
export default function (pi: ExtensionAPI) {
let runtime: ExtRuntime | undefined;
const getRuntime = () => (runtime ??= createRuntime());
pi.registerTool({
name: "my_tool",
parameters: Type.Object({/* typebox */}),
async execute(_id, params, signal) {
return await runTool(getRuntime(), myEffect(params), {
signal,
interruptMessage: "Cancelled.",
});
},
});
pi.on("session_shutdown", async () => {
const closing = runtime;
runtime = undefined;
await closing?.dispose(); // runs all finalizers: kills scoped child processes, etc.
});
}
```
Key facts:
- `runPromiseExit(effect, { signal })` — passing the tool's `AbortSignal` makes cancelling
the tool interrupt the fiber; scoped resources (child processes) are torn down.
- `runtime.runFork(effect)` for fire-and-forget background work (polling loops, §5).
- `dispose()` is the single teardown hook — put resource cleanup in Effect finalizers, not
in ad-hoc `session_shutdown` code.
- If an extension has **no** services (pure `Effect.tryPromise` calls, no fs/process), you
can skip `NodeServices.layer` and even skip `ManagedRuntime` — `Effect.runPromiseExit(effect, { signal })`
works standalone. Add the runtime only when you have a `Layer` to share.
---
## 3. Services, layers, errors (house style)
Mirror the subagents code. Class-style `Context.Service`, `Layer.effect`/`Layer.sync`,
`Data.TaggedError` for domain errors.
```ts
import { Context, Data, Effect, Layer } from "effect";
// Domain error — yieldable, tag-narrowable with Effect.catchTag:
export class GitError extends Data.TaggedError("GitError")<{
readonly message: string;
}> {}
// Service: the class value is the key AND the type.
export interface GitShape {
readonly status: Effect.Effect<string, GitError>;
}
export class Git extends Context.Service<Git, GitShape>()("gitinfo/Git") {}
// Layer building it (Layer.effect can use scoped resources; Layer.sync for pure):
export const GitLive: Layer.Layer<Git, never, ChildProcessSpawner> =
Layer.effect(
Git,
Effect.gen(function* () {
const spawner = yield* ChildProcessSpawner; // dependency, provided by NodeServices.layer
return Git.of({
status: spawner
.string(ChildProcess.make("git", ["status", "--porcelain"]))
.pipe(
Effect.mapError(
(cause) => new GitError({ message: String(cause) }),
),
),
});
}),
);
```
Provide dependencies with `Layer.provide`, compose with `Layer.mergeAll` (see §2 `AppLayer`).
Full rename table (`Effect.fork`→`forkChild`, `catchAll`→`catch`, `Either`→`Result`, …) and
the child-process / concurrency APIs live in `effect-v4-notes.md` — don't re-derive them.
---
## 4. Recipe: wrapping a Promise SDK (firecrawl-search)
The Firecrawl client is Promise-based. Wrap each call in `Effect.tryPromise` with a typed
error; the callback receives an `AbortSignal` tied to fiber interruption — forward it to any
SDK that accepts one so tool cancellation propagates.
```ts
import { Data, Effect } from "effect";
class FirecrawlError extends Data.TaggedError("FirecrawlError")<{
readonly cause: unknown;
}> {}
const search = (client: Firecrawl, query: string) =>
Effect.tryPromise({
try: (signal) => client.search(query, {/* …, signal? if supported */}),
catch: (cause) => new FirecrawlError({ cause }),
});
```
There's no `Layer` here unless you want the client as a service. For a single tool this is
fine to call directly through `runTool` (or even plain `Effect.runPromiseExit`, §2). Keep the
existing `.env`/API-key reading and result truncation as plain TS — no reason to Effect-ify
string trimming.
---
## 5. Recipe: child processes + timeout + polling (git-info, copy-all)
`git-info` shells out to `git`/`gh` with per-command timeouts and polls on an interval;
`copy-all` pipes text into `pbcopy`. Two viable levels — pick the lightest that fits.
**Simple, one-shot, small:** if all you do is "run a command, capture stdout, with a
timeout," the Effect win is `Effect.timeout` + interruption killing the child. Use the
spawner service:
```ts
import { Effect } from "effect";
import { ChildProcess } from "effect/unstable/process";
import { ChildProcessSpawner } from "effect/unstable/process/ChildProcessSpawner";
const gitStatus = Effect.gen(function* () {
const spawner = yield* ChildProcessSpawner;
return yield* spawner
.string(ChildProcess.make("git", ["status", "--porcelain"], { cwd }))
.pipe(Effect.timeout("3 seconds")); // fails with Cause.TimeoutError (tag "TimeoutError")
});
```
Note the import gotcha: `ChildProcessSpawner` the **class** comes from the
`.../ChildProcessSpawner` submodule; the `effect/unstable/process` index gives you the
namespace. Provide `NodeServices.layer` (or `ChildProcessSpawner.layer`). Full command
builder / streaming / kill semantics are in `effect-v4-notes.md §6`.
**Polling on an interval:** replace `setInterval` with a forked, scheduled effect so it
cancels cleanly on dispose:
```ts
import { Effect, Schedule } from "effect";
const pollLoop = refresh.pipe(
Effect.catchCause(() => Effect.void), // don't let one failure kill the loop
Effect.repeat(Schedule.spaced("3 seconds")),
);
const fiber = runtime.runFork(pollLoop); // interrupted by runtime.dispose()
```
**When to stay plain:** `copy-all` spawning `pbcopy` is a trivial one-shot with no
cancellation need — the existing `node:child_process` + Promise wrapper is honestly fine.
Migrate it only for consistency; if you do, `Effect.callback` around `child.once("exit", …)`
(see notes §4) is the minimal wrapper. Don't add a service/layer for a clipboard write.
---
## 6. Recipe: UI popups & rendering (ask-user, ui-customization, model-info)
These are the "leave it mostly plain" cases.
- `ask-user` is a TUI popup that resolves a Promise when the user picks. That Promise already
models the one async thing. Effect adds nothing; if you want uniformity, wrap the final
await in `Effect.tryPromise` at the boundary and stop there. Do **not** build a service.
- `ui-customization` and `model-info` are renderers / event bookkeepers driven by
`pi.on(...)` and cross-extension channels (`shared/dashboard-state.ts`). Channels are a
pi-native mechanism — keep them. State counting and formatting stay synchronous TS.
- If `model-info` has a periodic "live update" tick, the §5 polling pattern applies; but a
plain timer here is also acceptable since there's no resource to tear down.
The migration bar for these: adopt the toolchain (§1) so they typecheck under TS7 + the
Effect LS, and only touch runtime code that has a real async/resource concern.
---
## 7. Verify — scoped to the one extension
Run everything from inside the extension directory so you never trigger a root/global build
or format:
```bash
cd extensions/<ext>
npm install # first time / after dep or script edits — runs `prepare` (effect-tsgo patch)
npm run check # tsc --noEmit -p . ← ground-truth green/red
npm run test # only if the extension defines tests; keep them minimal
```
`extensions/subagents` is the known-green reference: `npm run check` there exits 0 against
the pinned versions. If a migrated extension fails `check` with `Effect.fork`/`ServiceMap`/
`Either` errors, it's using stale v3/early-beta APIs — consult the rename table in
`effect-v4-notes.md`.
---
## 8. Don'ts (keep it lean)
1. **Don't float versions.** Pin `effect` and `@effect/platform-node` to the exact same
`4.0.0-beta.98`; `unstable/*` can break between betas.
2. **Don't Effect-ify pure/UI code.** No service or layer for a clipboard write, a string
truncation, or a popup. §0 is the test: is there a typed-error / cancellation / resource /
retry concern? If not, leave it.
3. **Don't guess APIs.** If it's not in this guide or `effect-v4-notes.md` and you haven't
grepped it in `node_modules/effect/dist/*.d.ts`, don't write it. Watch for hallucinated
`ServiceMap` (reverted to `Context`), `Effect.fork` (→ `forkChild`), `Effect.either`
(→ `Effect.result`).
4. **Don't hand-roll teardown.** Put cleanup in Effect finalizers and let `runtime.dispose()`
in `session_shutdown` run them.
5. **Don't over-test.** A `check` that passes plus one focused runtime test (where behavior
is non-obvious) beats a wall of defensive unit tests.
6. **Don't run root scripts.** No repo-root `tsc`, `prettier`, or `npm run format`; stay
inside `extensions/<ext>`.
</content>