omnivore/packages/api/src/utils/youtube.ts

import UrlPattern from 'url-pattern'

export const videoIdFromYouTubeUrl = (
  urlString: string | undefined
): string | undefined => {
  if (!urlString) {
    return undefined
  }

  const url = new URL(urlString)
  if (
    !url.hostname.endsWith('youtube.com') &&
    !url.hostname.endsWith('youtu.be')
  ) {
    return undefined
  }

  const videoId = url.searchParams.get('v')
  if (videoId) {
    return videoId || undefined
  }

  const parsed = (() => {
    const parsedUrl = new URL(url)
    parsedUrl.search = ''
    return parsedUrl.toString()
  })()

  const shortVideo = new UrlPattern('http(s)\\://(www.)youtu.be/:videoId')
  const directVideo = new UrlPattern(
    '(http(s)\\://)(www.)youtube.com/v/:videoId'
  )
  const embedVideo = new UrlPattern(
    '(http(s)\\://)(www.)youtube.com/embed/:videoId'
  )

  let params = shortVideo.match(parsed) as Record<string, string>
  if (params && params.videoId) {
    return params.videoId
  }

  params = directVideo.match(parsed) as Record<string, string>
  if (params && params.videoId) {
    return params.videoId
  }

  params = embedVideo.match(parsed) as Record<string, string>
  if (params && params.videoId) {
    return params.videoId
  }

  return undefined
}

export const isYouTubeVideoURL = (url: string | undefined): boolean => {
  if (!url) {
    return false
  }
  const videoId = videoIdFromYouTubeUrl(url)
  return videoId != null
}