feat(protocol): add browser contracts
This commit is contained in:
parent
b47cfbee7c
commit
a7a1b3011e
8 changed files with 519 additions and 0 deletions
79
packages/protocol/src/browser-control.ts
Normal file
79
packages/protocol/src/browser-control.ts
Normal 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 }),
|
||||
),
|
||||
)
|
||||
}
|
||||
75
packages/protocol/src/browser-tunnel.ts
Normal file
75
packages/protocol/src/browser-tunnel.ts
Normal 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 }),
|
||||
),
|
||||
)
|
||||
}
|
||||
63
packages/protocol/src/groups/browser.ts
Normal file
63
packages/protocol/src/groups/browser.ts
Normal 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.",
|
||||
}),
|
||||
)
|
||||
|
|
@ -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,
|
||||
}) {}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue