package amazon

import (
	"errors"
	"strings"

	"github.com/aws/aws-sdk-go/service/ecs"
	"github.com/sirupsen/logrus"
)

func (c client) RegisterTaskDefinition(task *ecs.RegisterTaskDefinitionInput) (*string, error) {
	logrus.Debug("Register Task Definition")
	def, err := c.ECS.RegisterTaskDefinition(task)
	if err != nil {
		return nil, err
	}
	return def.TaskDefinition.TaskDefinitionArn, err
}

func (c client) CreateCluster() (*string, error) {
	logrus.Debug("Create cluster ", c.Cluster)
	response, err := c.ECS.CreateCluster(&ecs.CreateClusterInput{ClusterName: &c.Cluster})
	if err != nil {
		return nil, err
	}
	return response.Cluster.Status, nil
}

func (c client) DeleteCluster() error {
	logrus.Debug("Delete cluster ", c.Cluster)
	response, err := c.ECS.DeleteCluster(&ecs.DeleteClusterInput{Cluster: &c.Cluster})
	if err != nil {
		return err
	}
	if *response.Cluster.Status == "INACTIVE" {
		return nil
	}
	return errors.New("Failed to delete cluster, status: " + *response.Cluster.Status)
}

func (c client) ClusterExists() (bool, error) {
	logrus.Debug("Check if cluster was already created: ", c.Cluster)
	clusters, err := c.ECS.ListClusters(nil)
	if err != nil {
		return false, err
	}
	found := false
	for _, arn := range clusters.ClusterArns {
		found = found || strings.HasSuffix(*arn, "/"+c.Cluster)
	}
	return found, nil
}