// Override the startRecording method so we can pass the desired stream
// Checkout: https://github.com/katspaugh/wavesurfer.js/blob/fa2bcfe/src/plugins/record.ts

import RecordPlugin from "wavesurfer.js/dist/plugins/record";

const MIME_TYPES = [
  "audio/webm",
  "audio/wav",
  "audio/mpeg",
  "audio/mp4",
  "audio/mp3",
];
const findSupportedMimeType = () =>
  MIME_TYPES.find((mimeType) => MediaRecorder.isTypeSupported(mimeType));

class CustomRecordPlugin extends RecordPlugin {
  static create(options) {
    return new CustomRecordPlugin(options || {});
  }
  render(stream) {
    if (!this.wavesurfer) return () => undefined

    const container = this.wavesurfer.getWrapper()
    const canvas = document.createElement('canvas')
    canvas.width = container.clientWidth
    canvas.height = container.clientHeight
    canvas.style.zIndex = '10'
    container.appendChild(canvas)

    const canvasCtx = canvas.getContext('2d')
    const audioContext = new AudioContext()
    const source = audioContext.createMediaStreamSource(stream)
    const analyser = audioContext.createAnalyser()
    analyser.fftSize = 2 ** 5
    source.connect(analyser)
    const bufferLength = analyser.frequencyBinCount
    const dataArray = new Uint8Array(bufferLength)

    let animationId, previousTimeStamp;
    const BUFFER_SIZE = 2 ** 8
    const dataBuffer = new Array(BUFFER_SIZE).fill(canvas.height)

    const drawWaveform = (timeStamp) => {
      if (!canvasCtx) return

      analyser.getByteTimeDomainData(dataArray)
      canvasCtx.clearRect(0, 0, canvas.width, canvas.height)
      canvasCtx.fillStyle = 'black'

      if (previousTimeStamp === undefined) {
        previousTimeStamp = timeStamp
        dataBuffer.push(Math.min(...dataArray))
        dataBuffer.splice(0, 1)
      }
      const elapsed = timeStamp - previousTimeStamp;
      if (elapsed > 10) {
        previousTimeStamp = timeStamp
        dataBuffer.push(Math.min(...dataArray))
        dataBuffer.splice(0, 1)
      }

      // Drawing
      const sliceWidth = canvas.width / dataBuffer.length
      let x = 0

      for (let i = 0; i < dataBuffer.length; i++) {
        const valueNormalized = dataBuffer[i] / canvas.height
        const y = valueNormalized * canvas.height / 2
        const sliceHeight = canvas.height + 1 - y * 2

        canvasCtx.fillRect(x, y, sliceWidth * 2 / 3, sliceHeight)
        x += sliceWidth
      }

      animationId = requestAnimationFrame(drawWaveform)
    }

    drawWaveform()

    return () => {
      if (animationId) {
        cancelAnimationFrame(animationId)
      }

      if (source) {
        source.disconnect()
        source.mediaStream.getTracks().forEach((track) => track.stop())
      }

      if (audioContext) {
        audioContext.close()
      }

      canvas?.remove()
    }
  }
  startRecording(stream) {
    this.preventInteraction();
    this.cleanUp();

    const onStop = this.render(stream);
    const mediaRecorder = new MediaRecorder(stream, {
      mimeType: this.options.mimeType || findSupportedMimeType(),
      audioBitsPerSecond: this.options.audioBitsPerSecond,
    });
    const recordedChunks = [];

    mediaRecorder.addEventListener("dataavailable", (event) => {
      if (event.data.size > 0) {
        recordedChunks.push(event.data);
      }
    });

    mediaRecorder.addEventListener("stop", () => {
      onStop();
      this.loadBlob(recordedChunks, mediaRecorder.mimeType);
      this.emit("stopRecording");
    });

    mediaRecorder.start();

    this.emit("startRecording");

    this.mediaRecorder = mediaRecorder;
  }
}

export default CustomRecordPlugin;