Kit Langton
57 lines
1.8 KiB
TypeScript
57 lines
1.8 KiB
TypeScript
import { Context, Effect, Layer } from "effect"
|
|
import { InstanceState } from "@/effect/instance-state"
|
|
import { makeRuntime } from "@/effect/run-service"
|
|
|
|
export namespace Env {
|
|
type State = Record<string, string | undefined>
|
|
|
|
export interface Interface {
|
|
readonly get: (key: string) => Effect.Effect<string | undefined>
|
|
readonly all: () => Effect.Effect<State>
|
|
readonly set: (key: string, value: string) => Effect.Effect<void>
|
|
readonly remove: (key: string) => Effect.Effect<void>
|
|
}
|
|
|
|
export class Service extends Context.Service<Service, Interface>()("@opencode/Env") {}
|
|
|
|
export const layer = Layer.effect(
|
|
Service,
|
|
Effect.gen(function* () {
|
|
const state = yield* InstanceState.make<State>(Effect.fn("Env.state")(() => Effect.succeed({ ...process.env })))
|
|
|
|
const get = Effect.fn("Env.get")((key: string) => InstanceState.use(state, (env) => env[key]))
|
|
const all = Effect.fn("Env.all")(() => InstanceState.get(state))
|
|
const set = Effect.fn("Env.set")(function* (key: string, value: string) {
|
|
const env = yield* InstanceState.get(state)
|
|
env[key] = value
|
|
})
|
|
const remove = Effect.fn("Env.remove")(function* (key: string) {
|
|
const env = yield* InstanceState.get(state)
|
|
delete env[key]
|
|
})
|
|
|
|
return Service.of({ get, all, set, remove })
|
|
}),
|
|
)
|
|
|
|
export const defaultLayer = layer
|
|
|
|
const rt = makeRuntime(Service, defaultLayer)
|
|
|
|
export function get(key: string) {
|
|
return rt.runSync((svc) => svc.get(key))
|
|
}
|
|
|
|
export function all() {
|
|
return rt.runSync((svc) => svc.all())
|
|
}
|
|
|
|
export function set(key: string, value: string) {
|
|
return rt.runSync((svc) => svc.set(key, value))
|
|
}
|
|
|
|
export function remove(key: string) {
|
|
return rt.runSync((svc) => svc.remove(key))
|
|
}
|
|
}
|