Skip to content

feat: add provisioner key cli commands #13875

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 9 commits into from
Jul 18, 2024
Merged

feat: add provisioner key cli commands #13875

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
83e3868
add api routes
f0ssel Jul 9, 2024
e4cf813
feat: add provisioner key cli commands
f0ssel Jul 11, 2024
9fd47c8
fix imports
f0ssel Jul 16, 2024
3d1b075
fmt
f0ssel Jul 16, 2024
c67ed19
add tests
f0ssel Jul 16, 2024
adffefd
make golden
f0ssel Jul 16, 2024
da95666
hide commands
f0ssel Jul 16, 2024
90412c7
pr comments
f0ssel Jul 18, 2024
86c0d02
remove rows type
f0ssel Jul 18, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
feat: add provisioner key cli commands
  • Loading branch information
@f0ssel
f0ssel committed Jul 16, 2024
commit e4cf8131b7e88908d922fe9b72f73e1f6516f30d
2 changes: 2 additions & 0 deletions enterprise/cli/provisionerdaemons.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,8 +39,10 @@ func (r *RootCmd) provisionerDaemons() *serpent.Command {
Handler: func(inv *serpent.Invocation) error {
return inv.Command.HelpHandler(inv)
},
Aliases: []string{"provisioner"},
Children: []*serpent.Command{
r.provisionerDaemonStart(),
r.provisionerKeys(),
},
}

Expand Down
184 changes: 184 additions & 0 deletions enterprise/cli/provisionerkeys.go
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,184 @@
package cli

import (
"fmt"
"time"

"github.com/google/uuid"
"golang.org/x/xerrors"

agpl "github.com/coder/coder/v2/cli"
"github.com/coder/coder/v2/cli/cliui"
"github.com/coder/coder/v2/codersdk"
"github.com/coder/pretty"
"github.com/coder/serpent"
)

func (r *RootCmd) provisionerKeys() *serpent.Command {
cmd := &serpent.Command{
Use: "keys",
Short: "Manage provisioner keys",
Handler: func(inv *serpent.Invocation) error {
return inv.Command.HelpHandler(inv)
},
Aliases: []string{"key"},
Children: []*serpent.Command{
r.provisionerKeysCreate(),
r.provisionerKeysList(),
r.provisionerKeysDelete(),
},
}

return cmd
}

func (r *RootCmd) provisionerKeysCreate() *serpent.Command {
var (
orgContext = agpl.NewOrganizationContext()
)

client := new(codersdk.Client)
cmd := &serpent.Command{
Use: "create <name>",
Short: "Create a new provisioner key",
Middleware: serpent.Chain(
serpent.RequireNArgs(1),
r.InitClient(client),
),
Handler: func(inv *serpent.Invocation) error {
ctx := inv.Context()

org, err := orgContext.Selected(inv, client)
if err != nil {
return xerrors.Errorf("current organization: %w", err)
}

res, err := client.CreateProvisionerKey(ctx, org.ID, codersdk.CreateProvisionerKeyRequest{
Name: inv.Args[0],
})
if err != nil {
return xerrors.Errorf("create provisioner key: %w", err)
}

_, _ = fmt.Fprintf(inv.Stdout, "Successfully created provisioner key %s!\n\n%s", pretty.Sprint(cliui.DefaultStyles.Keyword, inv.Args[0], res.Key))

return nil
},
}

cmd.Options = serpent.OptionSet{}

return cmd
}

type provisionerKeysTableRow struct {
// For json output:
Key codersdk.ProvisionerKey `table:"-"`

// For table output:
Name string `json:"-" table:"name,default_sort"`
CreatedAt time.Time `json:"-" table:"created_at"`
OrganizationID uuid.UUID `json:"-" table:"organization_id"`
}

func provisionerKeysToRows(keys ...codersdk.ProvisionerKey) []provisionerKeysTableRow {
rows := make([]provisionerKeysTableRow, 0, len(keys))
for _, key := range keys {
rows = append(rows, provisionerKeysTableRow{
Name: key.Name,
CreatedAt: key.CreatedAt,
OrganizationID: key.OrganizationID,
})
}

return rows
}

func (r *RootCmd) provisionerKeysList() *serpent.Command {
var (
orgContext = agpl.NewOrganizationContext()
formatter = cliui.NewOutputFormatter(
cliui.TableFormat([]provisionerKeysTableRow{}, nil),
cliui.JSONFormat(),
)
)

client := new(codersdk.Client)
cmd := &serpent.Command{
Use: "list",
Short: "List provisioner keys",
Middleware: serpent.Chain(
serpent.RequireNArgs(0),
r.InitClient(client),
),
Handler: func(inv *serpent.Invocation) error {
ctx := inv.Context()

org, err := orgContext.Selected(inv, client)
if err != nil {
return xerrors.Errorf("current organization: %w", err)
}

keys, err := client.ListProvisionerKeys(ctx, org.ID)
if err != nil {
return xerrors.Errorf("list provisioner keys: %w", err)
}

if len(keys) == 0 {
_, _ = fmt.Fprintln(inv.Stdout, "No provisioner keys found")
return nil
}

rows := provisionerKeysToRows(keys...)
out, err := formatter.Format(inv.Context(), rows)
if err != nil {
return xerrors.Errorf("display provisioner keys: %w", err)
}

_, _ = fmt.Fprintln(inv.Stdout, out)

return nil
},
}

cmd.Options = serpent.OptionSet{}

return cmd
}

func (r *RootCmd) provisionerKeysDelete() *serpent.Command {
var (
orgContext = agpl.NewOrganizationContext()
)

client := new(codersdk.Client)
cmd := &serpent.Command{
Use: "delete <name>",
Short: "Delete a provisioner key",
Middleware: serpent.Chain(
serpent.RequireNArgs(1),
r.InitClient(client),
),
Handler: func(inv *serpent.Invocation) error {
ctx := inv.Context()

org, err := orgContext.Selected(inv, client)
if err != nil {
return xerrors.Errorf("current organization: %w", err)
}

err = client.DeleteProvisionerKey(ctx, org.ID, inv.Args[0])
if err != nil {
return xerrors.Errorf("delete provisioner key: %w", err)
}

_, _ = fmt.Fprintf(inv.Stdout, "Successfully deleted provisioner key %s!", pretty.Sprint(cliui.DefaultStyles.Keyword, inv.Args[0]))

return nil
},
}

cmd.Options = serpent.OptionSet{}

return cmd
}