mirror of
https://github.com/larksuite/cli.git
synced 2026-09-14 18:42:53 +08:00
evandance
87 lines
2.5 KiB
Go
87 lines
2.5 KiB
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
|
|
}
|