HomeScreenController.cs

using System.Collections.Generic;
using System.Linq;
using OneSignalDemo.Models;
using OneSignalDemo.Services;
using OneSignalDemo.UI.Dialogs;
using OneSignalDemo.UI.Sections;
using OneSignalDemo.ViewModels;
using UnityEngine;
using UnityEngine.SceneManagement;
using UnityEngine.UIElements;

namespace OneSignalDemo.UI
{
    public class HomeScreenController : MonoBehaviour
    {
        [SerializeField]
        private UIDocument _uiDocument;

        [SerializeField]
        private AppViewModel _viewModel;

        private VisualElement _root;
        private VisualElement _contentRoot;
        private VisualElement _loadingOverlay;
        private VisualElement _spinner;
        private IVisualElementScheduledItem _spinnerAnim;
        private LogViewController _logView;
        private ToastView _toastView;

        private AppSectionController _appSection;
        private UserSectionController _userSection;
        private PushSectionController _pushSection;
        private SendPushSectionController _sendPushSection;
        private InAppSectionController _inAppSection;
        private SendIamSectionController _sendIamSection;
        private AliasesSectionController _aliasesSection;
        private EmailsSectionController _emailsSection;
        private SmsSectionController _smsSection;
        private TagsSectionController _tagsSection;
        private OutcomesSectionController _outcomesSection;
        private TriggersSectionController _triggersSection;
        private TrackEventSectionController _trackEventSection;
        private LocationSectionController _locationSection;
        private LiveActivitiesSectionController _liveActivitiesSection;

        private void OnEnable()
        {
            ConfigureAndroidStatusBar();

            if (_viewModel == null)
            {
                _viewModel = FindAnyObjectByType<AppViewModel>();
            }

            _root = _uiDocument.rootVisualElement;
            _root.Clear();

            var themeSheet = Resources.Load<StyleSheet>("Theme");
            if (themeSheet != null)
                _root.styleSheets.Add(themeSheet);

            var logViewSheet = Resources.Load<StyleSheet>("LogView");
            if (logViewSheet != null)
                _root.styleSheets.Add(logViewSheet);

            BuildScreen();
            WireEvents();
        }

        private void OnDisable()
        {
            if (_viewModel != null)
            {
                _viewModel.OnStateChanged -= RefreshAll;
                _viewModel.OnToastMessage -= ShowToast;
            }
            _logView?.Destroy();
        }

        private void Update()
        {
            ApplySafeArea();
        }

        private static void ConfigureAndroidStatusBar()
        {
#if UNITY_ANDROID && !UNITY_EDITOR
            Screen.fullScreen = false;
            try
            {
                var unityPlayer = new AndroidJavaClass("com.unity3d.player.UnityPlayer");
                var activity = unityPlayer.GetStatic<AndroidJavaObject>("currentActivity");
                activity.Call(
                    "runOnUiThread",
                    new AndroidJavaRunnable(() =>
                    {
                        var window = activity.Call<AndroidJavaObject>("getWindow");
                        window.Call("addFlags", unchecked((int)0x80000000));
                        window.Call("clearFlags", 0x04000000);
                        window.Call("setStatusBarColor", unchecked((int)0xFFE54B4D));
                    })
                );
            }
            catch (System.Exception) { }
#endif
        }

        private void ApplySafeArea()
        {
            if (_root == null)
                return;

            float rootHeight = _root.resolvedStyle.height;
            if (float.IsNaN(rootHeight) || rootHeight <= 0 || Screen.height <= 0)
                return;

            var safe = Screen.safeArea;
            float scale = rootHeight / Screen.height;
            float top = (Screen.height - safe.yMax) * scale;
            float bottom = safe.y * scale;

            var screenRoot = _root.Q("screen_root");
            if (screenRoot != null)
                screenRoot.style.paddingBottom = bottom;
            var statusSpacer = _root.Q("status_bar_spacer");
            if (statusSpacer != null)
                statusSpacer.style.height = top;
        }

        private void BuildScreen()
        {
            var screenRoot = new VisualElement();
            screenRoot.name = "screen_root";
            screenRoot.AddToClassList("screen-root");

            var statusSpacer = new VisualElement();
            statusSpacer.name = "status_bar_spacer";
            statusSpacer.AddToClassList("status-bar-spacer");
            statusSpacer.style.height = 0;
            screenRoot.Add(statusSpacer);

            var appBar = new VisualElement();
            appBar.AddToClassList("app-bar");

            var logo = new VisualElement();
            logo.AddToClassList("app-bar-logo");
            var logoTexture = Resources.Load<Texture2D>("onesignal_logo");
            if (logoTexture != null)
                logo.style.backgroundImage = new StyleBackground(logoTexture);
            appBar.Add(logo);

            var appBarTitle = new Label("Unity");
            appBarTitle.AddToClassList("app-bar-title");
            appBar.Add(appBarTitle);

            screenRoot.Add(appBar);

            _logView = new LogViewController(screenRoot);

            var scrollView = new ScrollView(ScrollViewMode.Vertical);
            scrollView.AddToClassList("flex-grow");

            _contentRoot = new VisualElement();
            _contentRoot.AddToClassList("scroll-content");

            BuildSections();

            scrollView.Add(_contentRoot);
            screenRoot.Add(scrollView);

            _loadingOverlay = new VisualElement();
            _loadingOverlay.name = "loading_overlay";
            _loadingOverlay.AddToClassList("loading-overlay");
            _loadingOverlay.style.display = DisplayStyle.None;

            _spinner = new VisualElement();
            _spinner.AddToClassList("loading-spinner");
            _loadingOverlay.Add(_spinner);

            screenRoot.Add(_loadingOverlay);

            _toastView = new ToastView(screenRoot);

            _root.Add(screenRoot);
        }

        private void BuildSections()
        {
            _appSection = new AppSectionController(_viewModel);
            _contentRoot.Add(_appSection.Root);

            _userSection = new UserSectionController(_viewModel);
            _userSection.OnLoginTap = ShowLoginDialog;
            _userSection.OnLogoutTap = () => _viewModel.LogoutUser();
            _contentRoot.Add(_userSection.Root);

            _pushSection = new PushSectionController(_viewModel);
            _pushSection.OnInfoTap = () => ShowTooltip("push");
            _contentRoot.Add(_pushSection.Root);

            _sendPushSection = new SendPushSectionController(_viewModel);
            _sendPushSection.OnInfoTap = () => ShowTooltip("sendPushNotification");
            _sendPushSection.OnCustomTap = ShowCustomNotificationDialog;
            _contentRoot.Add(_sendPushSection.Root);

            _inAppSection = new InAppSectionController(_viewModel);
            _inAppSection.OnInfoTap = () => ShowTooltip("inAppMessaging");
            _contentRoot.Add(_inAppSection.Root);

            _sendIamSection = new SendIamSectionController(_viewModel);
            _sendIamSection.OnInfoTap = () => ShowTooltip("sendInAppMessage");
            _contentRoot.Add(_sendIamSection.Root);

            _aliasesSection = new AliasesSectionController(_viewModel);
            _aliasesSection.OnInfoTap = () => ShowTooltip("aliases");
            _aliasesSection.OnAddTap = ShowAddAliasDialog;
            _aliasesSection.OnAddMultipleTap = ShowAddMultipleAliasesDialog;
            _contentRoot.Add(_aliasesSection.Root);

            _emailsSection = new EmailsSectionController(_viewModel);
            _emailsSection.OnInfoTap = () => ShowTooltip("emails");
            _emailsSection.OnAddTap = ShowAddEmailDialog;
            _contentRoot.Add(_emailsSection.Root);

            _smsSection = new SmsSectionController(_viewModel);
            _smsSection.OnInfoTap = () => ShowTooltip("sms");
            _smsSection.OnAddTap = ShowAddSmsDialog;
            _contentRoot.Add(_smsSection.Root);

            _tagsSection = new TagsSectionController(_viewModel);
            _tagsSection.OnInfoTap = () => ShowTooltip("tags");
            _tagsSection.OnAddTap = ShowAddTagDialog;
            _tagsSection.OnAddMultipleTap = ShowAddMultipleTagsDialog;
            _tagsSection.OnRemoveSelectedTap = ShowRemoveSelectedTagsDialog;
            _contentRoot.Add(_tagsSection.Root);

            _outcomesSection = new OutcomesSectionController(_viewModel);
            _outcomesSection.OnInfoTap = () => ShowTooltip("outcomes");
            _outcomesSection.OnSendOutcomeTap = ShowOutcomeDialog;
            _contentRoot.Add(_outcomesSection.Root);

            _triggersSection = new TriggersSectionController(_viewModel);
            _triggersSection.OnInfoTap = () => ShowTooltip("triggers");
            _triggersSection.OnAddTap = ShowAddTriggerDialog;
            _triggersSection.OnAddMultipleTap = ShowAddMultipleTriggersDialog;
            _triggersSection.OnRemoveSelectedTap = ShowRemoveSelectedTriggersDialog;
            _contentRoot.Add(_triggersSection.Root);

            _trackEventSection = new TrackEventSectionController(_viewModel);
            _trackEventSection.OnInfoTap = () => ShowTooltip("trackEvent");
            _trackEventSection.OnTrackEventTap = ShowTrackEventDialog;
            _contentRoot.Add(_trackEventSection.Root);

            _locationSection = new LocationSectionController(_viewModel);
            _locationSection.OnInfoTap = () => ShowTooltip("location");
            _contentRoot.Add(_locationSection.Root);

#if UNITY_IOS
            _liveActivitiesSection = new LiveActivitiesSectionController(_viewModel);
            _liveActivitiesSection.OnInfoTap = () => ShowTooltip("liveActivities");
            _contentRoot.Add(_liveActivitiesSection.Root);
#endif

            var nextButton = SectionBuilder.CreatePrimaryButton(
                "NEXT SCREEN",
                "next_screen_button",
                () => SceneManager.LoadScene("Secondary")
            );
            nextButton.style.marginTop = 24;
            _contentRoot.Add(nextButton);
        }

        private void WireEvents()
        {
            _viewModel.OnStateChanged += RefreshAll;
            _viewModel.OnToastMessage += ShowToast;
        }

        private void RefreshAll()
        {
            _appSection?.Refresh();
            _userSection?.Refresh();
            _pushSection?.Refresh();
            _inAppSection?.Refresh();
            _aliasesSection?.Refresh();
            _emailsSection?.Refresh();
            _smsSection?.Refresh();
            _tagsSection?.Refresh();
            _triggersSection?.Refresh();
            _locationSection?.Refresh();
            _liveActivitiesSection?.Refresh();

            var showLoading = _viewModel.IsLoading;
            _loadingOverlay.style.display = showLoading ? DisplayStyle.Flex : DisplayStyle.None;

            if (showLoading && _spinnerAnim == null)
            {
                float angle = 0f;
                _spinnerAnim = _spinner
                    .schedule.Execute(() =>
                    {
                        angle = (angle + 12f) % 360f;
                        _spinner.style.rotate = new Rotate(Angle.Degrees(angle));
                    })
                    .Every(16);
            }
            else if (!showLoading && _spinnerAnim != null)
            {
                _spinnerAnim.Pause();
                _spinnerAnim = null;
            }
        }

        private void ShowToast(string message) => _toastView?.Show(message);

        private void ShowLoginDialog()
        {
            var dialog = new LoginDialog(
                externalId => _viewModel.LoginUser(externalId),
                _viewModel.IsLoggedIn
            );
            dialog.Show(_root);
        }

        private void ShowAddAliasDialog()
        {
            var dialog = new PairInputDialog(
                "Add Alias",
                "Label",
                "ID",
                "alias_key",
                "alias_value",
                "Add",
                (key, value) => _viewModel.AddAlias(key, value)
            );
            dialog.Show(_root);
        }

        private void ShowAddMultipleAliasesDialog()
        {
            var dialog = new MultiPairInputDialog(
                "Add Multiple Aliases",
                "Label",
                "ID",
                "Add all",
                pairs => _viewModel.AddAliases(pairs)
            );
            dialog.Show(_root);
        }

        private void ShowAddEmailDialog()
        {
            var dialog = new SingleInputDialog(
                "Add Email",
                "Email",
                "email_input",
                "Add",
                email => _viewModel.AddEmail(email)
            );
            dialog.Show(_root);
        }

        private void ShowAddSmsDialog()
        {
            var dialog = new SingleInputDialog(
                "Add SMS",
                "SMS",
                "sms_input",
                "Add",
                sms => _viewModel.AddSms(sms)
            );
            dialog.Show(_root);
        }

        private void ShowAddTagDialog()
        {
            var dialog = new PairInputDialog(
                "Add Tag",
                "Key",
                "Value",
                "tag_key",
                "tag_value",
                "Add",
                (key, value) => _viewModel.AddTag(key, value)
            );
            dialog.Show(_root);
        }

        private void ShowAddMultipleTagsDialog()
        {
            var dialog = new MultiPairInputDialog(
                "Add Multiple Tags",
                "Key",
                "Value",
                "Add all",
                pairs => _viewModel.AddTags(pairs)
            );
            dialog.Show(_root);
        }

        private void ShowRemoveSelectedTagsDialog()
        {
            var items = _viewModel.Tags.ToList();
            if (items.Count == 0)
                return;

            var dialog = new MultiSelectRemoveDialog(
                "Remove Tags",
                items,
                keys => _viewModel.RemoveSelectedTags(keys)
            );
            dialog.Show(_root);
        }

        private void ShowAddTriggerDialog()
        {
            var dialog = new PairInputDialog(
                "Add Trigger",
                "Key",
                "Value",
                "trigger_key",
                "trigger_value",
                "Add",
                (key, value) => _viewModel.AddTrigger(key, value)
            );
            dialog.Show(_root);
        }

        private void ShowAddMultipleTriggersDialog()
        {
            var dialog = new MultiPairInputDialog(
                "Add Multiple Triggers",
                "Key",
                "Value",
                "Add all",
                pairs => _viewModel.AddTriggers(pairs)
            );
            dialog.Show(_root);
        }

        private void ShowRemoveSelectedTriggersDialog()
        {
            var items = _viewModel.Triggers.ToList();
            if (items.Count == 0)
                return;

            var dialog = new MultiSelectRemoveDialog(
                "Remove Triggers",
                items,
                keys => _viewModel.RemoveSelectedTriggers(keys)
            );
            dialog.Show(_root);
        }

        private void ShowOutcomeDialog()
        {
            var dialog = new OutcomeDialog(
                (type, name, value) =>
                {
                    switch (type)
                    {
                        case OutcomeType.Normal:
                            _viewModel.SendOutcome(name);
                            break;
                        case OutcomeType.Unique:
                            _viewModel.SendUniqueOutcome(name);
                            break;
                        case OutcomeType.WithValue:
                            _viewModel.SendOutcomeWithValue(name, value);
                            break;
                    }
                }
            );
            dialog.Show(_root);
        }

        private void ShowTrackEventDialog()
        {
            var dialog = new TrackEventDialog((name, props) => _viewModel.TrackEvent(name, props));
            dialog.Show(_root);
        }

        private void ShowCustomNotificationDialog()
        {
            var dialog = new CustomNotificationDialog(
                (title, body) => _viewModel.SendCustomNotification(title, body)
            );
            dialog.Show(_root);
        }

        private void ShowTooltip(string key)
        {
            var tooltip = TooltipHelper.Instance.GetTooltip(key);
            if (tooltip != null)
            {
                var dialog = new TooltipDialog(tooltip);
                dialog.Show(_root);
            }
        }
    }
}