import { z } from "zod";
import { ContentFormatSchema } from "./content_format";
import { ExtensionPropertySchema } from "./extensions";
import { VanityExtensionSchema } from "./extensions/vanity";
import { extensionTypeRegex } from "./regex";

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

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

export 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(),
});

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

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

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

export 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(),
});

export 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(),
});

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

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

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

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

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

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

export 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.",
        ),
});

export 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(),
});

export 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(),
});