using GraphQL;
using Speckle.Sdk.Api;

namespace Speckle.Automate.Sdk.Integration;

public struct FunctionRun
{
  public string StatusMessage { get; set; }
}

public struct AutomationRun
{
  public string Status { get; set; }
  public IList<FunctionRun> FunctionRuns { get; set; }
}

public struct AutomationStatus
{
  public string Status { get; set; }
  public IList<AutomationRun> AutomationRuns { get; set; }
}

public struct ModelAutomationStatus
{
  public AutomationStatus AutomationStatus { get; set; }
}

public struct ProjectAutomationStatus
{
  public ModelAutomationStatus Model { get; set; }
}

public struct AutomationStatusResponseModel
{
  public ProjectAutomationStatus Project { get; set; }
}

public static class AutomationStatusOperations
{
  public static async Task<AutomationStatus> Get(string projectId, string modelId, IClient speckleClient)
  {
    //language=graphql
    GraphQLRequest query = new(
      """
      query AutomationRuns($projectId: String!, $modelId: String!) {
        project(id: $projectId) {
          model(id: $modelId) {
            automationStatus{
              id
              status
              statusMessage
              automationRuns {
                id
                automationId
                versionId
                createdAt
                updatedAt
                status
                functionRuns {
                  id
                  functionId
                  elapsed
                  status
                  contextView
                  statusMessage
                  results
                  resultVersions {
                    id
                  }
                }
              }
            }
          }
        }
      }
      """,
      variables: new { projectId, modelId }
    );
    AutomationStatusResponseModel response = await speckleClient.ExecuteGraphQLRequest<AutomationStatusResponseModel>(
      query
    );
    return response.Project.Model.AutomationStatus;
  }
}