agent_deployment.go

package main

import (
	"encoding/json"
	"errors"
	"fmt"
	"os"
	"path"

	"github.com/glasskube/distr/api"
	"github.com/glasskube/distr/internal/types"
	"github.com/google/uuid"
)

type AgentDeployment struct {
	ID          uuid.UUID        `json:"id"`
	RevisionID  uuid.UUID        `json:"revisionId"`
	ProjectName string           `json:"projectName"`
	DockerType  types.DockerType `json:"docker_type"`
}

func (d *AgentDeployment) FileName() string {
	return path.Join(agentDeploymentDir(), d.ID.String())
}

func agentDeploymentDir() string {
	return path.Join(ScratchDir(), "deployments")
}

func NewAgentDeployment(deployment api.DockerAgentDeployment) (*AgentDeployment, error) {
	if name, err := getProjectName(deployment.ComposeFile); err != nil {
		return nil, err
	} else {
		return &AgentDeployment{ID: deployment.ID, RevisionID: deployment.RevisionID, ProjectName: name, DockerType: deployment.DockerType}, nil
	}
}

func getProjectName(data []byte) (string, error) {
	if compose, err := DecodeComposeFile(data); err != nil {
		return "", err
	} else if name, ok := compose["name"].(string); !ok {
		return "", fmt.Errorf("name is not a string")
	} else {
		return name, nil
	}
}
func GetExistingDeployments() (map[uuid.UUID]AgentDeployment, error) {
	if entries, err := os.ReadDir(agentDeploymentDir()); err != nil {
		if errors.Is(err, os.ErrNotExist) {
			return nil, nil
		}
		return nil, err
	} else {
		fn := func(name string) (*AgentDeployment, error) {
			if file, err := os.Open(path.Join(agentDeploymentDir(), name)); err != nil {
				return nil, err
			} else {
				defer file.Close()
				var d AgentDeployment
				if err := json.NewDecoder(file).Decode(&d); err != nil {
					return nil, err
				}
				return &d, nil
			}
		}
		result := make(map[uuid.UUID]AgentDeployment, len(entries))
		for _, entry := range entries {
			if !entry.IsDir() {
				if d, err := fn(entry.Name()); err != nil {
					return nil, err
				} else {
					result[d.RevisionID] = *d
				}
			}
		}
		return result, nil
	}
}

func SaveDeployment(deployment AgentDeployment) error {
	if err := os.MkdirAll(path.Dir(deployment.FileName()), 0o700); err != nil {
		return err
	}

	file, err := os.Create(deployment.FileName())
	if err != nil {
		return err
	}
	defer file.Close()

	if err := json.NewEncoder(file).Encode(deployment); err != nil {
		return err
	}

	return nil
}

func DeleteDeployment(deployment AgentDeployment) error {
	return os.Remove(deployment.FileName())
}