bigbuffet-rw/packages/pl-api/lib/entities/media-attachment.ts

import { isBlurhashValid } from 'blurhash';
import * as v from 'valibot';

import { mimeSchema } from './utils';

const blurhashSchema = z.string().superRefine((value, ctx) => {
  const r = isBlurhashValid(value);

  if (!r.result) {
    ctx.addIssue({
      code: z.ZodIssueCode.custom,
      message: r.errorReason,
    });
  }
});

const baseAttachmentSchema = v.object({
  id: v.string(),
  type: v.string(),
  url: v.fallback(v.pipe(v.string(), v.url()), ''),
  preview_url: v.fallback(v.pipe(v.string(), v.url()), ''),
  remote_url: v.fallback(v.nullable(z.string().url()), null),
  description: v.fallback(v.string(), ''),
  blurhash: v.fallback(v.nullable(blurhashSchema), null),

  mime_type: v.fallback(v.nullable(mimeSchema), null),
});

const imageMetaSchema = v.object({
  width: v.number(),
  height: v.number(),
  size: z.string().regex(/\d+x\d+$/).nullable().catch(null),
  aspect: v.fallback(v.nullable(v.number()), null),
});

const imageAttachmentSchema = baseAttachmentSchema.extend({
  type: v.literal('image'),
  meta: v.fallback(v.object({
    original: v.fallback(v.optional(imageMetaSchema), undefined),
    small: v.fallback(v.optional(imageMetaSchema), undefined),
    focus: v.fallback(v.optional(v.object({
      x: z.number().min(-1).max(1),
      y: z.number().min(-1).max(1),
    })), undefined),
  }), {}),
});

const videoAttachmentSchema = baseAttachmentSchema.extend({
  type: v.literal('video'),
  meta: v.fallback(v.object({
    duration: v.fallback(v.optional(v.number()), undefined),
    original: v.fallback(v.optional(imageMetaSchema.extend({
      frame_rate: v.fallback(v.nullable(v.pipe(v.string(), v.regex(/\d+\/\d+$/))), null),
      duration: v.fallback(v.nullable(z.number().nonnegative()), null),
    })), undefined),
    small: v.fallback(v.optional(imageMetaSchema), undefined),
    // WIP: add rest
  }), {}),
});

const gifvAttachmentSchema = baseAttachmentSchema.extend({
  type: v.literal('gifv'),
  meta: v.fallback(v.object({
    duration: v.fallback(v.optional(v.number()), undefined),
    original: v.fallback(v.optional(imageMetaSchema), undefined),
  }), {}),
});

const audioAttachmentSchema = baseAttachmentSchema.extend({
  type: v.literal('audio'),
  meta: v.fallback(v.object({
    duration: v.fallback(v.optional(v.number()), undefined),
    colors: v.fallback(v.optional(v.object({
      background: v.fallback(v.optional(v.string()), undefined),
      foreground: v.fallback(v.optional(v.string()), undefined),
      accent: v.fallback(v.optional(v.string()), undefined),
      duration: v.fallback(v.optional(v.number()), undefined),
    })), undefined),
    original: v.fallback(v.optional(v.object({
      duration: v.fallback(v.optional(v.number()), undefined),
      bitrate: z.number().nonnegative().optional().catch(undefined),
    })), undefined),
  }), {}),
});

const unknownAttachmentSchema = baseAttachmentSchema.extend({
  type: v.literal('unknown'),
});

/** @see {@link https://docs.joinmastodon.org/entities/MediaAttachment} */
const mediaAttachmentSchema = z.preprocess((data: any) => {
  if (!data) return null;

  return {
    mime_type: data.pleroma?.mime_type,
    preview_url: data.url,
    ...data,
  };
}, z.discriminatedUnion('type', [
  imageAttachmentSchema,
  videoAttachmentSchema,
  gifvAttachmentSchema,
  audioAttachmentSchema,
  unknownAttachmentSchema,
]));

type MediaAttachment = v.InferOutput<typeof mediaAttachmentSchema>;

export { blurhashSchema, mediaAttachmentSchema, type MediaAttachment };