mlogclub
67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
package third
|
|
|
|
import (
|
|
"cs-agent/internal/wxwork"
|
|
"net/http"
|
|
|
|
"github.com/kataras/iris/v12"
|
|
"github.com/silenceper/wechat/v2/work/kf"
|
|
)
|
|
|
|
type WechatController struct {
|
|
Ctx iris.Context
|
|
}
|
|
|
|
// GetCallback GET请求用于校验回调是否配置正确
|
|
func (c *WechatController) GetCallback() {
|
|
cli, err := wxwork.GetWorkCli().GetKF()
|
|
if err != nil {
|
|
c.Ctx.StopWithError(http.StatusInternalServerError, err)
|
|
return
|
|
}
|
|
options := kf.SignatureOptions{}
|
|
if err := c.Ctx.ReadForm(&options); err != nil {
|
|
c.Ctx.StopWithError(http.StatusUnauthorized, err)
|
|
return
|
|
}
|
|
// 调用VerifyURL方法校验当前请求，如果合法则把解密后的内容作为响应返回给微信服务器
|
|
echo, err := cli.VerifyURL(options)
|
|
if err == nil {
|
|
c.Ctx.WriteString(echo)
|
|
} else {
|
|
c.Ctx.StopWithError(http.StatusUnauthorized, err)
|
|
}
|
|
}
|
|
|
|
// PostCallback POST请求用于接收回调
|
|
func (c *WechatController) PostCallback() {
|
|
cli, err := wxwork.GetWorkCli().GetKF()
|
|
if err != nil {
|
|
c.Ctx.StopWithError(http.StatusInternalServerError, err)
|
|
return
|
|
}
|
|
var (
|
|
message kf.CallbackMessage
|
|
body []byte
|
|
)
|
|
// 读取原始消息内容
|
|
body, err = c.Ctx.GetBody()
|
|
if err != nil {
|
|
c.Ctx.StopWithError(http.StatusInternalServerError, err)
|
|
return
|
|
}
|
|
// 解析原始数据
|
|
message, err = cli.GetCallbackMessage(body)
|
|
if err != nil {
|
|
c.Ctx.StopWithError(http.StatusInternalServerError, err)
|
|
return
|
|
}
|
|
|
|
if err := wxwork.ConsumeCallback(message); err != nil {
|
|
c.Ctx.StopWithError(http.StatusInternalServerError, err)
|
|
return
|
|
}
|
|
|
|
c.Ctx.WriteString("ok")
|
|
}
|