feat(protocol): add browser contracts

This commit is contained in:
LukeParkerDev 2026-07-29 15:39:56 +10:00
commit a7a1b3011e
8 changed files with 519 additions and 0 deletions

View file

@ -0,0 +1,79 @@
export * as BrowserControlProtocol from "./browser-control.js"
import { BrowserControl } from "@opencode-ai/schema/browser-control"
import { Effect, Schema } from "effect"
export const Path = "/api/browser/control"
export const Subprotocol = "opencode.browser.control.v1"
export const MaxMessageBytes = 8 * 1_024 * 1_024
class MessageError extends Schema.TaggedErrorClass<MessageError>()("BrowserControlProtocol.MessageError", {
kind: Schema.Literals(["invalid", "too_large"]),
message: Schema.String,
cause: Schema.optional(Schema.Defect()),
}) {}
const decoder = new TextDecoder("utf-8", { fatal: true })
const encoder = new TextEncoder()
const encodeClient = Schema.encodeSync(Schema.fromJsonString(BrowserControl.FromClient))
const encodeServer = Schema.encodeSync(Schema.fromJsonString(BrowserControl.FromServer))
const decodeClient = Schema.decodeUnknownEffect(Schema.fromJsonString(BrowserControl.FromClient), {
errors: "all",
onExcessProperty: "error",
})
const decodeServer = Schema.decodeUnknownEffect(Schema.fromJsonString(BrowserControl.FromServer), {
errors: "all",
onExcessProperty: "error",
})
export function encodeFromClient(input: BrowserControl.FromClient) {
return encode(input, encodeClient)
}
export function encodeFromServer(input: BrowserControl.FromServer) {
return encode(input, encodeServer)
}
function encode<Message>(input: Message, encodeMessage: (input: Message) => string) {
const output = encodeMessage(input)
if (encoder.encode(output).byteLength > MaxMessageBytes) {
throw new RangeError(`Browser control message must not exceed ${MaxMessageBytes} bytes.`)
}
return output
}
export function decodeFromClient(input: string | Uint8Array) {
return decode(input, decodeClient)
}
export function decodeFromServer(input: string | Uint8Array) {
return decode(input, decodeServer)
}
function decode<Message>(
input: string | Uint8Array,
decodeMessage: (input: unknown) => Effect.Effect<Message, unknown>,
): Effect.Effect<Message, MessageError> {
if (typeof input === "string" && encoder.encode(input).byteLength > MaxMessageBytes) {
return Effect.fail(new MessageError({ kind: "too_large", message: "Browser control message is too large." }))
}
if (typeof input !== "string" && input.byteLength > MaxMessageBytes) {
return Effect.fail(new MessageError({ kind: "too_large", message: "Browser control message is too large." }))
}
const text =
typeof input === "string"
? Effect.succeed(input)
: Effect.try({
try: () => decoder.decode(input),
catch: (cause) =>
new MessageError({ kind: "invalid", message: "Browser control message is not valid UTF-8.", cause }),
})
return text.pipe(
Effect.flatMap(decodeMessage),
Effect.mapError((cause) =>
cause instanceof MessageError
? cause
: new MessageError({ kind: "invalid", message: "Browser control message is invalid.", cause }),
),
)
}

View file

@ -0,0 +1,75 @@
export * as BrowserTunnelProtocol from "./browser-tunnel.js"
import { BrowserTunnel } from "@opencode-ai/schema/browser-tunnel"
import { Effect, Schema } from "effect"
export const Path = "/api/browser/tunnel"
export const Subprotocol = "opencode.browser.tunnel.v1"
export const MaxFrameBytes = 64 * 1_024
export const MaxHandshakeBytes = 16 * 1_024
class MessageError extends Schema.TaggedErrorClass<MessageError>()("BrowserTunnelProtocol.MessageError", {
kind: Schema.Literals(["invalid", "too_large"]),
message: Schema.String,
cause: Schema.optional(Schema.Defect()),
}) {}
const encoder = new TextEncoder()
const decoder = new TextDecoder("utf-8", { fatal: true })
const encodeClient = Schema.encodeSync(Schema.fromJsonString(BrowserTunnel.FromClient))
const encodeServer = Schema.encodeSync(Schema.fromJsonString(BrowserTunnel.FromServer))
const decodeClient = Schema.decodeUnknownEffect(Schema.fromJsonString(BrowserTunnel.FromClient), {
errors: "all",
onExcessProperty: "error",
})
const decodeServer = Schema.decodeUnknownEffect(Schema.fromJsonString(BrowserTunnel.FromServer), {
errors: "all",
onExcessProperty: "error",
})
export function encodeFromClient(input: BrowserTunnel.FromClient) {
return encode(encodeClient(input))
}
export function encodeFromServer(input: BrowserTunnel.FromServer) {
return encode(encodeServer(input))
}
function encode(input: string) {
if (encoder.encode(input).byteLength > MaxHandshakeBytes) {
throw new RangeError(`Browser tunnel handshake must not exceed ${MaxHandshakeBytes} bytes.`)
}
return input
}
export function decodeFromClient(input: string | Uint8Array) {
return decode(input, decodeClient)
}
export function decodeFromServer(input: string | Uint8Array) {
return decode(input, decodeServer)
}
function decode<Message>(
input: string | Uint8Array,
decodeMessage: (input: unknown) => Effect.Effect<Message, unknown>,
): Effect.Effect<Message, MessageError> {
if ((typeof input === "string" ? encoder.encode(input).byteLength : input.byteLength) > MaxHandshakeBytes) {
return Effect.fail(new MessageError({ kind: "too_large", message: "Browser tunnel handshake is too large." }))
}
const text =
typeof input === "string"
? Effect.succeed(input)
: Effect.try({
try: () => decoder.decode(input),
catch: (cause) => new MessageError({ kind: "invalid", message: "Invalid tunnel handshake UTF-8.", cause }),
})
return text.pipe(
Effect.flatMap(decodeMessage),
Effect.mapError((cause) =>
cause instanceof MessageError
? cause
: new MessageError({ kind: "invalid", message: "Browser tunnel handshake is invalid.", cause }),
),
)
}

View file

@ -0,0 +1,63 @@
import { Schema } from "effect"
import { HttpApiEndpoint, HttpApiGroup, OpenApi } from "effect/unstable/httpapi"
import { ConflictError, ServiceUnavailableError } from "../errors.js"
import { BrowserControlProtocol } from "../browser-control.js"
import { BrowserTunnelProtocol } from "../browser-tunnel.js"
import { HeaderOnlyAuthorization } from "../middleware/authorization.js"
const websocket = (identifier: string, summary: string, description: string, subprotocol: string) =>
OpenApi.annotations({
identifier,
summary,
description,
transform: (operation) => ({
...operation,
"x-websocket": true,
"x-websocket-subprotocol": subprotocol,
responses: {
...operation.responses,
403: { description: "WebSocket Origin is not allowed." },
426: { description: `WebSocket subprotocol ${subprotocol} is required.` },
},
}),
})
export const BrowserGroup = HttpApiGroup.make("server.browser")
.add(
HttpApiEndpoint.get("browser.control.connect", BrowserControlProtocol.Path, {
success: Schema.Boolean,
error: ConflictError,
})
.annotate(HeaderOnlyAuthorization, true)
.annotate(OpenApi.Exclude, true)
.annotateMerge(
websocket(
"v2.browser.control.connect",
"Connect Session browser host",
"Establish an authenticated WebSocket controlling the browser attachment for one Session.",
BrowserControlProtocol.Subprotocol,
),
),
)
.add(
HttpApiEndpoint.get("browser.tunnel.connect", BrowserTunnelProtocol.Path, {
success: Schema.Boolean,
error: ServiceUnavailableError,
})
.annotate(HeaderOnlyAuthorization, true)
.annotate(OpenApi.Exclude, true)
.annotateMerge(
websocket(
"v2.browser.tunnel.connect",
"Open browser network tunnel",
"Establish an authenticated WebSocket carrying one TCP stream dialed from the OpenCode server.",
BrowserTunnelProtocol.Subprotocol,
),
),
)
.annotateMerge(
OpenApi.annotations({
title: "browser",
description: "Desktop browser host control and server-network tunnel routes.",
}),
)

View file

@ -1,6 +1,11 @@
import { Context } from "effect"
import { HttpApiMiddleware } from "effect/unstable/httpapi"
import { UnauthorizedError } from "../errors.js"
export const HeaderOnlyAuthorization = Context.Reference<boolean>("@opencode/HttpApiAuthorization/HeaderOnly", {
defaultValue: () => false,
})
export class Authorization extends HttpApiMiddleware.Service<Authorization>()("@opencode/HttpApiAuthorization", {
error: UnauthorizedError,
}) {}