Hugging Face's logo

Spaces:
Shadowfilebot
/
nodelink
Runtime error

App Files Community
nodelink
File size: 2,534 Bytes 
b58c6cb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
 
import config from '../../config.js'
import constants from '../../constants.js'

import prism from 'prism-media'

class NodeLinkStream {
  constructor(stream, pipes) {
    pipes.unshift(stream)

    for (let i = 0; i < pipes.length - 1; i++) {
      const pipe = pipes[i]

      pipe.pipe(pipes[i + 1])
    }

    this.stream = pipes[pipes.length - 1]

    this.listeners = []
    this.pipes = pipes
  }

  _end() {
    this.listeners.forEach(({ event, listener }) => this.stream.removeListener(event, listener))
    this.listeners = []

    if (this.stream) { 
      this.stream.destroy()
      this.stream = null
    }
  
    this.pipes.forEach((_, i) => {
      if (this.pipes[i].destroy) this.pipes[i].destroy()
      delete this.pipes[i]
    })
  }

  on(event, listener) {
    this.listeners.push({ event, listener })

    this.stream.on(event, listener)
  }

  once(event, listener) {
    this.listeners.push({ event, listener })

    this.stream.once(event, listener)
  }

  emit(event, ...args) {
    this.stream.emit(event, ...args)
  }

  read() {
    return this.stream?.read()
  }

  resume() {
    this.stream?.resume()
  }

  destroy() {
    this._end()
  }

  setVolume(volume) {
    this.pipes.find((pipe) => pipe instanceof prism.VolumeTransformer)?.setVolume(volume)
  }
}

function createAudioResource(stream, type) {
  if ([ 'webm/opus', 'ogg/opus' ].includes(type)) {
    return new NodeLinkStream(stream, [
      new prism.opus[type === 'webm/opus' ? 'WebmDemuxer' : 'OggDemuxer'](),
      new prism.opus.Decoder({ frameSize: 960, channels: 2, rate: 48000 }),
      new prism.VolumeTransformer({ type: 's16le' }),
      new prism.opus.Encoder({
        rate: constants.opus.samplingRate,
        channels: constants.opus.channels,
        frameSize: constants.opus.frameSize
      })
    ])
  }

  const ffmpeg = new prism.FFmpeg({
    args: [
      '-loglevel', '0',
      '-analyzeduration', '0',
      '-hwaccel', 'auto',
      '-threads', config.filters.threads,
      '-filter_threads', config.filters.threads,
      '-filter_complex_threads', config.filters.threads,
      '-i', '-',
      '-f', 's16le',
      '-ar', '48000',
      '-ac', '2',
      '-crf', '0'
    ]
  })

  return new NodeLinkStream(stream, [
    ffmpeg, 
    new prism.VolumeTransformer({ type: 's16le' }),
    new prism.opus.Encoder({
      rate: constants.opus.samplingRate,
      channels: constants.opus.channels,
      frameSize: constants.opus.frameSize
    })
  ])
}

export default {
  NodeLinkStream,
  createAudioResource
}