api.go

package api

import (
	"github.com/caos/orbos/internal/git"
	"github.com/caos/orbos/internal/operator/common"
	"github.com/caos/orbos/internal/tree"
	"github.com/caos/orbos/mntr"
	"gopkg.in/yaml.v3"
)

const (
	orbiterFile = "orbiter.yml"
	boomFile    = "boom.yml"
	zitadelFile = "zitadel.yml"
)

type PushDesiredFunc func(monitor mntr.Monitor) error

func ExistsOrbiterYml(gitClient *git.Client) (bool, error) {
	return existsFileInGit(gitClient, orbiterFile)
}

func ReadOrbiterYml(gitClient *git.Client) (*tree.Tree, error) {
	return readFileInGit(gitClient, orbiterFile)
}

func PushOrbiterYml(monitor mntr.Monitor, msg string, gitClient *git.Client, desired *tree.Tree) (err error) {
	return pushFileInGit(monitor, msg, gitClient, desired, orbiterFile)
}

func PushOrbiterDesiredFunc(gitClient *git.Client, desired *tree.Tree) PushDesiredFunc {
	return func(monitor mntr.Monitor) error {
		monitor.Info("Writing orbiter desired state")
		return PushOrbiterYml(monitor, "Orbiter desired state written", gitClient, desired)
	}
}

func ExistsBoomYml(gitClient *git.Client) (bool, error) {
	return existsFileInGit(gitClient, boomFile)
}

func ReadBoomYml(gitClient *git.Client) (*tree.Tree, error) {
	return readFileInGit(gitClient, boomFile)
}

func PushBoomYml(monitor mntr.Monitor, msg string, gitClient *git.Client, desired *tree.Tree) (err error) {
	return pushFileInGit(monitor, msg, gitClient, desired, boomFile)
}

func PushBoomDesiredFunc(gitClient *git.Client, desired *tree.Tree) PushDesiredFunc {
	return func(monitor mntr.Monitor) error {
		monitor.Info("Writing boom desired state")
		return PushBoomYml(monitor, "Boom desired state written", gitClient, desired)
	}
}

func ExistsZitadelYml(gitClient *git.Client) (bool, error) {
	return existsFileInGit(gitClient, zitadelFile)
}

func ReadZitadelYml(gitClient *git.Client) (*tree.Tree, error) {
	return readFileInGit(gitClient, zitadelFile)
}

func PushZitadelYml(monitor mntr.Monitor, msg string, gitClient *git.Client, desired *tree.Tree) (err error) {
	return pushFileInGit(monitor, msg, gitClient, desired, zitadelFile)
}

func PushZitadelDesiredFunc(gitClient *git.Client, desired *tree.Tree) PushDesiredFunc {
	return func(monitor mntr.Monitor) error {
		monitor.Info("Writing zitadel desired state")
		return PushZitadelYml(monitor, "Zitadel desired state written", gitClient, desired)
	}
}

func pushFileInGit(monitor mntr.Monitor, msg string, gitClient *git.Client, desired *tree.Tree, path string) (err error) {
	monitor.OnChange = func(_ string, fields map[string]string) {
		err = gitClient.UpdateRemote(mntr.SprintCommit(msg, fields), git.File{
			Path:    path,
			Content: common.MarshalYAML(desired),
		})
		mntr.LogMessage(msg, fields)
	}
	monitor.Changed(msg)
	return err
}

func existsFileInGit(gitClient *git.Client, path string) (bool, error) {
	of := gitClient.Read(path)
	if of != nil && len(of) > 0 {
		return true, nil
	}
	return false, nil
}

func readFileInGit(gitClient *git.Client, path string) (*tree.Tree, error) {
	tree := &tree.Tree{}
	if err := yaml.Unmarshal(gitClient.Read(path), tree); err != nil {
		return nil, err
	}

	return tree, nil
}