fleet/cmd/fleetctl/gitops.go

package main

import (
	"errors"
	"fmt"
	"github.com/fleetdm/fleet/v4/pkg/spec"
	"github.com/urfave/cli/v2"
	"os"
	"path/filepath"
)

func gitopsCommand() *cli.Command {
	var (
		flFilename string
		flDryRun   bool
	)
	return &cli.Command{
		Name:      "gitops",
		Usage:     "Synchronize Fleet configuration with provided file. This command is intended to be used in a GitOps workflow.",
		UsageText: `fleetctl gitops [options]`,
		Flags: []cli.Flag{
			&cli.StringFlag{
				Name:        "f",
				EnvVars:     []string{"FILENAME"},
				Value:       "",
				Destination: &flFilename,
				Usage:       "The file with the GitOps configuration",
			},
			&cli.BoolFlag{
				Name:        "dry-run",
				EnvVars:     []string{"DRY_RUN"},
				Destination: &flDryRun,
				Usage:       "Do not apply the file, just validate it",
			},
			configFlag(),
			contextFlag(),
			debugFlag(),
		},
		Action: func(c *cli.Context) error {
			if flFilename == "" {
				return errors.New("-f must be specified")
			}
			b, err := os.ReadFile(flFilename)
			if err != nil {
				return err
			}
			fleetClient, err := clientFromCLI(c)
			if err != nil {
				return err
			}
			baseDir := filepath.Dir(flFilename)
			config, err := spec.GitOpsFromBytes(b, baseDir)
			if err != nil {
				return err
			}
			logf := func(format string, a ...interface{}) {
				_, _ = fmt.Fprintf(c.App.Writer, format, a...)
			}
			err = fleetClient.DoGitOps(c.Context, config, baseDir, logf, flDryRun)
			if err != nil {
				return err
			}
			if flDryRun {
				_, _ = fmt.Fprintf(c.App.Writer, "[!] gitops dry run succeeded\n")
			} else {
				_, _ = fmt.Fprintf(c.App.Writer, "[!] gitops succeeded\n")
			}
			return nil
		},
	}
}