hcloud-upload-image/cmd/hcloud-image/cmd/root.go

package cmd

import (
	"context"
	"fmt"
	"log/slog"
	"os"
	"time"

	"github.com/hetznercloud/hcloud-go/v2/hcloud"
	"github.com/spf13/cobra"

	hcloud_upload_image "github.com/apricote/hcloud-upload-image"
	"github.com/apricote/hcloud-upload-image/contextlogger"
	"github.com/apricote/hcloud-upload-image/control"
)

// The pre-authenticated client. Set in the root command PersistentPreRun
var client hcloud_upload_image.SnapshotClient

// rootCmd represents the base command when called without any subcommands
var rootCmd = &cobra.Command{
	Use:   "hcloud-image",
	Short: "A brief description of your application",
	Long: `A longer description that spans multiple lines and likely contains
examples and usage of using your application. For example:

Cobra is a CLI library for Go that empowers applications.
This application is a tool to generate the needed files
to quickly create a Cobra application.`,
	// Uncomment the following line if your bare application
	// has an action associated with it:
	// Run: func(cmd *cobra.Command, args []string) { },

	PersistentPreRun: func(cmd *cobra.Command, _ []string) {
		ctx := cmd.Context()

		// Add logger to command context
		logger := slog.Default()
		ctx = contextlogger.New(ctx, logger)
		cmd.SetContext(ctx)

		client = newClient(ctx)
	},
}

func newClient(ctx context.Context) hcloud_upload_image.SnapshotClient {
	logger := contextlogger.From(ctx)
	// Build hcloud-go client
	if os.Getenv("HCLOUD_TOKEN") == "" {
		logger.ErrorContext(ctx, "You need to set the HCLOUD_TOKEN environment variable to your Hetzner Cloud API Token.")
		os.Exit(1)
	}
	hcloudClient := hcloud.NewClient(
		hcloud.WithToken(os.Getenv("HCLOUD_TOKEN")),
		hcloud.WithApplication("hcloud-image", ""),
		hcloud.WithPollBackoffFunc(control.ExponentialBackoffWithLimit(2, 1*time.Second, 30*time.Second)),
		// hcloud.WithDebugWriter(os.Stderr),
	)

	return hcloud_upload_image.New(hcloudClient)
}

func Execute() {
	err := rootCmd.Execute()
	if err != nil {
		os.Exit(1)
	}
}

func init() {
	rootCmd.SetErrPrefix("\033[1;31mError:")
	rootCmd.SetFlagErrorFunc(func(command *cobra.Command, err error) error {
		return fmt.Errorf("fooo")
	})
}