package gateway

import (
	"context"
	"fmt"

	"github.com/railwayapp/cli/entity"
	"github.com/railwayapp/cli/errors"
	gqlgen "github.com/railwayapp/cli/lib/gql"
)

func (g *Gateway) GetDeploymentsForEnvironment(ctx context.Context, projectId, environmentId string) ([]*entity.Deployment, error) {
	gqlReq, err := g.NewRequestWithAuth(`
		query ($projectId: ID!, $environmentId: ID!) {
			allDeploymentsForEnvironment(projectId: $projectId, environmentId: $environmentId) {
				id
				status
				projectId
				meta
				staticUrl
			}
		}
	`)
	if err != nil {
		return nil, err
	}

	gqlReq.Var("projectId", projectId)
	gqlReq.Var("environmentId", environmentId)

	var resp struct {
		Deployments []*entity.Deployment `json:"allDeploymentsForEnvironment"`
	}
	if err := gqlReq.Run(ctx, &resp); err != nil {
		return nil, errors.DeploymentFetchingFailed
	}
	return resp.Deployments, nil
}

func (g *Gateway) GetLatestDeploymentForEnvironment(ctx context.Context, projectID, environmentID string) (*entity.Deployment, error) {
	deployments, err := g.GetDeploymentsForEnvironment(ctx, projectID, environmentID)
	if err != nil {
		return nil, err
	}
	if len(deployments) == 0 {
		return nil, errors.NoDeploymentsFound
	}
	for _, deploy := range deployments {
		if deploy.Status != entity.STATUS_REMOVED {
			return deploy, nil
		}
	}
	return nil, errors.NoDeploymentsFound
}

func (g *Gateway) GetDeploymentByID(ctx context.Context, req *entity.DeploymentByIDRequest) (*entity.Deployment, error) {
	gen, err := gqlgen.AsGQL(ctx, req.GQL)
	if err != nil {
		return nil, err
	}
	gqlReq, err := g.NewRequestWithAuth(fmt.Sprintf(`
		query ($projectId: ID!, $deploymentId: ID!) {
			deploymentById(projectId: $projectId, deploymentId: $deploymentId) {
				%s
			}
		}
	`, *gen))
	if err != nil {
		return nil, err
	}

	gqlReq.Var("projectId", req.ProjectID)
	gqlReq.Var("deploymentId", req.DeploymentID)

	var resp struct {
		Deployment *entity.Deployment `json:"deploymentById"`
	}
	if err := gqlReq.Run(ctx, &resp); err != nil {
		return nil, errors.DeploymentFetchingFailed
	}
	return resp.Deployment, nil
}