notification hub push registration service

src/Core/Services/IDeviceService.cs

@@ -6,5 +6,7 @@ namespace Bit.Core.Services
     public interface IDeviceService
     {
         Task SaveAsync(Device device);
+        Task ClearTokenAsync(Device device);
+        Task DeleteAsync(Device device);
     }
 }

src/Core/Services/IPushRegistrationService.cs

@@ -0,0 +1,12 @@
+using System;
+using System.Threading.Tasks;
+using Bit.Core.Models.Table;
+
+namespace Bit.Core.Services
+{
+    public interface IPushRegistrationService
+    {
+        Task CreateOrUpdateRegistrationAsync(Device device);
+        Task DeleteRegistrationAsync(Guid deviceId);
+    }
+}

src/Core/Services/Implementations/DeviceService.cs

@@ -8,11 +8,14 @@ namespace Bit.Core.Services
     public class DeviceService : IDeviceService
     {
         private readonly IDeviceRepository _deviceRepository;
+        private readonly IPushRegistrationService _pushRegistrationService;
 
         public DeviceService(
-            IDeviceRepository deviceRepository)
+            IDeviceRepository deviceRepository,
+            IPushRegistrationService pushRegistrationService)
         {
             _deviceRepository = deviceRepository;
+            _pushRegistrationService = pushRegistrationService;
         }
 
         public async Task SaveAsync(Device device)
@@ -26,6 +29,20 @@ namespace Bit.Core.Services
                 device.RevisionDate = DateTime.UtcNow;
                 await _deviceRepository.ReplaceAsync(device);
             }
+
+            await _pushRegistrationService.CreateOrUpdateRegistrationAsync(device);
+        }
+
+        public async Task ClearTokenAsync(Device device)
+        {
+            await _deviceRepository.ClearPushTokenByIdentifierAsync(device.Identifier);
+            await _pushRegistrationService.DeleteRegistrationAsync(device.Id);
+        }
+
+        public async Task DeleteAsync(Device device)
+        {
+            await _deviceRepository.DeleteAsync(device);
+            await _pushRegistrationService.DeleteRegistrationAsync(device.Id);
         }
     }
 }

src/Core/Services/Implementations/NotificationHubPushRegistrationService.cs

@@ -0,0 +1,66 @@
+using System;
+using System.Linq;
+using System.Collections.Generic;
+using System.Threading.Tasks;
+using Microsoft.Azure.NotificationHubs;
+using Bit.Core.Models.Table;
+
+namespace Bit.Core.Services
+{
+    public class NotificationHubPushRegistrationService : IPushRegistrationService
+    {
+        private readonly NotificationHubClient _client;
+
+        public NotificationHubPushRegistrationService(GlobalSettings globalSettings)
+        {
+            _client = NotificationHubClient.CreateClientFromConnectionString(globalSettings.NotificationHub.ConnectionString,
+                globalSettings.NotificationHub.HubName);
+        }
+
+        public async Task CreateOrUpdateRegistrationAsync(Device device)
+        {
+            if(string.IsNullOrWhiteSpace(device.PushToken))
+            {
+                return;
+            }
+
+            var installation = new Installation
+            {
+                InstallationId = device.Id.ToString(),
+                PushChannel = device.PushToken
+            };
+
+            installation.Tags = new List<string>
+            {
+                "userId:" + device.UserId.ToString()
+            };
+
+            if(!string.IsNullOrWhiteSpace(device.Identifier))
+            {
+                installation.Tags.Add("identifier:" + device.Identifier);
+            }
+
+            switch(device.Type)
+            {
+                case Enums.DeviceType.Android:
+                    installation.Platform = NotificationPlatform.Gcm;
+                    break;
+                case Enums.DeviceType.iOS:
+                    installation.Platform = NotificationPlatform.Apns;
+                    break;
+                case Enums.DeviceType.AndroidAmazon:
+                    installation.Platform = NotificationPlatform.Adm;
+                    break;
+                default:
+                    break;
+            }
+
+            await _client.CreateOrUpdateInstallationAsync(installation);
+        }
+
+        public async Task DeleteRegistrationAsync(Guid deviceId)
+        {
+            await _client.DeleteInstallationAsync(deviceId.ToString());
+        }
+    }
+}

src/Core/Services/NoopImplementations/NoopPushRegistrationService.cs

@@ -0,0 +1,19 @@
+using System;
+using System.Threading.Tasks;
+using Bit.Core.Models.Table;
+
+namespace Bit.Core.Services
+{
+    public class NoopPushRegistrationService : IPushRegistrationService
+    {
+        public Task CreateOrUpdateRegistrationAsync(Device device)
+        {
+            return Task.FromResult(0);
+        }
+
+        public Task DeleteRegistrationAsync(Guid deviceId)
+        {
+            return Task.FromResult(0);
+        }
+    }
+}