From 341ef0a943234ed35e50d8200ff00f5e143509a4 Mon Sep 17 00:00:00 2001 From: Kyle Spearrin Date: Tue, 28 Nov 2017 22:21:47 -0500 Subject: [PATCH] table storage event repo stub --- src/Core/Enums/EventType.cs | 14 ++ .../TableStorage/EventRepository.cs | 158 ++++++++++++++++++ src/Core/Utilities/CoreHelpers.cs | 10 ++ 3 files changed, 182 insertions(+) create mode 100644 src/Core/Enums/EventType.cs create mode 100644 src/Core/Repositories/TableStorage/EventRepository.cs diff --git a/src/Core/Enums/EventType.cs b/src/Core/Enums/EventType.cs new file mode 100644 index 000000000..756ce2a14 --- /dev/null +++ b/src/Core/Enums/EventType.cs @@ -0,0 +1,14 @@ +namespace Bit.Core.Enums +{ + public enum EventType : int + { + User_LoggedIn = 1000, + User_ChangedPassword = 1001, + User_Enabled2fa = 1002, + User_Disabled2fa = 1003, + + Cipher_Created = 2000, + Cipher_Edited = 2001, + Cipher_Deleted = 2002 + } +} diff --git a/src/Core/Repositories/TableStorage/EventRepository.cs b/src/Core/Repositories/TableStorage/EventRepository.cs new file mode 100644 index 000000000..d5a6929dc --- /dev/null +++ b/src/Core/Repositories/TableStorage/EventRepository.cs @@ -0,0 +1,158 @@ +using System; +using System.Collections.Generic; +using System.Linq; +using System.Threading.Tasks; +using Bit.Core.Enums; +using Bit.Core.Models.Table; +using Bit.Core.Utilities; +using Microsoft.WindowsAzure.Storage; +using Microsoft.WindowsAzure.Storage.Table; + +namespace Bit.Core.Repositories.TableStorage +{ + public class EventRepository + { + public EventRepository(GlobalSettings globalSettings) + { + var storageAccount = CloudStorageAccount.Parse(globalSettings.Storage.ConnectionString); + var tableClient = storageAccount.CreateCloudTableClient(); + Table = tableClient.GetTableReference("event"); + } + + protected CloudTable Table { get; set; } + + public ICollection GetManyByUser(Guid userId, DateTime startDate, DateTime endDate) + { + var start = CoreHelpers.DateTimeToTableStorageKey(startDate); + var end = CoreHelpers.DateTimeToTableStorageKey(endDate); + + return Table.CreateQuery().Where(e => + e.PartitionKey == $"UserId={userId}" && + e.RowKey.CompareTo($"{start}_") >= 0 && + e.RowKey.CompareTo($"{end}`") <= 0).ToList(); + } + + public ICollection GetManyByOrganization(Guid organizationId, DateTime startDate, DateTime endDate) + { + var start = CoreHelpers.DateTimeToTableStorageKey(startDate); + var end = CoreHelpers.DateTimeToTableStorageKey(endDate); + + return Table.CreateQuery().Where(e => + e.PartitionKey == $"OrganizationId={organizationId}" && + e.RowKey.CompareTo($"{start}_") >= 0 && + e.RowKey.CompareTo($"{end}`") <= 0).ToList(); + } + + public async Task CreateAsync(ITableEntity entity) + { + await Table.ExecuteAsync(TableOperation.Insert(entity)); + } + + public void CreateManyAsync(IEnumerable entities) + { + if(!entities?.Any() ?? true) + { + return; + } + + // A batch insert can only contain 100 entities at a time + var iterations = entities.Count() / 100; + for(var i = 0; i <= iterations; i++) + { + var batch = new TableBatchOperation(); + var batchEntities = entities.Skip(i * 100).Take(100); + if(!batchEntities.Any()) + { + break; + } + + foreach(var entity in batchEntities) + { + batch.InsertOrReplace(entity); + } + + Table.ExecuteBatch(batch); + } + } + } + + public class UserEvent : EventTableEntiity + { + public UserEvent(Guid userId, EventType type) + { + PartitionKey = $"UserId={userId}"; + RowKey = string.Format("Date={0}__Type={1}", + CoreHelpers.DateTimeToTableStorageKey(), type); + + UserId = userId; + Type = type; + } + + public UserEvent(Guid userId, Guid organizationId, EventType type) + { + PartitionKey = $"OrganizationId={organizationId}"; + RowKey = string.Format("Date={0}__UserId={1}__Type={2}", + CoreHelpers.DateTimeToTableStorageKey(), userId, type); + + OrganizationId = organizationId; + UserId = userId; + Type = type; + } + } + + public class CipherEvent : EventTableEntiity + { + public CipherEvent(Cipher cipher, EventType type) + { + if(cipher.OrganizationId.HasValue) + { + PartitionKey = $"OrganizationId={cipher.OrganizationId.Value}"; + } + else + { + PartitionKey = $"UserId={cipher.UserId.Value}"; + } + + RowKey = string.Format("Date={0}__CipherId={1}__Type={2}", + CoreHelpers.DateTimeToTableStorageKey(), cipher.Id, type); + + OrganizationId = cipher.OrganizationId; + UserId = cipher.UserId; + CipherId = cipher.Id; + Type = type; + } + } + + public class OrganizationEvent : EventTableEntiity + { + public OrganizationEvent(Guid organizationId, EventType type) + { + PartitionKey = $"OrganizationId={organizationId}"; + RowKey = string.Format("Date={0}__Type={1}", + CoreHelpers.DateTimeToTableStorageKey(), type); + + OrganizationId = organizationId; + Type = type; + } + + public OrganizationEvent(Guid organizationId, Guid userId, EventType type) + { + PartitionKey = $"OrganizationId={organizationId}"; + RowKey = string.Format("Date={0}__UserId={1}__Type={2}", + CoreHelpers.DateTimeToTableStorageKey(), userId, type); + + OrganizationId = organizationId; + UserId = userId; + Type = type; + } + } + + public class EventTableEntiity : TableEntity + { + public EventType Type { get; set; } + public Guid? UserId { get; set; } + public Guid? OrganizationId { get; set; } + public Guid? CipherId { get; set; } + public ICollection CipherIds { get; set; } + } +} diff --git a/src/Core/Utilities/CoreHelpers.cs b/src/Core/Utilities/CoreHelpers.cs index ad90b0354..486adfc6d 100644 --- a/src/Core/Utilities/CoreHelpers.cs +++ b/src/Core/Utilities/CoreHelpers.cs @@ -409,5 +409,15 @@ namespace Bit.Core.Utilities .Replace("http://", string.Empty) .Replace("https://", string.Empty); } + + public static string DateTimeToTableStorageKey(DateTime? date = null) + { + if(date == null) + { + date = DateTime.UtcNow; + } + + return date.Value.ToString("yyyy-MM-dd HH:mm:ss.fff"); + } } }