larksuite__cli/cmd/profile/remove.go

// Copyright (c) 2026 Lark Technologies Pte. Ltd.
// SPDX-License-Identifier: MIT

package profile

import (
	"fmt"
	"strings"

	"github.com/spf13/cobra"

	"github.com/larksuite/cli/errs"
	larkauth "github.com/larksuite/cli/internal/auth"
	"github.com/larksuite/cli/internal/cmdutil"
	"github.com/larksuite/cli/internal/core"
	"github.com/larksuite/cli/internal/output"
	"github.com/larksuite/cli/internal/recovery"
)

// NewCmdProfileRemove creates the profile remove subcommand.
func NewCmdProfileRemove(f *cmdutil.Factory) *cobra.Command {
	cmd := &cobra.Command{
		Use:   "remove <name>",
		Short: "Remove a profile",
		Args:  cobra.ExactArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			return profileRemoveRun(f, args[0])
		},
	}
	cmdutil.SetTips(cmd, []string{
		"AI agents: Do NOT remove profiles unless the user explicitly asks. This is destructive and clears all associated credentials.",
	})
	cmdutil.SetRisk(cmd, "write")
	return cmd
}

func profileRemoveRun(f *cmdutil.Factory, name string) error {
	multi, err := core.LoadOrNotConfigured()
	if err != nil {
		return err
	}

	idx := multi.FindAppIndex(name)
	if idx < 0 {
		return errs.NewValidationError(errs.SubtypeInvalidArgument, "profile %q not found, available profiles: %s", name, strings.Join(multi.ProfileNames(), ", "))
	}

	if len(multi.Apps) == 1 {
		return recovery.Attach(
			errs.NewValidationError(errs.SubtypeFailedPrecondition, "cannot remove the only profile"),
			recovery.Join("",
				recovery.Command(recovery.TargetProfileAdd, "add another profile first: lark-cli profile add"),
			).WithFallback("configure another profile through this distribution before removing the only profile"),
		)
	}

	app := &multi.Apps[idx]
	removedName := app.ProfileName()
	appId := app.AppId
	appSecret := app.AppSecret
	users := app.Users

	// Remove from slice
	multi.Apps = append(multi.Apps[:idx], multi.Apps[idx+1:]...)

	// Fix currentApp / previousApp references
	if multi.CurrentApp == removedName {
		multi.CurrentApp = multi.Apps[0].ProfileName()
	}
	if multi.PreviousApp == removedName {
		multi.PreviousApp = ""
	}

	if err := core.SaveMultiAppConfig(multi); err != nil {
		return errs.NewInternalError(errs.SubtypeStorage, "failed to save config: %v", err).WithCause(err)
	}

	// Best-effort credential cleanup after config commit
	core.RemoveSecretStore(appSecret, f.Keychain)
	for _, user := range users {
		larkauth.RemoveStoredToken(appId, user.UserOpenId)
	}

	output.PrintSuccess(f.IOStreams.ErrOut, fmt.Sprintf("Profile %q removed", removedName))
	return nil
}