import { z } from 'zod';

import { filteredArray } from './utils';

const translationPollSchema = z.object({
  id: z.string(),
  options: z.array(z.object({
    title: z.string(),
  })),
});

const translationMediaAttachment = z.object({
  id: z.string(),
  description: z.string().catch(''),
});

/** @see {@link https://docs.joinmastodon.org/entities/Translation/} */
const translationSchema = z.preprocess((translation: any) => {
  /**
   * handle Akkoma
   * @see {@link https://akkoma.dev/AkkomaGang/akkoma/src/branch/develop/lib/pleroma/web/mastodon_api/controllers/status_controller.ex#L504}
   */
  if (translation?.text) return {
    content: translation.text,
    detected_source_language: translation.detected_language,
    provider: '',
  };

  return translation;
}, z.object({
  id: z.string().nullable().catch(null),
  content: z.string().catch(''),
  spoiler_text: z.string().catch(''),
  poll: translationPollSchema.optional().catch(undefined),
  media_attachments: filteredArray(translationMediaAttachment),
  detected_source_language: z.string(),
  provider: z.string(),
}));

type Translation = z.infer<typeof translationSchema>;

export { translationSchema, type Translation };