Skip to content

Commit

Permalink
Raw headers (#193)
Browse files Browse the repository at this point in the history
* raw command arbitary headers

* pull in sdk
  • Loading branch information
0x4c6565 authored Feb 24, 2025
1 parent 9cb03ce commit e7cc3b2
Show file tree
Hide file tree
Showing 3 changed files with 20 additions and 5 deletions.
19 changes: 17 additions & 2 deletions cmd/raw.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,8 @@ package cmd
import (
"errors"
"fmt"
"io/ioutil"
"io"
"net/http"
"strings"

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

type rawCommandOutput string

func (r *rawCommandOutput) Deserialize(resp *connection.APIResponse) error {
defer resp.Response.Body.Close()
bodyBytes, err := ioutil.ReadAll(resp.Response.Body)
bodyBytes, err := io.ReadAll(resp.Response.Body)
if err != nil {
return err
}
Expand Down Expand Up @@ -88,6 +90,19 @@ func raw(c connection.Connection, cmd *cobra.Command, args []string) error {
req.Body = &commandData
}

if cmd.Flags().Changed("header") {
req.Headers = http.Header{}
headers, _ := cmd.Flags().GetStringArray("header")
for _, header := range headers {
headerParts := strings.SplitN(header, ":", 2)
if len(headerParts) != 2 {
return fmt.Errorf("invalid header format: %s", header)
}

req.Headers.Add(strings.TrimSpace(headerParts[0]), strings.TrimSpace(headerParts[1]))
}
}

resp, err := c.Invoke(req)
if err != nil {
return err
Expand Down
2 changes: 1 addition & 1 deletion go.mod
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ go 1.23.0
toolchain go1.23.3

require (
github.com/ans-group/sdk-go v1.22.1
github.com/ans-group/sdk-go v1.22.2
github.com/blang/semver v3.5.1+incompatible
github.com/golang/mock v1.6.0
github.com/iancoleman/strcase v0.3.0
Expand Down
4 changes: 2 additions & 2 deletions go.sum
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
github.com/ans-group/go-durationstring v1.2.0 h1:UJIuQATkp0t1rBvZsHRwki33YHV9E+Ulro+3NbMB7MM=
github.com/ans-group/go-durationstring v1.2.0/go.mod h1:QGF9Mdpq9058QXaut8r55QWu6lcHX6i/GvF1PZVkV6o=
github.com/ans-group/sdk-go v1.22.1 h1:3wfYuWXPldIzKT393+yxQBPjP+O0kPZHzsD1q6zm/+E=
github.com/ans-group/sdk-go v1.22.1/go.mod h1:Dx34ZUbyHNniHAKsDy/vp8q8hQC5L51ub2sv9We7d8E=
github.com/ans-group/sdk-go v1.22.2 h1:wN84TXwJlxc63hTzOtN8ns3VO7sZkNkrYgk4IPYaB20=
github.com/ans-group/sdk-go v1.22.2/go.mod h1:Dx34ZUbyHNniHAKsDy/vp8q8hQC5L51ub2sv9We7d8E=
github.com/blang/semver v3.5.1+incompatible h1:cQNTCjp13qL8KC3Nbxr/y2Bqb63oX6wdnnjpJbkM4JQ=
github.com/blang/semver v3.5.1+incompatible/go.mod h1:kRBLl5iJ+tD4TcOOxsy/0fnwebNt5EWlYSAyrTnjyyk=
github.com/cpuguy83/go-md2man/v2 v2.0.4/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o=
Expand Down

0 comments on commit e7cc3b2

Please sign in to comment.