server/packages/kit/plugin.ts

import type { Hono, MiddlewareHandler } from "hono";
import { createMiddleware } from "hono/factory";
import type { z } from "zod";
import { fromZodError, type ZodError } from "zod-validation-error";
import type { HonoEnv } from "~/types/api";
import type { ServerHooks } from "./hooks.ts";

export type HonoPluginEnv<ConfigType extends z.ZodTypeAny> = HonoEnv & {
    Variables: {
        pluginConfig: z.infer<ConfigType>;
    };
};

export class Plugin<ConfigSchema extends z.ZodTypeAny> {
    private handlers: Partial<ServerHooks> = {};
    private store: z.infer<ConfigSchema> | null = null;
    private routes: {
        path: string;
        fn: (app: Hono<HonoPluginEnv<ConfigSchema>>) => void;
    }[] = [];

    public constructor(private configSchema: ConfigSchema) {}

    public get middleware(): MiddlewareHandler<HonoPluginEnv<ConfigSchema>> {
        // Middleware that adds the plugin's configuration to the request object
        return createMiddleware<HonoPluginEnv<ConfigSchema>>(
            async (context, next) => {
                context.set("pluginConfig", this.getConfig());
                await next();
            },
        );
    }

    public registerRoute(
        path: string,
        fn: (app: Hono<HonoPluginEnv<ConfigSchema>>) => void,
    ): void {
        this.routes.push({
            path,
            fn,
        });
    }

    /**
     * Loads the plugin's configuration from the Versia Server configuration file.
     * This will be called when the plugin is loaded.
     * @param config Values the user has set in the configuration file.
     */
    protected async _loadConfig(config: z.input<ConfigSchema>): Promise<void> {
        try {
            this.store = await this.configSchema.parseAsync(config);
        } catch (error) {
            throw fromZodError(error as ZodError);
        }
    }

    protected _addToApp(app: Hono<HonoEnv>): void {
        for (const route of this.routes) {
            app.use(route.path, this.middleware);
            route.fn(app as unknown as Hono<HonoPluginEnv<ConfigSchema>>);
        }
    }

    public registerHandler<HookName extends keyof ServerHooks>(
        hook: HookName,
        handler: ServerHooks[HookName],
    ): void {
        this.handlers[hook] = handler;
    }

    public static [Symbol.hasInstance](instance: unknown): boolean {
        return (
            typeof instance === "object" &&
            instance !== null &&
            "registerHandler" in instance
        );
    }

    /**
     * Returns the internal configuration object.
     */
    private getConfig(): z.infer<ConfigSchema> {
        if (!this.store) {
            throw new Error("Configuration has not been loaded yet.");
        }

        return this.store;
    }
}