feat: base url + auth support

This commit is contained in:
2025-05-04 11:54:42 +02:00
parent 3551d0f4dc
commit ad2d978fb4
3 changed files with 94 additions and 28 deletions

View File

@ -7,18 +7,18 @@ import (
type Adddress struct {
Host string
Port string
Port string
Route string
}
func ParseAddress(address string) (*Adddress, error) {
parts := strings.SplitN(address, "/", 2)
parts := strings.Split(address, "/")
var route string
if len(parts) == 1 {
route = "/"
} else if len(parts) > 1 {
route = "/" + strings.Join(parts[1:], "")
route = "/" + strings.Join(parts[1:], "/")
} else {
return nil, fmt.Errorf("Failed to parse address: %s", address)
}
@ -30,18 +30,29 @@ func ParseAddress(address string) (*Adddress, error) {
if strings.Contains(addr, ":") {
p := strings.Split(addr, ":")
if len(p) >= 2 {
port = p[len(p) - 1]
port = p[len(p)-1]
addr = p[0]
}
}
return &Adddress{
Host: addr,
Port: port,
Port: port,
Route: route,
}, nil
}
func (a *Adddress) Extend(extension string) (*Adddress, error) {
route := strings.TrimRight(a.Route, "/")
ext := strings.TrimLeft(extension, "/")
return &Adddress{
Host: a.Host,
Port: a.Port,
Route: route + "/" + ext,
}, nil
}
func (a *Adddress) String() string {
return fmt.Sprintf("%s:%s", a.Host, a.Port)
}

View File

@ -4,32 +4,59 @@ import (
"encoding/json"
"fmt"
"io"
"maps"
"net"
"github.com/LandaMm/hsp-go/hsp"
)
type Client struct {
type ClientOptions struct {
Headers map[string]string
// TODO: in future support multiple types of auth (credentials, key etc.)
Auth string
BaseURL string
}
func NewClient(headers map[string]string) *Client {
type Client struct {
Options *ClientOptions
Base *hsp.Adddress
}
func NewClient(options *ClientOptions) *Client {
if options == nil {
options = &ClientOptions{}
}
var base *hsp.Adddress
if len(options.BaseURL) > 0 {
addr, err := hsp.ParseAddress(options.BaseURL)
if err == nil {
base = addr
}
}
return &Client{
Headers: headers,
Options: options,
Base: base,
}
}
func (c *Client) BuildHeaders(address *hsp.Adddress, df *hsp.DataFormat) map[string]string {
headers := make(map[string]string)
if len(c.Headers) > 0 {
maps.Copy(headers, c.Headers)
if len(c.Options.Headers) > 0 {
maps.Copy(headers, c.Options.Headers)
}
headers[hsp.H_ROUTE] = address.Route
headers[hsp.H_DATA_FORMAT] = df.String()
if len(c.Options.Auth) > 0 {
headers[hsp.H_AUTH] = c.Options.Auth
}
return headers
}
@ -75,7 +102,14 @@ func (c *Client) SingleHit(addr *hsp.Adddress, pkt *hsp.Packet) (*hsp.Packet, er
}
func (c *Client) SendText(address, text string) (*hsp.Response, error) {
addr, err := hsp.ParseAddress(address)
var addr *hsp.Adddress
var err error
if c.Base != nil {
addr, err = c.Base.Extend(address)
} else {
addr, err = hsp.ParseAddress(address)
}
if err != nil {
return nil, err
@ -96,7 +130,14 @@ func (c *Client) SendText(address, text string) (*hsp.Response, error) {
}
func (c *Client) SendJson(address string, data any) (*hsp.Response, error) {
addr, err := hsp.ParseAddress(address)
var addr *hsp.Adddress
var err error
if c.Base != nil {
addr, err = c.Base.Extend(address)
} else {
addr, err = hsp.ParseAddress(address)
}
if err != nil {
return nil, err
@ -120,7 +161,14 @@ func (c *Client) SendJson(address string, data any) (*hsp.Response, error) {
}
func (c *Client) SendBytes(address string, data []byte) (*hsp.Response, error) {
addr, err := hsp.ParseAddress(address)
var addr *hsp.Adddress
var err error
if c.Base != nil {
addr, err = c.Base.Extend(address)
} else {
addr, err = hsp.ParseAddress(address)
}
if err != nil {
return nil, err