molecule-cli/internal/cmd/config.go

// Package cmd implements the CLI command tree.
package cmd

import (
	"fmt"
	"os"
	"path/filepath"
	"strings"
	"text/tabwriter"

	"github.com/spf13/cobra"
	"github.com/spf13/viper"
)

// ---------------------------------------------------------------------------
// Config command group
// ---------------------------------------------------------------------------

var configCmd = &cobra.Command{
	Use:   "config",
	Short: "View and manage CLI and workspace configuration",
	Long: `molecule config list      — list all config keys (from file + env)
molecule config get <key> — print a single config value
molecule config set <key> <value> — write a key to the config file
molecule config init      — scaffold a default molecule.yaml in the current directory
molecule config view      — print the current config file with sources annotated`,
}

func init() {
	configCmd.AddCommand(
		configListCmd, configGetCmd, configSetCmd, configInitCmd, configViewCmd,
	)
}

// ===========================================================================
// mol config list
// ===========================================================================
var configListCmd = &cobra.Command{
	Use:   "list",
	Short: "List all known config keys and their effective values",
	RunE:  runConfigList,
}

func runConfigList(cmd *cobra.Command, _ []string) error {
	settings := viper.AllSettings()
	if len(settings) == 0 {
		fmt.Println("No config keys set. Use `molecule config set <key> <value>` or set env vars.")
		return nil
	}
	w := tabwriter.NewWriter(os.Stdout, 0, 4, 2, ' ', 0)
	fmt.Fprintln(w, "KEY\tVALUE\tSOURCE")
	for k, v := range settings {
		source := "default"
		if viper.InConfig(k) {
			source = "file"
		}
		if strings.HasPrefix(k, "MOLECULE_") || strings.HasPrefix(k, "MOL_") {
			source = "env"
		}
		fmt.Fprintf(w, "%s\t%v\t%s\n", k, v, source)
	}
	return w.Flush()
}

// ===========================================================================
// mol config get
// ===========================================================================
var configGetCmd = &cobra.Command{
	Use:   "get <key>",
	Short: "Print the effective value of a config key",
	Args:  cobra.ExactArgs(1),
	RunE:  runConfigGet,
}

func runConfigGet(cmd *cobra.Command, args []string) error {
	if !viper.IsSet(args[0]) {
		return fmt.Errorf("config get: key %q not set (check env var MOLECULE_%s)", args[0], args[0])
	}
	fmt.Println(viper.GetString(args[0]))
	return nil
}

// ===========================================================================
// mol config set
// ===========================================================================
var configSetCmd = &cobra.Command{
	Use:   "set <key> <value>",
	Short: "Write a config key to the config file (~/.config/molecule.yaml)",
	Args:  cobra.ExactArgs(2),
	RunE:  runConfigSet,
}

func runConfigSet(cmd *cobra.Command, args []string) error {
	key, value := args[0], args[1]
	configDir, err := os.UserConfigDir()
	if err != nil {
		configDir = "."
	}
	configFile := filepath.Join(configDir, "molecule.yaml")

	// Use atomic write to avoid viper's WriteConfig/SafeWriteConfig edge cases.
	// Read existing config or start fresh, then write back with the new key.
	v := viper.New()
	v.SetConfigFile(configFile)
	_ = v.ReadInConfig() // ignore not-found; we always write a fresh map
	v.Set(key, value)
	data, err := v.Marshal()
	if err != nil {
		return fmt.Errorf("config set: marshal: %w", err)
	}
	if err := os.WriteFile(configFile, data, 0o644); err != nil {
		return fmt.Errorf("config set: write %s: %w", configFile, err)
	}
	fmt.Printf("Set %s=%q in %s\n", key, value, configFile)
	return nil
}

// ===========================================================================
// mol config init
// ===========================================================================
var configInitCmd = &cobra.Command{
	Use:   "init",
	Short: "Scaffold a default molecule.yaml in the current directory",
	RunE:  runConfigInit,
}

func runConfigInit(cmd *cobra.Command, _ []string) error {
	const defaultConfig = `# molecule CLI config — https://github.com/Molecule-AI/molecule-cli
#
# All values can be overridden by environment variables:
#   MOLECULE_API_URL, MOLECULE_RUNTIME_URL, MOL_OUTPUT, MOL_VERBOSE, etc.

# Platform API base URL (env: MOLECULE_API_URL)
# api_url: http://localhost:8080

# Output format: table | json | yaml  (env: MOL_OUTPUT)
# output: table

# Verbose logging: true | false  (env: MOL_VERBOSE)
# verbose: false
`
	if _, err := os.Stat("molecule.yaml"); err == nil {
		return fmt.Errorf("config init: molecule.yaml already exists (not overwriting)")
	}
	if err := os.WriteFile("molecule.yaml", []byte(defaultConfig), 0o644); err != nil {
		return fmt.Errorf("config init: write molecule.yaml: %w", err)
	}
	fmt.Println("Scaffolded molecule.yaml — edit it and run molecule --config molecule.yaml, or move it to ~/.config/molecule.yaml")
	return nil
}

// ===========================================================================
// mol config view
// ===========================================================================
var configViewCmd = &cobra.Command{
	Use:   "view",
	Short: "Print the current config file with sources annotated",
	RunE:  runConfigView,
}

func runConfigView(cmd *cobra.Command, _ []string) error {
	if viper.ConfigFileUsed() == "" {
		fmt.Println("No config file in use. Set one with --config or molecule config init.")
		fmt.Println("\nActive env vars starting with MOLECULE_ or MOL_:")
		for _, env := range os.Environ() {
			if strings.HasPrefix(env, "MOLECULE_") || strings.HasPrefix(env, "MOL_") {
				fmt.Println("  ", env)
			}
		}
		return nil
	}
	data, err := os.ReadFile(viper.ConfigFileUsed())
	if err != nil {
		return fmt.Errorf("config view: read %s: %w", viper.ConfigFileUsed(), err)
	}
	fmt.Printf("# Config file: %s\n\n", viper.ConfigFileUsed())
	fmt.Print(string(data))
	return nil
}