Musify/backend/Core/Musify.Infrastructure/MassTransit/Consumers/RoutingSlipCleanUpConsumer.cs

using MassTransit;
using MassTransit.Courier.Contracts;
using Microsoft.Extensions.Logging;

namespace Musify.Infrastructure.MassTransit.Consumers;

public class RoutingSlipCleanUpConsumer(
    ILogger<RoutingSlipCleanUpConsumer> logger) :
    IConsumer<RoutingSlipCompleted>,
    IConsumer<RoutingSlipFaulted>
{
    public const string QueueName = "routing-slip-clean-up";

    public Task Consume(ConsumeContext<RoutingSlipCompleted> consumeContext)
        => DeleteFolder(consumeContext.Message.Variables);

    public Task Consume(ConsumeContext<RoutingSlipFaulted> consumeContext)
        => DeleteFolder(consumeContext.Message.Variables);

    private Task DeleteFolder(IDictionary<string, object> variables)
    {
        if (!variables.TryGetValue(RoutingSlipVariableNames.Workflow.TemporalDirectory, out var pathObject))
            return Task.CompletedTask;

        var pathString = pathObject.ToString();

        if (string.IsNullOrEmpty(pathString))
            return Task.CompletedTask;

        try
        {
            Directory.Delete(pathString, recursive: true);
            logger.LogInformation("Deleted temporal directory at path {Path}", pathString);
        }
        catch (DirectoryNotFoundException directoryNotFoundException)
        {
            logger.LogWarning(directoryNotFoundException, "Temporal directory at path {Path} was not found for deletion", pathString);
        }
        catch (Exception exception)
        {
            logger.LogError(exception, "Failed to delete temporal directory at path {Path}", pathString);
            throw;
        }

        return Task.CompletedTask;
    }
}