From c29a17d7e49187c12421ec22f99ff75343dff577 Mon Sep 17 00:00:00 2001 From: pancho horrillo Date: Fri, 25 Oct 2019 18:54:52 +0200 Subject: [PATCH] Fix handling of arguments Closes: #67 --- internal/cmd/set.go | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/internal/cmd/set.go b/internal/cmd/set.go index 37252c9..1f81a36 100644 --- a/internal/cmd/set.go +++ b/internal/cmd/set.go @@ -26,7 +26,7 @@ import ( "github.com/spf13/cobra" ) -//SetCmd is the command line interface for set kapow data operation +// SetCmd is the command line interface for set kapow data operation var SetCmd = &cobra.Command{ Use: "set [flags] resource [value]", Short: "Set a Kapow! resource value", @@ -37,16 +37,15 @@ var SetCmd = &cobra.Command{ var r io.Reader dataURL, _ := cmd.Flags().GetString("data-url") handler, _ := cmd.Flags().GetString("handler") + path, args := args[0], args[1:] - if len(args) >= 2 { - // We have a command line value create a stringReader - r = strings.NewReader(strings.Join(args, " ")) + if len(args) == 1 { + r = strings.NewReader(args[0]) } else { - // Use stdin r = os.Stdin } - if err := client.SetData(dataURL, handler, args[0], r); err != nil { + if err := client.SetData(dataURL, handler, path, r); err != nil { log.Fatal(err) } },