PM-31923 adding the whole report endpoints v2

src/Api/Dirt/Models/Request/AddOrganizationReportRequestModel.cs

@@ -0,0 +1,29 @@
+using System.Text.Json.Serialization;
+using Bit.Core.Dirt.Reports.ReportFeatures.Requests;
+
+namespace Bit.Api.Dirt.Models.Request;
+
+public class AddOrganizationReportRequestModel
+{
+    public string? ReportData { get; set; }
+    public string? ContentEncryptionKey { get; set; }
+    public string? SummaryData { get; set; }
+    public string? ApplicationData { get; set; }
+    [JsonPropertyName("metrics")]
+    public OrganizationReportMetrics? ReportMetrics { get; set; }
+    public long? FileSize { get; set; }
+
+    public AddOrganizationReportRequest ToData(Guid organizationId)
+    {
+        return new AddOrganizationReportRequest
+        {
+            OrganizationId = organizationId,
+            ReportData = ReportData,
+            ContentEncryptionKey = ContentEncryptionKey,
+            SummaryData = SummaryData,
+            ApplicationData = ApplicationData,
+            ReportMetrics = ReportMetrics,
+            FileSize = FileSize
+        };
+    }
+}

src/Api/Dirt/Models/Request/UpdateOrganizationReportApplicationDataRequestModel.cs

@@ -0,0 +1,18 @@
+using Bit.Core.Dirt.Reports.ReportFeatures.Requests;
+
+namespace Bit.Api.Dirt.Models.Request;
+
+public class UpdateOrganizationReportApplicationDataRequestModel
+{
+    public string? ApplicationData { get; set; }
+
+    public UpdateOrganizationReportApplicationDataRequest ToData(Guid organizationId, Guid reportId)
+    {
+        return new UpdateOrganizationReportApplicationDataRequest
+        {
+            OrganizationId = organizationId,
+            Id = reportId,
+            ApplicationData = ApplicationData
+        };
+    }
+}

src/Api/Dirt/Models/Request/UpdateOrganizationReportSummaryRequestModel.cs

@@ -0,0 +1,22 @@
+using System.Text.Json.Serialization;
+using Bit.Core.Dirt.Reports.ReportFeatures.Requests;
+
+namespace Bit.Api.Dirt.Models.Request;
+
+public class UpdateOrganizationReportSummaryRequestModel
+{
+    public string? SummaryData { get; set; }
+    [JsonPropertyName("metrics")]
+    public OrganizationReportMetrics? ReportMetrics { get; set; }
+
+    public UpdateOrganizationReportSummaryRequest ToData(Guid organizationId, Guid reportId)
+    {
+        return new UpdateOrganizationReportSummaryRequest
+        {
+            OrganizationId = organizationId,
+            ReportId = reportId,
+            SummaryData = SummaryData,
+            ReportMetrics = ReportMetrics
+        };
+    }
+}

src/Api/Dirt/Models/Request/UpdateOrganizationReportV2RequestModel.cs

@@ -0,0 +1,24 @@
+using Bit.Core.Dirt.Reports.ReportFeatures.Requests;
+
+namespace Bit.Api.Dirt.Models.Request;
+
+public class UpdateOrganizationReportV2RequestModel
+{
+    public required string ContentEncryptionKey { get; set; }
+    public required string SummaryData { get; set; }
+    public required string ApplicationData { get; set; }
+    public required OrganizationReportMetrics ReportMetrics { get; set; }
+
+    public UpdateOrganizationReportV2Request ToData(Guid organizationId, Guid reportId)
+    {
+        return new UpdateOrganizationReportV2Request
+        {
+            OrganizationId = organizationId,
+            ReportId = reportId,
+            ContentEncryptionKey = ContentEncryptionKey,
+            SummaryData = SummaryData,
+            ApplicationData = ApplicationData,
+            ReportMetrics = ReportMetrics
+        };
+    }
+}

src/Api/Dirt/Models/Response/OrganizationReportApplicationDataResponseModel.cs

@@ -0,0 +1,13 @@
+using Bit.Core.Dirt.Models.Data;
+
+namespace Bit.Api.Dirt.Models.Response;
+
+public class OrganizationReportApplicationDataResponseModel
+{
+    public OrganizationReportApplicationDataResponseModel(OrganizationReportApplicationDataResponse applicationDataResponse)
+    {
+        ApplicationData = applicationDataResponse.ApplicationData;
+    }
+
+    public string? ApplicationData { get; set; }
+}

src/Api/Dirt/Models/Response/OrganizationReportFileResponseModel.cs

@@ -0,0 +1,12 @@
+using Bit.Core.Enums;
+
+namespace Bit.Api.Dirt.Models.Response;
+
+public class OrganizationReportFileResponseModel
+{
+    public OrganizationReportFileResponseModel() { }
+
+    public string ReportFileUploadUrl { get; set; } = string.Empty;
+    public OrganizationReportResponseModel ReportResponse { get; set; } = null!;
+    public FileUploadType FileUploadType { get; set; }
+}

src/Api/Dirt/Models/Response/OrganizationReportResponseModel.cs

@@ -1,4 +1,6 @@
 using Bit.Core.Dirt.Entities;
+using Bit.Core.Dirt.Models.Data;
+using Bit.Core.Enums;
 
 namespace Bit.Api.Dirt.Models.Response;
 
@@ -10,11 +12,11 @@ public class OrganizationReportResponseModel
     public string? ContentEncryptionKey { get; set; }
     public string? SummaryData { get; set; }
     public string? ApplicationData { get; set; }
-    public int? PasswordCount { get; set; }
-    public int? PasswordAtRiskCount { get; set; }
-    public int? MemberCount { get; set; }
-    public DateTime? CreationDate { get; set; } = null;
-    public DateTime? RevisionDate { get; set; } = null;
+    public ReportFile? ReportFile { get; set; }
+    public string? ReportFileDownloadUrl { get; set; }
+    public FileUploadType? FileUploadType { get; set; }
+    public DateTime? CreationDate { get; set; }
+    public DateTime? RevisionDate { get; set; }
 
     public OrganizationReportResponseModel(OrganizationReport organizationReport)
     {
@@ -29,9 +31,7 @@ public OrganizationReportResponseModel(OrganizationReport organizationReport)
         ContentEncryptionKey = organizationReport.ContentEncryptionKey;
         SummaryData = organizationReport.SummaryData;
         ApplicationData = organizationReport.ApplicationData;
-        PasswordCount = organizationReport.PasswordCount;
-        PasswordAtRiskCount = organizationReport.PasswordAtRiskCount;
-        MemberCount = organizationReport.MemberCount;
+        ReportFile = organizationReport.GetReportFile();
         CreationDate = organizationReport.CreationDate;
         RevisionDate = organizationReport.RevisionDate;
     }

src/Api/Dirt/Models/Response/OrganizationReportSummaryDataResponseModel.cs

@@ -0,0 +1,23 @@
+using System.Text.Json.Serialization;
+using Bit.Core.Dirt.Models.Data;
+
+namespace Bit.Api.Dirt.Models.Response;
+
+public class OrganizationReportSummaryDataResponseModel
+{
+    public OrganizationReportSummaryDataResponseModel(OrganizationReportSummaryDataResponse summaryDataResponse)
+    {
+        EncryptedData = summaryDataResponse.SummaryData;
+        EncryptionKey = summaryDataResponse.ContentEncryptionKey;
+        Date = summaryDataResponse.RevisionDate;
+    }
+
+    [JsonPropertyName("encryptedData")]
+    public string EncryptedData { get; set; }
+
+    [JsonPropertyName("encryptionKey")]
+    public string EncryptionKey { get; set; }
+
+    [JsonPropertyName("date")]
+    public DateTime Date { get; set; }
+}

src/Core/Constants.cs

@@ -301,6 +301,7 @@ public static class FeatureFlagKeys
     public const string ArchiveVaultItems = "pm-19148-innovation-archive";
 
     /* DIRT Team */
+    public const string AccessIntelligenceVersion2 = "pm-31920-access-intelligence-azure-file-storage";
     public const string EventManagementForBlumira = "event-management-for-blumira";
     public const string EventManagementForDataDogAndCrowdStrike = "event-management-for-datadog-and-crowdstrike";
     public const string EventDiagnosticLogging = "pm-27666-siem-event-log-debugging";

src/Core/Dirt/Entities/OrganizationReport.cs

@@ -1,5 +1,5 @@
-#nullable enable
-
+using System.Text.Json;
+using Bit.Core.Dirt.Models.Data;
 using Bit.Core.Entities;
 using Bit.Core.Utilities;
 
@@ -29,6 +29,21 @@ public class OrganizationReport : ITableObject<Guid>
     public int? CriticalPasswordAtRiskCount { get; set; }
     public string? ReportFile { get; set; }
 
+    public ReportFile? GetReportFile()
+    {
+        if (string.IsNullOrWhiteSpace(ReportFile))
+        {
+            return null;
+        }
+
+        return JsonSerializer.Deserialize<ReportFile>(ReportFile);
+    }
+
+    public void SetReportFile(ReportFile data)
+    {
+        ReportFile = JsonSerializer.Serialize(data, JsonHelpers.IgnoreWritingNull);
+    }
+
     public void SetNewId()
     {
         Id = CoreHelpers.GenerateComb();

src/Core/Dirt/Models/Data/OrganizationReportMetricsData.cs

@@ -18,7 +18,7 @@ public class OrganizationReportMetricsData
     public int? CriticalPasswordCount { get; set; }
     public int? CriticalPasswordAtRiskCount { get; set; }
 
-    public static OrganizationReportMetricsData From(Guid organizationId, OrganizationReportMetricsRequest? request)
+    public static OrganizationReportMetricsData From(Guid organizationId, OrganizationReportMetrics? request)
     {
         if (request == null)
         {

src/Core/Dirt/Models/Data/ReportFile.cs

@@ -1,6 +1,4 @@
-#nullable enable
-
-using System.Diagnostics.CodeAnalysis;
+using System.Diagnostics.CodeAnalysis;
 using System.Text.Json.Serialization;
 using static System.Text.Json.Serialization.JsonNumberHandling;
 
@@ -28,5 +26,5 @@ public class ReportFile
     /// <summary>
     /// When true the uploaded file's length has been validated.
     /// </summary>
-    public bool Validated { get; set; } = true;
+    public bool Validated { get; set; } = false;
 }

src/Core/Dirt/Reports/ReportFeatures/AddOrganizationReportCommand.cs

@@ -41,7 +41,7 @@ public async Task<OrganizationReport> AddOrganizationReportAsync(AddOrganization
             throw new BadRequestException(errorMessage);
         }
 
-        var requestMetrics = request.Metrics ?? new OrganizationReportMetricsRequest();
+        var requestMetrics = request.ReportMetrics ?? new OrganizationReportMetrics();
 
         var organizationReport = new OrganizationReport
         {

src/Core/Dirt/Reports/ReportFeatures/CreateOrganizationReportCommand.cs

@@ -0,0 +1,121 @@
+using Bit.Core.Dirt.Entities;
+using Bit.Core.Dirt.Models.Data;
+using Bit.Core.Dirt.Reports.ReportFeatures.Interfaces;
+using Bit.Core.Dirt.Reports.ReportFeatures.Requests;
+using Bit.Core.Dirt.Repositories;
+using Bit.Core.Exceptions;
+using Bit.Core.Repositories;
+using Bit.Core.Utilities;
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.Logging;
+using ZiggyCreatures.Caching.Fusion;
+
+namespace Bit.Core.Dirt.Reports.ReportFeatures;
+
+public class CreateOrganizationReportCommand : ICreateOrganizationReportCommand
+{
+    private readonly IOrganizationRepository _organizationRepo;
+    private readonly IOrganizationReportRepository _organizationReportRepo;
+    private readonly ILogger<CreateOrganizationReportCommand> _logger;
+    private readonly IFusionCache _cache;
+
+    public CreateOrganizationReportCommand(
+        IOrganizationRepository organizationRepository,
+        IOrganizationReportRepository organizationReportRepository,
+        ILogger<CreateOrganizationReportCommand> logger,
+        [FromKeyedServices(OrganizationReportCacheConstants.CacheName)] IFusionCache cache)
+    {
+        _organizationRepo = organizationRepository;
+        _organizationReportRepo = organizationReportRepository;
+        _logger = logger;
+        _cache = cache;
+    }
+
+    public async Task<OrganizationReport> CreateAsync(AddOrganizationReportRequest request)
+    {
+        _logger.LogInformation(Constants.BypassFiltersEventId,
+            "Creating organization report for organization {organizationId}", request.OrganizationId);
+
+        var (isValid, errorMessage) = await ValidateRequestAsync(request);
+        if (!isValid)
+        {
+            _logger.LogInformation(Constants.BypassFiltersEventId,
+                "Failed to create organization {organizationId} report: {errorMessage}",
+                request.OrganizationId, errorMessage);
+            throw new BadRequestException(errorMessage);
+        }
+
+        var fileData = new ReportFile
+        {
+            Id = CoreHelpers.SecureRandomString(32, upper: false, special: false),
+            FileName = "report-data.json",
+            Size = request.FileSize ?? 0,
+            Validated = false
+        };
+
+        var organizationReport = new OrganizationReport
+        {
+            OrganizationId = request.OrganizationId,
+            CreationDate = DateTime.UtcNow,
+            ContentEncryptionKey = request.ContentEncryptionKey ?? string.Empty,
+            SummaryData = request.SummaryData,
+            ApplicationData = request.ApplicationData,
+            ApplicationCount = request.ReportMetrics?.ApplicationCount,
+            ApplicationAtRiskCount = request.ReportMetrics?.ApplicationAtRiskCount,
+            CriticalApplicationCount = request.ReportMetrics?.CriticalApplicationCount,
+            CriticalApplicationAtRiskCount = request.ReportMetrics?.CriticalApplicationAtRiskCount,
+            MemberCount = request.ReportMetrics?.MemberCount,
+            MemberAtRiskCount = request.ReportMetrics?.MemberAtRiskCount,
+            CriticalMemberCount = request.ReportMetrics?.CriticalMemberCount,
+            CriticalMemberAtRiskCount = request.ReportMetrics?.CriticalMemberAtRiskCount,
+            PasswordCount = request.ReportMetrics?.PasswordCount,
+            PasswordAtRiskCount = request.ReportMetrics?.PasswordAtRiskCount,
+            CriticalPasswordCount = request.ReportMetrics?.CriticalPasswordCount,
+            CriticalPasswordAtRiskCount = request.ReportMetrics?.CriticalPasswordAtRiskCount,
+            RevisionDate = DateTime.UtcNow
+        };
+        organizationReport.SetReportFile(fileData);
+
+        var data = await _organizationReportRepo.CreateAsync(organizationReport);
+
+        await _cache.RemoveByTagAsync(OrganizationReportCacheConstants.BuildCacheTagForOrganizationReports(request.OrganizationId));
+
+        _logger.LogInformation(Constants.BypassFiltersEventId,
+            "Successfully created organization report for organization {organizationId}, {organizationReportId}",
+            request.OrganizationId, data.Id);
+
+        return data;
+    }
+
+    private async Task<(bool IsValid, string errorMessage)> ValidateRequestAsync(
+        AddOrganizationReportRequest request)
+    {
+        var organization = await _organizationRepo.GetByIdAsync(request.OrganizationId);
+        if (organization == null)
+        {
+            return (false, "Invalid Organization");
+        }
+
+        if (string.IsNullOrWhiteSpace(request.ContentEncryptionKey))
+        {
+            return (false, "Content Encryption Key is required");
+        }
+
+        if (string.IsNullOrWhiteSpace(request.SummaryData))
+        {
+            return (false, "Summary Data is required");
+        }
+
+        if (string.IsNullOrWhiteSpace(request.ApplicationData))
+        {
+            return (false, "Application Data is required");
+        }
+
+        if (request.ReportMetrics == null)
+        {
+            return (false, "Report Metrics is required");
+        }
+
+        return (true, string.Empty);
+    }
+}