ArtPlayer/test/helpers/multiple-subtitles.js

import process from 'node:process'
import vm from 'node:vm'
import { transform } from 'esbuild'
import { verifyMultipleSubtitlesContract } from '../../refactor/scripts/multiple-subtitles-contract.mjs'
import { readMember } from '../../refactor/scripts/releases.mjs'
import { browserCandidate } from './browser-candidate.js'

export const subtitleVtt = text => `WEBVTT\n\n00:00.000 --> 00:02.000\n${text}\n`

export async function multipleSubtitlesCandidate() {
  return { name: 'candidate', version: 'candidate', ...await browserCandidate('artplayer-plugin-multiple-subtitles', process.env.ARTPLAYER_MULTIPLE_SUBTITLES_ARTIFACT) }
}

export async function multipleSubtitlesHistorical() {
  const { baseline, archives, sources } = await verifyMultipleSubtitlesContract()
  const name = 'artplayer-plugin-multiple-subtitles'
  const implementations = []
  for (const release of [baseline.release, ...baseline.previous]) {
    for (const field of ['main', 'legacy'])
      implementations.push({ name: `published-${release.version}-${field}`, version: release.version, code: readMember(archives.get(release.version), `package/${release.manifest[field].replace(/^\.\//, '')}`).toString() })
  }
  const source = sources.get(`packages/${name}/src/parser.js`)
    + sources.get(`packages/${name}/src/index.js`).replace(/^import .* from ['"].\/parser['"];?\s*$/m, '')
  implementations.push({ name: 'frozen-workspace-source', version: 'workspace', code: (await transform(source, { format: 'cjs', target: 'es2020' })).code })
  for (const suffix of ['js', 'legacy.js'])
    implementations.push({ name: `frozen-workspace-${suffix}`, version: 'workspace', code: sources.get(`packages/${name}/dist/${name}.${suffix}`) })
  return implementations
}

export function multipleSubtitlesEnvironment(implementation, { script = false, responses = {}, fetchResponse, coreUtils, onInit, onCreate, abortController = true } = {}) {
  const requests = []
  const requestOptions = []
  const blobs = new Map()
  const liveBlobs = new Map()
  const revoked = []
  const initialized = []
  const initStates = []
  const converted = []
  const listeners = new Map()
  const warnings = []
  const module = { exports: {} }
  const utils = {
    getExt: url => url.split('.').pop(),
    unescape: text => text.replaceAll('&lt;', '<').replaceAll('&gt;', '>').replaceAll('&amp;', '&'),
    srtToVtt(text) {
      converted.push(['srt', text])
      return subtitleVtt('SRT')
    },
    assToVtt(text) {
      converted.push(['ass', text])
      return subtitleVtt('ASS')
    },
    ...coreUtils,
  }
  const art = {
    isDestroy: false,
    constructor: { utils },
    option: { subtitle: { escape: true, style: { color: 'red' } } },
    subtitle: { init(option) {
      initialized.push(option)
      initStates.push({ destroyed: art.isDestroy })
      return onInit?.(option)
    } },
    on(name, callback) {
      if (!listeners.has(name))
        listeners.set(name, new Set())
      listeners.get(name).add(callback)
      return art
    },
    off(name, callback) {
      listeners.get(name)?.delete(callback)
      return art
    },
  }
  function responseFor(url) {
    const value = responses[url] ?? subtitleVtt(url)
    const bytes = typeof value === 'string' ? new TextEncoder().encode(value) : value
    return { ok: true, status: 200, arrayBuffer: async () => bytes.buffer.slice(bytes.byteOffset, bytes.byteOffset + bytes.byteLength) }
  }
  const context = {
    Blob,
    TextDecoder,
    AbortController: abortController ? AbortController : undefined,
    console: { log() {}, warn(...args) { warnings.push(args) }, error() {} },
    URL: {
      createObjectURL(blob) {
        onCreate?.(blob)
        const url = `blob:subtitle-${blobs.size + 1}`
        blobs.set(url, blob)
        liveBlobs.set(url, blob)
        return url
      },
      revokeObjectURL(url) {
        revoked.push(url)
        liveBlobs.delete(url)
      },
    },
    fetch: async (url, options) => {
      requests.push(url)
      requestOptions.push(options)
      return fetchResponse ? fetchResponse(url, options, responseFor) : responseFor(url)
    },
  }
  context.window = context
  if (!script) {
    context.module = module
    context.exports = module.exports
  }
  vm.runInNewContext(implementation.code, context, { timeout: 5000 })
  const exported = script ? context.artplayerPluginMultipleSubtitles : module.exports
  const factory = exported?.default || exported
  const latestText = async () => blobs.get(initialized.at(-1).url).text()
  function emit(name, ...args) {
    if (name === 'destroy')
      art.isDestroy = true
    for (const callback of [...(listeners.get(name) || [])])
      callback(...args)
  }
  return { factory, exported, art, utils, requests, requestOptions, blobs, liveBlobs, revoked, initialized, initStates, converted, listeners, warnings, emit, responseFor, latestText }
}