UnrealEngineUWP/Engine/Source/Programs/UnrealToolbox/Notifications/Notifications.cs

// Copyright Epic Games, Inc. All Rights Reserved.

using Avalonia;
using Avalonia.Controls;
using Avalonia.Controls.ApplicationLifetimes;
using Avalonia.Platform;
using DesktopNotifications;
using DesktopNotifications.FreeDesktop;
using DesktopNotifications.Windows;
using Microsoft.Extensions.Logging;
using System;
using UnrealToolbox;

namespace UnrealToolbox
{
	/// <summary>
	/// Toolbox notification manager interface
	/// </summary>
	public interface IToolboxNotificationManager 
	{
		/// <summary>
		/// Show a notification
		/// </summary>
		/// <param name="title"></param>
		/// <param name="body"></param>
		/// <param name="force"></param>
		void ShowNotification(string title, string body, bool force = false);
	}

	/// <summary>
	/// Toolbox notification manager implementation
	/// </summary>
	class ToolboxNotificationManager : IToolboxNotificationManager, IDisposable
	{

		private INotificationManager? _platformManager;
		readonly ILogger _logger;

		// spam prevention
		DateTime? _lastNotificationTime;
		string? _lastTitle;
		string? _lastBody;

		public ToolboxNotificationManager(ILogger<ToolboxNotificationManager> logger)
		{
			_logger = logger;
		}

		public void Start()
		{
			if (Environment.OSVersion.Platform == PlatformID.Win32NT)
			{
				WindowsApplicationContext context = WindowsApplicationContext.FromCurrentProcess();
				_platformManager = new WindowsNotificationManager(context);
			}
			else
			{
				throw new NotImplementedException();
			}

			// initialize and ensure with result
			_platformManager.Initialize().GetAwaiter().GetResult();

		}

		public void Dispose()
		{
			_platformManager?.Dispose();
		}

		/// <summary>
		/// Show a notification, 
		/// </summary>
		/// <param name="title"></param>
		/// <param name="body"></param>
		/// <param name="force"></param>
		public void ShowNotification(string title, string body, bool force = false)
		{
			// spawn 
			if (!force && _lastNotificationTime != null && !String.IsNullOrEmpty(_lastBody) && !String.IsNullOrEmpty(_lastTitle)) 
			{
				TimeSpan deltaTime = DateTime.Now - _lastNotificationTime.Value; 

				// don't show a new notification if already displayed one in last minute
				if (deltaTime.TotalSeconds < 60)
				{
					return;
				}

				// if the title and body are the same, wait 2 minutes
				if ((deltaTime.TotalSeconds < 120) && title == _lastTitle && body == _lastBody)
				{
					return;
				}
			}

			_lastTitle = title;
			_lastBody = body;
			_lastNotificationTime = DateTime.Now;

			Notification notification = new Notification
			{
				Title = title,
				Body = body
			};

			_platformManager?.ShowNotification(notification, DateTimeOffset.Now + TimeSpan.FromSeconds(30));
		}
	}
}