http.go

package server

import (
	"context"
	"crypto/hmac"
	"crypto/sha1"
	"encoding/hex"
	"net/http"
	"os"
	"strconv"
	"strings"
	"time"

	"github.com/guonaihong/gout/dataflow"

	"github.com/Mrs4s/go-cqhttp/coolq"
	"github.com/Mrs4s/go-cqhttp/global"
	"github.com/gin-gonic/gin"
	"github.com/guonaihong/gout"
	log "github.com/sirupsen/logrus"
	"github.com/tidwall/gjson"
)

type httpServer struct {
	engine *gin.Engine
	bot    *coolq.CQBot
	Http   *http.Server
}

type httpClient struct {
	bot     *coolq.CQBot
	secret  string
	addr    string
	timeout int32
}

var HttpServer = &httpServer{}
var Debug = false

func (s *httpServer) Run(addr, authToken string, bot *coolq.CQBot) {
	gin.SetMode(gin.ReleaseMode)
	s.engine = gin.New()
	s.bot = bot
	s.engine.Use(func(c *gin.Context) {
		if c.Request.Method != "GET" && c.Request.Method != "POST" {
			log.Warnf("已拒绝客户端 %v 的请求: 方法错误", c.Request.RemoteAddr)
			c.Status(404)
			return
		}
		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)
				return
			}
			if !gjson.ValidBytes(d) {
				log.Warnf("已拒绝客户端 %v 的请求: 非法Json", c.Request.RemoteAddr)
				c.Status(400)
				return
			}
			c.Set("json_body", gjson.ParseBytes(d))
		}
		c.Next()
	})

	if authToken != "" {
		s.engine.Use(func(c *gin.Context) {
			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()
			}
		})
	}

	s.engine.Any("/:action", s.HandleActions)

	go func() {
		log.Infof("CQ HTTP 服务器已启动: %v", addr)
		s.Http = &http.Server{
			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)
		}
	}()
}

func NewHttpClient() *httpClient {
	return &httpClient{}
}

func (c *httpClient) Run(addr, secret string, timeout int32, bot *coolq.CQBot) {
	c.bot = bot
	c.secret = secret
	c.addr = addr
	c.timeout = timeout
	if c.timeout < 5 {
		c.timeout = 5
	}
	bot.OnEventPush(c.onBotPushEvent)
	log.Infof("HTTP POST上报器已启动: %v", addr)
}

func (c *httpClient) onBotPushEvent(m coolq.MSG) {
	var res string
	err := gout.POST(c.addr).SetJSON(m).BindBody(&res).SetHeader(func() gout.H {
		h := gout.H{
			"X-Self-ID":  c.bot.Client.Uin,
			"User-Agent": "CQHttp/4.15.0",
		}
		if c.secret != "" {
			mac := hmac.New(sha1.New, []byte(c.secret))
			mac.Write([]byte(m.ToJson()))
			h["X-Signature"] = "sha1=" + hex.EncodeToString(mac.Sum(nil))
		}
		return h
	}()).SetTimeout(time.Second * time.Duration(c.timeout)).F().Retry().Attempt(5).
		WaitTime(time.Millisecond * 500).MaxWaitTime(time.Second * 5).
		Func(func(con *dataflow.Context) error {
			if con.Error != nil {
				log.Warnf("上报Event到 HTTP 服务器 %v 时出现错误: %v 将重试.", c.addr, con.Error)
				return con.Error
			}
			return nil
		}).Do()
	if err != nil {
		log.Warnf("上报Event数据 %v 到 %v 失败: %v", m.ToJson(), c.addr, err)
		return
	}
	log.Debugf("上报Event数据 %v 到 %v", m.ToJson(), c.addr)
	if gjson.Valid(res) {
		c.bot.CQHandleQuickOperation(gjson.Parse(m.ToJson()), gjson.Parse(res))
	}
}

func (s *httpServer) HandleActions(c *gin.Context) {
	global.RateLimit(context.Background())
	action := strings.ReplaceAll(c.Param("action"), "_async", "")
	log.Debugf("HTTPServer接收到API调用: %v", action)
	if f, ok := httpApi[action]; ok {
		f(s, c)
	} else {
		c.JSON(200, coolq.Failed(404))
	}
}

func GetLoginInfo(s *httpServer, c *gin.Context) {
	c.JSON(200, s.bot.CQGetLoginInfo())
}

func GetFriendList(s *httpServer, c *gin.Context) {
	c.JSON(200, s.bot.CQGetFriendList())
}

func GetGroupList(s *httpServer, c *gin.Context) {
	nc := getParamOrDefault(c, "no_cache", "false")
	c.JSON(200, s.bot.CQGetGroupList(nc == "true"))
}

func GetGroupInfo(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	nc := getParamOrDefault(c, "no_cache", "false")
	c.JSON(200, s.bot.CQGetGroupInfo(gid, nc == "true"))
}

func GetGroupMemberList(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	nc := getParamOrDefault(c, "no_cache", "false")
	c.JSON(200, s.bot.CQGetGroupMemberList(gid, nc == "true"))
}

func GetGroupMemberInfo(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	uid, _ := strconv.ParseInt(getParam(c, "user_id"), 10, 64)
	c.JSON(200, s.bot.CQGetGroupMemberInfo(gid, uid))
}

func GetGroupFileSystemInfo(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	c.JSON(200, s.bot.CQGetGroupFileSystemInfo(gid))
}

func GetGroupRootFiles(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	c.JSON(200, s.bot.CQGetGroupRootFiles(gid))
}

func GetGroupFilesByFolderId(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	folderId := getParam(c, "folder_id")
	c.JSON(200, s.bot.CQGetGroupFilesByFolderId(gid, folderId))
}

func GetGroupFileUrl(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	fid := getParam(c, "file_id")
	busid, _ := strconv.ParseInt(getParam(c, "busid"), 10, 32)
	c.JSON(200, s.bot.CQGetGroupFileUrl(gid, fid, int32(busid)))
}

func SendMessage(s *httpServer, c *gin.Context) {
	if getParam(c, "message_type") == "private" {
		SendPrivateMessage(s, c)
		return
	}
	if getParam(c, "message_type") == "group" {
		SendGroupMessage(s, c)
		return
	}
	if getParam(c, "group_id") != "" {
		SendGroupMessage(s, c)
		return
	}
	if getParam(c, "user_id") != "" {
		SendPrivateMessage(s, c)
	}
}

func SendPrivateMessage(s *httpServer, c *gin.Context) {
	uid, _ := strconv.ParseInt(getParam(c, "user_id"), 10, 64)
	msg, t := getParamWithType(c, "message")
	autoEscape := global.EnsureBool(getParam(c, "auto_escape"), false)
	if t == gjson.JSON {
		c.JSON(200, s.bot.CQSendPrivateMessage(uid, gjson.Parse(msg), autoEscape))
		return
	}
	c.JSON(200, s.bot.CQSendPrivateMessage(uid, msg, autoEscape))
}

func SendGroupMessage(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	msg, t := getParamWithType(c, "message")
	autoEscape := global.EnsureBool(getParam(c, "auto_escape"), false)
	if t == gjson.JSON {
		c.JSON(200, s.bot.CQSendGroupMessage(gid, gjson.Parse(msg), autoEscape))
		return
	}
	c.JSON(200, s.bot.CQSendGroupMessage(gid, msg, autoEscape))
}

func SendGroupForwardMessage(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	msg := getParam(c, "messages")
	c.JSON(200, s.bot.CQSendGroupForwardMessage(gid, gjson.Parse(msg)))
}

func GetImage(s *httpServer, c *gin.Context) {
	file := getParam(c, "file")
	c.JSON(200, s.bot.CQGetImage(file))
}

func GetMessage(s *httpServer, c *gin.Context) {
	mid, _ := strconv.ParseInt(getParam(c, "message_id"), 10, 32)
	c.JSON(200, s.bot.CQGetMessage(int32(mid)))
}

func GetGroupHonorInfo(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	c.JSON(200, s.bot.CQGetGroupHonorInfo(gid, getParam(c, "type")))
}

func ProcessFriendRequest(s *httpServer, c *gin.Context) {
	flag := getParam(c, "flag")
	approve := getParamOrDefault(c, "approve", "true")
	c.JSON(200, s.bot.CQProcessFriendRequest(flag, approve == "true"))
}

func ProcessGroupRequest(s *httpServer, c *gin.Context) {
	flag := getParam(c, "flag")
	subType := getParam(c, "sub_type")
	if subType == "" {
		subType = getParam(c, "type")
	}
	approve := getParamOrDefault(c, "approve", "true")
	c.JSON(200, s.bot.CQProcessGroupRequest(flag, subType, getParam(c, "reason"), approve == "true"))
}

func SetGroupCard(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	uid, _ := strconv.ParseInt(getParam(c, "user_id"), 10, 64)
	c.JSON(200, s.bot.CQSetGroupCard(gid, uid, getParam(c, "card")))
}

func SetSpecialTitle(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	uid, _ := strconv.ParseInt(getParam(c, "user_id"), 10, 64)
	c.JSON(200, s.bot.CQSetGroupSpecialTitle(gid, uid, getParam(c, "special_title")))
}

func SetGroupKick(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	uid, _ := strconv.ParseInt(getParam(c, "user_id"), 10, 64)
	msg := getParam(c, "message")
	block := getParamOrDefault(c, "reject_add_request", "false")
	c.JSON(200, s.bot.CQSetGroupKick(gid, uid, msg, block == "true"))
}

func SetGroupBan(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	uid, _ := strconv.ParseInt(getParam(c, "user_id"), 10, 64)
	i, _ := strconv.ParseInt(getParamOrDefault(c, "duration", "1800"), 10, 64)
	c.JSON(200, s.bot.CQSetGroupBan(gid, uid, uint32(i)))
}

func SetWholeBan(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	c.JSON(200, s.bot.CQSetGroupWholeBan(gid, getParamOrDefault(c, "enable", "true") == "true"))
}

func SetGroupName(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	c.JSON(200, s.bot.CQSetGroupName(gid, getParam(c, "group_name")))
}

func SetGroupAdmin(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	uid, _ := strconv.ParseInt(getParam(c, "user_id"), 10, 64)
	c.JSON(200, s.bot.CQSetGroupAdmin(gid, uid, getParamOrDefault(c, "enable", "true") == "true"))
}

func SendGroupNotice(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	c.JSON(200, s.bot.CQSetGroupMemo(gid, getParam(c, "content")))
}

func SetGroupLeave(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	c.JSON(200, s.bot.CQSetGroupLeave(gid))
}

func SetRestart(s *httpServer, c *gin.Context) {
	delay, _ := strconv.ParseInt(getParam(c, "delay"), 10, 64)
	c.JSON(200, coolq.MSG{"data": nil, "retcode": 0, "status": "async"})
	go func(delay int64) {
		time.Sleep(time.Duration(delay) * time.Millisecond)
		Restart <- struct{}{}
	}(delay)

}

func GetForwardMessage(s *httpServer, c *gin.Context) {
	resId := getParam(c, "message_id")
	if resId == "" {
		resId = getParam(c, "id")
	}
	c.JSON(200, s.bot.CQGetForwardMessage(resId))
}

func GetGroupSystemMessage(s *httpServer, c *gin.Context) {
	c.JSON(200, s.bot.CQGetGroupSystemMessages())
}

func DeleteMessage(s *httpServer, c *gin.Context) {
	mid, _ := strconv.ParseInt(getParam(c, "message_id"), 10, 32)
	c.JSON(200, s.bot.CQDeleteMessage(int32(mid)))
}

func CanSendImage(s *httpServer, c *gin.Context) {
	c.JSON(200, s.bot.CQCanSendImage())
}

func CanSendRecord(s *httpServer, c *gin.Context) {
	c.JSON(200, s.bot.CQCanSendRecord())
}

func GetStatus(s *httpServer, c *gin.Context) {
	c.JSON(200, s.bot.CQGetStatus())
}

func GetVersionInfo(s *httpServer, c *gin.Context) {
	c.JSON(200, s.bot.CQGetVersionInfo())
}

func ReloadEventFilter(s *httpServer, c *gin.Context) {
	c.JSON(200, s.bot.CQReloadEventFilter())
}

func GetVipInfo(s *httpServer, c *gin.Context) {
	uid, _ := strconv.ParseInt(getParam(c, "user_id"), 10, 64)
	c.JSON(200, s.bot.CQGetVipInfo(uid))
}

func GetStrangerInfo(s *httpServer, c *gin.Context) {
	uid, _ := strconv.ParseInt(getParam(c, "user_id"), 10, 64)
	c.JSON(200, s.bot.CQGetStrangerInfo(uid))
}

func GetGroupAtAllRemain(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	c.JSON(200, s.bot.CQGetAtAllRemain(gid))
}

func SetGroupAnonymousBan(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	d, _ := strconv.ParseInt(getParam(c, "duration"), 10, 64)
	flag := getParam(c, "flag")
	if flag == "" {
		flag = getParam(c, "anonymous_flag")
	}
	if flag == "" {
		o := gjson.Parse(getParam(c, "anonymous"))
		flag = o.Get("flag").String()
	}
	c.JSON(200, s.bot.CQSetGroupAnonymousBan(gid, flag, int32(d)))
}

func HandleQuickOperation(s *httpServer, c *gin.Context) {
	if c.Request.Method != "POST" {
		c.AbortWithStatus(404)
		return
	}
	if i, ok := c.Get("json_body"); ok {
		body := i.(gjson.Result)
		c.JSON(200, s.bot.CQHandleQuickOperation(body.Get("context"), body.Get("operation")))
	}
}

func OcrImage(s *httpServer, c *gin.Context) {
	img := getParam(c, "image")
	c.JSON(200, s.bot.CQOcrImage(img))
}

func GetWordSlices(s *httpServer, c *gin.Context) {
	content := getParam(c, "content")
	c.JSON(200, s.bot.CQGetWordSlices(content))
}

func SetGroupPortrait(s *httpServer, c *gin.Context) {
	gid, _ := strconv.ParseInt(getParam(c, "group_id"), 10, 64)
	file := getParam(c, "file")
	cache := getParam(c, "cache")
	c.JSON(200, s.bot.CQSetGroupPortrait(gid, file, cache))
}

func getParamOrDefault(c *gin.Context, k, def string) string {
	r := getParam(c, k)
	if r != "" {
		return r
	}
	return def
}

func getParam(c *gin.Context, k string) string {
	p, _ := getParamWithType(c, k)
	return p
}

func getParamWithType(c *gin.Context, k string) (string, gjson.Type) {
	if q := c.Query(k); q != "" {
		return q, gjson.Null
	}
	if c.Request.Method == "POST" {
		if h := c.Request.Header.Get("Content-Type"); h != "" {
			if strings.Contains(h, "application/x-www-form-urlencoded") {
				if p, ok := c.GetPostForm(k); ok {
					return p, gjson.Null
				}
			}
			if strings.Contains(h, "application/json") {
				if obj, ok := c.Get("json_body"); ok {
					res := obj.(gjson.Result).Get(k)
					if res.Exists() {
						switch res.Type {
						case gjson.JSON:
							return res.Raw, gjson.JSON
						case gjson.String:
							return res.Str, gjson.String
						case gjson.Number:
							return strconv.FormatInt(res.Int(), 10), gjson.Number // 似乎没有需要接受 float 类型的api
						case gjson.True:
							return "true", gjson.True
						case gjson.False:
							return "false", gjson.False
						}
					}
				}
			}
		}
	}
	return "", gjson.Null
}

var httpApi = map[string]func(s *httpServer, c *gin.Context){
	"get_login_info":             GetLoginInfo,
	"get_friend_list":            GetFriendList,
	"get_group_list":             GetGroupList,
	"get_group_info":             GetGroupInfo,
	"get_group_member_list":      GetGroupMemberList,
	"get_group_member_info":      GetGroupMemberInfo,
	"get_group_file_system_info": GetGroupFileSystemInfo,
	"get_group_root_files":       GetGroupRootFiles,
	"get_group_files_by_folder":  GetGroupFilesByFolderId,
	"get_group_file_url":         GetGroupFileUrl,
	"send_msg":                   SendMessage,
	"send_group_msg":             SendGroupMessage,
	"send_group_forward_msg":     SendGroupForwardMessage,
	"send_private_msg":           SendPrivateMessage,
	"delete_msg":                 DeleteMessage,
	"set_friend_add_request":     ProcessFriendRequest,
	"set_group_add_request":      ProcessGroupRequest,
	"set_group_card":             SetGroupCard,
	"set_group_special_title":    SetSpecialTitle,
	"set_group_kick":             SetGroupKick,
	"set_group_ban":              SetGroupBan,
	"set_group_whole_ban":        SetWholeBan,
	"set_group_name":             SetGroupName,
	"set_group_admin":            SetGroupAdmin,
	"set_restart":                SetRestart,
	"_send_group_notice":         SendGroupNotice,
	"set_group_leave":            SetGroupLeave,
	"get_image":                  GetImage,
	"get_forward_msg":            GetForwardMessage,
	"get_msg":                    GetMessage,
	"get_group_system_msg":       GetGroupSystemMessage,
	"get_group_honor_info":       GetGroupHonorInfo,
	"can_send_image":             CanSendImage,
	"can_send_record":            CanSendRecord,
	"get_status":                 GetStatus,
	"get_version_info":           GetVersionInfo,
	"_get_vip_info":              GetVipInfo,
	"get_stranger_info":          GetStrangerInfo,
	"reload_event_filter":        ReloadEventFilter,
	"set_group_portrait":         SetGroupPortrait,
	"set_group_anonymous_ban":    SetGroupAnonymousBan,
	".handle_quick_operation":    HandleQuickOperation,
	".ocr_image":                 OcrImage,
	"ocr_image":                  OcrImage,
	"get_group_at_all_remain":    GetGroupAtAllRemain,
	".get_word_slices":           GetWordSlices,
}

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