From 58ea1ad890be69bd45f5910d03bde25cb8ebedc0 Mon Sep 17 00:00:00 2001
From: LandaMm <qwer.009771@gmail.com>
Date: Sat, 19 Apr 2025 12:38:26 +0200
Subject: [PATCH] feat: build commonly used headers

---
 hsp/client/client.go | 90 +++++++++++++++++++++++++++++++++++---------
 1 file changed, 72 insertions(+), 18 deletions(-)

diff --git a/hsp/client/client.go b/hsp/client/client.go
index 454733f..3167a6f 100644
--- a/hsp/client/client.go
+++ b/hsp/client/client.go
@@ -1,6 +1,8 @@
 package client
 
 import (
+	"encoding/json"
+	"maps"
 	"net"
 
 	"github.com/LandaMm/hsp-go/hsp"
@@ -16,26 +18,20 @@ func NewClient() *Client {
 	}
 }
 
-func (c *Client) SendText(address, text string) (*hsp.Response, error) {
-	addr, err := hsp.ParseAddress(address)
-
-	if err != nil {
-		return nil, err
-	}
-
-	df := hsp.DataFormat{
-		Format: hsp.DF_TEXT,
-		Encoding: hsp.E_UTF8,
-	}
-
+func (c *Client) BuildHeaders(address *hsp.Adddress, df *hsp.DataFormat, mergeHeaders *map[string]string) map[string]string {
 	headers := make(map[string]string)
-	headers[hsp.H_ROUTE] = addr.Route
+
+	if mergeHeaders != nil {
+		maps.Copy(headers, *mergeHeaders)
+	}
+
+	headers[hsp.H_ROUTE] = address.Route
 	headers[hsp.H_DATA_FORMAT] = df.String()
 
-	payload := []byte(text)
-
-	pkt := hsp.BuildPacket(headers, payload)
+	return headers
+}
 
+func (c *Client) SingleHit(addr *hsp.Adddress, pkt *hsp.Packet) (*hsp.Packet, error) {
 	conn, err := net.Dial("tcp", addr.String())
 	if err != nil {
 		return nil, err
@@ -46,11 +42,69 @@ func (c *Client) SendText(address, text string) (*hsp.Response, error) {
 		return nil, err
 	}
 
-	pkt, err = c.Duplex.ReadPacket()
+	return c.Duplex.ReadPacket()
+}
+
+func (c *Client) SendText(address, text string) (*hsp.Response, error) {
+	addr, err := hsp.ParseAddress(address)
+
 	if err != nil {
 		return nil, err
 	}
 
-	return hsp.NewPacketResponse(pkt), nil
+	payload := []byte(text)
+
+	hdrs := c.BuildHeaders(addr, hsp.TextDataFormat(), nil)
+
+	pkt := hsp.BuildPacket(hdrs, payload)
+
+	rpkt, err := c.SingleHit(addr, pkt)
+	if err != nil {
+		return nil, err
+	}
+
+	return hsp.NewPacketResponse(rpkt), nil
 }
 
+func (c *Client) SendJson(address string, data any) (*hsp.Response, error) {
+	addr, err := hsp.ParseAddress(address)
+
+	if err != nil {
+		return nil, err
+	}
+
+	payload, err := json.Marshal(data)
+	if err != nil {
+		return nil, err
+	}
+
+	hdrs := c.BuildHeaders(addr, hsp.JsonDataFormat(), nil)
+
+	pkt := hsp.BuildPacket(hdrs, payload)
+
+	rpkt, err := c.SingleHit(addr, pkt)
+	if err != nil {
+		return nil, err
+	}
+
+	return hsp.NewPacketResponse(rpkt), nil
+}
+
+func (c *Client) SendBytes(address string, data []byte) (*hsp.Response, error) {
+	addr, err := hsp.ParseAddress(address)
+
+	if err != nil {
+		return nil, err
+	}
+
+	hdrs := c.BuildHeaders(addr, hsp.BytesDataFormat(), nil)
+
+	pkt := hsp.BuildPacket(hdrs, data)
+
+	rpkt, err := c.SingleHit(addr, pkt)
+	if err != nil {
+		return nil, err
+	}
+
+	return hsp.NewPacketResponse(rpkt), nil
+}