api: csrf token

ProtobufBot · Apr 3, 2021 · 7c060c0 · 7c060c0
1 parent 6b1e913
commit 7c060c0
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/service/bot/api_handler.go b/service/bot/api_handler.go
@@ -529,3 +529,12 @@ func GetCookiesWithDomain(c *client.QQClient, domain string) string
 func HandleGetCookies(cli *client.QQClient, req *onebot.GetCookiesReq) *onebot.GetCookiesResp {
 	return &onebot.GetCookiesResp{Cookies: GetCookiesWithDomain(cli, req.Domain)}
 }
+
+//go:linkname GetCSRFToken github.com/Mrs4s/MiraiGo/client.(*QQClient).getCSRFToken
+func GetCSRFToken(c *client.QQClient) int
+
+func HandleGetCSRFToken(cli *client.QQClient, req *onebot.GetCsrfTokenReq) *onebot.GetCsrfTokenResp {
+	return &onebot.GetCsrfTokenResp{
+		Token: int32(GetCSRFToken(cli)),
+	}
+}