fixed flags

This commit is contained in:
ston1th 2023-03-15 02:11:37 +01:00
commit 398ff981d1
3 changed files with 41 additions and 23 deletions

View file

@ -10,26 +10,20 @@ import (
"os"
apiclient "git.giftfish.de/ston1th/keyctl/pkg/api/client"
"git.giftfish.de/ston1th/keyctl/pkg/api/v1/client"
clientv1 "git.giftfish.de/ston1th/keyctl/pkg/api/v1/client"
"git.giftfish.de/ston1th/keyctl/pkg/api/v1/schema"
"git.giftfish.de/ston1th/keyctl/pkg/core"
)
var fs = flag.NewFlagSet("", flag.ExitOnError)
func exit(err error) {
fmt.Println(err)
os.Exit(1)
}
func Run(arg string) {
opts, err := apiclient.ClientOptionsFromEnv()
if err != nil {
exit(err)
}
apic := apiclient.NewClient(opts...)
c, err := client.NewClient(apic)
if err != nil {
exit(err)
}
var err error
ctx := context.Background()
switch arg {
case "new":
@ -39,18 +33,23 @@ func Run(arg string) {
Size: size,
Encoding: core.EncodingFromString(encoding),
}
c := client()
err = c.CreateKey(ctx, newKey)
case "del":
id := idFlags("key id")
c := client()
err = c.DeleteKey(ctx, id)
case "approve":
id := idFlags("approval id")
c := client()
err = c.AcceptApproval(ctx, id)
case "reject":
id := idFlags("approval id")
c := client()
err = c.RejectApproval(ctx, id)
case "get":
id, json := getFlags()
c := client()
key, err := c.GetKey(ctx, id)
if err != nil {
exit(err)
@ -61,6 +60,7 @@ func Run(arg string) {
fmt.Println(key.Key)
case "ls":
json := outputFlags()
c := client()
keys, err := c.GetKeys(ctx)
if err != nil {
exit(err)
@ -79,6 +79,7 @@ func Run(arg string) {
}
case "req":
json := outputFlags()
c := client()
approvals, err := c.GetApprovals(ctx)
if err != nil {
exit(err)
@ -102,6 +103,19 @@ func Run(arg string) {
os.Exit(0)
}
func client() (c *clientv1.Client) {
opts, err := apiclient.ClientOptionsFromEnv()
if err != nil {
exit(err)
}
apic := apiclient.NewClient(opts...)
c, err = clientv1.NewClient(apic)
if err != nil {
exit(err)
}
return
}
func jsonOutput(v any) {
b, err := json.Marshal(v)
if err != nil {
@ -112,28 +126,28 @@ func jsonOutput(v any) {
}
func idFlags(desc string) (id string) {
flag.StringVar(&id, "id", "", desc)
flag.Parse()
fs.StringVar(&id, "id", "", desc)
fs.Parse(os.Args[2:])
return
}
func getFlags() (id string, json bool) {
flag.StringVar(&id, "id", "", "key id")
flag.BoolVar(&json, "json", false, "json output")
flag.Parse()
fs.StringVar(&id, "id", "", "key id")
fs.BoolVar(&json, "json", false, "json output")
fs.Parse(os.Args[2:])
return
}
func outputFlags() (json bool) {
flag.BoolVar(&json, "json", false, "json output")
flag.Parse()
fs.BoolVar(&json, "json", false, "json output")
fs.Parse(os.Args[2:])
return
}
func newKeyFlags() (name, encoding string, size int) {
flag.StringVar(&name, "name", "", "key name")
flag.StringVar(&encoding, "enc", "hex", "key encoding scheme [hex|base32|base64|base64url]")
flag.IntVar(&size, "size", 16, "key size in bytes")
flag.Parse()
fs.StringVar(&name, "name", "", "key name")
fs.StringVar(&encoding, "enc", "hex", "key encoding scheme [hex|base32|base64|base64url]")
fs.IntVar(&size, "size", 16, "key size in bytes")
fs.Parse(os.Args[2:])
return
}