finish half of message

http.go

@@ -2,70 +2,63 @@ package nano
 
 import (
 	"bytes"
+	"encoding/base64"
 	"encoding/json"
 	"fmt"
 	"io"
+	"mime/multipart"
 	"net/http"
 	"net/url"
+	"os"
 	"reflect"
 	"strings"
+
+	base14 "github.com/fumiama/go-base16384"
 )
 
-// NewHTTPEndpointGetRequestWithAuth 新建带鉴权头的 HTTP GET 请求
-func NewHTTPEndpointGetRequestWithAuth(ep string, auth string) (req *http.Request, err error) {
-	req, err = http.NewRequest("GET", StandardAPI+ep, nil)
+func newHTTPEndpointRequestWithAuth(method, contenttype, ep string, auth string, body io.Reader) (req *http.Request, err error) {
+	req, err = http.NewRequest(method, StandardAPI+ep, body)
 	if err != nil {
 		return
 	}
 	req.Header.Add("Authorization", auth)
+	if contenttype == "" {
+		contenttype = "application/json"
+	}
+	req.Header.Add("Content-Type", contenttype)
 	return
 }
 
+// NewHTTPEndpointGetRequestWithAuth 新建带鉴权头的 HTTP GET 请求
+func NewHTTPEndpointGetRequestWithAuth(ep string, contenttype string, auth string) (*http.Request, error) {
+	return newHTTPEndpointRequestWithAuth("GET", contenttype, ep, auth, nil)
+}
+
 // NewHTTPEndpointPutRequestWithAuth 新建带鉴权头的 HTTP PUT 请求
-func NewHTTPEndpointPutRequestWithAuth(ep string, auth string, body io.Reader) (req *http.Request, err error) {
-	req, err = http.NewRequest("PUT", StandardAPI+ep, body)
-	if err != nil {
-		return
-	}
-	req.Header.Add("Authorization", auth)
-	return
+func NewHTTPEndpointPutRequestWithAuth(ep string, contenttype string, auth string, body io.Reader) (*http.Request, error) {
+	return newHTTPEndpointRequestWithAuth("PUT", contenttype, ep, auth, body)
 }
 
 // NewHTTPEndpointDeleteRequestWithAuth 新建带鉴权头的 HTTP DELETE 请求
-func NewHTTPEndpointDeleteRequestWithAuth(ep string, auth string, body io.Reader) (req *http.Request, err error) {
-	req, err = http.NewRequest("DELETE", StandardAPI+ep, body)
-	if err != nil {
-		return
-	}
-	req.Header.Add("Authorization", auth)
-	return
+func NewHTTPEndpointDeleteRequestWithAuth(ep string, contenttype string, auth string, body io.Reader) (*http.Request, error) {
+	return newHTTPEndpointRequestWithAuth("DELETE", contenttype, ep, auth, body)
 }
 
 // NewHTTPEndpointPostRequestWithAuth 新建带鉴权头的 HTTP POST 请求
-func NewHTTPEndpointPostRequestWithAuth(ep string, auth string, body io.Reader) (req *http.Request, err error) {
-	req, err = http.NewRequest("POST", StandardAPI+ep, body)
-	if err != nil {
-		return
-	}
-	req.Header.Add("Authorization", auth)
-	return
+func NewHTTPEndpointPostRequestWithAuth(ep string, contenttype string, auth string, body io.Reader) (*http.Request, error) {
+	return newHTTPEndpointRequestWithAuth("POST", contenttype, ep, auth, body)
 }
 
 // NewHTTPEndpointPatchRequestWithAuth 新建带鉴权头的 HTTP PATCH 请求
-func NewHTTPEndpointPatchRequestWithAuth(ep string, auth string, body io.Reader) (req *http.Request, err error) {
-	req, err = http.NewRequest("PATCH", StandardAPI+ep, body)
-	if err != nil {
-		return
-	}
-	req.Header.Add("Authorization", auth)
-	return
+func NewHTTPEndpointPatchRequestWithAuth(ep string, contenttype string, auth string, body io.Reader) (*http.Request, error) {
+	return newHTTPEndpointRequestWithAuth("PATCH", contenttype, ep, auth, body)
 }
 
 // WriteHTTPQueryIfNotNil 如果非空则将请求添加到 baseurl 后
 //
 // ex. WriteHTTPQueryIfNotNil("http://a.com/api", "a", 0, "b", 1, "c", 2) is http://a.com/api?b=1&c=2
 func WriteHTTPQueryIfNotNil(baseurl string, queries ...any) string {
-	if len(queries) == 0 {
+	if len(queries) < 2 {
 		return baseurl
 	}
 	hasstart := false
@@ -101,3 +94,82 @@ func WriteBodyFromJSON(ptr any) *bytes.Buffer {
 	_ = json.NewEncoder(buf).Encode(ptr)
 	return buf
 }
+
+// WriteBodyByMultipartFormData 使用 multipart/form-data 上传
+func WriteBodyByMultipartFormData(params ...any) (*bytes.Buffer, string, error) {
+	if len(params)%2 != 0 {
+		panic("invalid params to " + getThisFuncName())
+	}
+	fieldname := ""
+	buf := bytes.NewBuffer(make([]byte, 0, 65536))
+	w := multipart.NewWriter(buf)
+	defer w.Close()
+	for i, x := range params {
+		if i%2 == 0 { // 参数
+			fieldname = x.(string)
+			continue
+		}
+		rx := reflect.ValueOf(x)
+		if rx.IsZero() {
+			continue
+		}
+		r, err := w.CreateFormField(fieldname)
+		if err != nil {
+			return nil, "", err
+		}
+		if rx.Elem().Kind() == reflect.Struct { // 使用 json 编码
+			err = json.NewEncoder(r).Encode(x)
+			if err != nil {
+				return nil, "", err
+			}
+			continue
+		}
+		switch o := x.(type) {
+		case string:
+			if strings.HasPrefix(o, "file:///") { // 是文件路径
+				f, err := os.Open(o[8:])
+				if err != nil {
+					return nil, "", err
+				}
+				defer f.Close()
+				_, err = io.Copy(r, f)
+				if err != nil {
+					return nil, "", err
+				}
+				continue
+			}
+			if strings.HasPrefix(o, "base64://") { // 是 base64
+				_, err = io.Copy(r, base64.NewDecoder(base64.StdEncoding, bytes.NewBufferString(o[9:])))
+				if err != nil {
+					return nil, "", err
+				}
+				continue
+			}
+			if strings.HasPrefix(o, "base16384://") { // 是 base16384
+				_, err = io.Copy(r, base14.NewDecoder(bytes.NewBufferString(o[12:])))
+				if err != nil {
+					return nil, "", err
+				}
+				continue
+			}
+			_, err = io.WriteString(r, o)
+			if err != nil {
+				return nil, "", err
+			}
+			continue
+		case []byte:
+			_, err = r.Write(o)
+			if err != nil {
+				return nil, "", err
+			}
+			continue
+		default:
+			_, err = io.WriteString(r, fmt.Sprint(o))
+			if err != nil {
+				return nil, "", err
+			}
+			continue
+		}
+	}
+	return buf, w.FormDataContentType(), nil
+}