main.lua

---@type string
local addonName = ...
---@class PingSounds
local addon = select(2, ...)
local asyncConfig = {
  type = "everyFrame",
  maxTime = 8,
  maxTimeCombat = 8,
  errorHandler = geterrorhandler()
}
addon.asyncHandler = LibStub("LibAsync"):GetHandler(asyncConfig)

function addon:Async(func, name, singleton)
  addon.asyncHandler:Async(func, name, singleton)
end

function addon:AddonPrint(msg)
  print("|cFF3275c3"..addonName.."|r: "..msg)
end

local f = CreateFrame("Frame")
f:RegisterEvent("PLAYER_ENTERING_WORLD")
f:RegisterEvent("PLAYER_LOGOUT")
f:RegisterEvent("CHAT_MSG_PING")

local reportLoadingFinished = false
local isLoadingFinished = false
local runOnce

f:SetScript("OnEvent", function(self, event, ...)
  if event == "PLAYER_ENTERING_WORLD" then
    local isInitialLogin, isReload = ...
    if not addon.db.installed then
      addon:CheckSoundChannelVolumes()
    end
    if isInitialLogin or not addon.db.installed then
      addon:Login()
    else
      C_CVar.SetCVar("Sound_EnableSFX", 1)
      isLoadingFinished = true
    end
    f:UnregisterEvent("PLAYER_ENTERING_WORLD")
    if not runOnce then
      addon:AddAddonSounds()
      addon:HookPlaySoundFile()
      f:RegisterEvent("CVAR_UPDATE")
      runOnce = true
    end
  elseif event == "CVAR_UPDATE" then
    local cvar, value = ...
    if cvar == "Sound_EnableSFX" and addon.db.enabled then
      addon:AddonPrint(
        "To Restore normal sound effect behavior, use /pingsounds disable and relog or disable the AddOn completely")
    end
  elseif event == "PLAYER_LOGOUT" then
    if addon.db.enabled then
      C_CVar.SetCVar("Sound_EnableSFX", 0)
      C_CVar.SetCVar("Sound_EnableAmbience", 0)
    end
    for i = 1, C_AddOns.GetNumAddOns() do
      local name = C_AddOns.GetAddOnInfo(i)
      if name == addonName then
        local enabled = C_AddOns.GetAddOnEnableState(i)
        if enabled == 0 then
          addon.db.installed = false
        end
      end
    end
  elseif event == "CHAT_MSG_PING" then
    if addon.db.enabled and not isLoadingFinished then
      addon:AddonPrint("Ping sounds will play after the loading process has finished.")
      reportLoadingFinished = true
    end
  end
end)

function addon:Login()
  if addon.db.enabled then
    C_Timer.After(2, function()
      addon:SetMuteStatus(true)
    end)
  end
end

local function round(number, decimals)
  return tonumber((("%%.%df"):format(decimals)):format(number))
end

function addon:SetMuteStatus(mute)
  f:UnregisterEvent("CVAR_UPDATE")
  C_CVar.SetCVar("Sound_EnableSFX", 0)
  C_CVar.SetCVar("Sound_EnableAmbience", 0)
  addon.db.installed = false
  local setMuteStatusAllAsync = function()
    for i = 1, 9999999 do
      if mute then
        MuteSoundFile(i)
      else
        ---@diagnostic disable-next-line: redundant-parameter
        UnmuteSoundFile(i)
      end
      if i % 100 == 0 then
        coroutine.yield()
      end
    end
  end

  local unMuteSounds = function(sounds)
    for _, id in pairs(sounds) do
      ---@diagnostic disable-next-line: redundant-parameter
      UnmuteSoundFile(id)
    end
  end

  addon:Async(function()
    local start = GetTime()
    setMuteStatusAllAsync()
    unMuteSounds(addon.sounds)
    local elapsed = round(GetTime() - start, 2)
    if mute then
      C_CVar.SetCVar("Sound_EnableSFX", 1)
      C_CVar.SetCVar("Sound_EnableAmbience", 1)
    end
    f:RegisterEvent("CVAR_UPDATE")
    if reportLoadingFinished then
      addon:AddonPrint("Loading finished after "..elapsed.."s")
    end
    isLoadingFinished = true
    addon.db.installed = true
  end, "Mute")
end

function addon:CheckSoundChannelVolumes()
  local soundChannels = {
    {
      name = "Sound_MasterVolume",
      volume = 0.25,
    },
    {
      name = "Sound_SFXVolume",
      volume = 1,
    }
  }
  for _, channel in pairs(soundChannels) do
    local volume = C_CVar.GetCVarInfo(channel.name)
    if volume == "0" then
      C_CVar.SetCVar(channel.name, channel.volume)
      addon:AddonPrint("Restored "..channel.name.." to "..channel.volume)
    end
  end
end