Skip to content

Commit e7cc3b2

Browse files
authored
Raw headers (#193)
* raw command arbitary headers * pull in sdk
1 parent 9cb03ce commit e7cc3b2

3 files changed

Lines changed: 20 additions & 5 deletions

File tree

cmd/raw.go

Lines changed: 17 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,8 @@ package cmd
33
import (
44
"errors"
55
"fmt"
6-
"io/ioutil"
6+
"io"
7+
"net/http"
78
"strings"
89

910
"github.com/ans-group/cli/internal/pkg/helper"
@@ -35,14 +36,15 @@ func rawCmd(f connection.ConnectionFactory) *cobra.Command {
3536
cmd.Flags().String("method", "GET", "Method for request")
3637
cmd.Flags().StringP("request", "X", "GET", "Method for request (curl alias for 'method')")
3738
cmd.Flags().StringP("data", "d", "", "Data for request")
39+
cmd.Flags().StringArrayP("header", "H", []string{}, "Additional header for request")
3840
return cmd
3941
}
4042

4143
type rawCommandOutput string
4244

4345
func (r *rawCommandOutput) Deserialize(resp *connection.APIResponse) error {
4446
defer resp.Response.Body.Close()
45-
bodyBytes, err := ioutil.ReadAll(resp.Response.Body)
47+
bodyBytes, err := io.ReadAll(resp.Response.Body)
4648
if err != nil {
4749
return err
4850
}
@@ -88,6 +90,19 @@ func raw(c connection.Connection, cmd *cobra.Command, args []string) error {
8890
req.Body = &commandData
8991
}
9092

93+
if cmd.Flags().Changed("header") {
94+
req.Headers = http.Header{}
95+
headers, _ := cmd.Flags().GetStringArray("header")
96+
for _, header := range headers {
97+
headerParts := strings.SplitN(header, ":", 2)
98+
if len(headerParts) != 2 {
99+
return fmt.Errorf("invalid header format: %s", header)
100+
}
101+
102+
req.Headers.Add(strings.TrimSpace(headerParts[0]), strings.TrimSpace(headerParts[1]))
103+
}
104+
}
105+
91106
resp, err := c.Invoke(req)
92107
if err != nil {
93108
return err

go.mod

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ go 1.23.0
55
toolchain go1.23.3
66

77
require (
8-
github.com/ans-group/sdk-go v1.22.1
8+
github.com/ans-group/sdk-go v1.22.2
99
github.com/blang/semver v3.5.1+incompatible
1010
github.com/golang/mock v1.6.0
1111
github.com/iancoleman/strcase v0.3.0

go.sum

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
github.com/ans-group/go-durationstring v1.2.0 h1:UJIuQATkp0t1rBvZsHRwki33YHV9E+Ulro+3NbMB7MM=
22
github.com/ans-group/go-durationstring v1.2.0/go.mod h1:QGF9Mdpq9058QXaut8r55QWu6lcHX6i/GvF1PZVkV6o=
3-
github.com/ans-group/sdk-go v1.22.1 h1:3wfYuWXPldIzKT393+yxQBPjP+O0kPZHzsD1q6zm/+E=
4-
github.com/ans-group/sdk-go v1.22.1/go.mod h1:Dx34ZUbyHNniHAKsDy/vp8q8hQC5L51ub2sv9We7d8E=
3+
github.com/ans-group/sdk-go v1.22.2 h1:wN84TXwJlxc63hTzOtN8ns3VO7sZkNkrYgk4IPYaB20=
4+
github.com/ans-group/sdk-go v1.22.2/go.mod h1:Dx34ZUbyHNniHAKsDy/vp8q8hQC5L51ub2sv9We7d8E=
55
github.com/blang/semver v3.5.1+incompatible h1:cQNTCjp13qL8KC3Nbxr/y2Bqb63oX6wdnnjpJbkM4JQ=
66
github.com/blang/semver v3.5.1+incompatible/go.mod h1:kRBLl5iJ+tD4TcOOxsy/0fnwebNt5EWlYSAyrTnjyyk=
77
github.com/cpuguy83/go-md2man/v2 v2.0.4/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o=

0 commit comments

Comments
 (0)