85 lines
1.6 KiB
Go
85 lines
1.6 KiB
Go
package shell
|
|
|
|
import "context"
|
|
|
|
type Key struct {
|
|
Id string
|
|
Name string
|
|
}
|
|
|
|
type KeyRenameObject struct {
|
|
Id string
|
|
Now string
|
|
Overwrite bool
|
|
Was string
|
|
}
|
|
|
|
type keyListOutput struct {
|
|
Keys []*Key
|
|
}
|
|
|
|
type KeyOpt func(*RequestBuilder) error
|
|
type keyGen struct{}
|
|
|
|
var KeyGen keyGen
|
|
|
|
func (keyGen) Type(alg string) KeyOpt {
|
|
return func(rb *RequestBuilder) error {
|
|
rb.Option("type", alg)
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func (keyGen) Size(size int) KeyOpt {
|
|
return func(rb *RequestBuilder) error {
|
|
rb.Option("size", size)
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// KeyGen Create a new keypair
|
|
func (s *Shell) KeyGen(ctx context.Context, name string, options ...KeyOpt) (*Key, error) {
|
|
rb := s.Request("key/gen", name)
|
|
for _, opt := range options {
|
|
if err := opt(rb); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
var out Key
|
|
if err := rb.Exec(ctx, &out); err != nil {
|
|
return nil, err
|
|
}
|
|
return &out, nil
|
|
}
|
|
|
|
// KeyList List all local keypairs
|
|
func (s *Shell) KeyList(ctx context.Context) ([]*Key, error) {
|
|
var out keyListOutput
|
|
if err := s.Request("key/list").Exec(ctx, &out); err != nil {
|
|
return nil, err
|
|
}
|
|
return out.Keys, nil
|
|
}
|
|
|
|
// KeyRename Rename a keypair
|
|
func (s *Shell) KeyRename(ctx context.Context, old string, new string, force bool) (*KeyRenameObject, error) {
|
|
var out KeyRenameObject
|
|
if err := s.Request("key/rename", old, new).
|
|
Option("force", force).
|
|
Exec(ctx, &out); err != nil {
|
|
return nil, err
|
|
}
|
|
return &out, nil
|
|
}
|
|
|
|
// KeyRm remove a keypair
|
|
func (s *Shell) KeyRm(ctx context.Context, name string) ([]*Key, error) {
|
|
var out keyListOutput
|
|
if err := s.Request("key/rm", name).
|
|
Exec(ctx, &out); err != nil {
|
|
return nil, err
|
|
}
|
|
return out.Keys, nil
|
|
}
|