t3-code-android-nightly/.repos/alchemy-effect/packages/alchemy/test/Neon/fixtures/StorageHttpHandler.ts

import { ReadBucket } from "@/Neon/ReadBucket";
import { ReadBucketHttp } from "@/Neon/ReadBucketHttp";
import { ReadWriteBucket } from "@/Neon/ReadWriteBucket";
import { ReadWriteBucketHttp } from "@/Neon/ReadWriteBucketHttp";
import { CurrentRuntimeContext } from "@/RuntimeContext";
import { StorageBindingError } from "@/Neon/StorageBinding";
import * as Effect from "effect/Effect";
import * as Layer from "effect/Layer";
import { HttpServerRequest } from "effect/http/HttpServerRequest";
import * as HttpServerResponse from "effect/http/HttpServerResponse";
import { StorageBucket } from "./StorageResources.ts";

export const storageHttpHandler = Effect.gen(function* () {
  const bucket = yield* StorageBucket;
  const files = yield* ReadWriteBucket(bucket);
  const reader = yield* ReadBucket(bucket);
  const runtime = yield* CurrentRuntimeContext;
  if (!runtime)
    return yield* Effect.die(
      new StorageBindingError({
        message: "Storage HTTP fixture requires a runtime host",
      }),
    );
  return {
    fetch: Effect.gen(function* () {
      const request = yield* HttpServerRequest;
      if (request.url.endsWith("/write")) {
        yield* files.put("external.txt", "external roundtrip", {
          ContentType: "text/plain",
        });
        return HttpServerResponse.text("ok");
      }
      const item = yield* reader.head("external.txt");
      return yield* HttpServerResponse.json({
        size: item?.ContentLength,
        hasAccountKey: (yield* runtime.get("NEON_API_KEY")) !== undefined,
      });
    }).pipe(
      Effect.catch(() =>
        Effect.succeed(
          HttpServerResponse.text("Storage request failed", { status: 500 }),
        ),
      ),
    ),
  };
}).pipe(Effect.provide(Layer.mergeAll(ReadWriteBucketHttp, ReadBucketHttp)));