bknd/app/src/adapter/bun/bun.adapter.ts

/// <reference types="bun-types" />

import path from "node:path";
import {
   type RuntimeBkndConfig,
   createRuntimeApp,
   type RuntimeOptions,
   Connection,
} from "bknd/adapter";
import { registerLocalMediaAdapter } from ".";
import { config } from "bknd/core";
import type { ServeOptions } from "bun";
import { serveStatic } from "hono/bun";
import { sqlite } from "bknd/adapter/sqlite";
import type { App } from "App";

type BunEnv = Bun.Env;
export type BunBkndConfig<Env = BunEnv> = RuntimeBkndConfig<Env> & Omit<ServeOptions, "fetch">;

export async function createApp<Env = BunEnv>(
   { distPath, ...config }: BunBkndConfig<Env> = {},
   args: Env = {} as Env,
   opts?: RuntimeOptions,
) {
   const root = path.resolve(distPath ?? "./node_modules/bknd/dist", "static");
   registerLocalMediaAdapter();

   let connection: Connection | undefined;
   if (Connection.isConnection(config.connection)) {
      connection = config.connection;
   } else {
      connection = sqlite(config.connection ?? { url: ":memory:" });
   }

   return await createRuntimeApp(
      {
         ...config,
         connection,
         serveStatic: serveStatic({ root }),
      },
      args ?? (process.env as Env),
      opts,
   );
}

export function createHandler<Env = BunEnv>(
   config: BunBkndConfig<Env> = {},
   args: Env = {} as Env,
   opts?: RuntimeOptions,
) {
   let app: App | undefined;
   return async (req: Request) => {
      if (!app) {
         app = await createApp(config, args ?? (process.env as Env), opts);
      }
      return app.fetch(req);
   };
}

export function serve<Env = BunEnv>(
   {
      distPath,
      connection,
      initialConfig,
      options,
      port = config.server.default_port,
      onBuilt,
      buildConfig,
      adminOptions,
      ...serveOptions
   }: BunBkndConfig<Env> = {},
   args: Env = {} as Env,
   opts?: RuntimeOptions,
) {
   Bun.serve({
      ...serveOptions,
      port,
      fetch: createHandler(
         {
            connection,
            initialConfig,
            options,
            onBuilt,
            buildConfig,
            adminOptions,
            distPath,
         },
         args,
         opts,
      ),
   });

   console.info(`Server is running on http://localhost:${port}`);
}