api/federation/schemas/base.ts

import { ContentFormatSchema } from "@/federation/schemas/content_format";
import { ExtensionPropertySchema } from "@/federation/schemas/extensions";
import { CustomEmojiExtensionSchema } from "@/federation/schemas/extensions/custom_emojis";
import { VanityExtensionSchema } from "@/federation/schemas/extensions/vanity";
import { extensionTypeRegex } from "@/federation/schemas/regex";
import { z } from "zod";

const EntitySchema = z.object({
    id: z.string().uuid(),
    created_at: z.string(),
    uri: z.string().url(),
    type: z.string(),
    extensions: ExtensionPropertySchema.optional().nullable().nullable(),
});

const VisibilitySchema = z.enum(["public", "unlisted", "private", "direct"]);

const PublicationSchema = EntitySchema.extend({
    type: z.enum(["Note", "Patch"]),
    author: z.string().url(),
    content: ContentFormatSchema.optional().nullable(),
    attachments: z.array(ContentFormatSchema).optional().nullable(),
    replies_to: z.string().url().optional().nullable(),
    quotes: z.string().url().optional().nullable(),
    mentions: z.array(z.string().url()).optional().nullable(),
    subject: z.string().optional().nullable(),
    is_sensitive: z.boolean().optional().nullable(),
    visibility: VisibilitySchema,
    extensions: ExtensionPropertySchema.extend({
        "org.lysand:reactions": z
            .object({
                reactions: z.string(),
            })
            .optional()
            .nullable(),
        "org.lysand:polls": z
            .object({
                poll: z.object({
                    options: z.array(ContentFormatSchema),
                    votes: z.array(z.number().int().nonnegative()),
                    multiple_choice: z.boolean().optional().nullable(),
                    expires_at: z.string(),
                }),
            })
            .optional()
            .nullable(),
    })
        .optional()
        .nullable(),
});

const NoteSchema = PublicationSchema.extend({
    type: z.literal("Note"),
});

const PatchSchema = PublicationSchema.extend({
    type: z.literal("Patch"),
    patched_id: z.string().uuid(),
    patched_at: z.string(),
});

const ActorPublicKeyDataSchema = z.object({
    public_key: z.string(),
    actor: z.string().url(),
});

const UserSchema = EntitySchema.extend({
    type: z.literal("User"),
    display_name: z.string().optional().nullable(),
    username: z.string(),
    avatar: ContentFormatSchema.optional().nullable(),
    header: ContentFormatSchema.optional().nullable(),
    indexable: z.boolean(),
    public_key: ActorPublicKeyDataSchema,
    bio: ContentFormatSchema.optional().nullable(),
    fields: z
        .array(
            z.object({
                key: ContentFormatSchema,
                value: ContentFormatSchema,
            }),
        )
        .optional()
        .nullable(),
    featured: z.string().url(),
    followers: z.string().url(),
    following: z.string().url(),
    likes: z.string().url(),
    dislikes: z.string().url(),
    inbox: z.string().url(),
    outbox: z.string().url(),
    extensions: ExtensionPropertySchema.extend({
        "org.lysand:vanity": VanityExtensionSchema.optional().nullable(),
    })
        .optional()
        .nullable(),
});

const ActionSchema = EntitySchema.extend({
    type: z.union([
        z.literal("Like"),
        z.literal("Dislike"),
        z.literal("Follow"),
        z.literal("FollowAccept"),
        z.literal("FollowReject"),
        z.literal("Announce"),
        z.literal("Undo"),
    ]),
    author: z.string().url(),
});

const LikeSchema = ActionSchema.extend({
    type: z.literal("Like"),
    object: z.string().url(),
});

const UndoSchema = ActionSchema.extend({
    type: z.literal("Undo"),
    object: z.string().url(),
});

const DislikeSchema = ActionSchema.extend({
    type: z.literal("Dislike"),
    object: z.string().url(),
});

const FollowSchema = ActionSchema.extend({
    type: z.literal("Follow"),
    followee: z.string().url(),
});

const FollowAcceptSchema = ActionSchema.extend({
    type: z.literal("FollowAccept"),
    follower: z.string().url(),
});

const FollowRejectSchema = ActionSchema.extend({
    type: z.literal("FollowReject"),
    follower: z.string().url(),
});

const ExtensionSchema = EntitySchema.extend({
    type: z.literal("Extension"),
    extension_type: z
        .string()
        .regex(
            extensionTypeRegex,
            "extension_type must be in the format 'namespaced_url:extension_name/ExtensionType', e.g. 'org.lysand:reactions/Reaction'. Notably, only the type can have uppercase letters.",
        ),
});

const ReportSchema = ExtensionSchema.extend({
    extension_type: z.literal("org.lysand:reports/Report"),
    objects: z.array(z.string().url()),
    reason: z.string(),
    comment: z.string().optional().nullable(),
});

const ServerMetadataSchema = EntitySchema.omit({
    created_at: true,
    id: true,
    uri: true,
}).extend({
    type: z.literal("ServerMetadata"),
    name: z.string(),
    version: z.string(),
    description: z.string().optional().nullable(),
    website: z.string().optional().nullable(),
    moderators: z.array(z.string()).optional().nullable(),
    admins: z.array(z.string()).optional().nullable(),
    logo: ContentFormatSchema.optional().nullable(),
    banner: ContentFormatSchema.optional().nullable(),
    supported_extensions: z.array(z.string()),
    extensions: z.record(z.string(), z.any()).optional().nullable(),
});

export {
    EntitySchema,
    VisibilitySchema,
    PublicationSchema,
    NoteSchema,
    PatchSchema,
    ActorPublicKeyDataSchema,
    VanityExtensionSchema,
    UserSchema,
    ActionSchema,
    LikeSchema,
    UndoSchema,
    DislikeSchema,
    FollowSchema,
    FollowAcceptSchema,
    FollowRejectSchema,
    ExtensionSchema,
    ReportSchema,
    ServerMetadataSchema,
    ContentFormatSchema,
    CustomEmojiExtensionSchema,
    ExtensionPropertySchema,
};