at-container-registry/pkg/credhelper/cmd_logout.go

package credhelper

import (
	"fmt"
	"os"
	"sort"

	"github.com/charmbracelet/huh"
	"github.com/spf13/cobra"
)

func newLogoutCmd() *cobra.Command {
	return &cobra.Command{
		Use:   "logout [registry]",
		Short: "Remove account credentials",
		Long:  "Remove stored credentials for an account.\nDefault registry: " + cfg.DefaultRegistry,
		Args:  cobra.MaximumNArgs(1),
		RunE:  runLogout,
	}
}

func runLogout(cmd *cobra.Command, args []string) error {
	serverURL := cfg.DefaultRegistry
	if len(args) > 0 {
		serverURL = args[0]
	}

	appViewURL := buildAppViewURL(serverURL)

	sc, err := loadConfigForWrite()
	if err != nil {
		return err
	}

	reg := sc.findRegistry(appViewURL)
	if reg == nil || len(reg.Accounts) == 0 {
		fmt.Fprintf(os.Stderr, "No accounts configured for %s.\n", serverURL)
		return nil
	}

	// Determine which account to remove
	var target *Account

	if len(reg.Accounts) == 1 {
		for _, acct := range reg.Accounts {
			target = acct
		}
	} else {
		// Multiple accounts — select which to remove. Options display the
		// handle but carry the account's key (DID) as the value.
		accts := make([]*Account, 0, len(reg.Accounts))
		for _, acct := range reg.Accounts {
			accts = append(accts, acct)
		}
		sort.Slice(accts, func(i, j int) bool { return accts[i].Handle < accts[j].Handle })

		active := reg.activeAccount()
		var options []huh.Option[string]
		for _, acct := range accts {
			label := acct.Handle
			if acct == active {
				label += " (active)"
			}
			options = append(options, huh.NewOption(label, acct.key()))
		}

		var selected string
		err := huh.NewSelect[string]().
			Title("Which account to remove?").
			Options(options...).
			Value(&selected).
			Run()
		if err != nil {
			return err
		}
		target = reg.find(selected)
	}

	if target == nil {
		fmt.Fprintf(os.Stderr, "No account selected.\n")
		return nil
	}

	// Confirm
	var confirm bool
	err = huh.NewConfirm().
		Title(fmt.Sprintf("Remove %s from %s?", target.Handle, serverURL)).
		Value(&confirm).
		Run()
	if err != nil || !confirm {
		fmt.Fprintf(os.Stderr, "Cancelled.\n")
		return nil
	}

	sc.removeAccount(appViewURL, target.key())
	if err := sc.save(); err != nil {
		return fmt.Errorf("saving config: %w", err)
	}

	fmt.Printf("Removed %s from %s\n", target.Handle, serverURL)
	return nil
}