Compare commits
3
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
a2b81cd8a5 | ||
|
|
771f1918b5 | ||
|
|
6a3cb06c6c |
@@ -128,11 +128,11 @@ Still open and likely worth migrating:
|
|||||||
|
|
||||||
- [ ] `Plugin`
|
- [ ] `Plugin`
|
||||||
- [ ] `ToolRegistry`
|
- [ ] `ToolRegistry`
|
||||||
- [ ] `Pty`
|
- [x] `Pty`
|
||||||
- [ ] `Worktree`
|
- [ ] `Worktree`
|
||||||
- [ ] `Installation`
|
- [ ] `Installation`
|
||||||
- [ ] `Bus`
|
- [ ] `Bus`
|
||||||
- [x] `Command`
|
- [ ] `Command`
|
||||||
- [ ] `Config`
|
- [ ] `Config`
|
||||||
- [ ] `Session`
|
- [ ] `Session`
|
||||||
- [ ] `SessionProcessor`
|
- [ ] `SessionProcessor`
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
import z from "zod"
|
import z from "zod"
|
||||||
import type { ZodObject, ZodRawShape } from "zod"
|
import type { ZodType } from "zod"
|
||||||
import { Log } from "../util/log"
|
import { Log } from "../util/log"
|
||||||
|
|
||||||
export namespace BusEvent {
|
export namespace BusEvent {
|
||||||
@@ -9,7 +9,7 @@ export namespace BusEvent {
|
|||||||
|
|
||||||
const registry = new Map<string, Definition>()
|
const registry = new Map<string, Definition>()
|
||||||
|
|
||||||
export function define<Type extends string, Properties extends ZodObject<ZodRawShape>>(type: Type, properties: Properties) {
|
export function define<Type extends string, Properties extends ZodType>(type: Type, properties: Properties) {
|
||||||
const result = {
|
const result = {
|
||||||
type,
|
type,
|
||||||
properties,
|
properties,
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ export const GlobalBus = new EventEmitter<{
|
|||||||
event: [
|
event: [
|
||||||
{
|
{
|
||||||
directory?: string
|
directory?: string
|
||||||
payload: { type: string; properties: Record<string, unknown> }
|
payload: any
|
||||||
},
|
},
|
||||||
]
|
]
|
||||||
}>()
|
}>()
|
||||||
|
|||||||
@@ -1,18 +1,15 @@
|
|||||||
import { BusEvent } from "@/bus/bus-event"
|
import { BusEvent } from "@/bus/bus-event"
|
||||||
import { InstanceContext } from "@/effect/instance-context"
|
|
||||||
import { runPromiseInstance } from "@/effect/runtime"
|
|
||||||
import { SessionID, MessageID } from "@/session/schema"
|
import { SessionID, MessageID } from "@/session/schema"
|
||||||
import { Effect, Fiber, Layer, ServiceMap } from "effect"
|
|
||||||
import z from "zod"
|
import z from "zod"
|
||||||
import { Config } from "../config/config"
|
import { Config } from "../config/config"
|
||||||
|
import { Instance } from "../project/instance"
|
||||||
|
import { Identifier } from "../id/id"
|
||||||
import PROMPT_INITIALIZE from "./template/initialize.txt"
|
import PROMPT_INITIALIZE from "./template/initialize.txt"
|
||||||
import PROMPT_REVIEW from "./template/review.txt"
|
import PROMPT_REVIEW from "./template/review.txt"
|
||||||
import { MCP } from "../mcp"
|
import { MCP } from "../mcp"
|
||||||
import { Skill } from "../skill"
|
import { Skill } from "../skill"
|
||||||
import { Log } from "../util/log"
|
|
||||||
|
|
||||||
export namespace Command {
|
export namespace Command {
|
||||||
const log = Log.create({ service: "command" })
|
|
||||||
export const Event = {
|
export const Event = {
|
||||||
Executed: BusEvent.define(
|
Executed: BusEvent.define(
|
||||||
"command.executed",
|
"command.executed",
|
||||||
@@ -60,126 +57,95 @@ export namespace Command {
|
|||||||
REVIEW: "review",
|
REVIEW: "review",
|
||||||
} as const
|
} as const
|
||||||
|
|
||||||
export interface Interface {
|
const state = Instance.state(async () => {
|
||||||
readonly get: (name: string) => Effect.Effect<Info | undefined>
|
const cfg = await Config.get()
|
||||||
readonly list: () => Effect.Effect<Info[]>
|
|
||||||
}
|
|
||||||
|
|
||||||
export class Service extends ServiceMap.Service<Service, Interface>()("@opencode/Command") {}
|
const result: Record<string, Info> = {
|
||||||
|
[Default.INIT]: {
|
||||||
|
name: Default.INIT,
|
||||||
|
description: "create/update AGENTS.md",
|
||||||
|
source: "command",
|
||||||
|
get template() {
|
||||||
|
return PROMPT_INITIALIZE.replace("${path}", Instance.worktree)
|
||||||
|
},
|
||||||
|
hints: hints(PROMPT_INITIALIZE),
|
||||||
|
},
|
||||||
|
[Default.REVIEW]: {
|
||||||
|
name: Default.REVIEW,
|
||||||
|
description: "review changes [commit|branch|pr], defaults to uncommitted",
|
||||||
|
source: "command",
|
||||||
|
get template() {
|
||||||
|
return PROMPT_REVIEW.replace("${path}", Instance.worktree)
|
||||||
|
},
|
||||||
|
subtask: true,
|
||||||
|
hints: hints(PROMPT_REVIEW),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
export const layer = Layer.effect(
|
for (const [name, command] of Object.entries(cfg.command ?? {})) {
|
||||||
Service,
|
result[name] = {
|
||||||
Effect.gen(function* () {
|
name,
|
||||||
const instance = yield* InstanceContext
|
agent: command.agent,
|
||||||
|
model: command.model,
|
||||||
|
description: command.description,
|
||||||
|
source: "command",
|
||||||
|
get template() {
|
||||||
|
return command.template
|
||||||
|
},
|
||||||
|
subtask: command.subtask,
|
||||||
|
hints: hints(command.template),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for (const [name, prompt] of Object.entries(await MCP.prompts())) {
|
||||||
|
result[name] = {
|
||||||
|
name,
|
||||||
|
source: "mcp",
|
||||||
|
description: prompt.description,
|
||||||
|
get template() {
|
||||||
|
// since a getter can't be async we need to manually return a promise here
|
||||||
|
return new Promise<string>(async (resolve, reject) => {
|
||||||
|
const template = await MCP.getPrompt(
|
||||||
|
prompt.client,
|
||||||
|
prompt.name,
|
||||||
|
prompt.arguments
|
||||||
|
? // substitute each argument with $1, $2, etc.
|
||||||
|
Object.fromEntries(prompt.arguments?.map((argument, i) => [argument.name, `$${i + 1}`]))
|
||||||
|
: {},
|
||||||
|
).catch(reject)
|
||||||
|
resolve(
|
||||||
|
template?.messages
|
||||||
|
.map((message) => (message.content.type === "text" ? message.content.text : ""))
|
||||||
|
.join("\n") || "",
|
||||||
|
)
|
||||||
|
})
|
||||||
|
},
|
||||||
|
hints: prompt.arguments?.map((_, i) => `$${i + 1}`) ?? [],
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
const commands: Record<string, Info> = {}
|
// Add skills as invokable commands
|
||||||
|
for (const skill of await Skill.all()) {
|
||||||
|
// Skip if a command with this name already exists
|
||||||
|
if (result[skill.name]) continue
|
||||||
|
result[skill.name] = {
|
||||||
|
name: skill.name,
|
||||||
|
description: skill.description,
|
||||||
|
source: "skill",
|
||||||
|
get template() {
|
||||||
|
return skill.content
|
||||||
|
},
|
||||||
|
hints: [],
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
const load = Effect.fn("Command.load")(function* () {
|
return result
|
||||||
yield* Effect.promise(async () => {
|
})
|
||||||
const cfg = await Config.get()
|
|
||||||
|
|
||||||
commands[Default.INIT] = {
|
|
||||||
name: Default.INIT,
|
|
||||||
description: "create/update AGENTS.md",
|
|
||||||
source: "command",
|
|
||||||
get template() {
|
|
||||||
return PROMPT_INITIALIZE.replace("${path}", instance.worktree)
|
|
||||||
},
|
|
||||||
hints: hints(PROMPT_INITIALIZE),
|
|
||||||
}
|
|
||||||
commands[Default.REVIEW] = {
|
|
||||||
name: Default.REVIEW,
|
|
||||||
description: "review changes [commit|branch|pr], defaults to uncommitted",
|
|
||||||
source: "command",
|
|
||||||
get template() {
|
|
||||||
return PROMPT_REVIEW.replace("${path}", instance.worktree)
|
|
||||||
},
|
|
||||||
subtask: true,
|
|
||||||
hints: hints(PROMPT_REVIEW),
|
|
||||||
}
|
|
||||||
|
|
||||||
for (const [name, command] of Object.entries(cfg.command ?? {})) {
|
|
||||||
commands[name] = {
|
|
||||||
name,
|
|
||||||
agent: command.agent,
|
|
||||||
model: command.model,
|
|
||||||
description: command.description,
|
|
||||||
source: "command",
|
|
||||||
get template() {
|
|
||||||
return command.template
|
|
||||||
},
|
|
||||||
subtask: command.subtask,
|
|
||||||
hints: hints(command.template),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for (const [name, prompt] of Object.entries(await MCP.prompts())) {
|
|
||||||
commands[name] = {
|
|
||||||
name,
|
|
||||||
source: "mcp",
|
|
||||||
description: prompt.description,
|
|
||||||
get template() {
|
|
||||||
// since a getter can't be async we need to manually return a promise here
|
|
||||||
return new Promise<string>(async (resolve, reject) => {
|
|
||||||
const template = await MCP.getPrompt(
|
|
||||||
prompt.client,
|
|
||||||
prompt.name,
|
|
||||||
prompt.arguments
|
|
||||||
? // substitute each argument with $1, $2, etc.
|
|
||||||
Object.fromEntries(prompt.arguments?.map((argument, i) => [argument.name, `$${i + 1}`]))
|
|
||||||
: {},
|
|
||||||
).catch(reject)
|
|
||||||
resolve(
|
|
||||||
template?.messages
|
|
||||||
.map((message) => (message.content.type === "text" ? message.content.text : ""))
|
|
||||||
.join("\n") || "",
|
|
||||||
)
|
|
||||||
})
|
|
||||||
},
|
|
||||||
hints: prompt.arguments?.map((_, i) => `$${i + 1}`) ?? [],
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Add skills as invokable commands
|
|
||||||
for (const skill of await Skill.all()) {
|
|
||||||
// Skip if a command with this name already exists
|
|
||||||
if (commands[skill.name]) continue
|
|
||||||
commands[skill.name] = {
|
|
||||||
name: skill.name,
|
|
||||||
description: skill.description,
|
|
||||||
source: "skill",
|
|
||||||
get template() {
|
|
||||||
return skill.content
|
|
||||||
},
|
|
||||||
hints: [],
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
})
|
|
||||||
|
|
||||||
const loadFiber = yield* load().pipe(
|
|
||||||
Effect.catchCause((cause) => Effect.sync(() => log.error("init failed", { cause }))),
|
|
||||||
Effect.forkScoped,
|
|
||||||
)
|
|
||||||
|
|
||||||
const get = Effect.fn("Command.get")(function* (name: string) {
|
|
||||||
yield* Fiber.join(loadFiber)
|
|
||||||
return commands[name]
|
|
||||||
})
|
|
||||||
|
|
||||||
const list = Effect.fn("Command.list")(function* () {
|
|
||||||
yield* Fiber.join(loadFiber)
|
|
||||||
return Object.values(commands)
|
|
||||||
})
|
|
||||||
|
|
||||||
return Service.of({ get, list })
|
|
||||||
}),
|
|
||||||
)
|
|
||||||
|
|
||||||
export async function get(name: string) {
|
export async function get(name: string) {
|
||||||
return runPromiseInstance(Service.use((svc) => svc.get(name)))
|
return state().then((x) => x[name])
|
||||||
}
|
}
|
||||||
|
|
||||||
export async function list() {
|
export async function list() {
|
||||||
return runPromiseInstance(Service.use((svc) => svc.list()))
|
return state().then((x) => Object.values(x))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -124,7 +124,7 @@ export namespace Workspace {
|
|||||||
await parseSSE(res.body, stop, (event) => {
|
await parseSSE(res.body, stop, (event) => {
|
||||||
GlobalBus.emit("event", {
|
GlobalBus.emit("event", {
|
||||||
directory: space.id,
|
directory: space.id,
|
||||||
payload: event as { type: string; properties: Record<string, unknown> },
|
payload: event,
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
// Wait 250ms and retry if SSE connection fails
|
// Wait 250ms and retry if SSE connection fails
|
||||||
|
|||||||
@@ -1,10 +1,10 @@
|
|||||||
import { Effect, Layer, LayerMap, ServiceMap } from "effect"
|
import { Effect, Layer, LayerMap, ServiceMap } from "effect"
|
||||||
import { Command } from "@/command"
|
|
||||||
import { File } from "@/file/service"
|
import { File } from "@/file/service"
|
||||||
import { FileTime } from "@/file/time-service"
|
import { FileTime } from "@/file/time-service"
|
||||||
import { FileWatcher } from "@/file/watcher"
|
import { FileWatcher } from "@/file/watcher"
|
||||||
import { Format } from "@/format/service"
|
import { Format } from "@/format/service"
|
||||||
import { Permission } from "@/permission/service"
|
import { Permission } from "@/permission/service"
|
||||||
|
import { Pty } from "@/pty"
|
||||||
import { Instance } from "@/project/instance"
|
import { Instance } from "@/project/instance"
|
||||||
import { Vcs } from "@/project/vcs"
|
import { Vcs } from "@/project/vcs"
|
||||||
import { ProviderAuth } from "@/provider/auth-service"
|
import { ProviderAuth } from "@/provider/auth-service"
|
||||||
@@ -17,7 +17,6 @@ import { registerDisposer } from "./instance-registry"
|
|||||||
export { InstanceContext } from "./instance-context"
|
export { InstanceContext } from "./instance-context"
|
||||||
|
|
||||||
export type InstanceServices =
|
export type InstanceServices =
|
||||||
| Command.Service
|
|
||||||
| Question.Service
|
| Question.Service
|
||||||
| Permission.Service
|
| Permission.Service
|
||||||
| ProviderAuth.Service
|
| ProviderAuth.Service
|
||||||
@@ -26,6 +25,7 @@ export type InstanceServices =
|
|||||||
| FileTime.Service
|
| FileTime.Service
|
||||||
| Format.Service
|
| Format.Service
|
||||||
| File.Service
|
| File.Service
|
||||||
|
| Pty.Service
|
||||||
| Skill.Service
|
| Skill.Service
|
||||||
| Snapshot.Service
|
| Snapshot.Service
|
||||||
|
|
||||||
@@ -38,7 +38,6 @@ export type InstanceServices =
|
|||||||
function lookup(_key: string) {
|
function lookup(_key: string) {
|
||||||
const ctx = Layer.sync(InstanceContext, () => InstanceContext.of(Instance.current))
|
const ctx = Layer.sync(InstanceContext, () => InstanceContext.of(Instance.current))
|
||||||
return Layer.mergeAll(
|
return Layer.mergeAll(
|
||||||
Command.layer,
|
|
||||||
Question.layer,
|
Question.layer,
|
||||||
Permission.layer,
|
Permission.layer,
|
||||||
ProviderAuth.defaultLayer,
|
ProviderAuth.defaultLayer,
|
||||||
@@ -47,6 +46,7 @@ function lookup(_key: string) {
|
|||||||
FileTime.layer,
|
FileTime.layer,
|
||||||
Format.layer,
|
Format.layer,
|
||||||
File.layer,
|
File.layer,
|
||||||
|
Pty.layer,
|
||||||
Skill.defaultLayer,
|
Skill.defaultLayer,
|
||||||
Snapshot.defaultLayer,
|
Snapshot.defaultLayer,
|
||||||
).pipe(Layer.provide(ctx))
|
).pipe(Layer.provide(ctx))
|
||||||
|
|||||||
@@ -20,6 +20,10 @@ export function runPromiseInstance<A, E>(effect: Effect.Effect<A, E, InstanceSer
|
|||||||
return runtime.runPromise(effect.pipe(Effect.provide(Instances.get(Instance.directory))))
|
return runtime.runPromise(effect.pipe(Effect.provide(Instances.get(Instance.directory))))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
export function runSyncInstance<A, E>(effect: Effect.Effect<A, E, InstanceServices>) {
|
||||||
|
return runtime.runSync(effect.pipe(Effect.provide(Instances.get(Instance.directory))))
|
||||||
|
}
|
||||||
|
|
||||||
export function disposeRuntime() {
|
export function disposeRuntime() {
|
||||||
return runtime.dispose()
|
return runtime.dispose()
|
||||||
}
|
}
|
||||||
|
|||||||
+267
-203
@@ -1,13 +1,12 @@
|
|||||||
import { BusEvent } from "@/bus/bus-event"
|
import { BusEvent } from "@/bus/bus-event"
|
||||||
import { Bus } from "@/bus"
|
import { Bus } from "@/bus"
|
||||||
|
import { InstanceContext } from "@/effect/instance-context"
|
||||||
import { type IPty } from "bun-pty"
|
import { type IPty } from "bun-pty"
|
||||||
import z from "zod"
|
import z from "zod"
|
||||||
import { Log } from "../util/log"
|
import { Log } from "../util/log"
|
||||||
import { Instance } from "../project/instance"
|
|
||||||
import { lazy } from "@opencode-ai/util/lazy"
|
import { lazy } from "@opencode-ai/util/lazy"
|
||||||
import { Shell } from "@/shell/shell"
|
|
||||||
import { Plugin } from "@/plugin"
|
|
||||||
import { PtyID } from "./schema"
|
import { PtyID } from "./schema"
|
||||||
|
import { Effect, Layer, ServiceMap } from "effect"
|
||||||
|
|
||||||
export namespace Pty {
|
export namespace Pty {
|
||||||
const log = Log.create({ service: "pty" })
|
const log = Log.create({ service: "pty" })
|
||||||
@@ -90,232 +89,297 @@ export namespace Pty {
|
|||||||
subscribers: Map<unknown, Socket>
|
subscribers: Map<unknown, Socket>
|
||||||
}
|
}
|
||||||
|
|
||||||
const state = Instance.state(
|
export interface Interface {
|
||||||
() => new Map<PtyID, ActiveSession>(),
|
readonly list: () => Effect.Effect<Info[]>
|
||||||
async (sessions) => {
|
readonly get: (id: PtyID) => Effect.Effect<Info | undefined>
|
||||||
for (const session of sessions.values()) {
|
readonly create: (input: CreateInput) => Effect.Effect<Info>
|
||||||
|
readonly update: (id: PtyID, input: UpdateInput) => Effect.Effect<Info | undefined>
|
||||||
|
readonly remove: (id: PtyID) => Effect.Effect<void>
|
||||||
|
readonly resize: (id: PtyID, cols: number, rows: number) => Effect.Effect<void>
|
||||||
|
readonly write: (id: PtyID, data: string) => Effect.Effect<void>
|
||||||
|
readonly connect: (
|
||||||
|
id: PtyID,
|
||||||
|
ws: Socket,
|
||||||
|
cursor?: number,
|
||||||
|
) => Effect.Effect<{ onMessage: (message: string | ArrayBuffer) => void; onClose: () => void } | undefined>
|
||||||
|
}
|
||||||
|
|
||||||
|
export class Service extends ServiceMap.Service<Service, Interface>()("@opencode/Pty") {}
|
||||||
|
|
||||||
|
export const layer = Layer.effect(
|
||||||
|
Service,
|
||||||
|
Effect.gen(function* () {
|
||||||
|
const instance = yield* InstanceContext
|
||||||
|
const sessions = new Map<PtyID, ActiveSession>()
|
||||||
|
|
||||||
|
yield* Effect.addFinalizer(() =>
|
||||||
|
Effect.sync(() => {
|
||||||
|
for (const session of sessions.values()) {
|
||||||
|
try {
|
||||||
|
session.process.kill()
|
||||||
|
} catch {}
|
||||||
|
for (const [key, ws] of session.subscribers.entries()) {
|
||||||
|
try {
|
||||||
|
if (ws.data === key) ws.close()
|
||||||
|
} catch {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
sessions.clear()
|
||||||
|
}),
|
||||||
|
)
|
||||||
|
|
||||||
|
const removeSession = (id: PtyID) => {
|
||||||
|
const session = sessions.get(id)
|
||||||
|
if (!session) return
|
||||||
|
sessions.delete(id)
|
||||||
|
log.info("removing session", { id })
|
||||||
try {
|
try {
|
||||||
session.process.kill()
|
session.process.kill()
|
||||||
} catch {}
|
} catch {}
|
||||||
for (const [key, ws] of session.subscribers.entries()) {
|
for (const [key, ws] of session.subscribers.entries()) {
|
||||||
try {
|
try {
|
||||||
if (ws.data === key) ws.close()
|
if (ws.data === key) ws.close()
|
||||||
|
} catch {}
|
||||||
|
}
|
||||||
|
session.subscribers.clear()
|
||||||
|
Bus.publish(Event.Deleted, { id: session.info.id })
|
||||||
|
}
|
||||||
|
|
||||||
|
const list = Effect.fn("Pty.list")(function* () {
|
||||||
|
return Array.from(sessions.values()).map((s) => s.info)
|
||||||
|
})
|
||||||
|
|
||||||
|
const get = Effect.fn("Pty.get")(function* (id: PtyID) {
|
||||||
|
return sessions.get(id)?.info
|
||||||
|
})
|
||||||
|
|
||||||
|
const create = Effect.fn("Pty.create")(function* (input: CreateInput) {
|
||||||
|
return yield* Effect.promise(async () => {
|
||||||
|
const [{ Shell }, { Plugin }] = await Promise.all([import("@/shell/shell"), import("@/plugin")])
|
||||||
|
const id = PtyID.ascending()
|
||||||
|
const command = input.command || Shell.preferred()
|
||||||
|
const args = input.args || []
|
||||||
|
if (command.endsWith("sh")) {
|
||||||
|
args.push("-l")
|
||||||
|
}
|
||||||
|
|
||||||
|
const cwd = input.cwd || instance.directory
|
||||||
|
const shellEnv = await Plugin.trigger("shell.env", { cwd }, { env: {} })
|
||||||
|
const env = {
|
||||||
|
...process.env,
|
||||||
|
...input.env,
|
||||||
|
...shellEnv.env,
|
||||||
|
TERM: "xterm-256color",
|
||||||
|
OPENCODE_TERMINAL: "1",
|
||||||
|
} as Record<string, string>
|
||||||
|
|
||||||
|
if (process.platform === "win32") {
|
||||||
|
env.LC_ALL = "C.UTF-8"
|
||||||
|
env.LC_CTYPE = "C.UTF-8"
|
||||||
|
env.LANG = "C.UTF-8"
|
||||||
|
}
|
||||||
|
log.info("creating session", { id, cmd: command, args, cwd })
|
||||||
|
|
||||||
|
const spawn = await pty()
|
||||||
|
const ptyProcess = spawn(command, args, {
|
||||||
|
name: "xterm-256color",
|
||||||
|
cwd,
|
||||||
|
env,
|
||||||
|
})
|
||||||
|
|
||||||
|
const info = {
|
||||||
|
id,
|
||||||
|
title: input.title || `Terminal ${id.slice(-4)}`,
|
||||||
|
command,
|
||||||
|
args,
|
||||||
|
cwd,
|
||||||
|
status: "running",
|
||||||
|
pid: ptyProcess.pid,
|
||||||
|
} as const
|
||||||
|
const session: ActiveSession = {
|
||||||
|
info,
|
||||||
|
process: ptyProcess,
|
||||||
|
buffer: "",
|
||||||
|
bufferCursor: 0,
|
||||||
|
cursor: 0,
|
||||||
|
subscribers: new Map(),
|
||||||
|
}
|
||||||
|
sessions.set(id, session)
|
||||||
|
ptyProcess.onData((chunk) => {
|
||||||
|
session.cursor += chunk.length
|
||||||
|
|
||||||
|
for (const [key, ws] of session.subscribers.entries()) {
|
||||||
|
if (ws.readyState !== 1) {
|
||||||
|
session.subscribers.delete(key)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if (ws.data !== key) {
|
||||||
|
session.subscribers.delete(key)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
ws.send(chunk)
|
||||||
|
} catch {
|
||||||
|
session.subscribers.delete(key)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
session.buffer += chunk
|
||||||
|
if (session.buffer.length <= BUFFER_LIMIT) return
|
||||||
|
const excess = session.buffer.length - BUFFER_LIMIT
|
||||||
|
session.buffer = session.buffer.slice(excess)
|
||||||
|
session.bufferCursor += excess
|
||||||
|
})
|
||||||
|
ptyProcess.onExit(({ exitCode }) => {
|
||||||
|
if (session.info.status === "exited") return
|
||||||
|
log.info("session exited", { id, exitCode })
|
||||||
|
session.info.status = "exited"
|
||||||
|
Bus.publish(Event.Exited, { id, exitCode })
|
||||||
|
removeSession(id)
|
||||||
|
})
|
||||||
|
Bus.publish(Event.Created, { info })
|
||||||
|
return info
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
const update = Effect.fn("Pty.update")(function* (id: PtyID, input: UpdateInput) {
|
||||||
|
const session = sessions.get(id)
|
||||||
|
if (!session) return
|
||||||
|
if (input.title) {
|
||||||
|
session.info.title = input.title
|
||||||
|
}
|
||||||
|
if (input.size) {
|
||||||
|
session.process.resize(input.size.cols, input.size.rows)
|
||||||
|
}
|
||||||
|
Bus.publish(Event.Updated, { info: session.info })
|
||||||
|
return session.info
|
||||||
|
})
|
||||||
|
|
||||||
|
const remove = Effect.fn("Pty.remove")(function* (id: PtyID) {
|
||||||
|
removeSession(id)
|
||||||
|
})
|
||||||
|
|
||||||
|
const resize = Effect.fn("Pty.resize")(function* (id: PtyID, cols: number, rows: number) {
|
||||||
|
const session = sessions.get(id)
|
||||||
|
if (session && session.info.status === "running") {
|
||||||
|
session.process.resize(cols, rows)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
const write = Effect.fn("Pty.write")(function* (id: PtyID, data: string) {
|
||||||
|
const session = sessions.get(id)
|
||||||
|
if (session && session.info.status === "running") {
|
||||||
|
session.process.write(data)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
const connect = Effect.fn("Pty.connect")(function* (id: PtyID, ws: Socket, cursor?: number) {
|
||||||
|
const session = sessions.get(id)
|
||||||
|
if (!session) {
|
||||||
|
ws.close()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
log.info("client connected to session", { id })
|
||||||
|
|
||||||
|
const connectionKey = ws.data && typeof ws.data === "object" ? ws.data : ws
|
||||||
|
session.subscribers.delete(connectionKey)
|
||||||
|
session.subscribers.set(connectionKey, ws)
|
||||||
|
|
||||||
|
const cleanup = () => {
|
||||||
|
session.subscribers.delete(connectionKey)
|
||||||
|
}
|
||||||
|
|
||||||
|
const start = session.bufferCursor
|
||||||
|
const end = session.cursor
|
||||||
|
|
||||||
|
const from =
|
||||||
|
cursor === -1 ? end : typeof cursor === "number" && Number.isSafeInteger(cursor) ? Math.max(0, cursor) : 0
|
||||||
|
|
||||||
|
const data = (() => {
|
||||||
|
if (!session.buffer) return ""
|
||||||
|
if (from >= end) return ""
|
||||||
|
const offset = Math.max(0, from - start)
|
||||||
|
if (offset >= session.buffer.length) return ""
|
||||||
|
return session.buffer.slice(offset)
|
||||||
|
})()
|
||||||
|
|
||||||
|
if (data) {
|
||||||
|
try {
|
||||||
|
for (let i = 0; i < data.length; i += BUFFER_CHUNK) {
|
||||||
|
ws.send(data.slice(i, i + BUFFER_CHUNK))
|
||||||
|
}
|
||||||
} catch {
|
} catch {
|
||||||
// ignore
|
cleanup()
|
||||||
|
ws.close()
|
||||||
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
sessions.clear()
|
try {
|
||||||
},
|
ws.send(meta(end))
|
||||||
|
} catch {
|
||||||
|
cleanup()
|
||||||
|
ws.close()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return {
|
||||||
|
onMessage: (message: string | ArrayBuffer) => {
|
||||||
|
session.process.write(String(message))
|
||||||
|
},
|
||||||
|
onClose: () => {
|
||||||
|
log.info("client disconnected from session", { id })
|
||||||
|
cleanup()
|
||||||
|
},
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
return Service.of({ list, get, create, update, remove, resize, write, connect })
|
||||||
|
}),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
function runtime() {
|
||||||
|
return require("@/effect/runtime") as typeof import("@/effect/runtime")
|
||||||
|
}
|
||||||
|
|
||||||
|
function run<A, E>(effect: Effect.Effect<A, E, Service>) {
|
||||||
|
return runtime().runPromiseInstance(effect)
|
||||||
|
}
|
||||||
|
|
||||||
|
function runSync<A, E>(effect: Effect.Effect<A, E, Service>) {
|
||||||
|
return runtime().runSyncInstance(effect)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sync facades
|
||||||
export function list() {
|
export function list() {
|
||||||
return Array.from(state().values()).map((s) => s.info)
|
return runSync(Service.use((svc) => svc.list()))
|
||||||
}
|
}
|
||||||
|
|
||||||
export function get(id: PtyID) {
|
export function get(id: PtyID) {
|
||||||
return state().get(id)?.info
|
return runSync(Service.use((svc) => svc.get(id)))
|
||||||
}
|
|
||||||
|
|
||||||
export async function create(input: CreateInput) {
|
|
||||||
const id = PtyID.ascending()
|
|
||||||
const command = input.command || Shell.preferred()
|
|
||||||
const args = input.args || []
|
|
||||||
if (command.endsWith("sh")) {
|
|
||||||
args.push("-l")
|
|
||||||
}
|
|
||||||
|
|
||||||
const cwd = input.cwd || Instance.directory
|
|
||||||
const shellEnv = await Plugin.trigger("shell.env", { cwd }, { env: {} })
|
|
||||||
const env = {
|
|
||||||
...process.env,
|
|
||||||
...input.env,
|
|
||||||
...shellEnv.env,
|
|
||||||
TERM: "xterm-256color",
|
|
||||||
OPENCODE_TERMINAL: "1",
|
|
||||||
} as Record<string, string>
|
|
||||||
|
|
||||||
if (process.platform === "win32") {
|
|
||||||
env.LC_ALL = "C.UTF-8"
|
|
||||||
env.LC_CTYPE = "C.UTF-8"
|
|
||||||
env.LANG = "C.UTF-8"
|
|
||||||
}
|
|
||||||
log.info("creating session", { id, cmd: command, args, cwd })
|
|
||||||
|
|
||||||
const spawn = await pty()
|
|
||||||
const ptyProcess = spawn(command, args, {
|
|
||||||
name: "xterm-256color",
|
|
||||||
cwd,
|
|
||||||
env,
|
|
||||||
})
|
|
||||||
|
|
||||||
const info = {
|
|
||||||
id,
|
|
||||||
title: input.title || `Terminal ${id.slice(-4)}`,
|
|
||||||
command,
|
|
||||||
args,
|
|
||||||
cwd,
|
|
||||||
status: "running",
|
|
||||||
pid: ptyProcess.pid,
|
|
||||||
} as const
|
|
||||||
const session: ActiveSession = {
|
|
||||||
info,
|
|
||||||
process: ptyProcess,
|
|
||||||
buffer: "",
|
|
||||||
bufferCursor: 0,
|
|
||||||
cursor: 0,
|
|
||||||
subscribers: new Map(),
|
|
||||||
}
|
|
||||||
state().set(id, session)
|
|
||||||
ptyProcess.onData(
|
|
||||||
Instance.bind((chunk) => {
|
|
||||||
session.cursor += chunk.length
|
|
||||||
|
|
||||||
for (const [key, ws] of session.subscribers.entries()) {
|
|
||||||
if (ws.readyState !== 1) {
|
|
||||||
session.subscribers.delete(key)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
if (ws.data !== key) {
|
|
||||||
session.subscribers.delete(key)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
ws.send(chunk)
|
|
||||||
} catch {
|
|
||||||
session.subscribers.delete(key)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
session.buffer += chunk
|
|
||||||
if (session.buffer.length <= BUFFER_LIMIT) return
|
|
||||||
const excess = session.buffer.length - BUFFER_LIMIT
|
|
||||||
session.buffer = session.buffer.slice(excess)
|
|
||||||
session.bufferCursor += excess
|
|
||||||
}),
|
|
||||||
)
|
|
||||||
ptyProcess.onExit(
|
|
||||||
Instance.bind(({ exitCode }) => {
|
|
||||||
if (session.info.status === "exited") return
|
|
||||||
log.info("session exited", { id, exitCode })
|
|
||||||
session.info.status = "exited"
|
|
||||||
Bus.publish(Event.Exited, { id, exitCode })
|
|
||||||
remove(id)
|
|
||||||
}),
|
|
||||||
)
|
|
||||||
Bus.publish(Event.Created, { info })
|
|
||||||
return info
|
|
||||||
}
|
|
||||||
|
|
||||||
export async function update(id: PtyID, input: UpdateInput) {
|
|
||||||
const session = state().get(id)
|
|
||||||
if (!session) return
|
|
||||||
if (input.title) {
|
|
||||||
session.info.title = input.title
|
|
||||||
}
|
|
||||||
if (input.size) {
|
|
||||||
session.process.resize(input.size.cols, input.size.rows)
|
|
||||||
}
|
|
||||||
Bus.publish(Event.Updated, { info: session.info })
|
|
||||||
return session.info
|
|
||||||
}
|
|
||||||
|
|
||||||
export async function remove(id: PtyID) {
|
|
||||||
const session = state().get(id)
|
|
||||||
if (!session) return
|
|
||||||
state().delete(id)
|
|
||||||
log.info("removing session", { id })
|
|
||||||
try {
|
|
||||||
session.process.kill()
|
|
||||||
} catch {}
|
|
||||||
for (const [key, ws] of session.subscribers.entries()) {
|
|
||||||
try {
|
|
||||||
if (ws.data === key) ws.close()
|
|
||||||
} catch {
|
|
||||||
// ignore
|
|
||||||
}
|
|
||||||
}
|
|
||||||
session.subscribers.clear()
|
|
||||||
Bus.publish(Event.Deleted, { id: session.info.id })
|
|
||||||
}
|
}
|
||||||
|
|
||||||
export function resize(id: PtyID, cols: number, rows: number) {
|
export function resize(id: PtyID, cols: number, rows: number) {
|
||||||
const session = state().get(id)
|
runSync(Service.use((svc) => svc.resize(id, cols, rows)))
|
||||||
if (session && session.info.status === "running") {
|
|
||||||
session.process.resize(cols, rows)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
export function write(id: PtyID, data: string) {
|
export function write(id: PtyID, data: string) {
|
||||||
const session = state().get(id)
|
runSync(Service.use((svc) => svc.write(id, data)))
|
||||||
if (session && session.info.status === "running") {
|
|
||||||
session.process.write(data)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
export function connect(id: PtyID, ws: Socket, cursor?: number) {
|
export function connect(id: PtyID, ws: Socket, cursor?: number) {
|
||||||
const session = state().get(id)
|
return runSync(Service.use((svc) => svc.connect(id, ws, cursor)))
|
||||||
if (!session) {
|
}
|
||||||
ws.close()
|
|
||||||
return
|
|
||||||
}
|
|
||||||
log.info("client connected to session", { id })
|
|
||||||
|
|
||||||
// Use ws.data as the unique key for this connection lifecycle.
|
// Async facades
|
||||||
// If ws.data is undefined, fallback to ws object.
|
export async function create(input: CreateInput) {
|
||||||
const connectionKey = ws.data && typeof ws.data === "object" ? ws.data : ws
|
return run(Service.use((svc) => svc.create(input)))
|
||||||
|
}
|
||||||
|
|
||||||
// Optionally cleanup if the key somehow exists
|
export async function update(id: PtyID, input: UpdateInput) {
|
||||||
session.subscribers.delete(connectionKey)
|
return run(Service.use((svc) => svc.update(id, input)))
|
||||||
session.subscribers.set(connectionKey, ws)
|
}
|
||||||
|
|
||||||
const cleanup = () => {
|
export async function remove(id: PtyID) {
|
||||||
session.subscribers.delete(connectionKey)
|
return run(Service.use((svc) => svc.remove(id)))
|
||||||
}
|
|
||||||
|
|
||||||
const start = session.bufferCursor
|
|
||||||
const end = session.cursor
|
|
||||||
|
|
||||||
const from =
|
|
||||||
cursor === -1 ? end : typeof cursor === "number" && Number.isSafeInteger(cursor) ? Math.max(0, cursor) : 0
|
|
||||||
|
|
||||||
const data = (() => {
|
|
||||||
if (!session.buffer) return ""
|
|
||||||
if (from >= end) return ""
|
|
||||||
const offset = Math.max(0, from - start)
|
|
||||||
if (offset >= session.buffer.length) return ""
|
|
||||||
return session.buffer.slice(offset)
|
|
||||||
})()
|
|
||||||
|
|
||||||
if (data) {
|
|
||||||
try {
|
|
||||||
for (let i = 0; i < data.length; i += BUFFER_CHUNK) {
|
|
||||||
ws.send(data.slice(i, i + BUFFER_CHUNK))
|
|
||||||
}
|
|
||||||
} catch {
|
|
||||||
cleanup()
|
|
||||||
ws.close()
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
ws.send(meta(end))
|
|
||||||
} catch {
|
|
||||||
cleanup()
|
|
||||||
ws.close()
|
|
||||||
return
|
|
||||||
}
|
|
||||||
return {
|
|
||||||
onMessage: (message: string | ArrayBuffer) => {
|
|
||||||
session.process.write(String(message))
|
|
||||||
},
|
|
||||||
onClose: () => {
|
|
||||||
log.info("client disconnected from session", { id })
|
|
||||||
cleanup()
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1782,9 +1782,6 @@ NOTE: At any point in time through this workflow you should feel free to ask the
|
|||||||
export async function command(input: CommandInput) {
|
export async function command(input: CommandInput) {
|
||||||
log.info("command", input)
|
log.info("command", input)
|
||||||
const command = await Command.get(input.command)
|
const command = await Command.get(input.command)
|
||||||
if (!command) {
|
|
||||||
throw new NamedError.Unknown({ message: `Command not found: "${input.command}"` })
|
|
||||||
}
|
|
||||||
const agentName = command.agent ?? input.agent ?? (await Agent.defaultAgent())
|
const agentName = command.agent ?? input.agent ?? (await Agent.defaultAgent())
|
||||||
|
|
||||||
const raw = input.arguments.match(argsRegex) ?? []
|
const raw = input.arguments.match(argsRegex) ?? []
|
||||||
|
|||||||
@@ -16,7 +16,7 @@ const describeWatcher = FileWatcher.hasNativeBinding() && !process.env.CI ? desc
|
|||||||
// Helpers
|
// Helpers
|
||||||
// ---------------------------------------------------------------------------
|
// ---------------------------------------------------------------------------
|
||||||
|
|
||||||
type BusUpdate = { directory?: string; payload: { type: string; properties: Record<string, unknown> } }
|
type BusUpdate = { directory?: string; payload: { type: string; properties: WatcherEvent } }
|
||||||
type WatcherEvent = { file: string; event: "add" | "change" | "unlink" }
|
type WatcherEvent = { file: string; event: "add" | "change" | "unlink" }
|
||||||
|
|
||||||
/** Run `body` with a live FileWatcher service. */
|
/** Run `body` with a live FileWatcher service. */
|
||||||
@@ -40,18 +40,18 @@ function listen(directory: string, check: (evt: WatcherEvent) => boolean, hit: (
|
|||||||
if (done) return
|
if (done) return
|
||||||
if (evt.directory !== directory) return
|
if (evt.directory !== directory) return
|
||||||
if (evt.payload.type !== FileWatcher.Event.Updated.type) return
|
if (evt.payload.type !== FileWatcher.Event.Updated.type) return
|
||||||
const props = evt.payload.properties as WatcherEvent
|
if (!check(evt.payload.properties)) return
|
||||||
if (!check(props)) return
|
hit(evt.payload.properties)
|
||||||
hit(props)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
GlobalBus.on("event", on)
|
function cleanup() {
|
||||||
|
|
||||||
return () => {
|
|
||||||
if (done) return
|
if (done) return
|
||||||
done = true
|
done = true
|
||||||
GlobalBus.off("event", on)
|
GlobalBus.off("event", on)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
GlobalBus.on("event", on)
|
||||||
|
return cleanup
|
||||||
}
|
}
|
||||||
|
|
||||||
function wait(directory: string, check: (evt: WatcherEvent) => boolean) {
|
function wait(directory: string, check: (evt: WatcherEvent) => boolean) {
|
||||||
|
|||||||
Reference in New Issue
Block a user