using System.Threading;
using System.Threading.Tasks;
using StellaOps.Cli.Services.Models;

namespace StellaOps.Cli.Services;

/// <summary>
/// Client for orchestrator API operations.
/// Per CLI-ORCH-32-001.
/// </summary>
internal interface IOrchestratorClient
{
    /// <summary>
    /// Lists sources matching the query.
    /// </summary>
    Task<SourceListResponse> ListSourcesAsync(
        SourceListRequest request,
        CancellationToken cancellationToken);

    /// <summary>
    /// Gets a source by ID.
    /// </summary>
    Task<OrchestratorSource?> GetSourceAsync(
        string sourceId,
        string? tenant,
        CancellationToken cancellationToken);

    /// <summary>
    /// Pauses a source.
    /// </summary>
    Task<SourceOperationResult> PauseSourceAsync(
        SourcePauseRequest request,
        CancellationToken cancellationToken);

    /// <summary>
    /// Resumes a paused source.
    /// </summary>
    Task<SourceOperationResult> ResumeSourceAsync(
        SourceResumeRequest request,
        CancellationToken cancellationToken);

    /// <summary>
    /// Tests a source connection.
    /// </summary>
    Task<SourceTestResult> TestSourceAsync(
        SourceTestRequest request,
        CancellationToken cancellationToken);

    // CLI-ORCH-34-001: Backfill operations

    /// <summary>
    /// Starts a backfill operation for a source.
    /// </summary>
    Task<BackfillResult> StartBackfillAsync(
        BackfillRequest request,
        CancellationToken cancellationToken);

    /// <summary>
    /// Gets the status of a backfill operation.
    /// </summary>
    Task<BackfillResult?> GetBackfillAsync(
        string backfillId,
        string? tenant,
        CancellationToken cancellationToken);

    /// <summary>
    /// Lists backfill operations.
    /// </summary>
    Task<BackfillListResponse> ListBackfillsAsync(
        BackfillListRequest request,
        CancellationToken cancellationToken);

    /// <summary>
    /// Cancels a running backfill operation.
    /// </summary>
    Task<SourceOperationResult> CancelBackfillAsync(
        BackfillCancelRequest request,
        CancellationToken cancellationToken);

    // CLI-ORCH-34-001: Quota management

    /// <summary>
    /// Gets quotas for a tenant/source.
    /// </summary>
    Task<QuotaGetResponse> GetQuotasAsync(
        QuotaGetRequest request,
        CancellationToken cancellationToken);

    /// <summary>
    /// Sets a quota limit.
    /// </summary>
    Task<QuotaOperationResult> SetQuotaAsync(
        QuotaSetRequest request,
        CancellationToken cancellationToken);

    /// <summary>
    /// Resets a quota's usage counter.
    /// </summary>
    Task<QuotaOperationResult> ResetQuotaAsync(
        QuotaResetRequest request,
        CancellationToken cancellationToken);
}