discord.js-selfbot-v13/src/client/voice/receiver/Receiver.js

An unofficial discord.js fork for creating selfbots

'use strict';

const EventEmitter = require('events');
const prism = require('prism-media');
const PacketHandler = require('./PacketHandler');
const { Error } = require('../../../errors');
const { PCMInsertSilence } = require('../player/processing/PCMInsertSilence');

/**
 * Receives audio packets from a voice connection.
 * @example
 * const receiver = connection.createReceiver();
 * // opusStream is a ReadableStream - that means you could play it back to a voice channel if you wanted to!
 * const opusStream = receiver.createStream(user);
 */
class VoiceReceiver extends EventEmitter {
  constructor(connection) {
    super();
    this.connection = connection;
    this.packets = new PacketHandler(this);
    /**
     * Emitted whenever there is a warning
     * @event VoiceReceiver#debug
     * @param {Error|string} error The error or message to debug
     */
    this.packets.on('error', err => this.emit('debug', err));
  }

  /**
   * Options passed to `VoiceReceiver#createStream`.
   * @typedef {Object} ReceiveStreamOptions
   * @property {string} [mode='opus'] The mode for audio output. This defaults to opus, meaning discord.js won't decode
   * the packets for you. You can set this to 'pcm' so that the stream's output will be 16-bit little-endian stereo
   * audio
   * @property {string} [end='silence'] When the stream should be destroyed. If `silence`, this will be when the user
   * stops talking. Otherwise, if `manual`, this should be handled by you.
   * @property {boolean} [paddingSilence=false] Whether to add silence padding
   * If 'end' is set to 'silence', this property automatically defaults to `false`
   */

  /**
   * Creates a new audio receiving stream. If a stream already exists for a user, then that stream will be returned
   * rather than generating a new one.
   * @param {UserResolvable} user The user to start listening to.
   * @param {ReceiveStreamOptions} options Options.
   * @returns {ReadableStream}
   */
  createStream(user, { mode = 'opus', end = 'silence', paddingSilence = false } = {}) {
    user = this.connection.client.users.resolve(user);
    if (end === 'silence') paddingSilence = false;
    if (!user) throw new Error('VOICE_USER_MISSING');
    const stream = this.packets.makeStream(user.id, end); // Opus stream
    if (paddingSilence) {
      const decoder = new prism.opus.Decoder({ channels: 2, rate: 48000, frameSize: 960 });
      const pcmTransformer = new PCMInsertSilence();
      stream.pipe(decoder).pipe(pcmTransformer);
      if (mode === 'opus') {
        const encoder = new prism.opus.Encoder({ channels: 2, rate: 48000, frameSize: 960 });
        pcmTransformer.pipe(encoder);
        return encoder;
      }
      return pcmTransformer;
    } else {
      if (mode === 'pcm') {
        const decoder = new prism.opus.Decoder({ channels: 2, rate: 48000, frameSize: 960 });
        stream.pipe(decoder);
        return decoder;
      }
      return stream;
    }
  }

  /**
   * Creates a new video receiving stream. If a stream already exists for a user, then that stream will be returned
   * rather than generating a new one.
   * <info>Proof of concept - Requires a very good internet connection</info>
   * @param {UserResolvable} user The user to start listening to.
   * @param {WritableStream|string} output Output stream or file path to write the video stream to.
   * @returns {Recorder} The video stream for the specified user.
   */
  createVideoStream(user, output) {
    user = this.connection.client.users.resolve(user);
    if (!user) throw new Error('VOICE_USER_MISSING');
    const stream = this.packets.makeVideoStream(user.id, output);
    return stream;
  }

  /**
   * Emitted whenever there is a data packet received.
   * @event VoiceReceiver#receiverData
   * @param {{ userId: Snowflake, hasVideo: boolean }} ssrcData SSRC Data
   * @param {RtpPacket} header RTP Packet
   */
}

module.exports = VoiceReceiver;