mirror of
https://github.com/Radarr/Radarr
synced 2024-12-25 17:27:59 +00:00
52 lines
No EOL
1.7 KiB
C#
52 lines
No EOL
1.7 KiB
C#
using System;
|
|
using System.Linq;
|
|
using System.Collections.Generic;
|
|
using NzbDrone.Core.Model.Notification;
|
|
|
|
namespace NzbDrone.Core.Providers
|
|
{
|
|
class NotificationProvider : INotificationProvider
|
|
{
|
|
private readonly Dictionary<Guid, BasicNotification> _basicNotifications = new Dictionary<Guid, BasicNotification>();
|
|
private Dictionary<Guid, ProgressNotification> _progressNotification = new Dictionary<Guid, ProgressNotification>();
|
|
private readonly Object _lock = new object();
|
|
|
|
public void Register(ProgressNotification notification)
|
|
{
|
|
_progressNotification.Add(notification.Id, notification);
|
|
}
|
|
|
|
public void Register(BasicNotification notification)
|
|
{
|
|
_basicNotifications.Add(notification.Id, notification);
|
|
}
|
|
|
|
public List<BasicNotification> BasicNotifications
|
|
{
|
|
get { return new List<BasicNotification>(_basicNotifications.Values); }
|
|
}
|
|
|
|
public List<ProgressNotification> GetProgressNotifications
|
|
{
|
|
get
|
|
{
|
|
return new List<ProgressNotification>(_progressNotification.Values.Where(p => p.Status == ProgressNotificationStatus.InProgress));
|
|
}
|
|
}
|
|
|
|
public void Dismiss(Guid notificationId)
|
|
{
|
|
lock (_lock)
|
|
{
|
|
if (_basicNotifications.ContainsKey(notificationId))
|
|
{
|
|
_basicNotifications.Remove(notificationId);
|
|
}
|
|
else if (_progressNotification.ContainsKey(notificationId))
|
|
{
|
|
_progressNotification.Remove(notificationId);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
} |