Radarr/NzbDrone.Core/Providers/NotificationProvider.cs

78 lines
2.5 KiB
C#
Raw Normal View History

using System;
using System.Collections.Generic;
2011-04-10 02:44:01 +00:00
using System.Linq;
using NzbDrone.Core.Model.Notification;
namespace NzbDrone.Core.Providers
{
2011-04-08 23:48:47 +00:00
public class NotificationProvider
{
private static readonly Dictionary<Guid, BasicNotification> _basicNotifications =
2011-04-10 02:44:01 +00:00
new Dictionary<Guid, BasicNotification>();
private static readonly Object _lock = new object();
private static readonly Dictionary<Guid, ProgressNotification> _progressNotification =
2011-04-10 02:44:01 +00:00
new Dictionary<Guid, ProgressNotification>();
2011-04-08 23:55:23 +00:00
public virtual List<BasicNotification> BasicNotifications
{
get { return new List<BasicNotification>(_basicNotifications.Values); }
}
public virtual List<ProgressNotification> ProgressNotifications
{
2010-11-06 00:43:13 +00:00
get
{
2011-07-02 17:41:10 +00:00
lock (_lock)
{
var activeNotification =
_progressNotification.Values.Where(p => p.Status == ProgressNotificationStatus.InProgress).
ToList();
2011-07-02 17:41:10 +00:00
if (activeNotification.Count == 0)
{
//Get notifications that were recently done
activeNotification =
_progressNotification.Values.Where(p => p.CompletedTime >= DateTime.Now.AddSeconds(-3)).
OrderByDescending(c => c.CompletedTime).ToList();
2011-07-02 17:41:10 +00:00
}
2011-07-02 17:41:10 +00:00
return activeNotification.ToList();
}
2010-11-06 00:43:13 +00:00
}
}
2011-04-10 02:44:01 +00:00
public virtual void Register(ProgressNotification notification)
{
2011-07-02 17:41:10 +00:00
lock (_lock)
{
_progressNotification.Add(notification.Id, notification);
}
2011-04-10 02:44:01 +00:00
}
public virtual void Register(BasicNotification notification)
{
2011-07-02 17:41:10 +00:00
lock (_lock)
{
_basicNotifications.Add(notification.Id, notification);
}
2011-04-10 02:44:01 +00:00
}
2011-04-08 23:55:23 +00:00
public virtual void Dismiss(Guid notificationId)
{
lock (_lock)
{
if (_basicNotifications.ContainsKey(notificationId))
{
_basicNotifications.Remove(notificationId);
}
else if (_progressNotification.ContainsKey(notificationId))
{
_progressNotification.Remove(notificationId);
}
}
}
}
}