import { apiRoute, auth, handleZodError, parseUserAddress } from "@/api";
import { Account as AccountSchema } from "@versia/client/schemas";
import { RolePermission } from "@versia/client/schemas";
import { Instance, User } from "@versia/kit/db";
import { Users } from "@versia/kit/tables";
import { and, eq, isNull } from "drizzle-orm";
import { describeRoute } from "hono-openapi";
import { resolver, validator } from "hono-openapi/zod";
import { z } from "zod";
import { ApiError } from "~/classes/errors/api-error";
import { config } from "~/config.ts";
import { rateLimit } from "~/middlewares/rate-limit";

export default apiRoute((app) =>
    app.get(
        "/api/v1/accounts/lookup",
        describeRoute({
            summary: "Lookup account ID from Webfinger address",
            description:
                "Quickly lookup a username to see if it is available, skipping WebFinger resolution.",
            tags: ["Accounts"],
            responses: {
                200: {
                    description: "Account",
                    content: {
                        "application/json": {
                            schema: resolver(AccountSchema),
                        },
                    },
                },
                404: ApiError.accountNotFound().schema,
                422: ApiError.validationFailed().schema,
            },
        }),
        auth({
            auth: false,
            permissions: [RolePermission.Search],
        }),
        rateLimit(5),
        validator(
            "query",
            z.object({
                acct: AccountSchema.shape.acct.openapi({
                    description: "The username or Webfinger address to lookup.",
                    example: "lexi@beta.versia.social",
                }),
            }),
            handleZodError,
        ),
        async (context) => {
            const { acct } = context.req.valid("query");

            // Check if acct is matching format username@domain.com or @username@domain.com
            const { username, domain } = parseUserAddress(acct);

            // User is local
            if (!domain || domain === config.http.base_url.host) {
                const account = await User.fromSql(
                    and(eq(Users.username, username), isNull(Users.instanceId)),
                );

                if (account) {
                    return context.json(account.toApi(), 200);
                }

                return context.json(
                    { error: `Account with username ${username} not found` },
                    404,
                );
            }

            // User is remote
            // Try to fetch it from database
            const instance = await Instance.resolveFromHost(domain);

            if (!instance) {
                return context.json(
                    { error: `Instance ${domain} not found` },
                    404,
                );
            }

            const account = await User.fromSql(
                and(
                    eq(Users.username, username),
                    eq(Users.instanceId, instance.id),
                ),
            );

            if (account) {
                return context.json(account.toApi(), 200);
            }

            // Fetch from remote instance
            const uri = await User.webFinger(username, domain);

            if (!uri) {
                throw ApiError.accountNotFound();
            }

            const foundAccount = await User.resolve(uri);

            if (foundAccount) {
                return context.json(foundAccount.toApi(), 200);
            }

            throw ApiError.accountNotFound();
        },
    ),
);