[AC-2972] AC Team ownership: Events (#4647)

* Move Event domain under AC Team ownership
2025-07-02 00:22:50 -05:00 · 2024-08-19 08:00:17 +10:00
parent faa9afbe5e
commit 0230013b20
20 changed files with 2 additions and 0 deletions
--- a/src/Core/AdminConsole/Repositories/IEventRepository.cs
+++ b/src/Core/AdminConsole/Repositories/IEventRepository.cs
@ -0,0 +1,26 @@
+using Bit.Core.Models.Data;
+using Bit.Core.Vault.Entities;
+
+#nullable enable
+
+namespace Bit.Core.Repositories;
+
+public interface IEventRepository
+{
+    Task<PagedResult<IEvent>> GetManyByUserAsync(Guid userId, DateTime startDate, DateTime endDate,
+        PageOptions pageOptions);
+    Task<PagedResult<IEvent>> GetManyByOrganizationAsync(Guid organizationId, DateTime startDate, DateTime endDate,
+        PageOptions pageOptions);
+    Task<PagedResult<IEvent>> GetManyByOrganizationActingUserAsync(Guid organizationId, Guid actingUserId,
+        DateTime startDate, DateTime endDate, PageOptions pageOptions);
+    Task<PagedResult<IEvent>> GetManyByProviderAsync(Guid providerId, DateTime startDate, DateTime endDate,
+        PageOptions pageOptions);
+    Task<PagedResult<IEvent>> GetManyByProviderActingUserAsync(Guid providerId, Guid actingUserId,
+        DateTime startDate, DateTime endDate, PageOptions pageOptions);
+    Task<PagedResult<IEvent>> GetManyByCipherAsync(Cipher cipher, DateTime startDate, DateTime endDate,
+        PageOptions pageOptions);
+    Task CreateAsync(IEvent e);
+    Task CreateManyAsync(IEnumerable<IEvent> e);
+    Task<PagedResult<IEvent>> GetManyByOrganizationServiceAccountAsync(Guid organizationId, Guid serviceAccountId,
+        DateTime startDate, DateTime endDate, PageOptions pageOptions);
+}
--- a/src/Core/AdminConsole/Repositories/TableStorage/EventRepository.cs
+++ b/src/Core/AdminConsole/Repositories/TableStorage/EventRepository.cs
@ -0,0 +1,160 @@
+using Azure.Data.Tables;
+using Bit.Core.Models.Data;
+using Bit.Core.Settings;
+using Bit.Core.Utilities;
+using Bit.Core.Vault.Entities;
+
+#nullable enable
+
+namespace Bit.Core.Repositories.TableStorage;
+
+public class EventRepository : IEventRepository
+{
+    private readonly TableClient _tableClient;
+
+    public EventRepository(GlobalSettings globalSettings)
+        : this(globalSettings.Events.ConnectionString)
+    { }
+
+    public EventRepository(string storageConnectionString)
+    {
+        var tableClient = new TableServiceClient(storageConnectionString);
+        _tableClient = tableClient.GetTableClient("event");
+    }
+
+    public async Task<PagedResult<IEvent>> GetManyByUserAsync(Guid userId, DateTime startDate, DateTime endDate,
+        PageOptions pageOptions)
+    {
+        return await GetManyAsync($"UserId={userId}", "Date={{0}}", startDate, endDate, pageOptions);
+    }
+
+    public async Task<PagedResult<IEvent>> GetManyByOrganizationAsync(Guid organizationId,
+        DateTime startDate, DateTime endDate, PageOptions pageOptions)
+    {
+        return await GetManyAsync($"OrganizationId={organizationId}", "Date={0}", startDate, endDate, pageOptions);
+    }
+
+    public async Task<PagedResult<IEvent>> GetManyByOrganizationActingUserAsync(Guid organizationId, Guid actingUserId,
+        DateTime startDate, DateTime endDate, PageOptions pageOptions)
+    {
+        return await GetManyAsync($"OrganizationId={organizationId}",
+            $"ActingUserId={actingUserId}__Date={{0}}", startDate, endDate, pageOptions);
+    }
+
+    public async Task<PagedResult<IEvent>> GetManyByProviderAsync(Guid providerId,
+        DateTime startDate, DateTime endDate, PageOptions pageOptions)
+    {
+        return await GetManyAsync($"ProviderId={providerId}", "Date={0}", startDate, endDate, pageOptions);
+    }
+
+    public async Task<PagedResult<IEvent>> GetManyByProviderActingUserAsync(Guid providerId, Guid actingUserId,
+        DateTime startDate, DateTime endDate, PageOptions pageOptions)
+    {
+        return await GetManyAsync($"ProviderId={providerId}",
+            $"ActingUserId={actingUserId}__Date={{0}}", startDate, endDate, pageOptions);
+    }
+
+    public async Task<PagedResult<IEvent>> GetManyByCipherAsync(Cipher cipher, DateTime startDate, DateTime endDate,
+        PageOptions pageOptions)
+    {
+        var partitionKey = cipher.OrganizationId.HasValue ?
+            $"OrganizationId={cipher.OrganizationId}" : $"UserId={cipher.UserId}";
+        return await GetManyAsync(partitionKey, $"CipherId={cipher.Id}__Date={{0}}", startDate, endDate, pageOptions);
+    }
+
+    public async Task<PagedResult<IEvent>> GetManyByOrganizationServiceAccountAsync(Guid organizationId,
+        Guid serviceAccountId, DateTime startDate, DateTime endDate, PageOptions pageOptions)
+    {
+
+        return await GetManyAsync($"OrganizationId={organizationId}",
+            $"ServiceAccountId={serviceAccountId}__Date={{0}}", startDate, endDate, pageOptions);
+    }
+
+    public async Task CreateAsync(IEvent e)
+    {
+        if (!(e is EventTableEntity entity))
+        {
+            throw new ArgumentException(nameof(e));
+        }
+
+        await CreateEventAsync(entity);
+    }
+
+    public async Task CreateManyAsync(IEnumerable<IEvent>? e)
+    {
+        if (e is null || !e.Any())
+        {
+            return;
+        }
+
+        if (!e.Skip(1).Any())
+        {
+            await CreateAsync(e.First());
+            return;
+        }
+
+        var entities = e.OfType<EventTableEntity>();
+        var entityGroups = entities.GroupBy(ent => ent.PartitionKey);
+        foreach (var group in entityGroups)
+        {
+            var groupEntities = group.ToList();
+            if (groupEntities.Count == 1)
+            {
+                await CreateEventAsync(groupEntities.First());
+                continue;
+            }
+
+            // A batch insert can only contain 100 entities at a time
+            var iterations = groupEntities.Count / 100;
+            for (var i = 0; i <= iterations; i++)
+            {
+                var batch = new List<TableTransactionAction>();
+                var batchEntities = groupEntities.Skip(i * 100).Take(100);
+                if (!batchEntities.Any())
+                {
+                    break;
+                }
+
+                foreach (var entity in batchEntities)
+                {
+                    batch.Add(new TableTransactionAction(TableTransactionActionType.Add,
+                        entity.ToAzureEvent()));
+                }
+
+                await _tableClient.SubmitTransactionAsync(batch);
+            }
+        }
+    }
+
+    public async Task<PagedResult<IEvent>> GetManyAsync(string partitionKey, string rowKey,
+        DateTime startDate, DateTime endDate, PageOptions pageOptions)
+    {
+        var start = CoreHelpers.DateTimeToTableStorageKey(startDate);
+        var end = CoreHelpers.DateTimeToTableStorageKey(endDate);
+        var filter = MakeFilter(partitionKey, string.Format(rowKey, start), string.Format(rowKey, end));
+
+        var result = new PagedResult<IEvent>();
+        var query = _tableClient.QueryAsync<AzureEvent>(filter, pageOptions.PageSize);
+
+        await using (var enumerator = query.AsPages(pageOptions.ContinuationToken,
+            pageOptions.PageSize).GetAsyncEnumerator())
+        {
+            await enumerator.MoveNextAsync();
+
+            result.ContinuationToken = enumerator.Current.ContinuationToken;
+            result.Data.AddRange(enumerator.Current.Values.Select(e => e.ToEventTableEntity()));
+        }
+
+        return result;
+    }
+
+    private async Task CreateEventAsync(EventTableEntity entity)
+    {
+        await _tableClient.UpsertEntityAsync(entity.ToAzureEvent());
+    }
+
+    private string MakeFilter(string partitionKey, string rowStart, string rowEnd)
+    {
+        return $"PartitionKey eq '{partitionKey}' and RowKey le '{rowStart}' and RowKey ge '{rowEnd}'";
+    }
+}