@bureaudouble/bureau@0.0.253

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
import type { Database } from "./types.sqlite.d.ts";
import type { Kysely } from "npm:kysely@^0.27.5";

import { createRouter, type Router } from "jsr:@fartlabs/rt@^0.0.3";
import { join } from "jsr:/@std/path@^1.0.8/join";

import createRequiredTables from "./utils/createRequiredTables.ts";
import client from "./api/client.ts";
import event from "./api/log/event.ts";
import exit from "./api/log/exit.ts";
import visit from "./api/log/visit.ts";

export interface AnalyticsConfig {
  basePath: string;
  databaseKey: string;
  getDatabase: () => Promise<Kysely<Database["public"]>>;
  getIpData?: (ip: string) => Promise<
    | null
    | undefined
    | {
      latitude: string;
      longitude: string;
      country_code: string;
      region_code: string;
      city_name: string;
    }
  >;
}

export interface AnalyticsState {
  config: AnalyticsConfig;
  database: Kysely<Database["public"]>;
}

export default async (
  config?: AnalyticsConfig,
): Promise<Router<AnalyticsConfig & Deno.ServeHandlerInfo>> => {
  if (!config) return createRouter();
  let database: Awaited<ReturnType<AnalyticsConfig["getDatabase"]>>;
  const state =
    (fn: (ctx: any) => Promise<Response> | Response) => async (ctx: any) => {
      if (!database) {
        database = await config.getDatabase();
        await createRequiredTables({ database })
          .catch(console.error)
      }
      return fn((Object.assign(ctx.state, { config, database }), ctx));
    };
  const router = createRouter(
    (router) =>
      router
        .get(join(config.basePath, "/client.js"), state(client))
        .post(join(config.basePath, "/api/log/event"), state(event))
        .post(join(config.basePath, "/api/log/exit"), state(exit))
        .post(join(config.basePath, "/api/log/visit"), state(visit)),
    (): AnalyticsConfig & Deno.ServeHandlerInfo => ({
      ...config,
      remoteAddr: null!,
    }),
  );

  return router;
};