uncloud/cmd/ucind/main.go

package main

import (
	"context"
	"fmt"
	"os"
	"strings"

	"github.com/docker/docker/client"
	"github.com/psviderski/uncloud/cmd/ucind/cluster"
	"github.com/psviderski/uncloud/internal/ucind"
	"github.com/spf13/cobra"
)

func main() {
	var configPath string
	cmd := &cobra.Command{
		Use:   "ucind",
		Short: "A CLI tool for running Uncloud test clusters using Docker.",
		Long: "A CLI tool for running Uncloud test clusters using Docker.\n" +
			"Machines in a ucind cluster are Docker containers running a ucind image. All machines within a cluster " +
			"are connected to the same Docker network.",
		SilenceUsage:  true,
		SilenceErrors: true,
		PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
			cli, err := client.NewClientWithOpts(client.FromEnv, client.WithAPIVersionNegotiation())
			if err != nil {
				return fmt.Errorf("create Docker client: %w", err)
			}

			if strings.HasPrefix(configPath, "~/") {
				home, err := os.UserHomeDir()
				if err != nil {
					return fmt.Errorf("get user home directory to resolve %q: %w", configPath, err)
				}
				configPath = strings.Replace(configPath, "~", home, 1)
			}
			configUpdater := ucind.NewConfigUpdater(configPath)

			p := ucind.NewProvisioner(cli, configUpdater)
			// Persist the provisioner in the context so it can be used by subcommands.
			cmd.SetContext(context.WithValue(cmd.Context(), "provisioner", p))
			return nil
		},
	}

	// TODO: allow to override using UNCLOUD_CONFIG env var.
	cmd.PersistentFlags().StringVar(&configPath, "uncloud-config", "~/.config/uncloud/config.yaml",
		"path to the Uncloud configuration file.")
	_ = cmd.MarkPersistentFlagFilename("uncloud-config", "yaml", "yml")

	cmd.AddCommand(
		cluster.NewRootCommand(),
	)
	cobra.CheckErr(cmd.Execute())
}