Changes

2026-03-24 14:45:20 +01:00
parent 4f5230ddf6
commit 22a65b7efa
66 changed files with 1830 additions and 922 deletions
@@ -0,0 +1,100 @@
+using MassTransit;
+using Microsoft.Extensions.Logging;
+using Musify.Application.Contracts.Infrastructure;
+using Musify.Domain.Entities;
+using Musify.Infrastructure.Messaging.Activities.Arguments;
+using Musify.Infrastructure.Messaging.Activities.Logs;
+
+namespace Musify.Infrastructure.Messaging.Activities
+{
+    public class DownloadFileFromBucketActivity(
+        IStorageHandler storageHandler,
+        ILogger<DownloadFileFromBucketActivity> logger,
+        IProcessTrackingStore processTrackingStore)
+        : IActivity<DownloadFileFromBucketArguments, DownloadFileFromBucketLog>
+    {
+        public const string ExecuteEndpointName = "Download-File-From-Bucket";
+
+        public async Task<ExecutionResult> Execute(ExecuteContext<DownloadFileFromBucketArguments> executeContext)
+        {
+            var processId = await processTrackingStore.GetOrCreateProcessAsync(
+                "RoutingSlip",
+                executeContext.CorrelationId ?? executeContext.TrackingNumber,
+                executeContext.ConversationId,
+                executeContext.MessageId,
+                executeContext.CancellationToken);
+
+            var stepId = await processTrackingStore.StartStepAsync(
+                processId,
+                nameof(DownloadFileFromBucketActivity),
+                ProcessStepComponentType.Activity,
+                0,
+                executeContext.CancellationToken);
+
+            var destinationPath = executeContext.GetVariable<string>(executeContext.Arguments.DestinationFilePathVariableName);
+            if (string.IsNullOrWhiteSpace(destinationPath))
+            {
+                throw new InvalidOperationException(
+                    $"Download activity requires variable '{executeContext.Arguments.DestinationFilePathVariableName}' with destination file path.");
+            }
+
+            try
+            {
+                var getFile = await storageHandler.GetFileAsync(
+                    executeContext.Arguments.BucketName,
+                    executeContext.Arguments.KeyName,
+                    executeContext.CancellationToken);
+
+                if (!getFile.IsSuccess)
+                {
+                    var errorMessage = string.Join("; ", getFile.Errors);
+                    logger.LogWarning("Failed to download file from bucket. Bucket: {BucketName}, Key: {KeyName}, Errors: {Errors}",
+                        executeContext.Arguments.BucketName,
+                        executeContext.Arguments.KeyName,
+                        errorMessage);
+                    throw new Exception(errorMessage);
+                }
+
+                var destinationDirectory = Path.GetDirectoryName(destinationPath);
+                if (!string.IsNullOrWhiteSpace(destinationDirectory))
+                {
+                    Directory.CreateDirectory(destinationDirectory);
+                }
+
+                using var fileStream = getFile.Value;
+                if (fileStream.CanSeek)
+                    fileStream.Position = 0;
+
+                using var destinationStream = File.Create(destinationPath);
+                await fileStream.CopyToAsync(destinationStream, executeContext.CancellationToken);
+
+                await processTrackingStore.CompleteStepAsync(processId, stepId, executeContext.CancellationToken);
+                return executeContext.Completed();
+            }
+            catch (Exception exception)
+            {
+                logger.LogError(exception, "Error downloading file from bucket");
+                await processTrackingStore.FailStepAsync(processId, stepId, exception.Message, executeContext.CancellationToken);
+                throw;
+            }
+        }
+
+        public Task<CompensationResult> Compensate(CompensateContext<DownloadFileFromBucketLog> compensateContext)
+        {
+            try
+            {
+                if (File.Exists(compensateContext.Log.DestinationFilePath))
+                {
+                    File.Delete(compensateContext.Log.DestinationFilePath);
+                }
+
+                return Task.FromResult(compensateContext.Compensated());
+            }
+            catch (Exception exception)
+            {
+                logger.LogError(exception, "Error compensating downloaded file {DestinationFilePath}", compensateContext.Log.DestinationFilePath);
+                return Task.FromResult(compensateContext.Failed(exception));
+            }
+        }
+    }
+}