migu/common/modules/parser.js

import Metadata from 'matroska-metadata'
import Debug from 'debug'
import { arr2hex, hex2bin } from 'uint8-util'
import { fontRx } from './util.js'
import { SUPPORTS } from '@/modules/support.js'

const debug = Debug('torrent:parser')

export default class Parser {
  parsed = false
  /** @type {Metadata} */
  metadata = null
  client = null
  file = null
  destroyed = false

  constructor (client, file) {
    debug('Initializing parser for file: ' + file.name)
    this.client = client
    this.file = file
    this.metadata = new Metadata(file)

    this.metadata.getTracks().then(tracks => {
      if (this.destroyed) return
      debug('Tracks received: ' + tracks)
      if (!tracks.length) {
        this.parsed = true
        this.destroy()
      } else {
        this.client.dispatch('tracks', tracks)
      }
    })

    this.metadata.getChapters().then(chapters => {
      if (this.destroyed) return
      debug(`Found ${chapters.length} chapters`)
      this.client.dispatch('chapters', chapters)
    })

    this.metadata.getAttachments().then(files => {
      if (this.destroyed) return
      debug(`Found ${files.length} attachments`)
      for (const file of files) {
        if (fontRx.test(file.filename) || file.mimetype?.toLowerCase().includes('font')) {
          const data = hex2bin(arr2hex(file.data))
          if (SUPPORTS.isAndroid && data.length > 15_000_000) {
            debug('Skipping large font file on Android: ' + file.filename)
            continue
          }
          this.client.dispatch('file', data)
        }
      }
    })

    this.metadata.on('subtitle', (subtitle, trackNumber) => {
      if (this.destroyed) return
      debug(`Found subtitle for track: ${trackNumber}: ${subtitle.text}`)
      this.client.dispatch('subtitle', { subtitle, trackNumber })
    })

    if (this.file.name.endsWith('.mkv') || this.file.name.endsWith('.webm')) {
      this.file.on('iterator', ({ iterator }, cb) => {
        if (this.destroyed) return cb(iterator)
        cb(this.metadata.parseStream(iterator))
      })
    } else {
      debug('Unsupported file format: ' + this.file.name)
    }
  }

  destroy () {
    debug('Destroying Parser')
    this.destroyed = true
    // Add any additional cleanup code here
  }
}