rohit/probo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Move alias CLI under a resource-alias command

Browse Source 
Promote the trust-center alias subcommand to a top-level
resource-alias command so aliases can be managed for any resource
from the CLI.

Signed-off-by: Bryan Frimin <bryan@probo.com>
This commit is contained in:
Bryan Frimin
2026-06-22 10:59:39 +02:00
parent 5141478083
commit 6758a1cd50
5 changed files with 27 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

105
pkg/cmd/resource-alias/remove/remove.go Normal file
View File

@@ -0,0 +1,105 @@
// Copyright (c) 2026 Probo Inc <hello@probo.com>.
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
// PERFORMANCE OF THIS SOFTWARE.
package remove
import (
"fmt"
"github.com/charmbracelet/huh"
"github.com/spf13/cobra"
"go.probo.inc/probo/pkg/cli/api"
"go.probo.inc/probo/pkg/cmd/cmdutil"
)
const removeMutation = `
mutation($input: RemoveResourceAliasInput!) {
removeResourceAlias(input: $input) {
deletedResourceId
}
}
`
func NewCmdRemove(f *cmdutil.Factory) *cobra.Command {
var flagYes bool
cmd := &cobra.Command{
Use: "remove <resource-id>",
Short: "Remove a resource alias",
Args: cobra.ExactArgs(1),
RunE: func(cmd *cobra.Command, args []string) error {
if !flagYes {
if !f.IOStreams.IsInteractive() {
return fmt.Errorf("cannot remove alias: confirmation required, use --yes to confirm")
}
var confirmed bool
err := huh.NewConfirm().
Title(fmt.Sprintf("Remove alias for resource %s?", args[0])).
Value(&confirmed).
Run()
if err != nil {
return err
}
if !confirmed {
return nil
}
}
cfg, err := f.Config()
if err != nil {
return err
}
host, hc, err := cfg.DefaultHost()
if err != nil {
return err
}
client := api.NewClient(
host,
hc.Token,
"/api/console/v1/graphql",
cfg.HTTPTimeoutDuration(),
cmdutil.TokenRefreshOption(cfg, host, hc),
)
_, err = client.Do(
removeMutation,
map[string]any{
"input": map[string]any{
"resourceId": args[0],
},
},
)
if err != nil {
return err
}
_, _ = fmt.Fprintf(
f.IOStreams.Out,
"Removed alias for resource %s\n",
args[0],
)
return nil
},
}
cmd.Flags().BoolVarP(&flagYes, "yes", "y", false, "Skip confirmation prompt")
return cmd
}

35
pkg/cmd/resource-alias/resource_alias.go Normal file
View File

@@ -0,0 +1,35 @@
// Copyright (c) 2026 Probo Inc <hello@probo.com>.
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
// PERFORMANCE OF THIS SOFTWARE.
package resourcealias
import (
"github.com/spf13/cobra"
"go.probo.inc/probo/pkg/cmd/cmdutil"
"go.probo.inc/probo/pkg/cmd/resource-alias/remove"
"go.probo.inc/probo/pkg/cmd/resource-alias/set"
)
func NewCmdResourceAlias(f *cmdutil.Factory) *cobra.Command {
cmd := &cobra.Command{
Use: "resource-alias <command>",
Short: "Manage resource aliases",
Aliases: []string{"ra"},
}
cmd.AddCommand(set.NewCmdSet(f))
cmd.AddCommand(remove.NewCmdRemove(f))
return cmd
}

144
pkg/cmd/resource-alias/set/set.go Normal file
View File

@@ -0,0 +1,144 @@
// Copyright (c) 2026 Probo Inc <hello@probo.com>.
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
// PERFORMANCE OF THIS SOFTWARE.
package set
import (
"encoding/json"
"fmt"
"github.com/charmbracelet/huh"
"github.com/spf13/cobra"
"go.probo.inc/probo/pkg/cli/api"
"go.probo.inc/probo/pkg/cmd/cmdutil"
)
const setMutation = `
mutation($input: SetResourceAliasInput!) {
setResourceAlias(input: $input) {
resourceAlias {
resourceId
alias
}
}
}
`
type setResponse struct {
SetResourceAlias struct {
ResourceAlias struct {
ResourceID string `json:"resourceId"`
Alias string `json:"alias"`
} `json:"resourceAlias"`
} `json:"setResourceAlias"`
}
func NewCmdSet(f *cmdutil.Factory) *cobra.Command {
var (
flagResourceID string
flagAlias string
)
cmd := &cobra.Command{
Use: "set",
Short: "Set a resource alias",
Example: ` # Set an alias interactively
prb resource-alias set --resource-id prbdoc_...
# Set an alias non-interactively
prb resource-alias set --resource-id prbdoc_... --alias privacy-policy`,
RunE: func(cmd *cobra.Command, args []string) error {
cfg, err := f.Config()
if err != nil {
return err
}
host, hc, err := cfg.DefaultHost()
if err != nil {
return err
}
client := api.NewClient(
host,
hc.Token,
"/api/console/v1/graphql",
cfg.HTTPTimeoutDuration(),
cmdutil.TokenRefreshOption(cfg, host, hc),
)
if f.IOStreams.IsInteractive() {
if flagResourceID == "" {
err := huh.NewInput().
Title("Resource ID").
Value(&flagResourceID).
Run()
if err != nil {
return err
}
}
if flagAlias == "" {
err := huh.NewInput().
Title("Alias").
Value(&flagAlias).
Run()
if err != nil {
return err
}
}
}
if flagResourceID == "" {
return fmt.Errorf("resource ID is required; pass --resource-id or run interactively")
}
if flagAlias == "" {
return fmt.Errorf("alias is required; pass --alias or run interactively")
}
data, err := client.Do(
setMutation,
map[string]any{
"input": map[string]any{
"resourceId": flagResourceID,
"alias": flagAlias,
},
},
)
if err != nil {
return err
}
var resp setResponse
if err := json.Unmarshal(data, &resp); err != nil {
return fmt.Errorf("cannot parse response: %w", err)
}
a := resp.SetResourceAlias.ResourceAlias
_, _ = fmt.Fprintf(
f.IOStreams.Out,
"Set alias %q for resource %s\n",
a.Alias,
a.ResourceID,
)
return nil
},
}
cmd.Flags().StringVar(&flagResourceID, "resource-id", "", "ID of the resource to alias")
cmd.Flags().StringVar(&flagAlias, "alias", "", "Human-readable alias slug")
return cmd
}