Enterprise-Knowledge-Base/cli/cmd/profile/add.go

package profilecmd

import (
	"fmt"

	"github.com/spf13/cobra"

	"github.com/Tencent/WeKnora/cli/internal/cmdutil"
	"github.com/Tencent/WeKnora/cli/internal/config"
	"github.com/Tencent/WeKnora/cli/internal/iostreams"
)

type AddOptions struct {
	Host   string
	User   string
	Use    bool // --use: switch to this profile after adding
	DryRun bool
}

// profileAddFields enumerates the fields surfaced for `--format json` discovery on
// `profile add`. The result describes the newly-registered profile.
var profileAddFields = []string{
	"name", "host", "user", "current",
}

// addResult is the typed payload emitted under data on success.
type addResult struct {
	Name    string `json:"name"`
	Host    string `json:"host"`
	User    string `json:"user,omitempty"`
	Current bool   `json:"current"`
}

// NewCmdAdd builds `weknora profile add`. Registers a *credentialless*
// connection target - host + optional user only. Credentials for the new
// profile are attached separately by making it active and running
// `weknora auth login`, separating "where" the CLI talks to (the host) and
// "how" it authenticates (the credential). Pass --use to switch to the new
// profile immediately, then run `weknora auth login`.
func NewCmdAdd(f *cmdutil.Factory) *cobra.Command {
	opts := &AddOptions{}
	cmd := &cobra.Command{
		Use:   "add <name>",
		Short: "Register a new profile (host without credentials)",
		Long: `Add a new profile entry to config.yaml. Stores host (and optionally
user) but does NOT prompt for credentials. To attach credentials, make the
profile active (` + "`weknora profile use <n>`" + `, or ` + "`profile add <n> --use`" + `) and run
` + "`weknora auth login`" + `.

The first profile added is auto-selected as the current profile. Subsequent
adds leave the current profile untouched unless --use is passed.`,
		Example: `  weknora profile add staging --host https://staging.example.com
  weknora profile add prod    --host https://prod.example.com --user alice@example.com
  weknora profile add prod    --host https://prod.example.com --use   # switch to it now`,
		Args: cobra.ExactArgs(1),
		RunE: func(c *cobra.Command, args []string) error {
			fopts, err := cmdutil.CheckFormatFlag(c)
			if err != nil {
				return err
			}
			fopts.ResolveDefault(iostreams.IO.IsStdoutTTY())
			// Pure-local validation runs before the dry-run gate so --dry-run
			// rejects identically to the live path. Same typed errors as runAdd
			// (kept there for direct-call callers).
			name := args[0]
			if err := cmdutil.ValidateProfileName(name); err != nil {
				return err
			}
			host, err := cmdutil.NormalizeHost(opts.Host)
			if err != nil {
				return err
			}
			cfg, err := config.Load()
			if err != nil {
				return err
			}
			if _, exists := cfg.Profiles[name]; exists {
				return &cmdutil.Error{
					Code:    cmdutil.CodeResourceAlreadyExists,
					Message: fmt.Sprintf("profile %q already exists", name),
					Hint:    fmt.Sprintf("use a different name, or run `weknora profile remove %s` first", name),
				}
			}
			if handled, err := cmdutil.HandleDryRun(c, opts.DryRun, cmdutil.DryRunPlan{
				Action: "profile.add",
				Args: map[string]any{
					"name": name,
					"host": host,
				},
			}); handled {
				return err
			}
			return runAddWithConfig(opts, fopts, name, host, cfg)
		},
	}
	cmd.Flags().StringVar(&opts.Host, "host", "", "Server base URL, e.g. https://kb.example.com (required)")
	cmd.Flags().StringVar(&opts.User, "user", "", "Account email shown in 'profile list' (optional, cosmetic only)")
	cmd.Flags().BoolVar(&opts.Use, "use", false, "Switch to this profile after adding it")
	cmdutil.AddFormatFlag(cmd, profileAddFields...)
	cmdutil.AddDryRunFlag(cmd, &opts.DryRun)
	_ = cmd.MarkFlagRequired("host")
	cmdutil.SetAgentHelp(cmd, cmdutil.AgentHelp{
		UsedFor:       "Register a new profile (connection target) with a name and host URL. Does not store credentials; make it active (--use, or `profile use <n>`) and run `auth login` afterwards to authenticate.",
		RequiredFlags: []string{"<name> (positional)", "--host"},
		Output:        "envelope.data has name, host, user, current",
		Examples: []string{
			"weknora profile add prod --host https://kb.example.com --use",
		},
	})
	return cmd
}

// runAdd is the legacy direct-call entrypoint: revalidates inputs and loads
// config. Preserved for tests / external callers that bypass the cobra layer.
// The cobra RunE path validates earlier and delegates to runAddWithConfig.
func runAdd(opts *AddOptions, fopts *cmdutil.FormatOptions, name string) error {
	if err := cmdutil.ValidateProfileName(name); err != nil {
		return err
	}
	host, err := cmdutil.NormalizeHost(opts.Host)
	if err != nil {
		return err
	}

	cfg, err := config.Load()
	if err != nil {
		return err
	}
	if _, exists := cfg.Profiles[name]; exists {
		return &cmdutil.Error{
			Code:    cmdutil.CodeResourceAlreadyExists,
			Message: fmt.Sprintf("profile %q already exists", name),
			Hint:    fmt.Sprintf("use a different name, or run `weknora profile remove %s` first", name),
		}
	}
	return runAddWithConfig(opts, fopts, name, host, cfg)
}

// runAddWithConfig performs the side-effectful write. Inputs are assumed to be
// pre-validated (ValidateProfileName, NormalizeHost, dup-check) by the caller.
func runAddWithConfig(opts *AddOptions, fopts *cmdutil.FormatOptions, name, host string, cfg *config.Config) error {
	if cfg.Profiles == nil {
		cfg.Profiles = map[string]config.Profile{}
	}
	cfg.Profiles[name] = config.Profile{Host: host, User: opts.User}
	wasFirst := cfg.CurrentProfile == ""
	// The first profile is auto-selected; --use forces selection even when
	// another profile is already current.
	current := wasFirst || opts.Use
	if current {
		cfg.CurrentProfile = name
	}
	if err := config.Save(cfg); err != nil {
		return cmdutil.Wrapf(cmdutil.CodeLocalFileIO, err, "save config")
	}

	if fopts.WantsJSON() {
		return fopts.Emit(iostreams.IO.Out, addResult{Name: name, Host: host, User: opts.User, Current: current}, nil)
	}
	if current {
		fmt.Fprintf(iostreams.IO.Out, "✓ Added profile %s (now current). Run `weknora auth login` to attach credentials.\n", name)
	} else {
		fmt.Fprintf(iostreams.IO.Out, "✓ Added profile %s. Run `weknora profile use %s` then `weknora auth login` to attach credentials.\n", name, name)
	}
	return nil
}