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

Add trust-center alias CLI subcommand

Browse Source 
Provides set and remove subcommands for managing trust center aliases
from the command line.

Signed-off-by: Bryan Frimin <bryan@probo.com>
This commit is contained in:
Bryan Frimin
2026-06-22 08:40:03 +02:00
parent a11fdc9520
commit 982509416e
4 changed files with 285 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
pkg/cmd/trust-center/alias/alias.go Normal file
View File

@@ -0,0 +1,34 @@
// 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 alias
import (
"github.com/spf13/cobra"
"go.probo.inc/probo/pkg/cmd/cmdutil"
"go.probo.inc/probo/pkg/cmd/trust-center/alias/remove"
"go.probo.inc/probo/pkg/cmd/trust-center/alias/set"
)
func NewCmdAlias(f *cmdutil.Factory) *cobra.Command {
cmd := &cobra.Command{
Use: "alias <command>",
Short: "Manage trust center aliases",
}
cmd.AddCommand(set.NewCmdSet(f))
cmd.AddCommand(remove.NewCmdRemove(f))
return cmd
}

105
pkg/cmd/trust-center/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: RemoveTrustCenterAliasInput!) {
removeTrustCenterAlias(input: $input) {
deletedResourceId
}
}
`
func NewCmdRemove(f *cmdutil.Factory) *cobra.Command {
var flagYes bool
cmd := &cobra.Command{
Use: "remove <resource-id>",
Short: "Remove a trust center 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
}

144
pkg/cmd/trust-center/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: SetTrustCenterAliasInput!) {
setTrustCenterAlias(input: $input) {
alias {
resourceId
alias
}
}
}
`
type setResponse struct {
SetTrustCenterAlias struct {
Alias struct {
ResourceID string `json:"resourceId"`
Alias string `json:"alias"`
} `json:"alias"`
} `json:"setTrustCenterAlias"`
}
func NewCmdSet(f *cmdutil.Factory) *cobra.Command {
var (
flagResourceID string
flagAlias string
)
cmd := &cobra.Command{
Use: "set",
Short: "Set a trust center alias",
Example: ` # Set an alias interactively
prb trust-center alias set --resource-id prbdoc_...
# Set an alias non-interactively
prb trust-center 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 (document, file, or audit)").
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.SetTrustCenterAlias.Alias
_, _ = fmt.Fprintf(
f.IOStreams.Out,
"Set alias %q for resource %s\n",
a.Alias,
a.ResourceID,
)
return nil
},
}
cmd.Flags().StringVar(&flagResourceID, "resource-id", "", "Document, trust center file, or audit ID")
cmd.Flags().StringVar(&flagAlias, "alias", "", "Human-readable alias slug")
return cmd
}

2
pkg/cmd/trust-center/trust_center.go
View File

@@ -17,6 +17,7 @@ package trustcenter
import (
"github.com/spf13/cobra"
"go.probo.inc/probo/pkg/cmd/cmdutil"
"go.probo.inc/probo/pkg/cmd/trust-center/alias"
"go.probo.inc/probo/pkg/cmd/trust-center/file"
"go.probo.inc/probo/pkg/cmd/trust-center/reference"
"go.probo.inc/probo/pkg/cmd/trust-center/update"
@@ -33,6 +34,7 @@ func NewCmdTrustCenter(f *cmdutil.Factory) *cobra.Command {
cmd.AddCommand(view.NewCmdView(f))
cmd.AddCommand(update.NewCmdUpdate(f))
cmd.AddCommand(reference.NewCmdReference(f))
cmd.AddCommand(alias.NewCmdAlias(f))
cmd.AddCommand(file.NewCmdFile(f))
return cmd