Merge

sync

global/config.go

@@ -35,6 +35,7 @@ type JsonConfig struct {
 	PostMessageFormat   string                        `json:"post_message_format"`
 	Debug               bool                          `json:"debug"`
 	LogLevel            string                        `json:"log_level"`
+	WebUi               *GoCqWebUi                    `json:"web_ui"`
 }
 
 type CQHttpApiConfig struct {
@@ -78,6 +79,13 @@ type GoCQReverseWebsocketConfig struct {
 	ReverseReconnectInterval uint16 `json:"reverse_reconnect_interval"`
 }
 
+type GoCqWebUi struct {
+	Enabled   bool   `json:"enabled"`
+	Host      string `json:"host"`
+	WebUiPort uint64 `json:"web_ui_port"`
+	WebInput  bool   `json:"web_input"`
+}
+
 func DefaultConfig() *JsonConfig {
 	return &JsonConfig{
 		EnableDB: true,
@@ -121,6 +129,12 @@ func DefaultConfig() *JsonConfig {
 				ReverseReconnectInterval: 3000,
 			},
 		},
+		WebUi: &GoCqWebUi{
+			Enabled:   true,
+			Host:      "0.0.0.0",
+			WebInput:  false,
+			WebUiPort: 9999,
+		},
 	}
 }
 

global/fs.go

@@ -94,3 +94,16 @@ func FindFile(f, cache, PATH string) (data []byte, err error) {
 	}
 	return
 }
+
+func DelFile(path string) bool {
+	err := os.Remove(path)
+	if err != nil {
+		// 删除失败
+		log.Error(err)
+		return false
+	} else {
+		// 删除成功
+		log.Info(path + "删除成功")
+		return true
+	}
+}

go.mod

@@ -3,7 +3,7 @@ module github.com/Mrs4s/go-cqhttp
 go 1.14
 
 require (
-	github.com/Mrs4s/MiraiGo v0.0.0-20201002092940-afe4132a38d5
+	github.com/Mrs4s/MiraiGo v0.0.0-20201003051902-8a968656c116
 	github.com/fastly/go-utils v0.0.0-20180712184237-d95a45783239 // indirect
 	github.com/gin-gonic/gin v1.6.3
 	github.com/go-playground/validator/v10 v10.4.0 // indirect

go.sum

@@ -4,6 +4,8 @@ github.com/Mrs4s/MiraiGo v0.0.0-20201002083653-383954dc11e2 h1:Pjs3pQKXDxvaWf3zT
 github.com/Mrs4s/MiraiGo v0.0.0-20201002083653-383954dc11e2/go.mod h1:cwYPI2uq6nxNbx0nA6YuAKF1V5szSs6FPlGVLQvRUlo=
 github.com/Mrs4s/MiraiGo v0.0.0-20201002092940-afe4132a38d5 h1:NVXpGL7tssdjYBAsPRvCIL8f8V9e8n9YO8sSsOwdy44=
 github.com/Mrs4s/MiraiGo v0.0.0-20201002092940-afe4132a38d5/go.mod h1:cwYPI2uq6nxNbx0nA6YuAKF1V5szSs6FPlGVLQvRUlo=
+github.com/Mrs4s/MiraiGo v0.0.0-20201003051902-8a968656c116 h1:GQHsohWh9LxaWmHIVS2tKzZZSysJBrnS+Sh5jXZ+5zk=
+github.com/Mrs4s/MiraiGo v0.0.0-20201003051902-8a968656c116/go.mod h1:cwYPI2uq6nxNbx0nA6YuAKF1V5szSs6FPlGVLQvRUlo=
 github.com/bwmarrin/snowflake v0.3.0 h1:xm67bEhkKh6ij1790JB83OujPR5CzNe8QuQqAgISZN0=
 github.com/bwmarrin/snowflake v0.3.0/go.mod h1:NdZxfVWX+oR6y2K0o6qAYv6gIOP9rjG0/E9WsDpxqwE=
 github.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU=

main.go

@@ -2,32 +2,27 @@ package main
 
 import (
 	"bufio"
-	"bytes"
 	"crypto/md5"
 	"encoding/base64"
 	"encoding/json"
 	"fmt"
-	"image"
+	"github.com/Mrs4s/go-cqhttp/server"
 	"io"
 	"io/ioutil"
 	"os"
 	"os/signal"
 	"path"
 	"strconv"
-	"strings"
 	"time"
 
 	"github.com/Mrs4s/MiraiGo/binary"
 	"github.com/Mrs4s/MiraiGo/client"
 	"github.com/Mrs4s/go-cqhttp/coolq"
 	"github.com/Mrs4s/go-cqhttp/global"
-	"github.com/Mrs4s/go-cqhttp/server"
+	"github.com/lestrrat-go/file-rotatelogs"
-
-	rotatelogs "github.com/lestrrat-go/file-rotatelogs"
 	"github.com/rifflock/lfshook"
 	log "github.com/sirupsen/logrus"
-	easy "github.com/t-tomalak/logrus-easy-formatter"
+	"github.com/t-tomalak/logrus-easy-formatter"
-	asciiart "github.com/yinghau76/go-ascii-art"
 )
 
 func init() {
@@ -238,109 +233,26 @@ func main() {
 	cli.OnServerUpdated(func(bot *client.QQClient, e *client.ServerUpdatedEvent) {
 		log.Infof("收到服务器地址更新通知, 将在下一次重连时应用. ")
 	})
-	rsp, err := cli.Login()
+	if conf.WebUi == nil {
-	for {
+		conf.WebUi = &global.GoCqWebUi{
-		global.Check(err)
+			Enabled:   true,
-		if !rsp.Success {
+			WebInput:  false,
-			switch rsp.Error {
+			Host:      "0.0.0.0",
-			case client.NeedCaptcha:
+			WebUiPort: 9999,
-				_ = ioutil.WriteFile("captcha.jpg", rsp.CaptchaImage, 0644)
-				img, _, _ := image.Decode(bytes.NewReader(rsp.CaptchaImage))
-				fmt.Println(asciiart.New("image", img).Art)
-				log.Warn("请输入验证码 (captcha.jpg)： (Enter 提交)")
-				text, _ := console.ReadString('\n')
-				rsp, err = cli.SubmitCaptcha(strings.ReplaceAll(text, "\n", ""), rsp.CaptchaSign)
-				continue
-			case client.UnsafeDeviceError:
-				log.Warnf("账号已开启设备锁，请前往 -> %v <- 验证并重启Bot.", rsp.VerifyUrl)
-				log.Infof(" 按 Enter 继续....")
-				_, _ = console.ReadString('\n')
-				return
-			case client.OtherLoginError, client.UnknownLoginError:
-				log.Fatalf("登录失败: %v", rsp.ErrorMessage)
 		}
 	}
-		break
+	if conf.WebUi.WebUiPort <= 0 {
+		conf.WebUi.WebUiPort = 9999
 	}
-	log.Infof("登录成功 欢迎使用: %v", cli.Nickname)
+	if conf.WebUi.Host == "" {
-	time.Sleep(time.Second)
+		conf.WebUi.Host = "0.0.0.0"
-	log.Info("开始加载好友列表...")
-	global.Check(cli.ReloadFriendList())
-	log.Infof("共加载 %v 个好友.", len(cli.FriendList))
-	log.Infof("开始加载群列表...")
-	global.Check(cli.ReloadGroupList())
-	log.Infof("共加载 %v 个群.", len(cli.GroupList))
-	b := coolq.NewQQBot(cli, conf)
-	if conf.PostMessageFormat != "string" && conf.PostMessageFormat != "array" {
-		log.Warnf("post_message_format 配置错误, 将自动使用 string")
-		coolq.SetMessageFormat("string")
-	} else {
-		coolq.SetMessageFormat(conf.PostMessageFormat)
 	}
-	if conf.RateLimit.Enabled {
+	confErr := conf.Save("config.json")
-		global.InitLimiter(conf.RateLimit.Frequency, conf.RateLimit.BucketSize)
+	if confErr != nil {
+		log.Error("保存配置文件失败")
 	}
-	log.Info("正在加载事件过滤器.")
+	b := server.WebServer.Run(fmt.Sprintf("%s:%d", conf.WebUi.Host, conf.WebUi.WebUiPort), cli)
-	global.BootFilter()
+	c := server.Console
-	coolq.IgnoreInvalidCQCode = conf.IgnoreInvalidCQCode
-	coolq.ForceFragmented = conf.ForceFragmented
-	if conf.HttpConfig != nil && conf.HttpConfig.Enabled {
-		server.HttpServer.Run(fmt.Sprintf("%s:%d", conf.HttpConfig.Host, conf.HttpConfig.Port), conf.AccessToken, b)
-		for k, v := range conf.HttpConfig.PostUrls {
-			server.NewHttpClient().Run(k, v, conf.HttpConfig.Timeout, b)
-		}
-	}
-	if conf.WSConfig != nil && conf.WSConfig.Enabled {
-		server.WebsocketServer.Run(fmt.Sprintf("%s:%d", conf.WSConfig.Host, conf.WSConfig.Port), conf.AccessToken, b)
-	}
-	for _, rc := range conf.ReverseServers {
-		server.NewWebsocketClient(rc, conf.AccessToken, b).Run()
-	}
-	log.Info("资源初始化完成, 开始处理信息.")
-	log.Info("アトリは、高性能ですから!")
-	cli.OnDisconnected(func(bot *client.QQClient, e *client.ClientDisconnectedEvent) {
-		if conf.ReLogin.Enabled {
-			var times uint = 1
-			for {
-
-				if cli.Online {
-					log.Warn("Bot已登录")
-					return
-				}
-				if conf.ReLogin.MaxReloginTimes != 0 && times > conf.ReLogin.MaxReloginTimes {
-					break
-				}
-
-				log.Warnf("Bot已离线 (%v)，将在 %v 秒后尝试重连. 重连次数：%v",
-					e.Message, conf.ReLogin.ReLoginDelay, times)
-				times++
-				time.Sleep(time.Second * time.Duration(conf.ReLogin.ReLoginDelay))
-				rsp, err := cli.Login()
-				if err != nil {
-					log.Errorf("重连失败: %v", err)
-					continue
-				}
-				if !rsp.Success {
-					switch rsp.Error {
-					case client.NeedCaptcha:
-						log.Fatalf("重连失败: 需要验证码. (验证码处理正在开发中)")
-					case client.UnsafeDeviceError:
-						log.Fatalf("重连失败: 设备锁")
-					default:
-						log.Errorf("重连失败: %v", rsp.ErrorMessage)
-						continue
-					}
-				}
-				log.Info("重连成功")
-				return
-
-			}
-			log.Fatal("重连失败: 重连次数达到设置的上限值")
-		}
-		b.Release()
-		log.Fatalf("Bot已离线：%v", e.Message)
-	})
-	c := make(chan os.Signal, 1)
 	signal.Notify(c, os.Interrupt, os.Kill)
 	<-c
 	b.Release()

server/apiAdmin.go

@@ -0,0 +1,500 @@
+package server
+
+import (
+	"bufio"
+	"bytes"
+	"encoding/base64"
+	"encoding/json"
+	"fmt"
+	"github.com/Mrs4s/MiraiGo/client"
+	"github.com/Mrs4s/go-cqhttp/coolq"
+	"github.com/Mrs4s/go-cqhttp/global"
+	"github.com/gin-gonic/gin"
+	log "github.com/sirupsen/logrus"
+	"github.com/tidwall/gjson"
+	"github.com/yinghau76/go-ascii-art"
+	"image"
+	"io/ioutil"
+	"net/http"
+	"os"
+	"os/signal"
+	"strconv"
+	"strings"
+	"time"
+)
+
+var WebInput = make(chan string, 1) //长度1，用于阻塞
+
+var Console = make(chan os.Signal, 1)
+
+var JsonConfig *global.JsonConfig
+
+type webServer struct {
+	engine  *gin.Engine
+	bot     *coolq.CQBot
+	Cli     *client.QQClient
+	Conf    *global.JsonConfig //old config
+	Console *bufio.Reader
+}
+
+var WebServer = &webServer{}
+
+// admin 子站的 路由映射
+var HttpuriAdmin = map[string]func(s *webServer, c *gin.Context){
+	"do_restart":        AdminDoRestart,       //热重启
+	"get_web_write":     AdminWebWrite,        //获取是否验证码输入
+	"do_web_write":      AdminDoWebWrite,      //web上进行输入操作
+	"do_restart_docker": AdminDoRestartDocker, //直接停止（依赖supervisord/docker）重新拉起
+	"do_config_base":    AdminDoConfigBase,    //修改config.json中的基础部分
+	"do_config_http":    AdminDoConfigHttp,    //修改config.json的http部分
+	"do_config_ws":      AdminDoConfigWs,      //修改config.json的正向ws部分
+	"do_config_reverse": AdminDoConfigReverse, //修改config.json 中的反向ws部分
+	"do_config_json":    AdminDoConfigJson,    //直接修改 config.json配置
+	"get_config_json":   AdminDoConfigJson,    //拉取 当前的config.json配置
+}
+
+func Failed(code int, msg string) coolq.MSG {
+	return coolq.MSG{"data": nil, "retcode": code, "status": "failed", "msg": msg}
+}
+
+func (s *webServer) Run(addr string, cli *client.QQClient) *coolq.CQBot {
+	s.Cli = cli
+	s.Conf = GetConf()
+	JsonConfig = s.Conf
+	gin.SetMode(gin.ReleaseMode)
+	s.engine = gin.New()
+
+	s.engine.Use(AuthMiddleWare())
+
+	//通用路由
+	s.engine.Any("/admin/:action", s.admin)
+
+	go func() {
+		//开启端口监听
+		if s.Conf.WebUi.Enabled {
+			log.Infof("Admin API 服务器已启动: %v", addr)
+			err := s.engine.Run(addr)
+			if err != nil {
+				log.Error(err)
+				log.Infof("请检查端口是否被占用.")
+				time.Sleep(time.Second * 5)
+				os.Exit(1)
+			}
+		} else {
+			//关闭端口监听
+			c := make(chan os.Signal, 1)
+			signal.Notify(c, os.Interrupt, os.Kill)
+			<-c
+			os.Exit(1)
+		}
+	}()
+	s.Dologin()
+	s.UpServer()
+	b := s.bot //外部引入 bot对象，用于操作bot
+	return b
+}
+
+func (s *webServer) Dologin() {
+	s.Console = bufio.NewReader(os.Stdin)
+	conf := GetConf()
+	cli := s.Cli
+	rsp, err := cli.Login()
+	for {
+		global.Check(err)
+		var text string
+		if !rsp.Success {
+			switch rsp.Error {
+			case client.NeedCaptcha:
+				_ = ioutil.WriteFile("captcha.jpg", rsp.CaptchaImage, 0644)
+				img, _, _ := image.Decode(bytes.NewReader(rsp.CaptchaImage))
+				fmt.Println(asciiart.New("image", img).Art)
+				if conf.WebUi.WebInput {
+					log.Warnf("请输入验证码 (captcha.jpg)： (http://%s:%d/admin/do_web_write 输入)", conf.WebUi.Host, conf.WebUi.WebUiPort)
+					text = <-WebInput
+				} else {
+					log.Warn("请输入验证码 (captcha.jpg)： (Enter 提交)")
+					text, _ = s.Console.ReadString('\n')
+				}
+				rsp, err = cli.SubmitCaptcha(strings.ReplaceAll(text, "\n", ""), rsp.CaptchaSign)
+				global.DelFile("captcha.jpg")
+				continue
+			case client.UnsafeDeviceError:
+				log.Warnf("账号已开启设备锁，请前往 -> %v <- 验证并重启Bot.", rsp.VerifyUrl)
+				if conf.WebUi.WebInput {
+					log.Infof(" (http://%s:%d/admin/do_web_write 确认后继续)....", conf.WebUi.Host, conf.WebUi.WebUiPort)
+					text = <-WebInput
+				} else {
+					log.Infof(" 按 Enter 继续....")
+					_, _ = s.Console.ReadString('\n')
+				}
+				log.Info(text)
+				os.Exit(0)
+				return
+			case client.OtherLoginError, client.UnknownLoginError:
+				log.Fatalf("登录失败: %v", rsp.ErrorMessage)
+				return
+			}
+		}
+		break
+	}
+	log.Infof("登录成功 欢迎使用: %v", cli.Nickname)
+	time.Sleep(time.Second)
+	log.Info("开始加载好友列表...")
+	global.Check(cli.ReloadFriendList())
+	log.Infof("共加载 %v 个好友.", len(cli.FriendList))
+	log.Infof("开始加载群列表...")
+	global.Check(cli.ReloadGroupList())
+	log.Infof("共加载 %v 个群.", len(cli.GroupList))
+	s.bot = coolq.NewQQBot(cli, conf)
+	if conf.PostMessageFormat != "string" && conf.PostMessageFormat != "array" {
+		log.Warnf("post_message_format 配置错误, 将自动使用 string")
+		coolq.SetMessageFormat("string")
+		return
+	} else {
+		coolq.SetMessageFormat(conf.PostMessageFormat)
+	}
+	if conf.RateLimit.Enabled {
+		global.InitLimiter(conf.RateLimit.Frequency, conf.RateLimit.BucketSize)
+	}
+	log.Info("正在加载事件过滤器.")
+	global.BootFilter()
+	coolq.IgnoreInvalidCQCode = conf.IgnoreInvalidCQCode
+	coolq.ForceFragmented = conf.ForceFragmented
+	log.Info("资源初始化完成, 开始处理信息.")
+	log.Info("アトリは、高性能ですから!")
+	cli.OnDisconnected(func(bot *client.QQClient, e *client.ClientDisconnectedEvent) {
+		if conf.ReLogin.Enabled {
+			var times uint = 1
+			for {
+				if cli.Online {
+					log.Warn("Bot已登录")
+					return
+				}
+				if conf.ReLogin.MaxReloginTimes == 0 {
+				} else if times > conf.ReLogin.MaxReloginTimes {
+					break
+				}
+				log.Warnf("Bot已离线 (%v)，将在 %v 秒后尝试重连. 重连次数：%v",
+					e.Message, conf.ReLogin.ReLoginDelay, times)
+				times++
+				time.Sleep(time.Second * time.Duration(conf.ReLogin.ReLoginDelay))
+				rsp, err := cli.Login()
+				if err != nil {
+					log.Errorf("重连失败: %v", err)
+					continue
+				}
+				if !rsp.Success {
+					switch rsp.Error {
+					case client.NeedCaptcha:
+						log.Fatalf("重连失败: 需要验证码. (验证码处理正在开发中)")
+					case client.UnsafeDeviceError:
+						log.Fatalf("重连失败: 设备锁")
+					default:
+						log.Errorf("重连失败: %v", rsp.ErrorMessage)
+						continue
+					}
+				}
+				log.Info("重连成功")
+				return
+			}
+			log.Fatal("重连失败: 重连次数达到设置的上限值")
+		}
+		s.bot.Release()
+		log.Fatalf("Bot已离线：%v", e.Message)
+	})
+}
+
+func (s *webServer) admin(c *gin.Context) {
+	action := c.Param("action")
+	log.Debugf("WebServer接收到cgi调用: %v", action)
+	if f, ok := HttpuriAdmin[action]; ok {
+		f(s, c)
+	} else {
+		c.JSON(200, coolq.Failed(404))
+	}
+}
+
+// 获取当前配置文件信息
+func GetConf() *global.JsonConfig {
+	if JsonConfig != nil {
+		return JsonConfig
+	}
+	conf := global.Load("config.json")
+	return conf
+}
+
+// admin 控制器 登录验证
+func AuthMiddleWare() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		conf := GetConf()
+		//处理跨域问题
+		c.Header("Access-Control-Allow-Origin", "*")
+		c.Header("Access-Control-Allow-Headers", "Content-Type,AccessToken,X-CSRF-Token, Authorization, Token")
+		c.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, PATCH, DELETE")
+		c.Header("Access-Control-Expose-Headers", "Content-Length, Access-Control-Allow-Origin, Access-Control-Allow-Headers, Content-Type")
+		c.Header("Access-Control-Allow-Credentials", "true")
+		// 放行所有OPTIONS方法，因为有的模板是要请求两次的
+		if c.Request.Method == "OPTIONS" {
+			c.AbortWithStatus(http.StatusNoContent)
+		}
+		// 处理请求
+		if c.Request.Method != "GET" && c.Request.Method != "POST" {
+			log.Warnf("已拒绝客户端 %v 的请求: 方法错误", c.Request.RemoteAddr)
+			c.Status(404)
+			c.Abort()
+		}
+		if c.Request.Method == "POST" && strings.Contains(c.Request.Header.Get("Content-Type"), "application/json") {
+			d, err := c.GetRawData()
+			if err != nil {
+				log.Warnf("获取请求 %v 的Body时出现错误: %v", c.Request.RequestURI, err)
+				c.Status(400)
+				c.Abort()
+			}
+			if !gjson.ValidBytes(d) {
+				log.Warnf("已拒绝客户端 %v 的请求: 非法Json", c.Request.RemoteAddr)
+				c.Status(400)
+				c.Abort()
+			}
+			c.Set("json_body", gjson.ParseBytes(d))
+		}
+		authToken := conf.AccessToken
+		if auth := c.Request.Header.Get("Authorization"); auth != "" {
+			if strings.SplitN(auth, " ", 2)[1] != authToken {
+				c.AbortWithStatus(401)
+				return
+			}
+		} else if c.Query("access_token") != authToken {
+			c.AbortWithStatus(401)
+			return
+		} else {
+			c.Next()
+		}
+	}
+}
+
+func (s *webServer) DoReLogin() { // TODO: 协议层的 ReLogin
+	JsonConfig = nil
+	conf := GetConf()
+	OldConf := s.Conf
+	cli := client.NewClient(conf.Uin, conf.Password)
+	log.Info("开始尝试登录并同步消息...")
+	log.Infof("使用协议: %v", func() string {
+		switch client.SystemDeviceInfo.Protocol {
+		case client.AndroidPad:
+			return "Android Pad"
+		case client.AndroidPhone:
+			return "Android Phone"
+		case client.AndroidWatch:
+			return "Android Watch"
+		}
+		return "未知"
+	}())
+	cli.OnLog(func(c *client.QQClient, e *client.LogEvent) {
+		switch e.Type {
+		case "INFO":
+			log.Info("Protocol -> " + e.Message)
+		case "ERROR":
+			log.Error("Protocol -> " + e.Message)
+		case "DEBUG":
+			log.Debug("Protocol -> " + e.Message)
+		}
+	})
+	cli.OnServerUpdated(func(bot *client.QQClient, e *client.ServerUpdatedEvent) {
+		log.Infof("收到服务器地址更新通知, 将在下一次重连时应用. ")
+	})
+	s.Cli = cli
+	s.Dologin()
+	//关闭之前的 server
+	if OldConf.HttpConfig != nil && OldConf.HttpConfig.Enabled {
+		HttpServer.ShutDown()
+	}
+	//if OldConf.WSConfig != nil && OldConf.WSConfig.Enabled {
+	//	server.WsShutdown()
+	//}
+	//s.UpServer()
+	s.ReloadServer()
+	s.Conf = conf
+}
+
+func (s *webServer) UpServer() {
+	conf := GetConf()
+	if conf.HttpConfig != nil && conf.HttpConfig.Enabled {
+		go HttpServer.Run(fmt.Sprintf("%s:%d", conf.HttpConfig.Host, conf.HttpConfig.Port), conf.AccessToken, s.bot)
+		for k, v := range conf.HttpConfig.PostUrls {
+			NewHttpClient().Run(k, v, conf.HttpConfig.Timeout, s.bot)
+		}
+	}
+	if conf.WSConfig != nil && conf.WSConfig.Enabled {
+		go WebsocketServer.Run(fmt.Sprintf("%s:%d", conf.WSConfig.Host, conf.WSConfig.Port), conf.AccessToken, s.bot)
+	}
+	for _, rc := range conf.ReverseServers {
+		go NewWebsocketClient(rc, conf.AccessToken, s.bot).Run()
+	}
+}
+
+// 暂不支持ws服务的重启
+func (s *webServer) ReloadServer() {
+	conf := GetConf()
+	if conf.HttpConfig != nil && conf.HttpConfig.Enabled {
+		go HttpServer.Run(fmt.Sprintf("%s:%d", conf.HttpConfig.Host, conf.HttpConfig.Port), conf.AccessToken, s.bot)
+		for k, v := range conf.HttpConfig.PostUrls {
+			NewHttpClient().Run(k, v, conf.HttpConfig.Timeout, s.bot)
+		}
+	}
+	for _, rc := range conf.ReverseServers {
+		go NewWebsocketClient(rc, conf.AccessToken, s.bot).Run()
+	}
+}
+
+// 热重启
+func AdminDoRestart(s *webServer, c *gin.Context) {
+	s.bot = nil
+	s.Cli = nil
+	s.DoReLogin()
+	c.JSON(200, coolq.OK(coolq.MSG{}))
+	return
+}
+
+// 冷重启
+func AdminDoRestartDocker(s *webServer, c *gin.Context) {
+	Console <- os.Kill
+	c.JSON(200, coolq.OK(coolq.MSG{}))
+	return
+}
+
+// web输入 html 页面
+func AdminWebWrite(s *webServer, c *gin.Context) {
+	pic := global.ReadAllText("captcha.jpg")
+	var picbase64 string
+	var ispic = false
+	if pic != "" {
+		input := []byte(pic)
+		// base64编码
+		picbase64 = base64.StdEncoding.EncodeToString(input)
+		ispic = true
+	}
+	c.JSON(200, coolq.OK(coolq.MSG{
+		"ispic":     ispic,     //为空则为 设备锁 或者没有需要输入
+		"picbase64": picbase64, //web上显示图片
+	}))
+}
+
+// web输入 处理
+func AdminDoWebWrite(s *webServer, c *gin.Context) {
+	input := c.PostForm("input")
+	WebInput <- input
+	c.JSON(200, coolq.OK(coolq.MSG{}))
+}
+
+// 普通配置修改
+func AdminDoConfigBase(s *webServer, c *gin.Context) {
+	conf := GetConf()
+	conf.Uin, _ = strconv.ParseInt(c.PostForm("uin"), 10, 64)
+	conf.Password = c.PostForm("password")
+	if c.PostForm("enable_db") == "true" {
+		conf.EnableDB = true
+	} else {
+		conf.EnableDB = false
+	}
+	conf.AccessToken = c.PostForm("access_token")
+	if err := conf.Save("config.json"); err != nil {
+		log.Fatalf("保存 config.json 时出现错误: %v", err)
+		c.JSON(200, Failed(502, "保存 config.json 时出现错误:"+fmt.Sprintf("%v", err)))
+	} else {
+		JsonConfig = nil
+		c.JSON(200, coolq.OK(coolq.MSG{}))
+	}
+}
+
+// http配置修改
+func AdminDoConfigHttp(s *webServer, c *gin.Context) {
+	conf := GetConf()
+	p, _ := strconv.ParseUint(c.PostForm("port"), 10, 16)
+	conf.HttpConfig.Port = uint16(p)
+	conf.HttpConfig.Host = c.PostForm("host")
+	if c.PostForm("enable") == "true" {
+		conf.HttpConfig.Enabled = true
+	} else {
+		conf.HttpConfig.Enabled = false
+	}
+	t, _ := strconv.ParseInt(c.PostForm("timeout"), 10, 32)
+	conf.HttpConfig.Timeout = int32(t)
+	if c.PostForm("post_url") != "" {
+		conf.HttpConfig.PostUrls[c.PostForm("post_url")] = c.PostForm("post_secret")
+	}
+	if err := conf.Save("config.json"); err != nil {
+		log.Fatalf("保存 config.json 时出现错误: %v", err)
+		c.JSON(200, Failed(502, "保存 config.json 时出现错误:"+fmt.Sprintf("%v", err)))
+	} else {
+		JsonConfig = nil
+		c.JSON(200, coolq.OK(coolq.MSG{}))
+	}
+}
+
+// ws配置修改
+func AdminDoConfigWs(s *webServer, c *gin.Context) {
+	conf := GetConf()
+	p, _ := strconv.ParseUint(c.PostForm("port"), 10, 16)
+	conf.WSConfig.Port = uint16(p)
+	conf.WSConfig.Host = c.PostForm("host")
+	if c.PostForm("enable") == "true" {
+		conf.WSConfig.Enabled = true
+	} else {
+		conf.WSConfig.Enabled = false
+	}
+	if err := conf.Save("config.json"); err != nil {
+		log.Fatalf("保存 config.json 时出现错误: %v", err)
+		c.JSON(200, Failed(502, "保存 config.json 时出现错误:"+fmt.Sprintf("%v", err)))
+	} else {
+		JsonConfig = nil
+		c.JSON(200, coolq.OK(coolq.MSG{}))
+	}
+}
+
+// 反向ws配置修改
+func AdminDoConfigReverse(s *webServer, c *gin.Context) {
+	conf := GetConf()
+	conf.ReverseServers[0].ReverseApiUrl = c.PostForm("reverse_api_url")
+	conf.ReverseServers[0].ReverseUrl = c.PostForm("reverse_url")
+	conf.ReverseServers[0].ReverseEventUrl = c.PostForm("reverse_event_url")
+	t, _ := strconv.ParseUint(c.PostForm("reverse_reconnect_interval"), 10, 16)
+	conf.ReverseServers[0].ReverseReconnectInterval = uint16(t)
+	if c.PostForm("enable") == "true" {
+		conf.ReverseServers[0].Enabled = true
+	} else {
+		conf.ReverseServers[0].Enabled = false
+	}
+	if err := conf.Save("config.json"); err != nil {
+		log.Fatalf("保存 config.json 时出现错误: %v", err)
+		c.JSON(200, Failed(502, "保存 config.json 时出现错误:"+fmt.Sprintf("%v", err)))
+	} else {
+		JsonConfig = nil
+		c.JSON(200, coolq.OK(coolq.MSG{}))
+	}
+}
+
+// config.json配置修改
+func AdminDoConfigJson(s *webServer, c *gin.Context) {
+	conf := GetConf()
+	Json := c.PostForm("json")
+	err := json.Unmarshal([]byte(Json), &conf)
+	if err != nil {
+		log.Warnf("尝试加载配置文件 %v 时出现错误: %v", "config.json", err)
+		c.JSON(200, Failed(502, "保存 config.json 时出现错误:"+fmt.Sprintf("%v", err)))
+		return
+	}
+	if err := conf.Save("config.json"); err != nil {
+		log.Fatalf("保存 config.json 时出现错误: %v", err)
+		c.JSON(200, Failed(502, "保存 config.json 时出现错误:"+fmt.Sprintf("%v", err)))
+	} else {
+		JsonConfig = nil
+		c.JSON(200, coolq.OK(coolq.MSG{}))
+	}
+}
+
+// 拉取config.json配置
+func AdminGetConfigJson(s *webServer, c *gin.Context) {
+	conf := GetConf()
+	c.JSON(200, coolq.OK(coolq.MSG{"config": conf}))
+
+}

server/http.go

@@ -5,6 +5,7 @@ import (
 	"crypto/hmac"
 	"crypto/sha1"
 	"encoding/hex"
+	"net/http"
 	"os"
 	"strconv"
 	"strings"
@@ -21,6 +22,7 @@ import (
 type httpServer struct {
 	engine *gin.Engine
 	bot    *coolq.CQBot
+	Http *http.Server
 }
 
 type httpClient struct {
@@ -79,13 +81,23 @@ func (s *httpServer) Run(addr, authToken string, bot *coolq.CQBot) {
 
 	go func() {
 		log.Infof("CQ HTTP 服务器已启动: %v", addr)
-		err := s.engine.Run(addr)
+		s.Http=&http.Server{
-		if err != nil {
+			Addr:addr,
+			Handler:s.engine,
+		}
+		if err := s.Http.ListenAndServe(); err != nil && err != http.ErrServerClosed {
 			log.Error(err)
 			log.Infof("请检查端口是否被占用.")
 			time.Sleep(time.Second * 5)
 			os.Exit(1)
 		}
+		//err := s.engine.Run(addr)
+		//if err != nil {
+		//	log.Error(err)
+		//	log.Infof("请检查端口是否被占用.")
+		//	time.Sleep(time.Second * 5)
+		//	os.Exit(1)
+		//}
 	}()
 }
 
@@ -529,3 +541,16 @@ var httpApi = map[string]func(s *httpServer, c *gin.Context){
 		s.GetWordSlices(c)
 	},
 }
+
+func (s *httpServer) ShutDown() {
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+	if err := s.Http.Shutdown(ctx); err != nil {
+		log.Fatal("http Server Shutdown:", err)
+	}
+	select {
+	case <-ctx.Done():
+		log.Println("timeout of 5 seconds.")
+	}
+	log.Println("http Server exiting")
+}