From 2b7291d7f6c2070078dab1ac4329346714bb754d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?TOP=E7=B3=AF=E7=B1=B3?= <1130395124@qq.com> Date: Mon, 27 Mar 2023 23:47:46 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E9=A1=B9=E7=9B=AE=E7=BB=93?= =?UTF-8?q?=E6=9E=84?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- core/data.go | 87 +++++++++++++++++++++++++++++++++++++++ core/request.go | 61 ++++----------------------- main.go | 10 ++--- {utils => util}/base64.go | 2 +- {utils => util}/config.go | 2 +- {utils => util}/helper.go | 35 +--------------- 6 files changed, 103 insertions(+), 94 deletions(-) create mode 100644 core/data.go rename {utils => util}/base64.go (96%) rename {utils => util}/config.go (97%) rename {utils => util}/helper.go (56%) diff --git a/core/data.go b/core/data.go new file mode 100644 index 0000000..7aebaed --- /dev/null +++ b/core/data.go @@ -0,0 +1,87 @@ +package core + +import ( + "chatgpt/util" + "encoding/json" + "fmt" + "reflect" + "sort" + "strings" +) + +// RequestData +type RequestData struct { + Words string `json:"words"` + Time int `json:"time"` + Sign string `json:"sign"` +} + +// Response +// 响应体 +type Response struct { + Code int `json:"code"` + Msg string `json:"msg"` + Data string `json:"data"` + Sign string `json:"sign"` +} + +// ResponseData +// 响应数据 +type ResponseData map[string]interface{} + +// BuildResponse +// 构建响应体 +// +// @param privateKey +// @param code +// @param msg +// @param data +// @return string +func BuildResponse(privateKey string, code int, msg string, data ResponseData) string { + resp := Response{ + Code: code, + Msg: msg, + } + dataJson, _ := json.Marshal(data) + resp.Data = util.Base64{ + Content: []byte(dataJson), + }.Encode() + if privateKey != "" { + resp.Sign = MakeSign(resp, privateKey) + } + + respJson, _ := json.Marshal(resp) + + return string(respJson) +} + +// MakeSign +// 生成签名 +// +// @param obj +// @param privateKey +// @return string +func MakeSign(obj interface{}, privateKey string) string { + valueOfObj := reflect.ValueOf(obj) + keysCount := valueOfObj.NumField() + + keys := make([]string, keysCount) + for i := 0; i < keysCount; i++ { + keys[i] = valueOfObj.Type().Field(i).Name + } + sort.Strings(keys) + + str := "" + currentKey := "" + var currentValue reflect.Value + for j := 0; j < keysCount; j++ { + currentKey = strings.ToLower(keys[j]) + if currentKey == "sign" { + continue + } + currentValue = valueOfObj.FieldByName(keys[j]) + str = fmt.Sprintf("%s%s=%v&", str, currentKey, currentValue) + } + + return util.Md5(str + "key=" + privateKey) +} diff --git a/core/request.go b/core/request.go index fbf0e18..bb1f66b 100644 --- a/core/request.go +++ b/core/request.go @@ -1,7 +1,7 @@ package core import ( - "chatgpt/utils" + "chatgpt/util" "encoding/json" "fmt" "io/ioutil" @@ -10,69 +10,24 @@ import ( "sync" ) -// RequestData -type RequestData struct { - Words string `json:"words"` - Time int `json:"time"` - Sign string `json:"sign"` -} - -// Response -// 响应体 -type Response struct { - Code int `json:"code"` - Msg string `json:"msg"` - Data string `json:"data"` - Sign string `json:"sign"` -} - -// ResponseData -// 响应数据 -type ResponseData map[string]interface{} - -// BuildResponse -// 构建响应体 -// -// @param privateKey -// @param code -// @param msg -// @param data -// @return string -func BuildResponse(privateKey string, code int, msg string, data ResponseData) string { - resp := Response{ - Code: code, - Msg: msg, - } - dataJson, _ := json.Marshal(data) - resp.Data = utils.Base64{ - Content: []byte(dataJson), - }.Encode() - if privateKey != "" { - resp.Sign = utils.MakeSign(resp, privateKey) - } - - respJson, _ := json.Marshal(resp) - - return string(respJson) -} - +// wg var wg sync.WaitGroup // proxyAddr // 代理配置 -var proxyAddr string = utils.GetProxyServer() +var proxyAddr string = util.GetProxyServer() // apiKey // GPT Api Key -var apiKey string = utils.Conf.Get("gpt", "api_key") +var apiKey string = util.Conf.Get("gpt", "api_key") // privateKey // 签名密钥 -var privateKey string = utils.Conf.Get("init", "private_key") +var privateKey string = util.Conf.Get("init", "private_key") // TimeoutValue // 超时配置 -var TimeoutValue string = utils.Conf.Get("init", "timeout") +var TimeoutValue string = util.Conf.Get("init", "timeout") // Action // 处理请求 @@ -89,12 +44,12 @@ func Handler(w http.ResponseWriter, r *http.Request) { raw, _ := ioutil.ReadAll(r.Body) json.Unmarshal(raw, &requestData) - if privateKey != "" && requestData.Sign != utils.MakeSign(requestData, privateKey) { + if privateKey != "" && requestData.Sign != MakeSign(requestData, privateKey) { fmt.Fprintln(w, BuildResponse(privateKey, 0, "sign error", ResponseData{})) return } - question := utils.Base64{ + question := util.Base64{ Content: []byte(requestData.Words), }.Decode() if len(question) > 0 { diff --git a/main.go b/main.go index 1b553eb..49b78b0 100644 --- a/main.go +++ b/main.go @@ -2,18 +2,18 @@ package main import ( "chatgpt/core" - "chatgpt/utils" + "chatgpt/util" "net/http" ) // main func main() { - utils.Conf = &utils.Config{} + util.Conf = &util.Config{} http.HandleFunc("/", core.Handler) - addressArr := utils.GetListenAddress() + addressArr := util.GetListenAddress() if addressArr[0] == "https" { - certFile := utils.Conf.Get("init", "cert_file") - keyFile := utils.Conf.Get("init", "key_file") + certFile := util.Conf.Get("init", "cert_file") + keyFile := util.Conf.Get("init", "key_file") http.ListenAndServeTLS(addressArr[1], certFile, keyFile, nil) } else { http.ListenAndServe(addressArr[1], nil) diff --git a/utils/base64.go b/util/base64.go similarity index 96% rename from utils/base64.go rename to util/base64.go index 4704a27..24ae7a4 100644 --- a/utils/base64.go +++ b/util/base64.go @@ -1,4 +1,4 @@ -package utils +package util import "encoding/base64" diff --git a/utils/config.go b/util/config.go similarity index 97% rename from utils/config.go rename to util/config.go index a7a1404..6459e9b 100644 --- a/utils/config.go +++ b/util/config.go @@ -1,4 +1,4 @@ -package utils +package util import ( "github.com/go-ini/ini" diff --git a/utils/helper.go b/util/helper.go similarity index 56% rename from utils/helper.go rename to util/helper.go index 90f15cc..f5e8773 100644 --- a/utils/helper.go +++ b/util/helper.go @@ -1,10 +1,8 @@ -package utils +package util import ( "crypto/md5" "fmt" - "reflect" - "sort" "strings" ) @@ -54,34 +52,3 @@ func Md5(text string) string { byteData := hash.Sum(nil) return fmt.Sprintf("%x", byteData) } - -// MakeSign -// 生成签名 -// -// @param obj -// @param privateKey -// @return string -func MakeSign(obj interface{}, privateKey string) string { - valueOfObj := reflect.ValueOf(obj) - keysCount := valueOfObj.NumField() - - keys := make([]string, keysCount) - for i := 0; i < keysCount; i++ { - keys[i] = valueOfObj.Type().Field(i).Name - } - sort.Strings(keys) - - str := "" - currentKey := "" - var currentValue reflect.Value - for j := 0; j < keysCount; j++ { - currentKey = strings.ToLower(keys[j]) - if currentKey == "sign" { - continue - } - currentValue = valueOfObj.FieldByName(keys[j]) - str = fmt.Sprintf("%s%s=%v&", str, currentKey, currentValue) - } - - return Md5(str + "key=" + privateKey) -}