kodadot/nft-gallery

import { MediaType } from '~/components/rmrk/types'

export const mediaTypeElementSelectors: Record<
  Extract<
    MediaType,
    MediaType.IMAGE | MediaType.VIDEO | MediaType.OBJECT | MediaType.IFRAME
  >,
  string
> = {
  [MediaType.IMAGE]: 'img[data-nuxt-img]',
  [MediaType.VIDEO]: 'video',
  [MediaType.OBJECT]: 'object',
  [MediaType.IFRAME]: 'iframe[title="html-embed"]',
}

const mediaWithoutImage = [
  MediaType.VIDEO,
  MediaType.MODEL,
  MediaType.IFRAME,
  MediaType.AUDIO,
  MediaType.OBJECT,
]

export const determineElementType = (animationType, imageType) =>
  [MediaType.IFRAME, MediaType.VIDEO].includes(animationType)
    ? animationType
    : imageType

export function isImageVisible(type: MediaType) {
  return mediaWithoutImage.every(media => media !== type)
}

export async function getMimeType(mediaUrl: string) {
  try {
    const { headers } = await $fetch.raw(mediaUrl, {
      method: mediaUrl.includes('w.kodadot.xyz') ? 'GET' : 'HEAD',
    })
    return headers.get('content-type') || ''
  }
  catch (error) {
    return ''
  }
}

export async function processMedia(mediaUrl: string) {
  const mimeType = await getMimeType(mediaUrl)
  const type = resolveMedia(mimeType)

  return {
    mimeType,
    imageVisible: isImageVisible(type),
  }
}

export const isAudio = (mimeType: string): boolean => /^audio/.test(mimeType)

export function resolveMedia(mimeType?: string): MediaType {
  if (!mimeType) {
    return MediaType.UNKNOWN
  }

  if (/^application\/json/.test(mimeType)) {
    return MediaType.JSON
  }

  if (/^model\/gltf-binary/.test(mimeType)) {
    return MediaType.MODEL
  }

  if (/^text\/html/.test(mimeType)) {
    return MediaType.IFRAME
  }

  if (/^image\/svg\+xml/.test(mimeType)) {
    return MediaType.IMAGE
  }

  if (/^application\/pdf/.test(mimeType)) {
    return MediaType.OBJECT
  }

  if (isAudio(mimeType)) {
    return MediaType.AUDIO
  }

  const match = mimeType.match(/^[a-z]+/)

  if (!match) {
    return MediaType.UNKNOWN
  }

  const prefix = match[0].toUpperCase()

  let result = MediaType.UNKNOWN

  Object.entries(MediaType).forEach(([type, value]) => {
    if (type === prefix) {
      result = value
      return
    }
  })

  return result ?? MediaType.UNKNOWN
}