# SPDX-License-Identifier: GPL-2.0
# Copyright (C) 2018-present Frank Hartung (supervisedthinking@gmail.com)
# Copyright (C) 2022-present Fewtarius

. /etc/profile 

# Set common paths and defaults
export PULSE_RUNTIME_PATH=/run/pulse
	RR_AUDIO_DEVICE="hw:$(get_setting audio_device)"
	[ $RR_AUDIO_DEVICE = "hw:" ] && RR_PA_UDEV="true" || RR_PA_UDEV="false"
	echo "Set-Audio: Using audio device $RR_AUDIO_DEVICE"
    RR_PA_TSCHED="true"
    RR_AUDIO_VOLUME=$(get_setting "audio.volume" 2>/dev/null)
    MIXER="${DEVICE_AUDIO_MIXER}"
    RR_AUDIO_BACKEND="PulseAudio"
	

pulseaudio_sink_load() {

  if [ ${RR_AUDIO_BACKEND} = "PulseAudio" ];then
  systemctl restart pulseaudio 
    if [ "${RR_PA_TSCHED}" = "false" ]; then
      TSCHED="tsched=0"
      echo "Set-Audio: PulseAudio will disable timer-based audio scheduling"
    else
      TSCHED="tsched=1"
      echo "Set-Audio: PulseAudio will enable timer-based audio scheduling"
    fi

    if [ ! -z "$(pactl list modules short | grep module-null-sink)" ];then
      if [ "${RR_PA_UDEV}" = "true" ]; then
        pactl load-module module-udev-detect $TSCHED > /dev/null
        pactl set-sink-volume "$(pactl info | grep 'Default Sink:' | cut -d ' ' -f 3)" ${RR_AUDIO_VOLUME}%
        if [ ! -z "$(pactl list modules short | grep module-alsa-card)" ];then
          echo "Set-Audio: PulseAudio module-udev-detect loaded, setting a volume of "${RR_AUDIO_VOLUME}"%"
          echo "Set-Audio: PulseAudio will use sink "$(pactl list sinks short)
          echo "Set-Audio: PulseAudio module-udev-detect failed to load"
        fi
      else
        pactl load-module module-alsa-sink device="${RR_AUDIO_DEVICE}" name="temp_sink" ${TSCHED} > /dev/null
        pactl set-sink-volume alsa_output.temp_sink ${RR_AUDIO_VOLUME}%
        if [ ! -z "$(pactl list modules short | grep module-alsa-sink)" ];then
          echo "Set-Audio: PulseAudio module-alsa-sink loaded, setting a volume of "${RR_AUDIO_VOLUME}"%"
          echo "Set-Audio: PulseAudio will use sink "$(pactl list sinks short)
        else
          echo "Set-Audio: PulseAudio module-alsa-sink failed to load"
        fi
      fi
    fi
  fi
}

# Unload PulseAudio sink
pulseaudio_sink_unload() {
  
  if [ ${RR_AUDIO_BACKEND} = "PulseAudio" ]; then
    if [ "${RR_PA_UDEV}" = "true" ] && [ ! -z "$(pactl list modules short 2>/dev/null | grep module-alsa-card)" ]; then
      pactl set-sink-volume "$(pactl info 2>/dev/null | grep 'Default Sink:' | cut -d ' ' -f 3)" ${RR_AUDIO_VOLUME}% >/dev/null 2>&1 
      pactl unload-module module-udev-detect >/dev/null 2>&1
      pactl unload-module module-alsa-card >/dev/null 2>&1
      echo "Set-Audio: PulseAudio module-udev-detect unloaded"
    elif [ "${RR_PA_UDEV}" = "false" ] && [ ! -z "$(pactl list modules short 2>/dev/null| grep module-alsa-sink)" ]; then
      pactl set-sink-volume alsa_output.temp_sink ${RR_AUDIO_VOLUME}% >/dev/null 2>&1
      NUMBER="$(pactl list modules short 2>/dev/null | grep "name=temp_sink" | awk '{print $1;}')"
      if [ -n "${NUMBER}" ]; then
        pactl unload-module "${NUMBER}" >/dev/null 2>&1
      fi
      echo "Set-Audio: PulseAudio module-alsa-sink unloaded"
    else
      echo "Set-Audio: neither the PulseAudio module module-alsa-card or module-alsa-sink was found. Nothing to unload"
    fi

    # Restore ALSA Master volume to 100%
      amixer -M -q set ${MIXER},0 ${RR_AUDIO_VOLUME}% unmute >/dev/null 2>&1
      echo "Set-Audio: ALSA mixer restore volume to ${RR_AUDIO_VOLUME}%"
  fi
  systemctl stop pulseaudio >/dev/null 2>&1
}

# Start FluidSynth
fluidsynth_service_start() {
  
  if [ ${RR_AUDIO_BACKEND} = "PulseAudio" ] && [ ! "$(systemctl is-active fluidsynth)" = "active" ]; then
    systemctl start fluidsynth
    if [ "$(systemctl is-active fluidsynth)" = "active" ]; then 
      echo "Set-Audio: FluidSynth service loaded successfully"
    else
      echo "Set-Audio: FluidSynth service failed to load"
    fi
  fi
}

# Stop FluidSynth
fluidsynth_service_stop() {
  

  if [ "$(systemctl is-active fluidsynth)" = "active" ]; then
    systemctl stop fluidsynth
    if [ ! "$(systemctl is-active fluidsynth)" = "active" ]; then 
      echo "Set-Audio: FluidSynth service successfully stopped"
    else
      echo "Set-Audio: FluidSynth service failed to stop"
    fi
  fi
}

# SDL2: Set audio driver to Pulseaudio or ALSA
set_SDL_audiodriver() {
  
  if [ ${RR_AUDIO_BACKEND} = "PulseAudio" ]; then
    export SDL_AUDIODRIVER=pulseaudio
  else
    export SDL_AUDIODRIVER=alsa
  fi
  echo "Set-Audio: SDL2 set environment variable SDL_AUDIODRIVER="${SDL_AUDIODRIVER}
}

# RETROARCH: Set audio & midi driver
set_RA_audiodriver() {
  
  RETROARCH_HOME=/storage/.config/retroarch
  RETROARCH_CONFIG=${RETROARCH_HOME}/retroarch.cfg

  if [ -f ${RETROARCH_CONFIG} ]; then
    if [ ${RR_AUDIO_BACKEND} = "PulseAudio" ]; then
      sed -e "s/audio_driver = \"alsathread\"/audio_driver = \"pulse\"/" -i ${RETROARCH_CONFIG}
      sed -e "s/midi_driver = \"null\"/midi_driver = \"alsa\"/" -i          ${RETROARCH_CONFIG}
      sed -e "s/midi_output = \"Off\"/midi_output = \"FluidSynth\"/" -i     ${RETROARCH_CONFIG}
      echo "Set-Audio: Retroarch force audio driver to PulseAudio & MIDI output to FluidSynth"
    else
      sed -e "s/audio_driver = \"pulse\"/audio_driver = \"alsathread\"/" -i ${RETROARCH_CONFIG}
      sed -e "s/midi_driver = \"alsa\"/midi_driver = \"null\"/" -i          ${RETROARCH_CONFIG}
      sed -e "s/midi_output = \"FluidSynth\"/midi_output = \"Off\"/" -i     ${RETROARCH_CONFIG}
      echo "Set-Audio: Retroarch force audio driver to ALSA & disable MIDI output"
    fi
  fi
}

case "$1" in
	"pulseaudio")
		pulseaudio_sink_unload
		fluidsynth_service_stop
		pulseaudio_sink_load
	;;
	"fluidsynth")
		pulseaudio_sink_unload
		pulseaudio_sink_load
		fluidsynth_service_stop
		fluidsynth_service_start
	;;
	"alsa")
		pulseaudio_sink_unload
		fluidsynth_service_stop
		RR_AUDIO_BACKEND="alsa"
	;;
esac
		set_SDL_audiodriver