openAgent

1export * as Image from "./image"
2 
3import { makeLocationNode } from "./effect/app-node"
4import { Context, Effect, Layer, Schema } from "effect"
5import { Config } from "./config"
6import { FileSystem } from "./filesystem"
7 
8export class ResizerUnavailableError extends Schema.TaggedErrorClass<ResizerUnavailableError>()(
9  "Image.ResizerUnavailableError",
10  {},
11) {}
12 
13export class DecodeError extends Schema.TaggedErrorClass<DecodeError>()("Image.DecodeError", {
14  resource: Schema.String,
15}) {
16  override get message() {
17    return `Image could not be decoded: ${this.resource}`
18  }
19}
20 
21export class SizeError extends Schema.TaggedErrorClass<SizeError>()("Image.SizeError", {
22  resource: Schema.String,
23  width: Schema.Number,
24  height: Schema.Number,
25  bytes: Schema.Number,
26  maxWidth: Schema.Number,
27  maxHeight: Schema.Number,
28  maxBytes: Schema.Number,
29}) {
30  override get message() {
31    return `Image ${this.resource} is ${this.width}x${this.height} with base64 size ${this.bytes}, exceeding configured limits ${this.maxWidth}x${this.maxHeight}/${this.maxBytes} bytes`
32  }
33}
34 
35export interface Interface {
36  readonly normalize: (
37    resource: string,
38    content: FileSystem.Content & { readonly encoding: "base64" },
39  ) => Effect.Effect<
40    FileSystem.Content & { readonly encoding: "base64" },
41    ResizerUnavailableError | DecodeError | SizeError
42  >
43}
44 
45export class Service extends Context.Service<Service, Interface>()("@opencode/Image") {}
46 
47const layer = Layer.effect(
48  Service,
49  Effect.gen(function* () {
50    const config = yield* Config.Service
51    const loadAdapter = yield* Effect.cached(
52      Effect.tryPromise({
53        try: () => import("./image/photon"),
54        catch: () => new ResizerUnavailableError(),
55      }).pipe(Effect.flatMap((adapter) => adapter.make)),
56    )
57    const normalize = Effect.fn("Image.normalize")(function* (
58      resource: string,
59      content: FileSystem.Content & { readonly encoding: "base64" },
60    ) {
61      const image = Object.assign(
62        {},
63        ...(yield* config.entries()).flatMap((entry) =>
64          entry.type === "document" && entry.info.attachments?.image ? [entry.info.attachments.image] : [],
65        ),
66      )
67      const normalize = yield* loadAdapter
68      return yield* normalize(resource, content, {
69        autoResize: image.auto_resize ?? true,
70        maxWidth: image.max_width ?? 2_000,
71        maxHeight: image.max_height ?? 2_000,
72        maxBase64Bytes: image.max_base64_bytes ?? 5 * 1024 * 1024,
73      })
74    })
75    return Service.of({ normalize })
76  }),
77)
78 
79export const locationLayer = layer.pipe(Layer.provide(Config.locationLayer))
80 
81export const node = makeLocationNode({ service: Service, layer, deps: [Config.node] })
82