nhyatt/bitwarden
1
0
Fork 0
mirror of https://github.com/bitwarden/server.git synced 2025-07-01 08:02:49 -05:00
Code Activity

notification hub migrator util

Browse Source
This commit is contained in:
Kyle Spearrin
2017-05-29 23:35:41 -04:00
parent 335ecf73ae
commit c4dee589cc
3 changed files with 196 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
util/AnhMigrator/AnhMigrator.csproj Normal file
View File

@ -0,0 +1,18 @@
<Project Sdk="Microsoft.NET.Sdk">
<PropertyGroup>
<OutputType>Exe</OutputType>
<TargetFramework>net461</TargetFramework>
<RootNamespace>Bit.AnhMigrator</RootNamespace>
</PropertyGroup>
<ItemGroup Condition=" '$(TargetFramework)' == 'net461' ">
<Reference Include="System" />
<Reference Include="Microsoft.CSharp" />
</ItemGroup>
<ItemGroup>
<ProjectReference Include="..\..\src\Core\Core.csproj" />
</ItemGroup>
</Project>

166
util/AnhMigrator/Program.cs Normal file
View File

@ -0,0 +1,166 @@
using Bit.Core.Models.Table;
using Dapper;
using Microsoft.Azure.NotificationHubs;
using System;
using System.Collections.Generic;
using System.Data;
using System.Data.SqlClient;
using System.Linq;
using System.Threading.Tasks;
using Bit.Core.Enums;
using Newtonsoft.Json;
namespace Bit.AnhMigrator
{
class Program
{
private const string SqlConnectionString = "";
private const string AnhConnectionString = "";
private const string AnhHubName = "";
private static DateTime _startDate = DateTime.UtcNow.AddYears(-10);
private static string _viewId = "";
private static NotificationHubClient _client;
static void Main(string[] args)
{
_client = NotificationHubClient.CreateClientFromConnectionString(
AnhConnectionString,
AnhHubName);
//RegisterAsync(args).Wait();
//ViewAsync(args).Wait();
Console.Read();
}
private static async Task RegisterAsync(string[] args)
{
IDictionary<Guid, List<OrganizationUser>> orgUsersDict;
using(var connection = new SqlConnection(SqlConnectionString))
{
var results = await connection.QueryAsync<OrganizationUser>(
"SELECT * FROM [dbo].[OrganizationUser] WHERE [Status] = 2 AND [UserId] IS NOT NULL",
commandType: CommandType.Text);
orgUsersDict = results
.GroupBy(ou => ou.UserId.Value)
.ToDictionary(ou => ou.Key, ou => ou.ToList());
}
IEnumerable<Device> devices;
using(var connection = new SqlConnection(SqlConnectionString))
{
devices = await connection.QueryAsync<Device>(
"SELECT * FROM [dbo].[Device] WHERE [PushToken] IS NOT NULL AND [RevisionDate] > @StartDate",
new { StartDate = _startDate },
commandType: CommandType.Text);
}
var i = 0;
foreach(var device in devices)
{
i++;
var installation = new Installation
{
InstallationId = device.Id.ToString(),
PushChannel = device.PushToken,
Templates = new Dictionary<string, InstallationTemplate>()
};
installation.Tags = new List<string>
{
$"userId:{device.UserId}"
};
if(!string.IsNullOrWhiteSpace(device.Identifier))
{
installation.Tags.Add("deviceIdentifier:" + device.Identifier);
}
if(orgUsersDict.ContainsKey(device.UserId))
{
foreach(var orgUser in orgUsersDict[device.UserId])
{
installation.Tags.Add("organizationId:" + orgUser.OrganizationId);
}
}
string payloadTemplate = null, messageTemplate = null, badgeMessageTemplate = null;
switch(device.Type)
{
case DeviceType.Android:
payloadTemplate = "{\"data\":{\"type\":\"#(type)\",\"payload\":\"$(payload)\"}}";
messageTemplate = "{\"data\":{\"type\":\"#(type)\"}," +
"\"notification\":{\"title\":\"$(title)\",\"body\":\"$(message)\"}}";
installation.Platform = NotificationPlatform.Gcm;
break;
case DeviceType.iOS:
payloadTemplate = "{\"data\":{\"type\":\"#(type)\",\"payload\":\"$(payload)\"}," +
"\"aps\":{\"alert\":null,\"badge\":null,\"content-available\":1}}";
messageTemplate = "{\"data\":{\"type\":\"#(type)\"}," +
"\"aps\":{\"alert\":\"$(message)\",\"badge\":null,\"content-available\":1}}";
badgeMessageTemplate = "{\"data\":{\"type\":\"#(type)\"}," +
"\"aps\":{\"alert\":\"$(message)\",\"badge\":\"#(badge)\",\"content-available\":1}}";
installation.Platform = NotificationPlatform.Apns;
break;
case DeviceType.AndroidAmazon:
payloadTemplate = "{\"data\":{\"type\":\"#(type)\",\"payload\":\"$(payload)\"}}";
messageTemplate = "{\"data\":{\"type\":\"#(type)\",\"message\":\"$(message)\"}}";
installation.Platform = NotificationPlatform.Adm;
break;
default:
break;
}
BuildInstallationTemplate(installation, "payload", payloadTemplate, device.UserId, device.Identifier);
BuildInstallationTemplate(installation, "message", messageTemplate, device.UserId, device.Identifier);
BuildInstallationTemplate(installation, "badgeMessage", badgeMessageTemplate ?? messageTemplate, device.UserId,
device.Identifier);
await _client.CreateOrUpdateInstallationAsync(installation);
Console.WriteLine("Added install #" + i + " (" + installation.InstallationId + ")");
}
}
private static void BuildInstallationTemplate(Installation installation, string templateId, string templateBody,
Guid userId, string deviceIdentifier)
{
if(templateBody == null)
{
return;
}
var fullTemplateId = $"template:{templateId}";
var template = new InstallationTemplate
{
Body = templateBody,
Tags = new List<string>
{
fullTemplateId,
$"{fullTemplateId}_userId:{userId}"
}
};
if(!string.IsNullOrWhiteSpace(deviceIdentifier))
{
template.Tags.Add($"{fullTemplateId}_deviceIdentifier:{deviceIdentifier}");
}
installation.Templates.Add(fullTemplateId, template);
}
private static async Task ViewAsync(string[] args)
{
var install = await _client.GetInstallationAsync(_viewId);
var json = JsonConvert.SerializeObject(install, Formatting.Indented);
Console.WriteLine(json);
}
}
}