| | 1 | | using Cysharp.Threading.Tasks; |
| | 2 | | using DCL; |
| | 3 | | using DCL.Components; |
| | 4 | | using DCL.Controllers; |
| | 5 | | using DCL.Models; |
| | 6 | | using DCL.Tasks; |
| | 7 | | using System; |
| | 8 | | using System.Collections.Generic; |
| | 9 | | using System.Threading; |
| | 10 | |
|
| | 11 | | namespace DCLPlugins.ECS6.HidePortableExperiencesUiFeatureToggle |
| | 12 | | { |
| | 13 | | public class HidePortableExperiencesUiController : IDisposable |
| | 14 | | { |
| | 15 | | private readonly IWorldState worldState; |
| | 16 | | private readonly BaseDictionary<int, bool> isSceneUiEnabled; |
| | 17 | | private readonly BaseHashSet<string> portableExperiencesIds; |
| 4 | 18 | | private CancellationTokenSource disableUiCancellationToken = new (); |
| | 19 | |
|
| 4 | 20 | | public HidePortableExperiencesUiController( |
| | 21 | | IWorldState worldState, |
| | 22 | | BaseDictionary<int, bool> isSceneUiEnabled, |
| | 23 | | BaseHashSet<string> portableExperiencesIds) |
| | 24 | | { |
| 4 | 25 | | this.worldState = worldState; |
| 4 | 26 | | this.isSceneUiEnabled = isSceneUiEnabled; |
| 4 | 27 | | this.portableExperiencesIds = portableExperiencesIds; |
| | 28 | |
|
| 4 | 29 | | isSceneUiEnabled.OnAdded += OnSceneUiVisibilityAdded; |
| 4 | 30 | | isSceneUiEnabled.OnSet += OnSceneUiVisibilitySet; |
| 4 | 31 | | portableExperiencesIds.OnAdded += OnPortableExperienceAdded; |
| 4 | 32 | | } |
| | 33 | |
|
| | 34 | | public void Dispose() |
| | 35 | | { |
| 4 | 36 | | isSceneUiEnabled.OnAdded -= OnSceneUiVisibilityAdded; |
| 4 | 37 | | isSceneUiEnabled.OnSet -= OnSceneUiVisibilitySet; |
| 4 | 38 | | portableExperiencesIds.OnAdded -= OnPortableExperienceAdded; |
| | 39 | |
|
| 4 | 40 | | disableUiCancellationToken.SafeCancelAndDispose(); |
| 4 | 41 | | } |
| | 42 | |
|
| | 43 | | private void OnSceneUiVisibilitySet(IEnumerable<KeyValuePair<int, bool>> scenesVisibility) |
| | 44 | | { |
| 0 | 45 | | foreach ((int sceneNumber, bool visible) in scenesVisibility) |
| 0 | 46 | | OnSceneUiVisibilityAdded(sceneNumber, visible); |
| 0 | 47 | | } |
| | 48 | |
|
| | 49 | | private void OnSceneUiVisibilityAdded(int sceneNumber, bool visible) |
| | 50 | | { |
| 5 | 51 | | IParcelScene currentScene = GetCurrentScene(); |
| | 52 | |
|
| 5 | 53 | | if (currentScene != null) |
| | 54 | | { |
| 5 | 55 | | if (visible |
| | 56 | | && currentScene.sceneData.scenePortableExperienceFeatureToggles == ScenePortableExperienceFeatureTog |
| | 57 | | { |
| 1 | 58 | | disableUiCancellationToken = disableUiCancellationToken.SafeRestart(); |
| 1 | 59 | | DisableUiVisibilityOnNextFrame(sceneNumber, disableUiCancellationToken.Token).Forget(); |
| 1 | 60 | | return; |
| | 61 | | } |
| | 62 | | } |
| | 63 | |
|
| 4 | 64 | | IParcelScene pxScene = worldState.GetScene(sceneNumber); |
| 4 | 65 | | if (pxScene == null) return; |
| 4 | 66 | | if (!pxScene.isPortableExperience) return; |
| | 67 | |
|
| 4 | 68 | | UIScreenSpace sceneUIComponent = pxScene.componentsManagerLegacy.GetSceneSharedComponent<UIScreenSpace>(); |
| | 69 | |
|
| 4 | 70 | | if (sceneUIComponent != null) |
| 4 | 71 | | sceneUIComponent.canvas.enabled = visible; |
| 4 | 72 | | } |
| | 73 | |
|
| | 74 | | private void OnPortableExperienceAdded(string pxId) |
| | 75 | | { |
| 1 | 76 | | IParcelScene currentScene = GetCurrentScene(); |
| | 77 | |
|
| 1 | 78 | | if (currentScene != null |
| | 79 | | && currentScene.sceneData.scenePortableExperienceFeatureToggles != ScenePortableExperienceFeatureToggles |
| 0 | 80 | | return; |
| | 81 | |
|
| 1 | 82 | | IParcelScene pxScene = worldState.GetPortableExperienceScene(pxId); |
| 1 | 83 | | if (pxScene == null) return; |
| | 84 | |
|
| 1 | 85 | | isSceneUiEnabled.AddOrSet(pxScene.sceneData.sceneNumber, false); |
| 1 | 86 | | } |
| | 87 | |
|
| | 88 | | private IParcelScene GetCurrentScene() => |
| 6 | 89 | | worldState.GetScene(worldState.GetCurrentSceneNumber()); |
| | 90 | |
|
| | 91 | | private async UniTaskVoid DisableUiVisibilityOnNextFrame(int sceneNumber, CancellationToken cancellationToken) |
| | 92 | | { |
| 3 | 93 | | await UniTask.NextFrame(cancellationToken: cancellationToken); |
| 1 | 94 | | isSceneUiEnabled.AddOrSet(sceneNumber, false); |
| 1 | 95 | | } |
| | 96 | | } |
| | 97 | | } |