server/classes/database/instance.ts

import { getLogger } from "@logtape/logtape";
import {
    EntityValidator,
    type ResponseError,
    type ValidationError,
} from "@versia/federation";
import type { InstanceMetadata } from "@versia/federation/types";
import chalk from "chalk";
import {
    type InferInsertModel,
    type InferSelectModel,
    type SQL,
    desc,
    eq,
    inArray,
} from "drizzle-orm";
import { db } from "~/drizzle/db";
import { Instances } from "~/drizzle/schema";
import { config } from "~/packages/config-manager/index.ts";
import { BaseInterface } from "./base.ts";
import { User } from "./user.ts";

export type AttachmentType = InferSelectModel<typeof Instances>;

export class Instance extends BaseInterface<typeof Instances> {
    async reload(): Promise<void> {
        const reloaded = await Instance.fromId(this.data.id);

        if (!reloaded) {
            throw new Error("Failed to reload instance");
        }

        this.data = reloaded.data;
    }

    public static async fromId(id: string | null): Promise<Instance | null> {
        if (!id) {
            return null;
        }

        return await Instance.fromSql(eq(Instances.id, id));
    }

    public static async fromIds(ids: string[]): Promise<Instance[]> {
        return await Instance.manyFromSql(inArray(Instances.id, ids));
    }

    public static async fromSql(
        sql: SQL<unknown> | undefined,
        orderBy: SQL<unknown> | undefined = desc(Instances.id),
    ): Promise<Instance | null> {
        const found = await db.query.Instances.findFirst({
            where: sql,
            orderBy,
        });

        if (!found) {
            return null;
        }
        return new Instance(found);
    }

    public static async manyFromSql(
        sql: SQL<unknown> | undefined,
        orderBy: SQL<unknown> | undefined = desc(Instances.id),
        limit?: number,
        offset?: number,
        extra?: Parameters<typeof db.query.Instances.findMany>[0],
    ): Promise<Instance[]> {
        const found = await db.query.Instances.findMany({
            where: sql,
            orderBy,
            limit,
            offset,
            with: extra?.with,
        });

        return found.map((s) => new Instance(s));
    }

    async update(
        newInstance: Partial<AttachmentType>,
    ): Promise<AttachmentType> {
        await db
            .update(Instances)
            .set(newInstance)
            .where(eq(Instances.id, this.id));

        const updated = await Instance.fromId(this.data.id);

        if (!updated) {
            throw new Error("Failed to update instance");
        }

        this.data = updated.data;
        return updated.data;
    }

    save(): Promise<AttachmentType> {
        return this.update(this.data);
    }

    async delete(ids?: string[]): Promise<void> {
        if (Array.isArray(ids)) {
            await db.delete(Instances).where(inArray(Instances.id, ids));
        } else {
            await db.delete(Instances).where(eq(Instances.id, this.id));
        }
    }

    public static async insert(
        data: InferInsertModel<typeof Instances>,
    ): Promise<Instance> {
        const inserted = (
            await db.insert(Instances).values(data).returning()
        )[0];

        const instance = await Instance.fromId(inserted.id);

        if (!instance) {
            throw new Error("Failed to insert instance");
        }

        return instance;
    }

    get id() {
        return this.data.id;
    }

    static async fetchMetadata(url: string): Promise<{
        metadata: InstanceMetadata;
        protocol: "versia" | "activitypub";
    } | null> {
        const origin = new URL(url).origin;
        const wellKnownUrl = new URL("/.well-known/versia", origin);
        const logger = getLogger("federation");

        const requester = await User.getFederationRequester();

        try {
            const { ok, raw, data } = await requester
                .get(wellKnownUrl, {
                    // @ts-expect-error Bun extension
                    proxy: config.http.proxy.address,
                })
                .catch((e) => ({
                    ...(e as ResponseError).response,
                }));

            if (!(ok && raw.headers.get("content-type")?.includes("json"))) {
                // If the server doesn't have a Versia well-known endpoint, it's not a Versia instance
                // Try to resolve ActivityPub metadata instead
                const data = await Instance.fetchActivityPubMetadata(url);

                if (!data) {
                    return null;
                }

                return {
                    metadata: data,
                    protocol: "activitypub",
                };
            }

            try {
                const metadata = await new EntityValidator().InstanceMetadata(
                    data,
                );

                return { metadata, protocol: "versia" };
            } catch (error) {
                logger.error`Instance ${chalk.bold(
                    origin,
                )} has invalid metadata: ${(error as ValidationError).message}`;
                return null;
            }
        } catch (error) {
            logger.error`Failed to fetch Versia metadata for instance ${chalk.bold(
                origin,
            )} - Error! ${error}`;
            return null;
        }
    }

    private static async fetchActivityPubMetadata(
        url: string,
    ): Promise<InstanceMetadata | null> {
        const origin = new URL(url).origin;
        const wellKnownUrl = new URL("/.well-known/nodeinfo", origin);

        // Go to endpoint, then follow the links to the actual metadata

        const logger = getLogger("federation");
        const requester = await User.getFederationRequester();

        try {
            const {
                raw: response,
                ok,
                data: wellKnown,
            } = await requester
                .get<{
                    links: { rel: string; href: string }[];
                }>(wellKnownUrl, {
                    // @ts-expect-error Bun extension
                    proxy: config.http.proxy.address,
                })
                .catch((e) => ({
                    ...(
                        e as ResponseError<{
                            links: { rel: string; href: string }[];
                        }>
                    ).response,
                }));

            if (!ok) {
                logger.error`Failed to fetch ActivityPub metadata for instance ${chalk.bold(
                    origin,
                )} - HTTP ${response.status}`;
                return null;
            }

            if (!wellKnown.links) {
                logger.error`Failed to fetch ActivityPub metadata for instance ${chalk.bold(
                    origin,
                )} - No links found`;
                return null;
            }

            const metadataUrl = wellKnown.links.find(
                (link: { rel: string }) =>
                    link.rel ===
                    "http://nodeinfo.diaspora.software/ns/schema/2.0",
            );

            if (!metadataUrl) {
                logger.error`Failed to fetch ActivityPub metadata for instance ${chalk.bold(
                    origin,
                )} - No metadata URL found`;
                return null;
            }

            const {
                raw: metadataResponse,
                ok: ok2,
                data: metadata,
            } = await requester
                .get<{
                    metadata: {
                        nodeName?: string;
                        title?: string;
                        nodeDescription?: string;
                        description?: string;
                    };
                    software: { version: string };
                }>(metadataUrl.href, {
                    // @ts-expect-error Bun extension
                    proxy: config.http.proxy.address,
                })
                .catch((e) => ({
                    ...(
                        e as ResponseError<{
                            metadata: {
                                nodeName?: string;
                                title?: string;
                                nodeDescription?: string;
                                description?: string;
                            };
                            software: { version: string };
                        }>
                    ).response,
                }));

            if (!ok2) {
                logger.error`Failed to fetch ActivityPub metadata for instance ${chalk.bold(
                    origin,
                )} - HTTP ${metadataResponse.status}`;
                return null;
            }

            return {
                name:
                    metadata.metadata.nodeName || metadata.metadata.title || "",
                description:
                    metadata.metadata.nodeDescription ||
                    metadata.metadata.description,
                type: "InstanceMetadata",
                software: {
                    name: "Unknown ActivityPub software",
                    version: metadata.software.version,
                },
                created_at: new Date().toISOString(),
                public_key: {
                    key: "",
                    algorithm: "ed25519",
                },
                host: new URL(url).host,
                compatibility: {
                    extensions: [],
                    versions: [],
                },
            };
        } catch (error) {
            logger.error`Failed to fetch ActivityPub metadata for instance ${chalk.bold(
                origin,
            )} - Error! ${error}`;
            return null;
        }
    }

    static async resolve(url: string): Promise<Instance> {
        const logger = getLogger("federation");
        const host = new URL(url).host;

        const existingInstance = await Instance.fromSql(
            eq(Instances.baseUrl, host),
        );

        if (existingInstance) {
            return existingInstance;
        }

        const output = await Instance.fetchMetadata(url);

        if (!output) {
            logger.error`Failed to resolve instance ${chalk.bold(host)}`;
            throw new Error("Failed to resolve instance");
        }

        const { metadata, protocol } = output;

        return Instance.insert({
            baseUrl: host,
            name: metadata.name,
            version: metadata.software.version,
            logo: metadata.logo,
            protocol,
        });
    }

    static getCount(): Promise<number> {
        return db.$count(Instances);
    }
}