Контроль музыки, пробел

Assets/Code/Infrastructure/Audio/AudioService.cs

@@ -17,9 +17,11 @@ namespace Code.Infrastructure.Audio
             AudioSource audioSource = null);
 
         UniTaskVoid PlayRepeatedly(AudioNames groupName, int id,
-            CancellationTokenSource cancellationTokenSource, AudioSourceType sourceType = AudioSourceType.NoBlend, AudioSource audioSource = null);
+            AudioSourceType sourceType = AudioSourceType.NoBlend, AudioSource audioSource = null, CancellationTokenSource cancellationTokenSource = null);
 
         bool CheckClip(AudioNames groupName, int id);
+        void CancelMusic(AudioNames groupName, int id);
+        void ReleaseCancellationSource(AudioNames groupName, int id);
     }
 
     public partial class AudioService : IAudioClipsEventHandle, IAudioService
@@ -67,12 +69,22 @@ namespace Code.Infrastructure.Audio
             
             _pool.ReleaseSource(audioSource);
         }
-
+        
+        private Dictionary<KeyValuePair<AudioNames, int>, CancellationTokenSource> _cancellationTokenSources = new();
+        
+        public void CancelMusic(AudioNames groupName, int id) => _cancellationTokenSources[new KeyValuePair<AudioNames, int>(groupName, id)].Cancel();
+        public void ReleaseCancellationSource(AudioNames groupName, int id) => _cancellationTokenSources[new KeyValuePair<AudioNames, int>(groupName, id)] = null;
         public async UniTaskVoid PlayRepeatedly(AudioNames groupName, int id,
-            CancellationTokenSource cancellationTokenSource, AudioSourceType sourceType = AudioSourceType.NoBlend, AudioSource audioSource = null)
+            AudioSourceType sourceType = AudioSourceType.NoBlend, AudioSource audioSource = null, CancellationTokenSource cancellationTokenSource = null)
         {
             if (!CheckClip(groupName, id)) return;
             
+            if(cancellationTokenSource == null)
+            {
+                cancellationTokenSource = new CancellationTokenSource();
+                _cancellationTokenSources[new KeyValuePair<AudioNames, int>(groupName, id)] = cancellationTokenSource;
+            }   
+            
             if (!audioSource) audioSource = _pool.GetSource(sourceType);
             if(!audioSource.resource) audioSource.resource = GetClip(groupName, id);
             
@@ -90,10 +102,12 @@ namespace Code.Infrastructure.Audio
                 
                 _pool.ReleaseSource(audioSource);
                 
+                ReleaseCancellationSource(groupName,id);
+                
                 return;
             }
             
-            PlayRepeatedly(groupName, id, cancellationTokenSource, sourceType, audioSource).Forget();
+            PlayRepeatedly(groupName, id, sourceType, audioSource, cancellationTokenSource).Forget();
         }
 
         public AudioClip GetClip(AudioNames groupName, int id) => _audioClips[groupName][id];