ArtPlayer/test/helpers/danmuku-candidate.js

import assert from 'node:assert/strict'
import fs from 'node:fs'
import path from 'node:path'
import process from 'node:process'
import { fileURLToPath } from 'node:url'
import { build } from 'esbuild'
import { ensureArchive, readMember, refactorDir } from '../../refactor/scripts/releases.mjs'
import { getEntryFile } from '../../scripts/projects.js'
import { danmukuEnvironment } from './danmuku.js'
import { resolveSource } from './load.js'

const workspace = fileURLToPath(new URL('../../', import.meta.url))
let sourceWorker

export async function danmukuWorkerCode(implementation, env, worker) {
  const blob = env.urls.get(worker.url)
  if (blob)
    return blob.text()
  // Production worker-loader emits a percent-encoded data URL. Execute those
  // exact shipped bytes instead of silently substituting the current source.
  const dataPrefix = 'data:text/javascript;charset=utf-8,'
  if (typeof worker.url === 'string' && worker.url.startsWith(dataPrefix))
    return decodeURIComponent(worker.url.slice(dataPrefix.length))
  assert.equal(implementation.name, 'candidate-source', 'Artifact Worker tests must execute the Worker embedded in that artifact')
  sourceWorker ??= build({
    entryPoints: [resolveSource('packages/artplayer-plugin-danmuku/src/worker')],
    bundle: true,
    write: false,
    format: 'iife',
    platform: 'browser',
    target: 'es2020',
  }).then(result => result.outputFiles[0].text)
  return sourceWorker
}

export async function danmukuCandidate() {
  const coreRelease = JSON.parse(fs.readFileSync(path.join(refactorDir, 'baselines/releases.json'), 'utf8')).releases.find(item => item.name === 'artplayer')
  const coreCode = readMember(await ensureArchive(coreRelease), 'package/dist/artplayer.js').toString()
  const artifact = process.env.ARTPLAYER_DANMUKU_ARTIFACT
  if (artifact)
    return { name: path.basename(artifact), code: fs.readFileSync(path.resolve(artifact), 'utf8'), coreCode }
  const project = path.join(workspace, 'packages/artplayer-plugin-danmuku')
  const compiled = await build({ entryPoints: [getEntryFile(project)], bundle: true, write: false, format: 'cjs', platform: 'browser', target: 'es2020', plugins: [{ name: 'danmuku-controlled-assets', setup(build) {
    build.onResolve({ filter: /\?(?:worker|raw|inline)/ }, args => ({ path: path.resolve(args.resolveDir, args.path), namespace: 'controlled-asset' }))
    build.onLoad({ filter: /.*/, namespace: 'controlled-asset' }, (args) => {
      if (args.path.includes('?worker'))
        return { contents: 'export default Worker', loader: 'js' }
      return { contents: `export default ${JSON.stringify(fs.readFileSync(args.path.split('?')[0], 'utf8'))}`, loader: 'js' }
    })
  } }] })
  return { name: 'candidate-source', code: compiled.outputFiles[0].text, coreCode }
}

export function danmukuCandidateEnvironment(implementation, settings = {}) {
  // Inject before module evaluation: source worker imports capture the constructor.
  const initialWorker = settings.initialWorkerError
    ? { ...implementation, code: `globalThis.Worker = class { constructor() { throw globalThis.__danmukuCandidateWorkerError } };\n${implementation.code}` }
    : implementation
  const env = danmukuEnvironment(initialWorker, settings)
  const createElement = env.context.document.createElement.bind(env.context.document)
  const removeChild = function (child) {
    const index = this.children.indexOf(child)
    if (index < 0)
      throw new Error('Cannot remove a node from a different parent')
    this.children.splice(index, 1)
    child.parentElement = null
    return child
  }
  env.context.document.createElement = (...args) => {
    const element = createElement(...args)
    element.removeChild = removeChild
    return element
  }
  for (const element of Object.values(env.art.template)) {
    if (element?.children)
      element.removeChild = removeChild
  }
  if (settings.initialWorkerError)
    env.context.__danmukuCandidateWorkerError = settings.initialWorkerError
  const consoleWarnings = []
  env.context.console.warn = (...args) => consoleWarnings.push(args)
  env.context.AbortController = AbortController
  env.context.AbortSignal = AbortSignal
  env.context.window.AbortController = AbortController
  env.context.window.AbortSignal = AbortSignal
  env.context.Worker.prototype.removeEventListener = function (name, callback) {
    this.listeners.set(name, (this.listeners.get(name) || []).filter(listener => listener !== callback))
  }
  const fetchCalls = []
  env.context.fetch = async (url, init) => {
    env.requests.push(url)
    fetchCalls.push({ url, init })
    if (settings.fetch)
      return settings.fetch(url, init)
    return { ok: true, status: 200, text: async () => '<i><d p="10,1,25,16777215,1,0,u,1">xml</d></i>' }
  }
  return { ...env, fetchCalls, consoleWarnings }
}