import { Flag } from "@/flag/flag"
import { Database } from "@opencode-ai/core/database/database"
import { Effect } from "effect"
import { HttpRouter, HttpServerRequest, HttpServerResponse } from "effect/unstable/http"
import * as Fence from "@/server/shared/fence"

const ignoredMethods = new Set(["GET", "HEAD", "OPTIONS"])

export const fenceLayer = HttpRouter.middleware<{ requires: Database.Service; handles: unknown }>()(
  Effect.gen(function* () {
    const { db } = yield* Database.Service
    return (effect) =>
      Effect.gen(function* () {
        const request = yield* HttpServerRequest.HttpServerRequest
        if (!Flag.OPENCODE_WORKSPACE_ID || ignoredMethods.has(request.method)) return yield* effect

        const previous = yield* Fence.load(db)
        const response = yield* effect
        const current = Fence.diff(previous, yield* Fence.load(db))
        if (Object.keys(current).length === 0) return response

        return HttpServerResponse.setHeader(response, Fence.HEADER, JSON.stringify(current))
      })
  }),
).layer