better-auth-dashboard/src/plugin/server.ts

A better-Auth powered admin dashboard.

import type { AuthContext, BetterAuthPlugin } from "better-auth";
import {
  APIError,
  createAuthEndpoint,
  getSessionFromCtx,
  getEndpoints,
} from "better-auth/api";

// eslint-disable-next-line @typescript-eslint/no-empty-object-type
type DashboardPluginConfig = {};

export type DashboardPlugin = typeof dashboardPlugin;

export function dashboardPlugin(config: DashboardPluginConfig = {}) {
  const plugin: BetterAuthPlugin = {
    id: "better-auth-dashboard",
    schema: {
      routeProtection: {
        fields: {
          route: {
            type: "string",
            unique: true,
            required: true,
            input: true,
          },
          isRoleProtected: {
            type: "boolean",
            input: true,
            required: true,
          },
          roleProtection: {
            type: "string",
            input: true,
            required: true,
          },
          isAuthenticatedProtected: {
            type: "boolean",
            input: true,
            required: true,
          },
        },
      },
    },
    endpoints: {
      routeProtection: createAuthEndpoint(
        "/dashboard/route-protection",
        {
          method: "GET",
        },
        async (ctx) => {
          const session = await getSessionFromCtx(ctx);

          if (!session || session.user.role != "admin") {
            throw new APIError("UNAUTHORIZED", {
              message: `Invalid or missing session.`,
            });
          }

          return ctx.json({
            message: "Hello World",
          });
        }
      ),
      getEndpoints: createAuthEndpoint(
        `/dashboard/get-endpoints`,
        {
          method: "GET",
        },
        async (ctx) => {
          const session = await getSessionFromCtx(ctx);

          if (!session || session.user.role != "admin") {
            throw new APIError("UNAUTHORIZED", {
              message: `Invalid or missing session.`,
            });
          }

          const endpoints = getEndpoints(ctx.context, ctx.context.options);
          const paths: string[] = endpoints.middlewares.map((x) => x.path);
          return ctx.json({
            paths,
          });
        }
      ),
    },
  };

  return plugin;
}

function hasAdminPlugin(context: AuthContext) {
  return Boolean(context.options.plugins?.find((x) => x.id === "admin"));
}