Kit Langton
50 lines
1.6 KiB
TypeScript
50 lines
1.6 KiB
TypeScript
import { WorkspaceRef } from "@/effect/instance-ref"
|
|
import { InstanceStore } from "@/project/instance-store"
|
|
import { Effect, Layer } from "effect"
|
|
import { HttpRouter, HttpServerResponse } from "effect/unstable/http"
|
|
import { HttpApiMiddleware } from "effect/unstable/httpapi"
|
|
import { WorkspaceRouteContext } from "./workspace-routing"
|
|
|
|
export class InstanceContextMiddleware extends HttpApiMiddleware.Service<
|
|
InstanceContextMiddleware,
|
|
{
|
|
requires: WorkspaceRouteContext
|
|
}
|
|
>()("@opencode/ExperimentalHttpApiInstanceContext") {}
|
|
|
|
function decode(input: string): string {
|
|
try {
|
|
return decodeURIComponent(input)
|
|
} catch {
|
|
return input
|
|
}
|
|
}
|
|
|
|
function provideInstanceContext<E>(
|
|
effect: Effect.Effect<HttpServerResponse.HttpServerResponse, E>,
|
|
store: InstanceStore.Interface,
|
|
): Effect.Effect<HttpServerResponse.HttpServerResponse, E, WorkspaceRouteContext> {
|
|
return Effect.gen(function* () {
|
|
const route = yield* WorkspaceRouteContext
|
|
return yield* store.provide(
|
|
{ directory: decode(route.directory) },
|
|
effect.pipe(Effect.provideService(WorkspaceRef, route.workspaceID)),
|
|
)
|
|
})
|
|
}
|
|
|
|
export const instanceContextLayer = Layer.effect(
|
|
InstanceContextMiddleware,
|
|
Effect.gen(function* () {
|
|
const store = yield* InstanceStore.Service
|
|
return InstanceContextMiddleware.of((effect) => provideInstanceContext(effect, store))
|
|
}),
|
|
)
|
|
|
|
export const instanceRouterMiddleware = HttpRouter.middleware()(
|
|
Effect.gen(function* () {
|
|
const store = yield* InstanceStore.Service
|
|
return (effect) => provideInstanceContext(effect, store)
|
|
}),
|
|
)
|