LevelModuleWaveMusic.cs

using System.Collections;
using ThunderRoad;
using UnityEngine;
using UnityEngine.Events;

namespace TOR {
    public class LevelModuleWaveMusic : LevelModule {
		public string musicWavePath;
		AudioContainer musicWaveAsset;
		Coroutine observe;

		GameObject obj;
        AudioSource audioSource;

        public string dynamicMusic = "";

        public override IEnumerator OnLoadCoroutine() {
            if (GlobalSettings.TORSoundtrack) {
                yield return Catalog.LoadAssetCoroutine(musicWavePath, delegate (AudioContainer value) {
                    musicWaveAsset = value;

                    obj = new GameObject("TOR_MusicPlayer");
                    audioSource = obj.AddComponent<AudioSource>();
                    audioSource.outputAudioMixerGroup = ThunderRoadSettings.GetAudioMixerGroup(AudioMixerName.Music);
                    audioSource.spatialBlend = 0;
                    audioSource.playOnAwake = false;

                    observe = level.StartCoroutine(Observe());
                }, "ModuleWaveMusic");
            } else {
                if (!Catalog.gameData.useDynamicMusic) yield break;

                if (level.options.TryGetValue(LevelOption.DynamicMusicId.Get(), out string musicOverride)) {
                    dynamicMusic = musicOverride;
                }
                if (!string.IsNullOrEmpty(dynamicMusic) && ThunderBehaviourSingleton<MusicManager>.HasInstance) {
                    MusicManager musicManager = ThunderBehaviourSingleton<MusicManager>.Instance;
                    musicManager.UnLoadMusic();
                    yield return musicManager.LoadMusics(dynamicMusic);
                    musicManager.Volume = 1f;
                    musicManager = null;
                }
            }
            yield break;
		}

		IEnumerator Observe() {
			var canExit = false;
			while (true) {
				yield return Utils.waitSeconds_01;

				foreach (var waveSpawner in WaveSpawner.instances) {
					var musicSelector = waveSpawner.gameObject.AddComponent<MusicSelector>();
					musicSelector.musicWaveAsset = musicWaveAsset;
					musicSelector.audioSource = audioSource;
					musicSelector.waveSpawner = waveSpawner;
					musicSelector.Setup();
					canExit = true;
				}
				if (canExit) {
					level.StopCoroutine(observe);
					yield break;
				}
			}
		}

		public override void OnUnload() {
            base.OnUnload();
            if (ThunderBehaviourSingleton<MusicManager>.HasInstance) {
                ThunderBehaviourSingleton<MusicManager>.Instance.UnLoadMusic();
            }
            if (musicWaveAsset) {
				Catalog.ReleaseAsset(musicWaveAsset);
			}
		}

		public class MusicSelector : ThunderBehaviour {
            public override ManagedLoops EnabledManagedLoops => ManagedLoops.Update;

            public WaveSpawner waveSpawner;
			public AudioSource audioSource;
			public AudioContainer musicWaveAsset;

            public void Setup() {
				waveSpawner.OnWaveBeginEvent.AddListener(new UnityAction(OnWaveBeginEvent));
				waveSpawner.OnWaveAnyEndEvent.AddListener(new UnityAction(OnWaveAnyEndEvent));
            }

			void OnWaveBeginEvent() {
                if (musicWaveAsset && audioSource) {
					audioSource.clip = musicWaveAsset.GetRandomAudioClip();
                    audioSource.loop = false;
					audioSource.Play();
				}
			}

			void OnWaveAnyEndEvent() {
				audioSource?.Stop();
			}

            protected override void ManagedUpdate() {
                if (waveSpawner.isRunning && musicWaveAsset && audioSource && !audioSource.isPlaying) {
					audioSource.clip = musicWaveAsset.GetRandomAudioClip();
                    audioSource.Play();
				}
			}
		}
	}
}