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

using MassTransit;
using MassTransit.Courier.Contracts;
using Microsoft.Extensions.Logging;
using Musify.Infrastructure.MassTransit.RoutingSlip;

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> context)
        => DeleteFolder(context.Message.Variables);

    public Task Consume(ConsumeContext<RoutingSlipFaulted> context)
        => DeleteFolder(context.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)
        {
            logger.LogWarning("Temporal directory at path {Path} was not found for deletion", pathString);
        }

        return Task.CompletedTask;
    }
}