From fdf5a517ed017ba25e8fe5a76a679bdece7fa8a0 Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Wed, 18 Dec 2024 09:57:31 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E7=9B=B4=E8=BF=9E=E7=BD=91=E5=85=83?=
 =?UTF-8?q?=E8=AF=B7=E6=B1=82=E5=B8=A6UA=E6=A0=87=E8=AF=86OMC/version?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/framework/utils/fetch/fetch.go | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/src/framework/utils/fetch/fetch.go b/src/framework/utils/fetch/fetch.go
index be7cc07c..d0075b0f 100644
--- a/src/framework/utils/fetch/fetch.go
+++ b/src/framework/utils/fetch/fetch.go
@@ -12,8 +12,13 @@ import (
 	"os"
 	"strings"
 	"time"
+
+	libGlobal "be.ems/lib/global"
 )
 
+// userAgent 自定义 User-Agent
+var userAgent = fmt.Sprintf("OMC/%s", libGlobal.Version)
+
 // Get 发送 GET 请求
 // timeout 超时时间（毫秒）
 func Get(url string, headers map[string]string, timeout int) ([]byte, error) {
@@ -29,6 +34,8 @@ func Get(url string, headers map[string]string, timeout int) ([]byte, error) {
 		return nil, err
 	}
 
+	req.Header.Set("User-Agent", userAgent)
+	req.Header.Set("Content-Type", "application/json;charset=UTF-8")
 	for key, value := range headers {
 		req.Header.Set(key, value)
 	}
@@ -60,8 +67,8 @@ func Post(url string, data url.Values, headers map[string]string) ([]byte, error
 		return nil, err
 	}
 
+	req.Header.Set("User-Agent", userAgent)
 	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
-
 	for key, value := range headers {
 		req.Header.Set(key, value)
 	}
@@ -100,8 +107,8 @@ func PostJSON(url string, data any, headers map[string]string) ([]byte, error) {
 		return nil, err
 	}
 
-	req.Header.Set("Content-Type", "application/json")
-
+	req.Header.Set("User-Agent", userAgent)
+	req.Header.Set("Content-Type", "application/json;charset=UTF-8")
 	for key, value := range headers {
 		req.Header.Set(key, value)
 	}
@@ -156,6 +163,7 @@ func PostUploadFile(url string, params map[string]string, file *os.File) ([]byte
 		return nil, fmt.Errorf("failed to create HTTP request: %v", err)
 	}
 
+	req.Header.Set("User-Agent", userAgent)
 	req.Header.Set("Content-Type", writer.FormDataContentType())
 
 	client := &http.Client{}
@@ -193,6 +201,8 @@ func PutJSON(url string, data any, headers map[string]string) ([]byte, error) {
 		return nil, err
 	}
 
+	req.Header.Set("User-Agent", userAgent)
+	req.Header.Set("Content-Type", "application/json;charset=UTF-8")
 	for key, value := range headers {
 		req.Header.Set(key, value)
 	}
@@ -224,6 +234,8 @@ func Delete(url string, headers map[string]string) ([]byte, error) {
 		return nil, err
 	}
 
+	req.Header.Set("User-Agent", userAgent)
+	req.Header.Set("Content-Type", "application/json;charset=UTF-8")
 	for key, value := range headers {
 		req.Header.Set(key, value)
 	}