添加日志、初始化

.gitignore

@@ -1,2 +1,4 @@
 /tmp
 /scripts
+/ProjectWIND
+/data

main.go

@@ -2,16 +2,174 @@ package main
 
 import (
 	"ProjectWIND/protocol"
+	"ProjectWIND/typed"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io"
 	"log"
+	"os"
 )
 
 func init() {
+	// 初始化日志记录器
+	log.SetFlags(log.Ldate | log.Ltime)
+	log.SetPrefix("[WIND] ")
 
-}
-
-func main() {
-	_, err := protocol.WebSocketHandler()
+	// 检查并创建必要的目录和文件
+	if _, err := os.Stat("./data/"); os.IsNotExist(err) {
+		// 如果不存在，则创建该文件夹
+		err := os.Mkdir("./data/", 0755)
 		if err != nil {
 			log.Fatal(err)
 		}
 	}
+
+	// 检查./data/文件夹中是否存在config.json文件
+	if _, err := os.Stat("./data/config.json"); os.IsNotExist(err) {
+		// 如果不存在，则创建该文件
+		file, err := os.Create("./data/config.json")
+		if err != nil {
+			log.Fatal(err)
+		}
+		defer func(file *os.File) {
+			err := file.Close()
+			if err != nil {
+				log.Fatal(err)
+			}
+		}(file)
+	}
+	// 检查并更新配置文件
+	err := checkAndUpdateConfig("./data/config.json")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	// 检查./data/文件夹中是否存在app/文件夹
+	if _, err := os.Stat("./data/app/"); os.IsNotExist(err) {
+		// 如果不存在，则创建该文件夹
+		err := os.Mkdir("./data/app/", 0755)
+		if err != nil {
+			log.Fatal(err)
+		}
+	}
+
+	// 检查./data/文件夹中是否存在database/文件夹
+	if _, err := os.Stat("./data/database/"); os.IsNotExist(err) {
+		// 如果不存在，则创建该文件夹
+		err := os.Mkdir("./data/database/", 0755)
+		if err != nil {
+			log.Fatal(err)
+		}
+	}
+
+	// 检查./data/文件夹中是否存在log/文件夹
+	if _, err := os.Stat("./data/log/"); os.IsNotExist(err) {
+		// 如果不存在，则创建该文件夹
+		err := os.Mkdir("./data/log/", 0755)
+		if err != nil {
+			log.Fatal(err)
+		}
+	}
+}
+
+func main() {
+	//读取参数
+	if len(os.Args) == 1 || os.Args[0] == "help" || os.Args[0] == "-h" || os.Args[0] == "--help" {
+		fmt.Println(`Usage: input command "start"or"-s" to start the server.`)
+		return
+	}
+	command := os.Args[1]
+	switch command {
+	case "start":
+	case "-s":
+		{
+			log.Println("Starting ProjectWIND...")
+			// 启动 WebSocket 处理程序
+			log.Println("Starting WebSocket handler...")
+			_, err := protocol.WebSocketHandler()
+			if err != nil {
+				// 如果发生错误，记录错误并退出程序
+				log.Fatal(err)
+			}
+			return
+		}
+	default:
+		{
+			fmt.Println("Invalid command.")
+			return
+		}
+	}
+}
+func checkAndUpdateConfig(configPath string) error {
+	var config typed.ConfigInfo
+	// 定义默认配置
+	var defaultConfig typed.ConfigInfo
+	defaultConfig.CoreName = "windCore"
+	defaultConfig.WebUIPort = 3211
+	defaultConfig.ProtocolAddr = make(map[string]string)
+	defaultConfig.ServiceName = "wind"
+	// 读取配置文件
+	file, err := os.Open(configPath)
+	if err != nil {
+		return err
+	}
+	defer func(file *os.File) {
+		err := file.Close()
+		if err != nil {
+			log.Printf("Failed to close config file: %v", err)
+		}
+	}(file)
+
+	// 解码JSON配置
+	decoder := json.NewDecoder(file)
+	err = decoder.Decode(&config)
+	if err != nil {
+		if !errors.Is(err, io.EOF) {
+			return err
+		}
+	}
+
+	// 检查并更新配置
+	if config.ProtocolAddr == nil {
+		config.ProtocolAddr = defaultConfig.ProtocolAddr
+	}
+	if config.WebUIPort == 0 {
+		config.WebUIPort = defaultConfig.WebUIPort
+	}
+	if config.CoreName == "" {
+		config.CoreName = defaultConfig.CoreName
+	}
+	if config.ServiceName == "" {
+		config.ServiceName = defaultConfig.ServiceName
+	}
+	if config.PasswordHash == "" {
+		config.PasswordHash = ""
+	}
+
+	formattedJSON, err := json.MarshalIndent(config, "", "  ")
+	if err != nil {
+		fmt.Println("Error formatting JSON:", err)
+		return err
+	}
+
+	// 将格式化后的JSON字符串写入文件
+	file, err = os.Create("./data/config.json")
+	if err != nil {
+		fmt.Println("Error creating file:", err)
+		return err
+	}
+	defer func(file *os.File) {
+		err := file.Close()
+		if err != nil {
+			log.Printf("Failed to close config file: %v", err)
+		}
+	}(file)
+
+	_, err = file.Write(formattedJSON)
+	if err != nil {
+		fmt.Println("Error writing to file:", err)
+		return err
+	}
+	return nil
+}

protocol/api.go

@@ -12,7 +12,7 @@ func init() {
 
 func SendMessage(messageType string, message string, targetId int64, autoEscape bool) (bool, error) {
 	// 构建发送消息的JSON数据
-	var messageData typed.APIRequest
+	var messageData typed.APIRequestInfo
 	messageData.Action = "send_msg"
 	switch messageType {
 	case "private":

protocol/web_socket.go

@@ -10,7 +10,7 @@ import (
 )
 
 const (
-	severURL = "ws://110.41.4.138:3001"
+	severURL = "your_url"
 )
 
 func WebSocketHandler() (*websocket.Conn, error) {
@@ -32,6 +32,8 @@ func WebSocketHandler() (*websocket.Conn, error) {
 		}
 	}(conn)
 
+	log.Println("[INFO] New connection established.")
+
 	for {
 		messageType, message, err := conn.ReadMessage()
 		if err != nil {

typed/typed.go

@@ -1,5 +1,13 @@
 package typed
 
+type ConfigInfo struct {
+	CoreName     string            `json:"core_name"`
+	ProtocolAddr map[string]string `json:"protocol_addr"`
+	WebUIPort    uint16            `json:"webui_port"`
+	PasswordHash string            `json:"password_hash"`
+	ServiceName  string            `json:"service_name"`
+}
+
 type MessageEventInfo struct {
 	Time        int64         `json:"time,omitempty"`
 	SelfId      int64         `json:"self_id,omitempty"`
@@ -101,7 +109,7 @@ type MessageDataInfo struct {
 	Data    string `json:"data,omitempty"`
 }
 
-type APIRequest struct {
+type APIRequestInfo struct {
 	Action string     `json:"action,omitempty"`
 	Params ParamsInfo `json:"params"`
 	Echo   string     `json:"echo,omitempty"`