fix(http): refactor JSON request body

Author: iseki0

api/client.go

@@ -1,7 +1,6 @@
 package api
 
 import (
-	"bytes"
 	"context"
 	"encoding/json"
 	"github.com/murphysecurity/murphysec/utils/must"
@@ -115,8 +114,10 @@ func (c *Client) POST(url *url.URL, body io.Reader) *http.Request {
 }
 
 func (c *Client) PostJson(url *url.URL, data any) *http.Request {
-	u := c.POST(url, bytes.NewReader(must.A(json.Marshal(data))))
-	u.Header.Set("Content-Type", "application/json")
+	var body = NewJsonRequestBody(data)
+	u := c.POST(url, body)
+	u.GetBody = body.GetBody
+	u.Header.Set("Content-Type", "application/json; charset=utf-8")
 	return u
 }
 

api/json_request_body.go

@@ -0,0 +1,75 @@
+package api
+
+import (
+	"bufio"
+	"encoding/json"
+	"errors"
+	"io"
+	"sync/atomic"
+)
+
+type JsonRequestBody interface {
+	GetBody() (io.ReadCloser, error)
+	io.ReadCloser
+}
+
+func NewJsonRequestBody(V any) JsonRequestBody {
+	return &jsonReqBody{v: V}
+}
+
+type jsonReqBody struct {
+	v      any
+	pr     *io.PipeReader
+	closed atomic.Bool
+}
+
+func (j *jsonReqBody) GetBody() (io.ReadCloser, error) {
+	return &jsonReqBody{v: j.v}, nil
+}
+
+func (j *jsonReqBody) Read(p []byte) (n int, err error) {
+	if j.closed.Load() {
+		return 0, io.ErrClosedPipe
+	}
+	if j.pr == nil {
+		var pw *io.PipeWriter
+		j.pr, pw = io.Pipe()
+		go func() {
+			defer func() {
+				r := recover()
+				if r != nil {
+					switch r.(type) {
+					case error:
+						_ = pw.CloseWithError(r.(error))
+					default:
+						_ = pw.CloseWithError(errors.New("panic in json encoding routing"))
+					}
+				}
+			}()
+			var bw = bufio.NewWriter(pw)
+			defer func() { _ = pw.Close() }()
+			var e = json.NewEncoder(pw).Encode(j.v)
+			if e != nil {
+				_ = pw.CloseWithError(e)
+				return
+			}
+			e = bw.Flush()
+			if e != nil {
+				_ = pw.CloseWithError(e)
+				return
+			}
+			_ = pw.Close()
+		}()
+	}
+	return j.pr.Read(p)
+}
+
+func (j *jsonReqBody) Close() error {
+	if j.closed.Load() {
+		return io.ErrClosedPipe
+	}
+	j.closed.Store(true)
+	return j.pr.Close()
+}
+
+var _ io.ReadCloser = &jsonReqBody{}