api.ts

import { Status } from "std/http/mod.ts";
import { ConsoleHandler } from "std/log/handlers.ts";
import { getLogger, setup as setupLog } from "std/log/mod.ts";

import {
  Application as OakApplication,
  type Context,
  isHttpError,
  Router,
} from "oak";

import { oakCors } from "https://deno.land/x/cors@v1.2.2/mod.ts";

import { Buffer } from "node:buffer";

import { stringify as losslessJsonStringify } from "npm:lossless-json";
import { getAddress, keccak256, toHex } from "viem";

import type { PrismaClient } from "./prisma/shim.ts";

import { validateEventRequest } from "./constants/apiSchema.ts";
import { serializeEventResponse } from "./messages/EventResponse.ts";
import { formatAbiItemPrototype } from "./utils/abiUtils.ts";
import {
  ApiBehindReverseProxyEnvKey,
  ApiUrlEnvKey,
  combinedEnv,
} from "./utils/envUtils.ts";
import {
  ApiLoggerName,
  defaultLogFormatter,
  getInternalLoggers,
  getLoggingLevel,
} from "./utils/logUtils.ts";
import { runWithPrisma } from "./utils/runUtils.ts";

export function api(prisma: PrismaClient) {
  const hexToBuffer = (hex: string): Buffer => {
    return Buffer.from(hex.replace("0x", ""), "hex");
  };

  const getBlockNumberFromHash = async (blockHash?: string | number) => {
    if (typeof blockHash !== "string") return blockHash;
    return (await prisma.event.findFirst({
      where: { blockHash: hexToBuffer(blockHash) },
    }))
      ?.blockNumber;
  };

  const logger = getLogger(ApiLoggerName);
  const abortController = new AbortController();

  const router = new Router();

  router.get("/", (ctx) => {
    // TODO: show JSON Schema
    ctx.response.body = "Not yet implemented";
    ctx.response.status = Status.NotImplemented;
  });
  router.post("/", async (ctx) => {
    const request = await ctx.request.body({ type: "json" }).value;

    if (!validateEventRequest(request)) {
      ctx.response.status = Status.BadRequest;
      ctx.response.body = validateEventRequest.errors;
      return;
    }

    const blockFrom = await getBlockNumberFromHash(request.blockFrom);
    const blockTo = await getBlockNumberFromHash(request.blockTo);

    ctx.response.body = losslessJsonStringify((await prisma.event.findMany({
      where: {
        blockHash: (request.blockHash && hexToBuffer(request.blockHash)) as
          | Buffer
          | undefined,
        blockNumber: request.blockIndex ?? { gte: blockFrom, lte: blockTo },
        logIndex: request.logIndex,
        txHash:
          (request.transactionHash && hexToBuffer(request.transactionHash)) as
            | Buffer
            | undefined,
        sourceAddress:
          (request.sourceAddress && hexToBuffer(request.sourceAddress)) as
            | Buffer
            | undefined,
        abiHash: ((request.abiHash && hexToBuffer(request.abiHash)) ??
          (request.abiSignature &&
            Buffer.from(
              keccak256(
                new TextEncoder().encode(request.abiSignature),
                "bytes",
              ),
            ))) as Buffer | undefined,
      },
      include: { Abi: true },
    })).map((evt) =>
      serializeEventResponse({
        address: evt.sourceAddress,
        sigHash: evt.abiHash,
        abi: evt.Abi.json,
        topics: [evt.topic3, evt.topic2, evt.topic1].reduce(
          (acc, x) => x != undefined ? [x, ...acc] : [],
          [] as Uint8Array[],
        ),
        data: evt.data,
        logIndex: BigInt(evt.logIndex),
        blockNumber: BigInt(evt.blockNumber),
        blockHash: evt.blockHash,
        txHash: evt.txHash,
      })
    ));
  });

  router.get("/sources", (ctx) => {
    // TODO: show JSON Schema
    ctx.response.body = "Not yet implemented";
    ctx.response.status = Status.NotImplemented;
  });
  router.post("/sources", async (ctx) => {
    // TODO: parameters
    ctx.response.body = (
      await prisma.eventSource.findMany({
        include: { Abi: true },
      })
    ).map((item) => ({
      address: getAddress(toHex(item.address)),
      abi: formatAbiItemPrototype(JSON.parse(item.Abi.json)),
      abiHash: toHex(item.abiHash),
    }));
  });

  router.get("/abi", (ctx) => {
    // TODO: show JSON Schema
    ctx.response.body = "Not yet implemented";
    ctx.response.status = Status.NotImplemented;
  });
  router.post("/abi", async (ctx) => {
    // TODO: parameters
    ctx.response.body = (await prisma.eventAbi.findMany()).reduce(
      (acc, item) => {
        const abi = JSON.parse(item.json);
        Object.assign(acc, {
          [toHex(item.hash)]: {
            signature: formatAbiItemPrototype(abi),
            abi: abi,
          },
        });
        return acc;
      },
      {},
    );
  });

  router.post("/webhook", async (ctx) => {
    // TODO: parameters
    ctx.response.body = (await prisma.emitDestination.findMany())
      .map((item) => ({
        id: item.id,
        sourceAddress: getAddress(toHex(item.sourceAddress)),
        abiHash: toHex(item.abiHash),
        webhookUrl: item.webhookUrl,
        topic1: item.topic1 ? toHex(item.topic1) : undefined,
        topic2: item.topic2 ? toHex(item.topic2) : undefined,
        topic3: item.topic3 ? toHex(item.topic3) : undefined,
      }));
  });

  const app = new OakApplication();
  app.proxy = combinedEnv[ApiBehindReverseProxyEnvKey] === "true";
  app.use(async (context, next) => {
    function formatResponse(
      ctx: Context<Record<string, unknown>, Record<string, unknown>>,
    ) {
      return `${ctx.response.status} ${ctx.request.method} ${ctx.request.url.pathname} ${ctx.request.ip} ${
        JSON.stringify(ctx.state["requestBody"])
      }: ${
        typeof ctx.response.body === "function"
          ? ctx.response.body()
          : ctx.response.body
      }`;
    }

    try {
      context.state["requestBody"] = await context.request.body({
        type: "json",
      }).value;
      await next();
      logger.info(formatResponse(context));
    } catch (err) {
      if (isHttpError(err)) {
        context.response.status = err.status;
      } else {
        context.response.status = 500;
      }
      context.response.body = { error: err.message };
      context.response.type = "json";
      logger.error(formatResponse(context));
    } finally {
      delete context.state["requestBody"];
    }
  });
  app.use(oakCors());
  app.use(router.routes());
  app.use(router.allowedMethods());

  const listenUrl = new URL(combinedEnv[ApiUrlEnvKey]);
  logger.info(`API server listening on ${listenUrl}.`);
  const runningPromise = app.listen({
    port: Number(listenUrl.port) || 80,
    hostname: listenUrl.hostname,
    signal: abortController.signal,
  });

  async function cleanup() {
    logger.warning("Stopping API server.");
    abortController.abort();
    return await runningPromise;
  }

  return { runningPromise, cleanup };
}

if (import.meta.main) {
  setupLog({
    handlers: {
      console: new ConsoleHandler(getLoggingLevel(), {
        formatter: defaultLogFormatter,
      }),
    },

    loggers: {
      ...getInternalLoggers({
        level: getLoggingLevel(),
        handlers: ["console"],
      }),
      [ApiLoggerName]: {
        level: getLoggingLevel(),
        handlers: ["console"],
      },
    },
  });
  await runWithPrisma(api);
}