check_command.go

package resource

import (
	"fmt"
	"io"
	"sort"
	"strconv"
)

type CheckCommand struct {
	github GitHub
	writer io.Writer
}

func NewCheckCommand(github GitHub, writer io.Writer) *CheckCommand {
	return &CheckCommand{
		github: github,
		writer: writer,
	}
}

func (c *CheckCommand) Run(request CheckRequest) ([]Version, error) {
	fmt.Fprintln(c.writer, "getting deployments list")
	deployments, etag, err := c.github.ListDeployments(request.Version.ETag)

	if err != nil {
		return []Version{}, err
	}

	if etag != "" && etag == request.Version.ETag {
		return []Version{request.Version}, nil
	}

	var latestVersions []Version

	for _, deployment := range deployments {
		if len(request.Source.Environments) > 0 {
			found := false
			for _, env := range request.Source.Environments {
				if env == *deployment.Environment {
					found = true
				}
			}

			if !found {
				continue
			}
		}

		id := *deployment.ID
		lastID, err := strconv.ParseInt(request.Version.ID, 10, 64)
		if err != nil || id >= lastID {
			latestVersions = append(latestVersions, Version{
				ID:   strconv.FormatInt(id, 10),
				ETag: etag,
			})
		}
	}

	if len(latestVersions) == 0 {
		return []Version{}, nil
	}

	sort.Slice(latestVersions[:], func(i, j int) bool {
		iID, _ := strconv.Atoi(latestVersions[i].ID)
		jID, _ := strconv.Atoi(latestVersions[j].ID)
		return iID < jID
	})

	latestVersion := latestVersions[len(latestVersions)-1]

	if request.Version.ID == "" {
		return []Version{
			latestVersion,
		}, nil
	}

	return latestVersions, nil
}