修改项目结构

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)
+}

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
+// wg
-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)
-}
-
 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 {

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")
+		certFile := util.Conf.Get("init", "cert_file")
-		keyFile := utils.Conf.Get("init", "key_file")
+		keyFile := util.Conf.Get("init", "key_file")
 		http.ListenAndServeTLS(addressArr[1], certFile, keyFile, nil)
 	} else {
 		http.ListenAndServe(addressArr[1], nil)

util/base64.go

@@ -1,4 +1,4 @@
-package utils
+package util
 
 import "encoding/base64"
 

util/config.go

@@ -1,4 +1,4 @@
-package utils
+package util
 
 import (
 	"github.com/go-ini/ini"

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)
-}