using Big.Application;
using Big.Domain;

namespace Big.Tests;

/// <summary>An in-memory <see cref="IRegistrationStore"/> for the application-layer tests. Upserts
/// keyed on the registration id, mirroring the production store (kept distinct from the production
/// <c>InMemoryRegistrationStore</c> so tests can seed and inspect save counts).</summary>
internal sealed class FakeRegistrationStore : IRegistrationStore
{
    private readonly Dictionary<RegistrationId, Registration> _byId = [];

    public int SaveCount { get; private set; }

    public Task SaveAsync(Registration registration, CancellationToken ct = default)
    {
        SaveCount++;
        _byId[registration.Id] = registration;
        return Task.CompletedTask;
    }

    public Task<Registration?> GetAsync(RegistrationId id, CancellationToken ct = default)
        => Task.FromResult(_byId.GetValueOrDefault(id));

    public void Seed(Registration registration) => _byId[registration.Id] = registration;
}

/// <summary>A fake Workflow Client that records the registration it was asked to start a process for
/// and returns a fixed process-instance id. An optional callback runs at start time, letting a test
/// assert ordering (e.g. that the registration was persisted before the process started).</summary>
internal sealed class FakeWorkflowClient(string processInstanceId = "proc-1", Action<RegistrationId>? onStart = null)
    : IWorkflowClient
{
    public RegistrationId? StartedFor { get; private set; }

    public Task<string> StartRegistrationProcessAsync(RegistrationId registrationId, CancellationToken ct = default)
    {
        onStart?.Invoke(registrationId);
        StartedFor = registrationId;
        return Task.FromResult(processInstanceId);
    }
}

/// <summary>A fake ACL client that records the bsn it was asked to open a zaak for and returns a
/// fixed zaak URL.</summary>
internal sealed class FakeAclClient(Uri? zaakUrl = null) : IAclClient
{
    public static readonly Uri DefaultZaakUrl = new("http://openzaak/zaken/api/v1/zaken/abc");

    private readonly Uri _zaakUrl = zaakUrl ?? DefaultZaakUrl;

    public string? OpenedForBsn { get; private set; }
    public int CallCount { get; private set; }

    public Uri? ApprovedZaakUrl { get; private set; }
    public int ApproveCallCount { get; private set; }

    public Task<Uri> OpenZaakAsync(string bsn, CancellationToken ct = default)
    {
        CallCount++;
        OpenedForBsn = bsn;
        return Task.FromResult(_zaakUrl);
    }

    public Task ApproveZaakAsync(Uri zaakUrl, CancellationToken ct = default)
    {
        ApproveCallCount++;
        ApprovedZaakUrl = zaakUrl;
        return Task.CompletedTask;
    }
}