feat: with auth-basic header

adapter_fasthttp.go

@@ -39,6 +39,16 @@ func (a *fastHttpReqAdapter) SetHeader(key, value string) {
 	a.req.Header.Set(key, value)
 }
 
+func (a *fastHttpReqAdapter) GetHeader(key string) (string, bool) {
+	data := a.req.Header.Peek(key)
+	if data == nil || len(data) == 0 {
+		return "", false
+	}
+	bts := make([]byte, len(data))
+	copy(bts, data)
+	return string(bts), true
+}
+
 func (a *fastHttpReqAdapter) SetMethod(method string) {
 	a.req.Header.SetMethod(method)
 }
@@ -151,3 +161,21 @@ func (a *fastHttpResAdapter) StatusText() string {
 func (a *fastHttpResAdapter) Body() io.ReadCloser {
 	return io.NopCloser(bytes.NewReader(a.res.Body()))
 }
+
+func (a *fastHttpResAdapter) AddHeader(key, value string) {
+	a.res.Header.Add(key, value)
+}
+
+func (a *fastHttpResAdapter) SetHeader(key, value string) {
+	a.res.Header.Set(key, value)
+}
+
+func (a *fastHttpResAdapter) GetHeader(key string) (string, bool) {
+	data := a.res.Header.Peek(key)
+	if data == nil || len(data) == 0 {
+		return "", false
+	}
+	bts := make([]byte, len(data))
+	copy(bts, data)
+	return string(bts), true
+}

adapter_native.go

@@ -32,6 +32,11 @@ func (a *nativeReqAdapter) SetHeader(key, value string) {
 	a.req.Header.Set(key, value)
 }
 
+func (a *nativeReqAdapter) GetHeader(key string) (string, bool) {
+	s := a.req.Header.Get(key)
+	return s, len(s) > 0
+}
+
 func (a *nativeReqAdapter) SetMethod(method string) {
 	a.req.Method = method
 }
@@ -112,3 +117,16 @@ func (a *nativeResAdapter) StatusText() string {
 func (a *nativeResAdapter) Body() io.ReadCloser {
 	return a.res.Body
 }
+
+func (a *nativeResAdapter) AddHeader(key, value string) {
+	a.res.Header.Add(key, value)
+}
+
+func (a *nativeResAdapter) SetHeader(key, value string) {
+	a.res.Header.Set(key, value)
+}
+
+func (a *nativeResAdapter) GetHeader(key string) (string, bool) {
+	s := a.res.Header.Get(key)
+	return s, len(s) > 0
+}

call_req.go

@@ -72,6 +72,10 @@ func (c *Call[T]) WithHeaderFunc(fn func() (key, value string, override bool)) *
 	return c.withReq(WithHeaderFunc[T](fn))
 }
 
+func (c *Call[T]) WithBasicAuth(user, pass string) *Call[T] {
+	return c.withReq(WithBasicAuth[T](user, pass))
+}
+
 func (c *Call[T]) WithContentType(ct ContentType) *Call[T] {
 	return c.withReq(WithContentType[T](ct))
 }

endpoint.go

@@ -7,10 +7,16 @@ import (
 )
 
 type (
-	Request interface {
-		SetMethod(string)
+	Header interface {
 		SetHeader(k, v string)
 		AddHeader(k, v string)
+		GetHeader(k string) (string, bool)
+	}
+
+	Request interface {
+		Header
+
+		SetMethod(string)
 		SetURL(*url.URL)
 		// SetBodyStream sets the stream of body data belonging to a request. bodySize parameter is needed
 		// when using fasthttp implementation.
@@ -24,6 +30,8 @@ type (
 	}
 
 	Response interface {
+		Header
+
 		Status() int
 		StatusText() string
 		Body() io.ReadCloser

errors.go

@@ -5,4 +5,5 @@ import "github.com/pkg/errors"
 var (
 	ErrAssertion            = errors.New("assertion was unmet")
 	ErrUnexpectedStatusCode = errors.Wrap(ErrAssertion, "unexpected status code")
+	ErrInsufficientParams   = errors.New("insufficient params")
 )

examples/mock/main.go

@@ -37,6 +37,7 @@ func main() {
 	call := withttp.NewCall[Order](withttp.WithFasthttp()).
 		WithURL("https://github.com/").
 		WithMethod(http.MethodGet).
+		WithBasicAuth("pepito", "secret").
 		WithHeader("User-Agent", "withttp/0.1.0 See https://github.com/sonirico/withttp", false).
 		WithHeaderFunc(func() (key, value string, override bool) {
 			key = "X-Date"
@@ -58,4 +59,6 @@ func main() {
 	if err != nil {
 		panic(err)
 	}
+
+	log.Println(call.Req.GetHeader("Authorization"))
 }

utils_nalloc.go

@@ -0,0 +1,19 @@
+package withttp
+
+import (
+	"reflect"
+	"unsafe"
+)
+
+func S2B(s string) (bts []byte) {
+	sh := (*reflect.StringHeader)(unsafe.Pointer(&s))
+	bh := (*reflect.SliceHeader)(unsafe.Pointer(&bts))
+	bh.Data = sh.Data
+	bh.Cap = sh.Len
+	bh.Len = sh.Len
+	return
+}
+
+func B2S(data []byte) string {
+	return *(*string)(unsafe.Pointer(&data))
+}

with_req.go

@@ -3,10 +3,14 @@ package withttp
 import (
 	"bytes"
 	"context"
+	"encoding/base64"
+	"fmt"
 	"io"
 	"net/url"
 	"sync"
 
+	"github.com/pkg/errors"
+
 	"github.com/sonirico/withttp/codec"
 )
 
@@ -16,6 +20,16 @@ func WithHeader[T any](k, v string, override bool) CallReqOptionFunc[T] {
 	}
 }
 
+func WithBasicAuth[T any](user, pass string) CallReqOptionFunc[T] {
+	return func(_ *Call[T], req Request) error {
+		header, err := CreateAuthorizationHeader(authHeaderKindBasic, user, pass)
+		if err != nil {
+			return err
+		}
+		return ConfigureHeader(req, "authorization", header, true)
+	}
+}
+
 func WithHeaderFunc[T any](fn func() (string, string, bool)) CallReqOptionFunc[T] {
 	return func(_ *Call[T], req Request) error {
 		k, v, override := fn()
@@ -30,6 +44,38 @@ func WithContentType[T any](ct ContentType) CallReqOptionFunc[T] {
 	}
 }
 
+type authHeaderKind string
+
+var (
+	authHeaderKindBasic authHeaderKind = "Basic"
+)
+
+func (a authHeaderKind) Codec() func(...string) (string, error) {
+	switch a {
+	case authHeaderKindBasic:
+		return func(s ...string) (string, error) {
+			if len(s) < 2 {
+				return "", errors.Wrapf(ErrAssertion, "header kind: %s", a)
+			}
+			user := s[0]
+			pass := s[1]
+
+			return base64.StdEncoding.EncodeToString(S2B(user + ":" + pass)), nil
+		}
+	default:
+		panic("unknown auth header kind")
+	}
+}
+
+func CreateAuthorizationHeader(kind authHeaderKind, user, pass string) (string, error) {
+	fn := kind.Codec()
+	header, err := fn(user, pass)
+	if err != nil {
+		return header, err
+	}
+	return fmt.Sprintf("%s %s", kind, header), nil
+}
+
 func ConfigureHeader(req Request, key, value string, override bool) error {
 	if override {
 		req.SetHeader(key, value)