// useVolume.ts

import { ref } from 'vue'

/**
 * 實(shí)時(shí)獲取麥克風(fēng)音量
 * 此方法是通過后去音頻數(shù)據(jù)來計(jì)算出音量,并非系統(tǒng)實(shí)際音量
 * @param _audioContext 如果傳入AudioContext,則使用它,否則新建一個(gè)音頻來獲取音量大小
 * @returns
 */
export const useVolume = () => {
  const audioContext = ref<AudioContext | null>()
  const analyser = ref<AnalyserNode | null>(null)
  const isStart = ref<boolean>(false)
  // 音量[0, 100]
  const volume = ref<number>(0)

  /**
   * 開啟音量檢測(cè)
   * @param _source 數(shù)據(jù)源
   * @param _audioContext audio上下文 (用于在錄音的過程中同時(shí)去檢測(cè),這樣就不需要再次創(chuàng)建多一個(gè)audio上下文)
   * @param deviceId 麥克風(fēng)設(shè)備ID,用于指定檢測(cè)那個(gè)麥克風(fēng)的音量
   */
  const startVolumeCheck = async (
    _source?: MediaStreamAudioSourceNode | null,
    _audioContext?: AudioContext | null,
    deviceId?: string | null,
  ) => {
    if (!audioContext.value) {
      if (_audioContext) {
        audioContext.value = _audioContext
      } else {
        audioContext.value = new AudioContext()
      }
    }

    isStart.value = true
    const mediaOptions = deviceId
      ? { audio: { deviceId: deviceId } }
      : { audio: true }
    let source = _source
    if (!source) {
      const stream = await navigator.mediaDevices.getUserMedia(mediaOptions)
      source = audioContext.value.createMediaStreamSource(stream)
    }

    analyser.value = audioContext.value.createAnalyser()
    analyser.value!.fftSize = 32
    source.connect(analyser.value!)

    // 創(chuàng)建數(shù)據(jù)緩沖區(qū)
    const bufferLength = analyser.value!.frequencyBinCount
    const dataArray = new Uint8Array(bufferLength)

    // 定義更新音量的函數(shù)
    function updateVolume() {
      analyser.value!.getByteFrequencyData(dataArray)

      let sum = 0
      for (let i = 0; i < bufferLength; i++) {
        sum += dataArray[i]
      }
      volume.value = Math.floor((sum / bufferLength / 255) * 100)

      if (isStart.value) {
        requestAnimationFrame(updateVolume)
      } else {
        volume.value = 0
      }
    }

    // 開始更新音量
    updateVolume()
  }
  
  // 暫停檢測(cè)
  const stopVolumeCheck = () => {
    isStart.value = false
    analyser.value?.disconnect()
  }

  return {
    startVolumeCheck,
    stopVolumeCheck,
    volume,
    isStart,
  }
}