增加微信通知功能

This commit is contained in:
2026-09-16 18:06:18 +08:00
parent b48688cbab
commit f386899625
23 changed files with 3539 additions and 18 deletions
+7
View File
@@ -53,6 +53,13 @@ acme_directory =
# Nginx 扑底站点按 SNI 读取该目录。Go 进程需要对此目录有写权限。
ssl_cert_dir = /www/wwwroot/ssl-certs
# ==================== 微信公众号(服务号) ====================
# 服务号 AppSecret / EncodingAESKey 入库加密密钥:任意随机串即可,代码内部做 SHA-256 派生。
# 注意:更换该值后,历史已保存的公众号密钥将无法解密,需在「通知设置 → 微信配置」重新保存一次。
wechat_mp_secret_key = 8f3c2a1d94b74e28a6c5f0187d3e9b4c72a5d0e6f13b8c47d92a6e5f0b7c3148
# 微信服务器回调地址基址(默认回落 payment_callback_base,即 https://api.yunzer.cn):
# wechat_mp_callback_base = https://api.yunzer.cn
# ==================== 支付模块 ====================
# 渠道敏感参数(商户密钥/私钥/Secret)入库加密密钥:任意随机串即可,代码内部做 SHA-256 派生。
# 注意:更换该值后,历史已保存的渠道参数将无法解密,需在各渠道配置页重新保存一次。
+124
View File
@@ -0,0 +1,124 @@
package controllers
import (
"io"
"strconv"
"strings"
"time"
"server/services/wechatmp"
beego "github.com/beego/beego/v2/server/web"
)
// WechatMpCallbackController 微信公众号服务器回调(微信服务器调用,无需平台鉴权)
//
// 公众号后台「设置与开发 → 服务器配置」填写:
// URL(服务器地址):https://<你的域名>/api/wechat/mp/callback
// Token:与平台「通知设置 → 微信配置」中的 Token 保持一致
// 消息加解密方式:明文 / 兼容 / 安全(三种均支持)
type WechatMpCallbackController struct {
beego.Controller
}
// Callback 服务器地址验证(GET,回显 echostr)与消息/事件推送(POST,被动回复)
func (c *WechatMpCallbackController) Callback() {
cfg, err := wechatmp.LoadEnabledConfig()
if err != nil {
c.Ctx.Output.SetStatus(403)
c.Ctx.Output.Body([]byte("wechat mp not configured"))
return
}
q := c.Ctx.Request.URL.Query()
timestamp := q.Get("timestamp")
nonce := q.Get("nonce")
// ---------------- GET:服务器地址校验 ----------------
if strings.EqualFold(c.Ctx.Input.Method(), "GET") {
echostr := q.Get("echostr")
if cfg.EncryptMode == wechatmp.EncryptModeSafe {
if !wechatmp.CheckMsgSignature(cfg.Token, timestamp, nonce, echostr, q.Get("msg_signature")) {
c.Ctx.Output.SetStatus(403)
c.Ctx.Output.Body([]byte("invalid signature"))
return
}
plain, derr := wechatmp.DecryptMessage(cfg.AESKey, cfg.AppID, echostr)
if derr != nil {
c.Ctx.Output.SetStatus(403)
c.Ctx.Output.Body([]byte("decrypt failed"))
return
}
c.Ctx.Output.Body([]byte(plain))
return
}
if !wechatmp.CheckSignature(cfg.Token, timestamp, nonce, q.Get("signature")) {
c.Ctx.Output.SetStatus(403)
c.Ctx.Output.Body([]byte("invalid signature"))
return
}
c.Ctx.Output.Body([]byte(echostr))
return
}
// ---------------- POST:消息 / 事件 ----------------
body, _ := io.ReadAll(c.Ctx.Request.Body)
rawBody := string(body)
if strings.TrimSpace(rawBody) == "" {
c.Ctx.Output.Body([]byte(""))
return
}
needEncryptReply := false
if strings.Contains(rawBody, "<Encrypt>") {
// 安全/兼容模式:外层为加密 XML,先验签再解密
outer, perr := wechatmp.ParseInboundXML(rawBody)
if perr != nil {
c.Ctx.Output.Body([]byte(""))
return
}
if !wechatmp.CheckMsgSignature(cfg.Token, timestamp, nonce, outer.Encrypt, q.Get("msg_signature")) {
c.Ctx.Output.Body([]byte(""))
return
}
plain, derr := wechatmp.DecryptMessage(cfg.AESKey, cfg.AppID, outer.Encrypt)
if derr != nil {
c.Ctx.Output.Body([]byte(""))
return
}
rawBody = plain
needEncryptReply = true
} else if !wechatmp.CheckSignature(cfg.Token, timestamp, nonce, q.Get("signature")) {
// 明文模式:签名校验不通过直接拒绝(微信会重试)
c.Ctx.Output.Body([]byte(""))
return
}
msg, merr := wechatmp.ParseInboundXML(rawBody)
if merr != nil {
c.Ctx.Output.Body([]byte(""))
return
}
reply := wechatmp.HandleInbound(cfg, msg)
if strings.TrimSpace(reply) == "" {
// 无需回复:返回空串,微信视为处理成功
c.Ctx.Output.Body([]byte(""))
return
}
now := time.Now().Unix()
plainReply := wechatmp.BuildTextReply(msg.FromUserName, msg.ToUserName, reply, now)
c.Ctx.Output.Header("Content-Type", "application/xml; charset=utf-8")
if needEncryptReply {
ts := strconv.FormatInt(now, 10)
enc, eerr := wechatmp.BuildEncryptedReply(cfg, plainReply, ts, nonce)
if eerr != nil {
c.Ctx.Output.Body([]byte(""))
return
}
c.Ctx.Output.Body([]byte(enc))
return
}
c.Ctx.Output.Body([]byte(plainReply))
}
+202
View File
@@ -0,0 +1,202 @@
package controllers
import (
"encoding/json"
"fmt"
"io"
"strings"
"time"
"server/models"
"server/pkg/jwtutil"
"server/services/wechatmp"
beego "github.com/beego/beego/v2/server/web"
)
// =============================================================
// 租户端「微信公众号绑定」:
// 扫码关注公众号 → 公众号被动回复验证码 → 本端输入验证码完成绑定
// 绑定后(公众号认证后)可接收平台/租户侧的提醒、公告推送
// =============================================================
// BackendWechatMpController 租户端微信公众号绑定
type BackendWechatMpController struct {
beego.Controller
}
func (c *BackendWechatMpController) claims() (*jwtutil.Claims, error) {
auth := c.Ctx.Request.Header.Get("Authorization")
if auth == "" {
return nil, fmt.Errorf("未登录")
}
parts := strings.SplitN(auth, " ", 2)
if len(parts) != 2 || parts[0] != "Bearer" {
return nil, fmt.Errorf("认证信息格式错误")
}
claims, err := jwtutil.ParseToken(parts[1])
if err != nil {
return nil, fmt.Errorf("无效的token")
}
if claims.UserType != "backend" && claims.UserType != "app" {
return nil, fmt.Errorf("无权访问")
}
return claims, nil
}
func (c *BackendWechatMpController) jsonErr(httpStatus, bizCode int, msg string) {
c.Ctx.Output.SetStatus(httpStatus)
c.Data["json"] = map[string]interface{}{"code": bizCode, "msg": msg}
_ = c.ServeJSON()
}
func (c *BackendWechatMpController) ok(data interface{}, msg string) {
if msg == "" {
msg = "success"
}
resp := map[string]interface{}{"code": 200, "msg": msg}
if data != nil {
resp["data"] = data
}
c.Data["json"] = resp
_ = c.ServeJSON()
}
func (c *BackendWechatMpController) readBody(out interface{}) bool {
raw, err := io.ReadAll(c.Ctx.Request.Body)
if err != nil {
c.jsonErr(400, 400, "参数错误")
return false
}
if err := json.Unmarshal(raw, out); err != nil {
c.jsonErr(400, 400, "参数错误")
return false
}
return true
}
// BindStart POST /backend/wechatMp/bind/start
func (c *BackendWechatMpController) BindStart() {
claims, err := c.claims()
if err != nil {
c.jsonErr(401, 401, err.Error())
return
}
uid := uint64(claims.UserID)
tid := uint64(claims.TenantId)
if uid == 0 || tid == 0 {
c.jsonErr(401, 401, "未获取到用户信息")
return
}
scene, _, qrURL, expireAt, err := wechatmp.CreateVerifyCode(models.WechatBindTypeTenantUser, uid, tid, 0)
if err != nil {
c.jsonErr(400, 400, "生成二维码失败:"+err.Error())
return
}
c.ok(map[string]interface{}{
"scene": scene,
"qrcode_url": qrURL,
"expire_at": expireAt,
"expires_in": int(time.Until(expireAt).Seconds()),
}, "生成成功")
}
// BindStatus GET /backend/wechatMp/bind/status?scene=xxx
func (c *BackendWechatMpController) BindStatus() {
claims, err := c.claims()
if err != nil {
c.jsonErr(401, 401, err.Error())
return
}
scene := strings.TrimSpace(c.GetString("scene"))
row, err := wechatmp.GetVerifyCode(scene)
if err != nil {
c.jsonErr(404, 404, err.Error())
return
}
if row.BindType != models.WechatBindTypeTenantUser || row.BindID != uint64(claims.UserID) {
c.jsonErr(403, 403, "该二维码不是当前账号发起,请重新获取")
return
}
c.ok(map[string]interface{}{
"status": row.Status,
"expire_at": row.ExpireAt,
}, "获取成功")
}
// BindConfirm POST /backend/wechatMp/bind/confirm {scene, code}
func (c *BackendWechatMpController) BindConfirm() {
claims, err := c.claims()
if err != nil {
c.jsonErr(401, 401, err.Error())
return
}
var p struct {
Scene string `json:"scene"`
Code string `json:"code"`
}
if !c.readBody(&p) {
return
}
if strings.TrimSpace(p.Code) == "" {
c.jsonErr(400, 400, "请输入公众号返回的验证码")
return
}
follower, err := wechatmp.ConfirmVerifyCode(
models.WechatBindTypeTenantUser,
uint64(claims.UserID),
uint64(claims.TenantId),
p.Scene,
p.Code,
)
if err != nil {
c.jsonErr(400, 400, err.Error())
return
}
c.ok(map[string]interface{}{
"openid": follower.OpenID,
"nickname": follower.Nickname,
"bind_time": follower.BindTime,
}, "绑定成功")
}
// MyBinding GET /backend/wechatMp/my
func (c *BackendWechatMpController) MyBinding() {
claims, err := c.claims()
if err != nil {
c.jsonErr(401, 401, err.Error())
return
}
follower, err := wechatmp.GetFollowerByUser(models.WechatBindTypeTenantUser, uint64(claims.UserID))
if err != nil {
c.jsonErr(500, 500, "查询失败:"+err.Error())
return
}
if follower == nil || follower.BindTid != uint64(claims.TenantId) {
c.ok(map[string]interface{}{"bound": false}, "获取成功")
return
}
c.ok(map[string]interface{}{
"bound": true,
"openid": follower.OpenID,
"nickname": follower.Nickname,
"avatar": follower.Avatar,
"subscribe": follower.Subscribe,
"bind_time": follower.BindTime,
"subscribe_time": follower.SubscribeTime,
}, "获取成功")
}
// Unbind POST /backend/wechatMp/unbind
func (c *BackendWechatMpController) Unbind() {
claims, err := c.claims()
if err != nil {
c.jsonErr(401, 401, err.Error())
return
}
if err := wechatmp.UnbindByUser(models.WechatBindTypeTenantUser, uint64(claims.UserID)); err != nil {
c.jsonErr(500, 500, "解绑失败:"+err.Error())
return
}
c.ok(nil, "已解绑")
}
+599
View File
@@ -0,0 +1,599 @@
package controllers
import (
"encoding/json"
"fmt"
"io"
"strings"
"time"
"server/models"
"server/pkg/jwtutil"
"server/services/wechatmp"
"github.com/beego/beego/v2/client/orm"
beego "github.com/beego/beego/v2/server/web"
)
// =============================================================
// 平台端「微信公众号(服务号)」:
// 配置(AppID/Secret/Token/AESKey)、绑定(扫码关注取验证码)、粉丝管理、消息推送
// 未认证阶段可完整跑通「扫码关注 → 公众号回复验证码 → 平台核销绑定」;
// 认证后开启模板消息推送(提醒/公告)。
// =============================================================
// PlatformWechatMpController 平台端微信公众号管理
type PlatformWechatMpController struct {
beego.Controller
}
func (c *PlatformWechatMpController) claims() (*jwtutil.Claims, error) {
auth := c.Ctx.Request.Header.Get("Authorization")
if auth == "" {
return nil, fmt.Errorf("未登录")
}
parts := strings.SplitN(auth, " ", 2)
if len(parts) != 2 || parts[0] != "Bearer" {
return nil, fmt.Errorf("认证信息格式错误")
}
claims, err := jwtutil.ParseToken(parts[1])
if err != nil {
return nil, fmt.Errorf("无效的token")
}
if claims.UserType != "platform" {
return nil, fmt.Errorf("无权访问")
}
return claims, nil
}
func (c *PlatformWechatMpController) jsonErr(httpStatus, bizCode int, msg string) {
c.Ctx.Output.SetStatus(httpStatus)
c.Data["json"] = map[string]interface{}{"code": bizCode, "msg": msg}
_ = c.ServeJSON()
}
func (c *PlatformWechatMpController) ok(data interface{}, msg string) {
if msg == "" {
msg = "success"
}
resp := map[string]interface{}{"code": 200, "msg": msg}
if data != nil {
resp["data"] = data
}
c.Data["json"] = resp
_ = c.ServeJSON()
}
func (c *PlatformWechatMpController) readBody(out interface{}) bool {
raw, err := io.ReadAll(c.Ctx.Request.Body)
if err != nil {
c.jsonErr(400, 400, "参数错误")
return false
}
if err := json.Unmarshal(raw, out); err != nil {
c.jsonErr(400, 400, "参数错误")
return false
}
return true
}
// ============================ 配置 ============================
// GetConfig GET /platform/wechatMp/config
func (c *PlatformWechatMpController) GetConfig() {
if _, err := c.claims(); err != nil {
c.jsonErr(401, 401, err.Error())
return
}
data := map[string]interface{}{
"app_id": "",
"app_secret": "",
"token": "",
"aes_key": "",
"encrypt_mode": wechatmp.EncryptModePlain,
"template_id": "",
"verified": 0,
"enabled": 0,
"remark": "",
"configured": false,
"callback_url": wechatmp.CallbackURL(),
}
if cfg, err := wechatmp.LoadConfig(); err == nil {
data["app_id"] = cfg.AppID
data["app_secret"] = wechatmp.MaskSecret(cfg.AppSecret)
data["token"] = cfg.Token
data["aes_key"] = wechatmp.MaskSecret(cfg.AESKey)
data["encrypt_mode"] = cfg.EncryptMode
data["template_id"] = cfg.TemplateID
data["remark"] = cfg.Remark
data["configured"] = cfg.AppID != ""
if cfg.Verified {
data["verified"] = 1
}
if cfg.Enabled {
data["enabled"] = 1
}
}
c.ok(data, "获取成功")
}
// SaveConfig POST /platform/wechatMp/config
func (c *PlatformWechatMpController) SaveConfig() {
if _, err := c.claims(); err != nil {
c.jsonErr(401, 401, err.Error())
return
}
var p struct {
AppID string `json:"app_id"`
AppSecret string `json:"app_secret"`
Token string `json:"token"`
AESKey string `json:"aes_key"`
EncryptMode string `json:"encrypt_mode"`
TemplateID string `json:"template_id"`
Verified bool `json:"verified"`
Enabled bool `json:"enabled"`
Remark string `json:"remark"`
}
if !c.readBody(&p) {
return
}
if strings.TrimSpace(p.AppID) == "" {
c.jsonErr(400, 400, "AppID 不能为空")
return
}
err := wechatmp.SaveConfig(wechatmp.SaveInput{
AppID: p.AppID,
AppSecret: p.AppSecret,
Token: p.Token,
AESKey: p.AESKey,
EncryptMode: p.EncryptMode,
TemplateID: p.TemplateID,
Verified: p.Verified,
Enabled: p.Enabled,
Remark: p.Remark,
})
if err != nil {
c.jsonErr(400, 400, "保存失败:"+err.Error())
return
}
c.ok(nil, "保存成功")
}
// TestConnection POST /platform/wechatMp/test
// 使用当前配置获取 access_token,验证 AppID/AppSecret 是否正确
func (c *PlatformWechatMpController) TestConnection() {
if _, err := c.claims(); err != nil {
c.jsonErr(401, 401, err.Error())
return
}
cfg, err := wechatmp.LoadConfig()
if err != nil || cfg.AppID == "" || cfg.AppSecret == "" {
c.jsonErr(400, 400, "请先保存 AppID 与 AppSecret")
return
}
token, err := wechatmp.GetAccessToken(cfg)
if err != nil {
msg := "连接失败:" + err.Error()
if hint, ok := wechatmp.ExplainError(err); ok {
msg += ";" + hint
}
c.jsonErr(400, 400, msg)
return
}
c.ok(map[string]interface{}{
"access_token_suffix": wechatmp.MaskSecret(token),
"callback_url": wechatmp.CallbackURL(),
"verified": boolToInt8(cfg.Verified),
}, "连接成功:AppID 与 AppSecret 校验通过")
}
// ============================ 绑定(当前平台用户) ============================
// BindStart POST /platform/wechatMp/bind/start
// 生成带参二维码;用户扫码关注后公众号将被动回复验证码
func (c *PlatformWechatMpController) BindStart() {
claims, err := c.claims()
if err != nil {
c.jsonErr(401, 401, err.Error())
return
}
uid := uint64(claims.UserID)
if uid == 0 {
c.jsonErr(401, 401, "未获取到用户信息")
return
}
scene, _, qrURL, expireAt, err := wechatmp.CreateVerifyCode(models.WechatBindTypePlatformUser, uid, 0, 0)
if err != nil {
c.jsonErr(400, 400, "生成二维码失败:"+err.Error())
return
}
c.ok(map[string]interface{}{
"scene": scene,
"qrcode_url": qrURL,
"expire_at": expireAt,
"expires_in": int(time.Until(expireAt).Seconds()),
"callback_url": wechatmp.CallbackURL(),
}, "生成成功")
}
// BindStatus GET /platform/wechatMp/bind/status?scene=xxx
func (c *PlatformWechatMpController) BindStatus() {
claims, err := c.claims()
if err != nil {
c.jsonErr(401, 401, err.Error())
return
}
scene := strings.TrimSpace(c.GetString("scene"))
row, err := wechatmp.GetVerifyCode(scene)
if err != nil {
c.jsonErr(404, 404, err.Error())
return
}
if row.BindType != models.WechatBindTypePlatformUser || row.BindID != uint64(claims.UserID) {
c.jsonErr(403, 403, "该二维码不是当前账号发起,请重新获取")
return
}
c.ok(map[string]interface{}{
"status": row.Status,
"expire_at": row.ExpireAt,
}, "获取成功")
}
// BindConfirm POST /platform/wechatMp/bind/confirm {scene, code}
func (c *PlatformWechatMpController) BindConfirm() {
claims, err := c.claims()
if err != nil {
c.jsonErr(401, 401, err.Error())
return
}
var p struct {
Scene string `json:"scene"`
Code string `json:"code"`
}
if !c.readBody(&p) {
return
}
if strings.TrimSpace(p.Code) == "" {
c.jsonErr(400, 400, "请输入公众号返回的验证码")
return
}
follower, err := wechatmp.ConfirmVerifyCode(models.WechatBindTypePlatformUser, uint64(claims.UserID), 0, p.Scene, p.Code)
if err != nil {
c.jsonErr(400, 400, err.Error())
return
}
c.ok(map[string]interface{}{
"openid": follower.OpenID,
"nickname": follower.Nickname,
"bind_time": follower.BindTime,
}, "绑定成功")
}
// MyBinding GET /platform/wechatMp/my
func (c *PlatformWechatMpController) MyBinding() {
claims, err := c.claims()
if err != nil {
c.jsonErr(401, 401, err.Error())
return
}
follower, err := wechatmp.GetFollowerByUser(models.WechatBindTypePlatformUser, uint64(claims.UserID))
if err != nil {
c.jsonErr(500, 500, "查询失败:"+err.Error())
return
}
if follower == nil {
c.ok(map[string]interface{}{"bound": false}, "获取成功")
return
}
c.ok(map[string]interface{}{
"bound": true,
"openid": follower.OpenID,
"nickname": follower.Nickname,
"avatar": follower.Avatar,
"subscribe": follower.Subscribe,
"bind_time": follower.BindTime,
"subscribe_time": follower.SubscribeTime,
}, "获取成功")
}
// Unbind POST /platform/wechatMp/unbind
func (c *PlatformWechatMpController) Unbind() {
claims, err := c.claims()
if err != nil {
c.jsonErr(401, 401, err.Error())
return
}
if err := wechatmp.UnbindByUser(models.WechatBindTypePlatformUser, uint64(claims.UserID)); err != nil {
c.jsonErr(500, 500, "解绑失败:"+err.Error())
return
}
c.ok(nil, "已解绑")
}
// ============================ 粉丝管理 ============================
// ListFollowers GET /platform/wechatMp/followers?page=&pageSize=&keyword=&bind_type=&subscribe=
func (c *PlatformWechatMpController) ListFollowers() {
if _, err := c.claims(); err != nil {
c.jsonErr(401, 401, err.Error())
return
}
page, _ := c.GetInt("page", 1)
pageSize, _ := c.GetInt("pageSize", 20)
if page < 1 {
page = 1
}
if pageSize < 1 || pageSize > 200 {
pageSize = 20
}
qs := models.Orm.QueryTable(new(models.WechatMpFollower)).Filter("delete_time__isnull", true)
if kw := strings.TrimSpace(c.GetString("keyword")); kw != "" {
cond := orm.NewCondition().
Or("nickname__icontains", kw).
Or("openid__icontains", kw).
Or("remark__icontains", kw)
qs = qs.SetCond(cond)
}
if bindType := strings.TrimSpace(c.GetString("bind_type")); bindType != "" {
qs = qs.Filter("bind_type", bindType)
}
if sub := strings.TrimSpace(c.GetString("subscribe")); sub == "0" || sub == "1" {
qs = qs.Filter("subscribe", sub)
}
total, err := qs.Count()
if err != nil {
c.jsonErr(500, 500, "获取失败:"+err.Error())
return
}
var rows []models.WechatMpFollower
if _, err := qs.OrderBy("-id").Limit(pageSize, (page-1)*pageSize).All(&rows); err != nil {
c.jsonErr(500, 500, "获取失败:"+err.Error())
return
}
// 绑定对象名称(平台用户 / 租户用户)
nameMap := loadWechatBindNames(rows)
list := make([]map[string]interface{}, 0, len(rows))
for i := range rows {
r := rows[i]
nickname := r.Nickname
if strings.TrimSpace(nickname) == "" {
nickname = maskOpenID(r.OpenID)
}
list = append(list, map[string]interface{}{
"id": r.ID,
"openid": r.OpenID,
"nickname": nickname,
"avatar": r.Avatar,
"sex": r.Sex,
"city": r.City,
"subscribe": r.Subscribe,
"subscribe_time": r.SubscribeTime,
"unsubscribe_time": r.UnsubscribeTime,
"bind_type": r.BindType,
"bind_id": r.BindID,
"bind_tid": r.BindTid,
"bind_name": nameMap[r.BindType+"#"+fmt.Sprintf("%d", r.BindID)+"#"+fmt.Sprintf("%d", r.BindTid)],
"bind_time": r.BindTime,
"remark": r.Remark,
"create_time": r.CreateTime,
})
}
c.ok(map[string]interface{}{"list": list, "total": total, "page": page, "pageSize": pageSize}, "获取成功")
}
// SetFollowerRemark POST /platform/wechatMp/follower/remark {id, remark}
func (c *PlatformWechatMpController) SetFollowerRemark() {
if _, err := c.claims(); err != nil {
c.jsonErr(401, 401, err.Error())
return
}
var p struct {
ID uint64 `json:"id"`
Remark string `json:"remark"`
}
if !c.readBody(&p) {
return
}
if p.ID == 0 {
c.jsonErr(400, 400, "参数错误")
return
}
now := time.Now()
if _, err := models.Orm.QueryTable(new(models.WechatMpFollower)).
Filter("id", p.ID).
Update(map[string]interface{}{"remark": strings.TrimSpace(p.Remark), "update_time": now}); err != nil {
c.jsonErr(500, 500, "保存失败:"+err.Error())
return
}
c.ok(nil, "保存成功")
}
// UnbindFollower POST /platform/wechatMp/follower/unbind {id}
// 仅清除绑定关系(保留粉丝记录)
func (c *PlatformWechatMpController) UnbindFollower() {
if _, err := c.claims(); err != nil {
c.jsonErr(401, 401, err.Error())
return
}
var p struct {
ID uint64 `json:"id"`
}
if !c.readBody(&p) {
return
}
if p.ID == 0 {
c.jsonErr(400, 400, "参数错误")
return
}
now := time.Now()
if _, err := models.Orm.QueryTable(new(models.WechatMpFollower)).
Filter("id", p.ID).
Update(map[string]interface{}{
"bind_type": "", "bind_id": 0, "bind_tid": 0, "update_time": now,
}); err != nil {
c.jsonErr(500, 500, "解绑失败:"+err.Error())
return
}
c.ok(nil, "已解绑")
}
// ============================ 推送 ============================
// Push POST /platform/wechatMp/push
// {title, content, remark, url, bind_type, bind_id, tid}
func (c *PlatformWechatMpController) Push() {
if _, err := c.claims(); err != nil {
c.jsonErr(401, 401, err.Error())
return
}
var p struct {
Title string `json:"title"`
Content string `json:"content"`
Remark string `json:"remark"`
URL string `json:"url"`
BindType string `json:"bind_type"`
BindID uint64 `json:"bind_id"`
Tid uint64 `json:"tid"`
}
if !c.readBody(&p) {
return
}
if strings.TrimSpace(p.Title) == "" && strings.TrimSpace(p.Content) == "" {
c.jsonErr(400, 400, "请填写推送内容")
return
}
cfg, err := wechatmp.LoadEnabledConfig()
if err != nil {
c.jsonErr(400, 400, "请先启用并配置微信公众号:"+err.Error())
return
}
sent, failed, perr := wechatmp.PushToFollowers(cfg, wechatmp.PushScope{
BindType: strings.TrimSpace(p.BindType),
BindID: p.BindID,
Tid: p.Tid,
}, wechatmp.PushMessage{
Title: strings.TrimSpace(p.Title),
Content: strings.TrimSpace(p.Content),
Remark: strings.TrimSpace(p.Remark),
URL: strings.TrimSpace(p.URL),
})
if perr != nil {
c.jsonErr(400, 400, perr.Error())
return
}
c.ok(map[string]interface{}{"sent": sent, "failed": failed},
fmt.Sprintf("推送完成:成功 %d 条,失败 %d 条", sent, failed))
}
// PushTest POST /platform/wechatMp/push/test {title, content, remark}
// 推送给当前登录平台用户绑定的微信
func (c *PlatformWechatMpController) PushTest() {
claims, err := c.claims()
if err != nil {
c.jsonErr(401, 401, err.Error())
return
}
var p struct {
Title string `json:"title"`
Content string `json:"content"`
Remark string `json:"remark"`
}
if !c.readBody(&p) {
return
}
cfg, err := wechatmp.LoadEnabledConfig()
if err != nil {
c.jsonErr(400, 400, "请先启用并配置微信公众号:"+err.Error())
return
}
follower, err := wechatmp.GetFollowerByUser(models.WechatBindTypePlatformUser, uint64(claims.UserID))
if err != nil || follower == nil {
c.jsonErr(400, 400, "当前账号尚未绑定微信,请先扫码绑定")
return
}
if follower.Subscribe != 1 {
c.jsonErr(400, 400, "该微信已取消关注,请重新关注后再试")
return
}
if p.Title == "" {
p.Title = "平台消息测试"
}
if p.Content == "" {
p.Content = "这是一条来自平台的测试消息,收到说明推送链路正常。"
}
if err := wechatmp.PushToOpenID(cfg, follower.OpenID, wechatmp.PushMessage{
Title: strings.TrimSpace(p.Title),
Content: strings.TrimSpace(p.Content),
Remark: strings.TrimSpace(p.Remark),
}); err != nil {
c.jsonErr(400, 400, "推送失败:"+err.Error())
return
}
c.ok(nil, "测试消息已发送")
}
// ============================ 辅助 ============================
func boolToInt8(b bool) int8 {
if b {
return 1
}
return 0
}
func maskOpenID(openid string) string {
if len(openid) <= 8 {
return openid
}
return openid[:4] + "****" + openid[len(openid)-4:]
}
// loadWechatBindNames 批量加载粉丝绑定对象名称:key = bindType#bindID#bindTid
func loadWechatBindNames(rows []models.WechatMpFollower) map[string]string {
names := map[string]string{}
if len(rows) == 0 {
return names
}
adminIDs := make([]uint64, 0)
tenantIDs := make([]uint64, 0)
for _, r := range rows {
switch r.BindType {
case models.WechatBindTypePlatformUser:
adminIDs = append(adminIDs, r.BindID)
case models.WechatBindTypeTenantUser:
tenantIDs = append(tenantIDs, r.BindID)
}
}
if len(adminIDs) > 0 {
var admins []models.AdminUser
if _, err := models.Orm.QueryTable(new(models.AdminUser)).Filter("id__in", adminIDs).All(&admins, "ID", "Account", "Name"); err == nil {
for _, a := range admins {
name := a.Account
if a.Name != nil && strings.TrimSpace(*a.Name) != "" {
name = *a.Name
}
names[models.WechatBindTypePlatformUser+"#"+fmt.Sprintf("%d", a.ID)+"#0"] = name
}
}
}
if len(tenantIDs) > 0 {
var users []models.SystemTenantUser
if _, err := models.Orm.QueryTable(new(models.SystemTenantUser)).Filter("uid__in", tenantIDs).All(&users, "UID", "Tid", "Account", "Name"); err == nil {
for _, u := range users {
name := ""
if u.Account != nil {
name = *u.Account
}
if u.Name != nil && strings.TrimSpace(*u.Name) != "" {
name = *u.Name
}
names[models.WechatBindTypeTenantUser+"#"+fmt.Sprintf("%d", u.Uid)+"#"+fmt.Sprintf("%d", u.Tid)] = name
}
}
}
return names
}
+101
View File
@@ -0,0 +1,101 @@
-- =============================================================
-- 微信公众号(服务号)绑定与推送
--
-- 业务说明:
-- 1. 服务号配置(yz_platform_wechat_mp_config):AppID / AppSecret / Token / EncodingAESKey,
-- 其中 AppSecret 与 EncodingAESKey 以 AES-256-GCM 加密后存储
-- (密钥取 app.conf 的 wechat_mp_secret_key,代码内部 SHA-256 派生)。
-- 2. 关注用户(yz_platform_wechat_mp_follower):用户关注/取关事件沉淀,openid 维度唯一;
-- 通过「扫码关注 → 公众号回复验证码 → 平台输入验证码」流程绑定到
-- 平台用户(platform_user)或租户用户(tenant_user),认证后可定向推送。
-- 3. 验证码(yz_platform_wechat_mp_verify_code):一次绑定会话一条,scene 绑定二维码场景值,
-- 默认 5 分钟有效;状态:0待扫码 1已扫码待核销 2已核销 3已过期。
--
-- 未认证服务号限制:仅支持服务器配置、关注/扫码事件、被动回复、带参数二维码;
-- 模板消息/订阅通知/客服消息需认证后用。
--
-- 本脚本可重复执行:建表用 IF NOT EXISTS,旧表名(yz_wechat_*)做存在性改名兼容。
-- 运行:mysql --default-character-set=utf8mb4 -h<host> -P<port> -u<user> -p <db> < create_platform_wechat_mp.sql
-- =============================================================
-- 0) 兼容早期表名(yz_wechat_mp_config / yz_wechat_follower / yz_wechat_verify_code):
-- 旧表存在且新表不存在时直接改名,保留已配置数据。
SET @old := (SELECT COUNT(*) FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME = 'yz_wechat_mp_config');
SET @new := (SELECT COUNT(*) FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME = 'yz_platform_wechat_mp_config');
SET @ddl := IF(@old > 0 AND @new = 0, 'RENAME TABLE `yz_wechat_mp_config` TO `yz_platform_wechat_mp_config`', 'DO 0');
PREPARE stmt FROM @ddl; EXECUTE stmt; DEALLOCATE PREPARE stmt;
SET @old := (SELECT COUNT(*) FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME = 'yz_wechat_follower');
SET @new := (SELECT COUNT(*) FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME = 'yz_platform_wechat_mp_follower');
SET @ddl := IF(@old > 0 AND @new = 0, 'RENAME TABLE `yz_wechat_follower` TO `yz_platform_wechat_mp_follower`', 'DO 0');
PREPARE stmt FROM @ddl; EXECUTE stmt; DEALLOCATE PREPARE stmt;
SET @old := (SELECT COUNT(*) FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME = 'yz_wechat_verify_code');
SET @new := (SELECT COUNT(*) FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME = 'yz_platform_wechat_mp_verify_code');
SET @ddl := IF(@old > 0 AND @new = 0, 'RENAME TABLE `yz_wechat_verify_code` TO `yz_platform_wechat_mp_verify_code`', 'DO 0');
PREPARE stmt FROM @ddl; EXECUTE stmt; DEALLOCATE PREPARE stmt;
-- 1) 服务号配置(单行)
CREATE TABLE IF NOT EXISTS `yz_platform_wechat_mp_config` (
`id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT '自增主键ID',
`app_id` varchar(64) NOT NULL DEFAULT '' COMMENT '公众号 AppID',
`app_secret` varchar(512) NOT NULL DEFAULT '' COMMENT 'AppSecret(AES 加密存储)',
`token` varchar(128) NOT NULL DEFAULT '' COMMENT '服务器配置 Token(自定义,用于签名校验)',
`aes_key` varchar(512) NOT NULL DEFAULT '' COMMENT 'EncodingAESKey(43位,AES 加密存储)',
`encrypt_mode` varchar(16) NOT NULL DEFAULT 'plain' COMMENT '消息加解密方式:plain明文 compatible兼容 safe安全',
`template_id` varchar(128) NOT NULL DEFAULT '' COMMENT '模板消息ID(认证后在公众号后台申请后填入)',
`verified` tinyint(4) NOT NULL DEFAULT 0 COMMENT '公众号是否已认证:1是 0否(认证后才可主动推送)',
`enabled` tinyint(4) NOT NULL DEFAULT 0 COMMENT '是否启用:1启用 0停用',
`remark` varchar(255) NOT NULL DEFAULT '' COMMENT '备注',
`create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',
`update_time` datetime DEFAULT NULL COMMENT '更新时间',
PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='微信公众号(服务号)配置表';
-- 2) 关注用户(粉丝)
CREATE TABLE IF NOT EXISTS `yz_platform_wechat_mp_follower` (
`id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT '自增主键ID',
`openid` varchar(64) NOT NULL DEFAULT '' COMMENT '微信 openid(公众号维度用户标识)',
`unionid` varchar(64) NOT NULL DEFAULT '' COMMENT '微信 unionid(开放平台绑定后才有)',
`nickname` varchar(128) NOT NULL DEFAULT '' COMMENT '昵称(未认证公众号可能拉取不到,留空展示 openid 缩写)',
`avatar` varchar(512) NOT NULL DEFAULT '' COMMENT '头像地址',
`sex` tinyint(4) NOT NULL DEFAULT 0 COMMENT '性别:0未知 1男 2女',
`city` varchar(64) NOT NULL DEFAULT '' COMMENT '城市',
`province` varchar(64) NOT NULL DEFAULT '' COMMENT '省份',
`country` varchar(64) NOT NULL DEFAULT '' COMMENT '国家',
`subscribe` tinyint(4) NOT NULL DEFAULT 1 COMMENT '关注状态:1已关注 0已取关',
`subscribe_time` datetime DEFAULT NULL COMMENT '关注时间',
`unsubscribe_time` datetime DEFAULT NULL COMMENT '取关时间',
`bind_type` varchar(32) NOT NULL DEFAULT '' COMMENT '绑定对象:platform_user平台用户 tenant_user租户用户 空=未绑定',
`bind_id` bigint(20) NOT NULL DEFAULT 0 COMMENT '绑定对象ID(平台用户ID / 租户用户ID)',
`bind_tid` bigint(20) NOT NULL DEFAULT 0 COMMENT '绑定租户ID(租户用户时记录)',
`bind_time` datetime DEFAULT NULL COMMENT '绑定时间',
`remark` varchar(255) NOT NULL DEFAULT '' COMMENT '备注(平台可编辑)',
`create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',
`update_time` datetime DEFAULT NULL COMMENT '更新时间',
`delete_time` datetime DEFAULT NULL COMMENT '删除时间(软删除)',
PRIMARY KEY (`id`),
UNIQUE KEY `uk_openid` (`openid`),
KEY `idx_bind` (`bind_type`,`bind_id`),
KEY `idx_subscribe` (`subscribe`),
KEY `idx_delete_time` (`delete_time`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='微信公众号关注用户(粉丝)表';
-- 3) 关注验证码(绑定会话)
CREATE TABLE IF NOT EXISTS `yz_platform_wechat_mp_verify_code` (
`id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT '自增主键ID',
`scene` varchar(64) NOT NULL DEFAULT '' COMMENT '二维码场景值(随机串,唯一)',
`code` varchar(8) NOT NULL DEFAULT '' COMMENT '验证码(6位数字,公众号被动回复给用户)',
`openid` varchar(64) NOT NULL DEFAULT '' COMMENT '扫码用户 openid(扫码后写入)',
`status` tinyint(4) NOT NULL DEFAULT 0 COMMENT '状态:0待扫码 1已扫码待核销 2已核销 3已过期',
`bind_type` varchar(32) NOT NULL DEFAULT '' COMMENT '发起绑定方:platform_user / tenant_user',
`bind_id` bigint(20) NOT NULL DEFAULT 0 COMMENT '发起绑定方ID',
`bind_tid` bigint(20) NOT NULL DEFAULT 0 COMMENT '发起绑定租户ID',
`expire_at` datetime DEFAULT NULL COMMENT '有效期截止(默认5分钟)',
`create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',
`update_time` datetime DEFAULT NULL COMMENT '更新时间',
PRIMARY KEY (`id`),
UNIQUE KEY `uk_scene` (`scene`),
KEY `idx_openid` (`openid`),
KEY `idx_create_time` (`create_time`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='微信公众号关注取码/绑定验证码表';
+5
View File
@@ -159,6 +159,11 @@ func Init(_ string) {
// 平台端推广佣金(docs/sql/create_platform_commission.sql)
new(PlatformCommissionRule),
new(PlatformCommissionOrder),
// 微信公众号(服务号)绑定与推送(docs/sql/create_platform_wechat_mp.sql)
new(WechatMpConfig),
new(WechatMpFollower),
new(WechatMpVerifyCode),
)
// 创建全局 Ormer
+85
View File
@@ -0,0 +1,85 @@
package models
import "time"
// =============================================================
// 微信公众号(服务号)绑定与推送(表结构见 docs/sql/create_platform_wechat_mp.sql)
// 1. 配置:AppID / AppSecret / Token / EncodingAESKey(敏感字段加密存储)
// 2. 粉丝:关注/取关事件沉淀,可绑定平台用户或租户用户,认证后定向推送
// 3. 验证码:扫码关注 → 公众号被动回复验证码 → 平台核销完成绑定
// =============================================================
// 绑定对象类型
const (
WechatBindTypePlatformUser = "platform_user" // 平台用户(yz_system_admin_user)
WechatBindTypeTenantUser = "tenant_user" // 租户用户(yz_system_tenant_user)
)
// 验证码状态
const (
WechatVerifyStatusWaiting = 0 // 待扫码
WechatVerifyStatusScanned = 1 // 已扫码待核销
WechatVerifyStatusUsed = 2 // 已核销
WechatVerifyStatusExpired = 3 // 已过期
)
// WechatMpConfig 微信公众号(服务号)配置 yz_platform_wechat_mp_config(单行)
type WechatMpConfig struct {
ID uint64 `orm:"column(id);pk;auto" json:"id"`
AppID string `orm:"column(app_id);size(64)" json:"app_id"`
AppSecret string `orm:"column(app_secret);size(512)" json:"-"` // 加密存储,不回显
Token string `orm:"column(token);size(128)" json:"token"`
AESKey string `orm:"column(aes_key);size(512)" json:"-"` // 加密存储,不回显
EncryptMode string `orm:"column(encrypt_mode);size(16)" json:"encrypt_mode"` // plain / compatible / safe
TemplateID string `orm:"column(template_id);size(128)" json:"template_id"`
Verified int8 `orm:"column(verified);default(0)" json:"verified"`
Enabled int8 `orm:"column(enabled);default(0)" json:"enabled"`
Remark string `orm:"column(remark);size(255)" json:"remark"`
CreateTime time.Time `orm:"column(create_time);type(datetime);auto_now_add" json:"create_time"`
UpdateTime *time.Time `orm:"column(update_time);type(datetime);null" json:"update_time"`
}
func (m *WechatMpConfig) TableName() string { return "yz_platform_wechat_mp_config" }
// WechatMpFollower 微信公众号关注用户(粉丝) yz_platform_wechat_mp_follower
type WechatMpFollower struct {
ID uint64 `orm:"column(id);pk;auto" json:"id"`
OpenID string `orm:"column(openid);size(64)" json:"openid"`
UnionID string `orm:"column(unionid);size(64)" json:"unionid"`
Nickname string `orm:"column(nickname);size(128)" json:"nickname"`
Avatar string `orm:"column(avatar);size(512)" json:"avatar"`
Sex int8 `orm:"column(sex);default(0)" json:"sex"`
City string `orm:"column(city);size(64)" json:"city"`
Province string `orm:"column(province);size(64)" json:"province"`
Country string `orm:"column(country);size(64)" json:"country"`
Subscribe int8 `orm:"column(subscribe);default(1)" json:"subscribe"`
SubscribeTime *time.Time `orm:"column(subscribe_time);type(datetime);null" json:"subscribe_time"`
UnsubscribeTime *time.Time `orm:"column(unsubscribe_time);type(datetime);null" json:"unsubscribe_time"`
BindType string `orm:"column(bind_type);size(32)" json:"bind_type"`
BindID uint64 `orm:"column(bind_id)" json:"bind_id"`
BindTid uint64 `orm:"column(bind_tid)" json:"bind_tid"`
BindTime *time.Time `orm:"column(bind_time);type(datetime);null" json:"bind_time"`
Remark string `orm:"column(remark);size(255)" json:"remark"`
CreateTime time.Time `orm:"column(create_time);type(datetime);auto_now_add" json:"create_time"`
UpdateTime *time.Time `orm:"column(update_time);type(datetime);null" json:"update_time"`
DeleteTime *time.Time `orm:"column(delete_time);type(datetime);null" json:"delete_time"`
}
func (m *WechatMpFollower) TableName() string { return "yz_platform_wechat_mp_follower" }
// WechatMpVerifyCode 公众号关注取码/绑定验证码 yz_platform_wechat_mp_verify_code
type WechatMpVerifyCode struct {
ID uint64 `orm:"column(id);pk;auto" json:"id"`
Scene string `orm:"column(scene);size(64)" json:"scene"`
Code string `orm:"column(code);size(8)" json:"code"`
OpenID string `orm:"column(openid);size(64)" json:"openid"`
Status int8 `orm:"column(status);default(0)" json:"status"`
BindType string `orm:"column(bind_type);size(32)" json:"bind_type"`
BindID uint64 `orm:"column(bind_id)" json:"bind_id"`
BindTid uint64 `orm:"column(bind_tid)" json:"bind_tid"`
ExpireAt *time.Time `orm:"column(expire_at);type(datetime);null" json:"expire_at"`
CreateTime time.Time `orm:"column(create_time);type(datetime);auto_now_add" json:"create_time"`
UpdateTime *time.Time `orm:"column(update_time);type(datetime);null" json:"update_time"`
}
func (m *WechatMpVerifyCode) TableName() string { return "yz_platform_wechat_mp_verify_code" }
+1
View File
@@ -13,6 +13,7 @@ func Register() {
RegisterRoleRoutes()
RegisterPaymentRoutes()
RegisterProductRoutes()
RegisterWechatMpRoutes()
}
// RegisterProductRoutes 注册租户端「产品购买」路由。
+17
View File
@@ -0,0 +1,17 @@
package backend
import (
"server/controllers"
beego "github.com/beego/beego/v2/server/web"
)
// RegisterWechatMpRoutes 注册租户端「微信公众号绑定」路由。
// 扫码关注公众号 → 公众号被动回复验证码 → 本端核销绑定(接口见 controllers/backend_wechat_mp.go)。
func RegisterWechatMpRoutes() {
beego.Router("/backend/wechatMp/bind/start", &controllers.BackendWechatMpController{}, "post:BindStart")
beego.Router("/backend/wechatMp/bind/status", &controllers.BackendWechatMpController{}, "get:BindStatus")
beego.Router("/backend/wechatMp/bind/confirm", &controllers.BackendWechatMpController{}, "post:BindConfirm")
beego.Router("/backend/wechatMp/my", &controllers.BackendWechatMpController{}, "get:MyBinding")
beego.Router("/backend/wechatMp/unbind", &controllers.BackendWechatMpController{}, "post:Unbind")
}
+3
View File
@@ -11,6 +11,9 @@ func Register() {
// 支付模块(渠道配置 / 支付订单 / 对账 / 佣金 / 渠道回调)
RegisterPaymentRoutes()
// 微信公众号(服务号):配置 / 绑定 / 粉丝 / 推送 / 微信服务器回调
RegisterWechatMpRoutes()
// 平台登录相关
beego.Router("/platform/login", &controllers.PlatformAuthController{}, "post:LoginPlatform")
beego.Router("/platform/currentUser", &controllers.PlatformAuthController{}, "get:GetCurrentUser")
+34
View File
@@ -0,0 +1,34 @@
package platform
import (
"server/controllers"
beego "github.com/beego/beego/v2/server/web"
)
// RegisterWechatMpRoutes 注册微信公众号(服务号)相关路由(表结构见 docs/sql/create_platform_wechat_mp.sql)。
// 注意:/api/wechat/mp/callback 供微信服务器回调,无平台鉴权(控制器内完成签名校验)。
func RegisterWechatMpRoutes() {
// 微信服务器回调(URL 填入公众号后台「服务器配置」,Token 与平台配置一致)
beego.Router("/api/wechat/mp/callback", &controllers.WechatMpCallbackController{}, "get:Callback;post:Callback")
// 服务号配置
beego.Router("/platform/wechatMp/config", &controllers.PlatformWechatMpController{}, "get:GetConfig;post:SaveConfig")
beego.Router("/platform/wechatMp/test", &controllers.PlatformWechatMpController{}, "post:TestConnection")
// 绑定(平台用户)
beego.Router("/platform/wechatMp/bind/start", &controllers.PlatformWechatMpController{}, "post:BindStart")
beego.Router("/platform/wechatMp/bind/status", &controllers.PlatformWechatMpController{}, "get:BindStatus")
beego.Router("/platform/wechatMp/bind/confirm", &controllers.PlatformWechatMpController{}, "post:BindConfirm")
beego.Router("/platform/wechatMp/my", &controllers.PlatformWechatMpController{}, "get:MyBinding")
beego.Router("/platform/wechatMp/unbind", &controllers.PlatformWechatMpController{}, "post:Unbind")
// 粉丝管理
beego.Router("/platform/wechatMp/followers", &controllers.PlatformWechatMpController{}, "get:ListFollowers")
beego.Router("/platform/wechatMp/follower/remark", &controllers.PlatformWechatMpController{}, "post:SetFollowerRemark")
beego.Router("/platform/wechatMp/follower/unbind", &controllers.PlatformWechatMpController{}, "post:UnbindFollower")
// 消息推送(认证后可用:提醒 / 公告)
beego.Router("/platform/wechatMp/push", &controllers.PlatformWechatMpController{}, "post:Push")
beego.Router("/platform/wechatMp/push/test", &controllers.PlatformWechatMpController{}, "post:PushTest")
}
+282
View File
@@ -0,0 +1,282 @@
package wechatmp
import (
"bytes"
"encoding/json"
"errors"
"fmt"
"io"
"net/http"
"net/url"
"time"
)
// =============================================================
// 微信公众平台服务端 API 调用(access_token / 带参二维码 / 模板消息 / 订阅通知 / 用户信息)
// =============================================================
const apiBase = "https://api.weixin.qq.com"
var httpClient = &http.Client{Timeout: 10 * time.Second}
// APIError 微信接口业务错误(HTTP 200 但 errcode 非 0)。
// Error() 输出保留 errcode / errmsg 原始信息,便于日志检索与前端展示。
type APIError struct {
API string // 调用的接口(如「获取 access_token」)
Code int
Msg string
}
func (e *APIError) Error() string {
return fmt.Sprintf("%s 失败:errcode=%d errmsg=%s", e.API, e.Code, e.Msg)
}
// Explain 返回常见 errcode 的中文排查建议;未知错误码返回空串。
func (e *APIError) Explain() string {
switch e.Code {
case 40013:
return "AppID 无效:请核对微信后台「设置与开发 → 基本配置」中的 AppID"
case 40125:
return "AppSecret 错误:请在微信后台重置后复制新的 AppSecret 填入本页"
case 40164, 61004:
return "服务器出口 IP 不在白名单:请在微信后台「基本配置 → IP白名单」加入本服务器出口 IP(服务器执行 curl ifconfig.me 可查看)"
case 45009:
return "接口调用频次超限:请稍后重试"
case 48001:
return "接口未获授权:需认证服务号,或该账号未开通此接口能力"
}
return ""
}
// ExplainError 提取微信接口错误的中文排查建议;无可用建议时返回 ("", false)。
func ExplainError(err error) (string, bool) {
var ae *APIError
if errors.As(err, &ae) {
if hint := ae.Explain(); hint != "" {
return hint, true
}
}
return "", false
}
// GetAccessToken 获取 access_token(内存缓存,提前 5 分钟刷新)
func GetAccessToken(cfg *Config) (string, error) {
if cfg == nil || cfg.AppID == "" || cfg.AppSecret == "" {
return "", fmt.Errorf("AppID / AppSecret 未配置")
}
tokenMu.Lock()
if tokenCache.appID == cfg.AppID && tokenCache.token != "" && time.Now().Before(tokenCache.expiresAt) {
t := tokenCache.token
tokenMu.Unlock()
return t, nil
}
tokenMu.Unlock()
api := fmt.Sprintf("%s/cgi-bin/token?grant_type=client_credential&appid=%s&secret=%s",
apiBase, url.QueryEscape(cfg.AppID), url.QueryEscape(cfg.AppSecret))
var resp struct {
AccessToken string `json:"access_token"`
ExpiresIn int `json:"expires_in"`
ErrCode int `json:"errcode"`
ErrMsg string `json:"errmsg"`
}
if err := httpGetJSON(api, &resp); err != nil {
return "", err
}
if resp.AccessToken == "" {
return "", &APIError{API: "获取 access_token", Code: resp.ErrCode, Msg: resp.ErrMsg}
}
ttl := time.Duration(resp.ExpiresIn) * time.Second
if ttl <= tokenMinTTL {
ttl = time.Hour
}
tokenMu.Lock()
tokenCache = tokenEntry{appID: cfg.AppID, token: resp.AccessToken, expiresAt: time.Now().Add(ttl - tokenMinTTL)}
tokenMu.Unlock()
return resp.AccessToken, nil
}
// CreateQRCode 创建带参数临时二维码(字符串场景值),返回二维码图片地址。
// scene 会通过「关注事件的 EventKey(qrscene_<scene>)/ 扫码事件的 EventKey(<scene>)」回传。
func CreateQRCode(cfg *Config, scene string, expireSeconds int) (string, error) {
token, err := GetAccessToken(cfg)
if err != nil {
return "", err
}
if expireSeconds <= 0 || expireSeconds > 2592000 {
expireSeconds = 300
}
payload := map[string]interface{}{
"expire_seconds": expireSeconds,
"action_name": "QR_STR_SCENE",
"action_info": map[string]interface{}{
"scene": map[string]string{"scene_str": scene},
},
}
api := fmt.Sprintf("%s/cgi-bin/qrcode/create?access_token=%s", apiBase, url.QueryEscape(token))
var resp struct {
Ticket string `json:"ticket"`
ExpireSeconds int `json:"expire_seconds"`
URL string `json:"url"`
ErrCode int `json:"errcode"`
ErrMsg string `json:"errmsg"`
}
if err := httpPostJSON(api, payload, &resp); err != nil {
return "", err
}
if resp.Ticket == "" {
return "", &APIError{API: "创建带参二维码", Code: resp.ErrCode, Msg: resp.ErrMsg}
}
return "https://mp.weixin.qq.com/cgi-bin/showqrcode?ticket=" + url.QueryEscape(resp.Ticket), nil
}
// SendTemplateMessage 发送模板消息(需认证服务号 + 模板ID)
// data 键为模板字段名(first / keyword1 / remark ...),值为文本内容。
func SendTemplateMessage(cfg *Config, openid, jumpURL string, data map[string]string) error {
if cfg.TemplateID == "" {
return fmt.Errorf("未配置模板消息 ID(认证后在公众号后台申请)")
}
token, err := GetAccessToken(cfg)
if err != nil {
return err
}
fields := make(map[string]map[string]string, len(data))
for k, v := range data {
fields[k] = map[string]string{"value": v}
}
payload := map[string]interface{}{
"touser": openid,
"template_id": cfg.TemplateID,
"data": fields,
}
if jumpURL != "" {
payload["url"] = jumpURL
}
api := fmt.Sprintf("%s/cgi-bin/message/template/send?access_token=%s", apiBase, url.QueryEscape(token))
var resp struct {
ErrCode int `json:"errcode"`
ErrMsg string `json:"errmsg"`
MsgID int64 `json:"msgid"`
}
if err := httpPostJSON(api, payload, &resp); err != nil {
return err
}
if resp.ErrCode != 0 {
return &APIError{API: "发送模板消息", Code: resp.ErrCode, Msg: resp.ErrMsg}
}
return nil
}
// SendSubscribeMessage 发送订阅通知(一次性订阅,需用户先在页面完成订阅授权)
func SendSubscribeMessage(cfg *Config, openid, templateID, page string, data map[string]string) error {
if templateID == "" {
return fmt.Errorf("未配置订阅通知模板 ID")
}
token, err := GetAccessToken(cfg)
if err != nil {
return err
}
fields := make(map[string]map[string]string, len(data))
for k, v := range data {
fields[k] = map[string]string{"value": v}
}
payload := map[string]interface{}{
"touser": openid,
"template_id": templateID,
"data": fields,
}
if page != "" {
payload["page"] = page
}
api := fmt.Sprintf("%s/cgi-bin/message/subscribe/bizsend?access_token=%s", apiBase, url.QueryEscape(token))
var resp struct {
ErrCode int `json:"errcode"`
ErrMsg string `json:"errmsg"`
}
if err := httpPostJSON(api, payload, &resp); err != nil {
return err
}
if resp.ErrCode != 0 {
return &APIError{API: "发送订阅通知", Code: resp.ErrCode, Msg: resp.ErrMsg}
}
return nil
}
// UserInfo 用户基本信息(昵称/头像等;未认证公众号可能接口受限,调用失败由上层忽略)
type UserInfo struct {
OpenID string `json:"openid"`
Nickname string `json:"nickname"`
Sex int `json:"sex"`
City string `json:"city"`
Province string `json:"province"`
Country string `json:"country"`
HeadImgURL string `json:"headimgurl"`
Subscribe int `json:"subscribe"`
}
// FetchUserInfo 拉取用户基本信息(失败不算错,由调用方容错)
func FetchUserInfo(cfg *Config, openid string) (*UserInfo, error) {
token, err := GetAccessToken(cfg)
if err != nil {
return nil, err
}
api := fmt.Sprintf("%s/cgi-bin/user/info?access_token=%s&openid=%s&lang=zh_CN",
apiBase, url.QueryEscape(token), url.QueryEscape(openid))
var resp struct {
UserInfo
ErrCode int `json:"errcode"`
ErrMsg string `json:"errmsg"`
}
if err := httpGetJSON(api, &resp); err != nil {
return nil, err
}
if resp.ErrCode != 0 {
return nil, &APIError{API: "获取用户信息", Code: resp.ErrCode, Msg: resp.ErrMsg}
}
return &resp.UserInfo, nil
}
// ============================ HTTP 辅助 ============================
func httpGetJSON(api string, out interface{}) error {
resp, err := httpClient.Get(api)
if err != nil {
return fmt.Errorf("请求微信接口失败: %w", err)
}
defer resp.Body.Close()
body, err := io.ReadAll(resp.Body)
if err != nil {
return fmt.Errorf("读取微信响应失败: %w", err)
}
if err := json.Unmarshal(body, out); err != nil {
return fmt.Errorf("解析微信响应失败: %w(body: %s)", err, truncate(string(body), 200))
}
return nil
}
func httpPostJSON(api string, payload interface{}, out interface{}) error {
bs, err := json.Marshal(payload)
if err != nil {
return err
}
resp, err := httpClient.Post(api, "application/json; charset=utf-8", bytes.NewReader(bs))
if err != nil {
return fmt.Errorf("请求微信接口失败: %w", err)
}
defer resp.Body.Close()
body, err := io.ReadAll(resp.Body)
if err != nil {
return fmt.Errorf("读取微信响应失败: %w", err)
}
if err := json.Unmarshal(body, out); err != nil {
return fmt.Errorf("解析微信响应失败: %w(body: %s)", err, truncate(string(body), 200))
}
return nil
}
func truncate(s string, n int) string {
if len(s) <= n {
return s
}
return s[:n] + "..."
}
+351
View File
@@ -0,0 +1,351 @@
package wechatmp
import (
"crypto/rand"
"encoding/hex"
"errors"
"fmt"
"math/big"
"strings"
"time"
"server/models"
"github.com/beego/beego/v2/client/orm"
)
// =============================================================
// 关注绑定:扫码关注 → 公众号被动回复验证码 → 平台核销完成绑定
// 一个绑定会话 = 一条 yz_platform_wechat_mp_verify_code(scene + 6位验证码 + 过期时间)
// =============================================================
// DefaultVerifyTTL 验证码默认有效期
const DefaultVerifyTTL = 5 * time.Minute
var (
ErrSceneNotScanned = errors.New("尚未检测到扫码关注,请先使用微信扫码并关注公众号")
ErrCodeUsed = errors.New("该验证码已使用,请重新获取二维码")
ErrCodeExpired = errors.New("二维码已过期,请重新获取")
ErrCodeMismatch = errors.New("验证码不正确")
ErrSceneInvalid = errors.New("二维码无效或已失效")
)
// CreateVerifyCode 创建绑定会话并生成带参二维码。
// bindType 为 platform_user / tenant_user,bindID 为发起方用户 ID。
func CreateVerifyCode(bindType string, bindID, bindTid uint64, ttl time.Duration) (scene, code, qrURL string, expireAt time.Time, err error) {
cfg, err := LoadEnabledConfig()
if err != nil {
return "", "", "", time.Time{}, err
}
if ttl <= 0 {
ttl = DefaultVerifyTTL
}
scene = randomScene()
code = randomCode6()
now := time.Now()
expireAt = now.Add(ttl)
row := &models.WechatMpVerifyCode{
Scene: scene,
Code: code,
Status: models.WechatVerifyStatusWaiting,
BindType: bindType,
BindID: bindID,
BindTid: bindTid,
ExpireAt: &expireAt,
CreateTime: now,
UpdateTime: &now,
}
if _, err = models.Orm.Insert(row); err != nil {
return "", "", "", time.Time{}, fmt.Errorf("创建绑定会话失败: %w", err)
}
qrURL, err = CreateQRCode(cfg, scene, int(ttl.Seconds())+120)
if err != nil {
return "", "", "", time.Time{}, err
}
return scene, code, qrURL, expireAt, nil
}
// GetVerifyCode 按 scene 查询绑定会话(过期时惰性标记)
func GetVerifyCode(scene string) (*models.WechatMpVerifyCode, error) {
scene = strings.TrimSpace(scene)
if scene == "" {
return nil, ErrSceneInvalid
}
var row models.WechatMpVerifyCode
if err := models.Orm.QueryTable(new(models.WechatMpVerifyCode)).Filter("scene", scene).One(&row); err != nil {
return nil, ErrSceneInvalid
}
if row.Status != models.WechatVerifyStatusUsed && row.ExpireAt != nil && row.ExpireAt.Before(time.Now()) {
row.Status = models.WechatVerifyStatusExpired
now := time.Now()
_, _ = models.Orm.QueryTable(new(models.WechatMpVerifyCode)).
Filter("id", row.ID).
Update(map[string]interface{}{"status": row.Status, "update_time": now})
}
return &row, nil
}
// ConfirmVerifyCode 核销验证码并完成绑定,返回粉丝记录
func ConfirmVerifyCode(bindType string, bindID, bindTid uint64, scene, code string) (*models.WechatMpFollower, error) {
row, err := GetVerifyCode(scene)
if err != nil {
return nil, err
}
if row.BindType != bindType || row.BindID != bindID {
return nil, errors.New("该二维码不是当前账号发起,请重新获取")
}
switch row.Status {
case models.WechatVerifyStatusWaiting:
return nil, ErrSceneNotScanned
case models.WechatVerifyStatusUsed:
return nil, ErrCodeUsed
case models.WechatVerifyStatusExpired:
return nil, ErrCodeExpired
}
if strings.TrimSpace(code) != row.Code {
return nil, ErrCodeMismatch
}
now := time.Now()
if _, err := models.Orm.QueryTable(new(models.WechatMpVerifyCode)).
Filter("id", row.ID).
Update(map[string]interface{}{"status": models.WechatVerifyStatusUsed, "update_time": now}); err != nil {
return nil, err
}
return BindFollower(row.OpenID, bindType, bindID, bindTid)
}
// BindFollower 将 openid 绑定到指定账号。
// 同一账号只保留一个微信:旧的 openid 绑定关系会被清除。
func BindFollower(openid, bindType string, bindID, bindTid uint64) (*models.WechatMpFollower, error) {
openid = strings.TrimSpace(openid)
if openid == "" {
return nil, errors.New("openid 为空")
}
now := time.Now()
_, _ = models.Orm.QueryTable(new(models.WechatMpFollower)).
Filter("bind_type", bindType).
Filter("bind_id", bindID).
Filter("openid__ne", openid).
Update(map[string]interface{}{
"bind_type": "", "bind_id": 0, "bind_tid": 0, "update_time": now,
})
var row models.WechatMpFollower
err := models.Orm.QueryTable(new(models.WechatMpFollower)).Filter("openid", openid).One(&row)
if err == orm.ErrNoRows {
row = models.WechatMpFollower{
OpenID: openid,
Subscribe: 1,
SubscribeTime: &now,
CreateTime: now,
}
}
row.BindType = bindType
row.BindID = bindID
row.BindTid = bindTid
row.BindTime = &now
row.UpdateTime = &now
if row.ID == 0 {
id, ierr := models.Orm.Insert(&row)
if ierr != nil {
return nil, ierr
}
row.ID = uint64(id)
return &row, nil
}
if _, uerr := models.Orm.Update(&row, "BindType", "BindID", "BindTid", "BindTime", "UpdateTime"); uerr != nil {
return nil, uerr
}
return &row, nil
}
// UnbindByUser 清除某账号的微信绑定(按绑定对象维度)
func UnbindByUser(bindType string, bindID uint64) error {
if bindID == 0 {
return nil
}
now := time.Now()
_, err := models.Orm.QueryTable(new(models.WechatMpFollower)).
Filter("bind_type", bindType).
Filter("bind_id", bindID).
Update(map[string]interface{}{
"bind_type": "", "bind_id": 0, "bind_tid": 0, "update_time": now,
})
return err
}
// GetFollowerByUser 查询某账号已绑定的粉丝记录(未绑定返回 nil, nil)
func GetFollowerByUser(bindType string, bindID uint64) (*models.WechatMpFollower, error) {
if bindID == 0 {
return nil, nil
}
var row models.WechatMpFollower
err := models.Orm.QueryTable(new(models.WechatMpFollower)).
Filter("bind_type", bindType).
Filter("bind_id", bindID).
Filter("delete_time__isnull", true).
OrderBy("-id").
One(&row)
if err == orm.ErrNoRows {
return nil, nil
}
if err != nil {
return nil, err
}
return &row, nil
}
// UpsertSubscribe 关注事件:创建/恢复粉丝记录
func UpsertSubscribe(openid string, at time.Time) (*models.WechatMpFollower, error) {
openid = strings.TrimSpace(openid)
if openid == "" {
return nil, errors.New("openid 为空")
}
var row models.WechatMpFollower
err := models.Orm.QueryTable(new(models.WechatMpFollower)).Filter("openid", openid).One(&row)
if err == orm.ErrNoRows {
row = models.WechatMpFollower{OpenID: openid, Subscribe: 1, SubscribeTime: &at, CreateTime: at, UpdateTime: &at}
id, ierr := models.Orm.Insert(&row)
if ierr != nil {
return nil, ierr
}
row.ID = uint64(id)
return &row, nil
}
if err != nil {
return nil, err
}
row.Subscribe = 1
row.SubscribeTime = &at
row.UnsubscribeTime = nil
row.UpdateTime = &at
if _, err := models.Orm.Update(&row, "Subscribe", "SubscribeTime", "UnsubscribeTime", "UpdateTime"); err != nil {
return nil, err
}
return &row, nil
}
// MarkUnsubscribe 取关事件
func MarkUnsubscribe(openid string, at time.Time) error {
openid = strings.TrimSpace(openid)
if openid == "" {
return nil
}
_, err := models.Orm.QueryTable(new(models.WechatMpFollower)).
Filter("openid", openid).
Update(map[string]interface{}{
"subscribe": 0, "unsubscribe_time": at, "update_time": at,
})
return err
}
// RefreshFollowerProfile 尝试拉取粉丝昵称/头像(未认证公众号可能受限,失败静默忽略)
func RefreshFollowerProfile(cfg *Config, openid string) {
if cfg == nil || openid == "" {
return
}
info, err := FetchUserInfo(cfg, openid)
if err != nil || info == nil {
return
}
now := time.Now()
up := map[string]interface{}{
"sex": info.Sex,
"city": info.City,
"province": info.Province,
"country": info.Country,
"update_time": now,
}
if info.Nickname != "" {
up["nickname"] = info.Nickname
}
if info.HeadImgURL != "" {
up["avatar"] = info.HeadImgURL
}
_, _ = models.Orm.QueryTable(new(models.WechatMpFollower)).Filter("openid", openid).Update(up)
}
// HandleInbound 处理入站消息/事件,返回需要被动回复的文本(空字符串表示不回复)
func HandleInbound(cfg *Config, msg *InboundMessage) string {
if msg == nil {
return ""
}
openid := strings.TrimSpace(msg.FromUserName)
now := time.Now()
switch strings.ToLower(strings.TrimSpace(msg.MsgType)) {
case "event":
switch strings.ToUpper(strings.TrimSpace(msg.Event)) {
case "subscribe":
_, _ = UpsertSubscribe(openid, now)
scene := strings.TrimPrefix(strings.TrimSpace(msg.EventKey), "qrscene_")
if scene != "" {
return handleScanScene(cfg, scene, openid)
}
return "欢迎关注云泽平台!\n如需接收系统提醒、公告推送,请回到系统页面点击「绑定微信」,使用微信扫码获取专属验证码。"
case "scan":
return handleScanScene(cfg, strings.TrimSpace(msg.EventKey), openid)
case "unsubscribe":
_ = MarkUnsubscribe(openid, now)
return ""
}
case "text":
return "如需绑定账号:请回到系统页面点击「绑定微信」,展示专属二维码后使用微信扫码,公众号会自动回复验证码。\n绑定成功后即可接收系统提醒、公告推送。"
}
return ""
}
// handleScanScene 扫码(关注/已关注)后的统一处理:写入验证码会话 + 回复验证码
func handleScanScene(cfg *Config, scene, openid string) string {
if strings.TrimSpace(scene) == "" {
return "未能识别二维码信息,请在系统页面重新获取二维码后再扫码。"
}
row, err := GetVerifyCode(scene)
if err != nil {
return "二维码无效或已失效,请在系统页面重新获取。"
}
switch row.Status {
case models.WechatVerifyStatusUsed:
return "该验证码已使用,请在系统页面重新获取二维码。"
case models.WechatVerifyStatusExpired:
return "二维码已过期,请在系统页面重新获取。"
}
now := time.Now()
if _, uerr := models.Orm.QueryTable(new(models.WechatMpVerifyCode)).
Filter("id", row.ID).
Update(map[string]interface{}{
"openid": openid, "status": models.WechatVerifyStatusScanned, "update_time": now,
}); uerr != nil {
return "系统繁忙,请稍后重新扫码。"
}
// 拉取昵称头像(失败不影响绑定流程)
RefreshFollowerProfile(cfg, openid)
minutes := int(time.Until(*row.ExpireAt).Minutes())
if minutes < 1 {
minutes = 1
}
return fmt.Sprintf("您的验证码是:%s\n请回到系统页面输入完成绑定(%d 分钟内有效)。", row.Code, minutes)
}
// ============================ 随机值 ============================
func randomScene() string {
b := make([]byte, 16)
if _, err := rand.Read(b); err != nil {
return fmt.Sprintf("s%d", time.Now().UnixNano())
}
return hex.EncodeToString(b)
}
func randomCode6() string {
n, err := rand.Int(rand.Reader, big.NewInt(900000))
if err != nil {
return "123456"
}
return fmt.Sprintf("%06d", n.Int64()+100000)
}
+207
View File
@@ -0,0 +1,207 @@
package wechatmp
import (
"errors"
"strings"
"sync"
"time"
"server/models"
"github.com/beego/beego/v2/client/orm"
beego "github.com/beego/beego/v2/server/web"
)
// =============================================================
// 微信公众号(服务号)配置与 access_token
// =============================================================
// 消息加解密方式
const (
EncryptModePlain = "plain" // 明文模式
EncryptModeCompatible = "compatible" // 兼容模式
EncryptModeSafe = "safe" // 安全模式
)
// Config 服务号运行配置(敏感字段已解密)
type Config struct {
ID uint64
AppID string
AppSecret string // 明文
Token string
AESKey string // 明文(43 位 EncodingAESKey)
EncryptMode string
TemplateID string
Verified bool
Enabled bool
Remark string
}
// ErrNotConfigured 未配置或未启用
var ErrNotConfigured = errors.New("微信公众号未配置或未启用")
// LoadConfig 读取服务号配置(单行,敏感字段解密);未配置返回 ErrNotConfigured
func LoadConfig() (*Config, error) {
var row models.WechatMpConfig
err := models.Orm.QueryTable(new(models.WechatMpConfig)).OrderBy("id").One(&row)
if err != nil {
if err == orm.ErrNoRows {
return nil, ErrNotConfigured
}
return nil, err
}
cfg := &Config{
ID: row.ID,
AppID: strings.TrimSpace(row.AppID),
Token: strings.TrimSpace(row.Token),
EncryptMode: strings.TrimSpace(row.EncryptMode),
TemplateID: strings.TrimSpace(row.TemplateID),
Verified: row.Verified == 1,
Enabled: row.Enabled == 1,
Remark: row.Remark,
}
if cfg.EncryptMode == "" {
cfg.EncryptMode = EncryptModePlain
}
if row.AppSecret != "" {
if plain, derr := DecryptSecret(row.AppSecret); derr == nil {
cfg.AppSecret = plain
} else {
return nil, derr
}
}
if row.AESKey != "" {
if plain, derr := DecryptSecret(row.AESKey); derr == nil {
cfg.AESKey = plain
} else {
return nil, derr
}
}
return cfg, nil
}
// LoadEnabledConfig 读取并校验启用状态
func LoadEnabledConfig() (*Config, error) {
cfg, err := LoadConfig()
if err != nil {
return nil, err
}
if !cfg.Enabled || cfg.AppID == "" || cfg.AppSecret == "" {
return nil, ErrNotConfigured
}
return cfg, nil
}
// SaveInput 保存配置入参(敏感字段支持掩码表示不改)
type SaveInput struct {
AppID string
AppSecret string // 掩码 / 空串 = 保持不变
Token string
AESKey string // 掩码 / 空串 = 保持不变
EncryptMode string
TemplateID string
Verified bool
Enabled bool
Remark string
}
// SaveConfig 保存服务号配置(不存在则创建)
func SaveConfig(in SaveInput) error {
if strings.TrimSpace(in.AppID) == "" {
return errors.New("AppID 不能为空")
}
mode := strings.TrimSpace(in.EncryptMode)
if mode == "" {
mode = EncryptModePlain
}
var row models.WechatMpConfig
err := models.Orm.QueryTable(new(models.WechatMpConfig)).OrderBy("id").One(&row)
now := time.Now()
isNew := false
if err != nil {
if err != orm.ErrNoRows {
return err
}
isNew = true
row = models.WechatMpConfig{}
}
// 敏感字段:掩码/空串表示保持原值
secret := strings.TrimSpace(in.AppSecret)
if secret != "" && !IsMasked(secret) {
enc, eerr := EncryptSecret(secret)
if eerr != nil {
return eerr
}
row.AppSecret = enc
}
aesKey := strings.TrimSpace(in.AESKey)
if aesKey != "" && !IsMasked(aesKey) {
enc, eerr := EncryptSecret(aesKey)
if eerr != nil {
return eerr
}
row.AESKey = enc
}
row.AppID = strings.TrimSpace(in.AppID)
row.Token = strings.TrimSpace(in.Token)
row.EncryptMode = mode
row.TemplateID = strings.TrimSpace(in.TemplateID)
row.Remark = strings.TrimSpace(in.Remark)
if in.Verified {
row.Verified = 1
} else {
row.Verified = 0
}
if in.Enabled {
row.Enabled = 1
} else {
row.Enabled = 0
}
row.UpdateTime = &now
if isNew {
row.CreateTime = now
_, err = models.Orm.Insert(&row)
return err
}
_, err = models.Orm.Update(&row)
return err
}
// CallbackURL 微信服务器回调地址(供公众号后台「服务器配置」填写)。
// 优先读取 app.conf 的 wechat_mp_callback_base,其次回落 payment_callback_base(同为对外 API 域名)。
func CallbackURL() string {
base, _ := beego.AppConfig.String("wechat_mp_callback_base")
if strings.TrimSpace(base) == "" {
base, _ = beego.AppConfig.String("payment_callback_base")
}
base = strings.TrimRight(strings.TrimSpace(base), "/")
if base == "" {
return "/api/wechat/mp/callback"
}
return base + "/api/wechat/mp/callback"
}
// ============================ access_token 缓存 ============================
type tokenEntry struct {
appID string
token string
expiresAt time.Time
}
var (
tokenMu sync.Mutex
tokenCache tokenEntry
tokenMinTTL = 5 * time.Minute
)
// resetTokenCacheForTest 仅供测试清理缓存
func resetTokenCacheForTest() {
tokenMu.Lock()
defer tokenMu.Unlock()
tokenCache = tokenEntry{}
}
+106
View File
@@ -0,0 +1,106 @@
package wechatmp
import (
"crypto/aes"
"crypto/cipher"
"crypto/rand"
"crypto/sha256"
"encoding/base64"
"errors"
"io"
"strings"
beego "github.com/beego/beego/v2/server/web"
)
// =============================================================
// 微信公众号敏感字段(AppSecret / EncodingAESKey)的对称加密与掩码
//
// yz_platform_wechat_mp_config.app_secret / aes_key 以 AES-256-GCM 加密后存储,
// 格式:base64( "YZW1:" + nonce + ciphertext )。
// 密钥来源:app.conf 的 wechat_mp_secret_key(任意长度字符串,内部做 SHA-256 派生);
// 未配置时退回内置兜底密钥(保证开箱可用),上线前建议在 app.conf 配置该值。
// 注意:更换密钥后,历史密文将无法解密,需要重新保存一次配置。
// =============================================================
const (
cipherPrefix = "YZW1:"
fallbackSecretKey = "yunzer_wechat_mp_default_secret_key_v1"
)
func configKey() []byte {
raw, _ := beego.AppConfig.String("wechat_mp_secret_key")
if strings.TrimSpace(raw) == "" {
raw = fallbackSecretKey
}
sum := sha256.Sum256([]byte(raw))
return sum[:]
}
// EncryptSecret 加密敏感字段;空串原样返回
func EncryptSecret(plain string) (string, error) {
if plain == "" {
return "", nil
}
block, err := aes.NewCipher(configKey())
if err != nil {
return "", err
}
gcm, err := cipher.NewGCM(block)
if err != nil {
return "", err
}
nonce := make([]byte, gcm.NonceSize())
if _, err = io.ReadFull(rand.Reader, nonce); err != nil {
return "", err
}
ciphertext := gcm.Seal(nil, nonce, []byte(plain), nil)
buf := append([]byte(cipherPrefix), nonce...)
buf = append(buf, ciphertext...)
return base64.StdEncoding.EncodeToString(buf), nil
}
// DecryptSecret 解密敏感字段;空串原样返回
func DecryptSecret(encoded string) (string, error) {
if encoded == "" {
return "", nil
}
raw, err := base64.StdEncoding.DecodeString(encoded)
if err != nil {
return "", errors.New("密文不是有效的 base64")
}
if len(raw) < len(cipherPrefix) || string(raw[:len(cipherPrefix)]) != cipherPrefix {
return "", errors.New("密文格式不正确(缺少加密前缀)")
}
block, err := aes.NewCipher(configKey())
if err != nil {
return "", err
}
gcm, err := cipher.NewGCM(block)
if err != nil {
return "", err
}
nonce := raw[len(cipherPrefix) : len(cipherPrefix)+gcm.NonceSize()]
ciphertext := raw[len(cipherPrefix)+gcm.NonceSize():]
plain, err := gcm.Open(nil, nonce, ciphertext, nil)
if err != nil {
return "", errors.New("解密失败(wechat_mp_secret_key 是否被更换过?)")
}
return string(plain), nil
}
// MaskSecret 生成掩码:保留末 4 位
func MaskSecret(v string) string {
if v == "" {
return ""
}
if len(v) <= 4 {
return "******"
}
return "******" + v[len(v)-4:]
}
// IsMasked 判断前端回传的值是否仍是掩码(表示「不修改原值」)
func IsMasked(v string) bool {
return strings.HasPrefix(v, "******")
}
+214
View File
@@ -0,0 +1,214 @@
package wechatmp
import (
"crypto/aes"
"crypto/cipher"
"crypto/rand"
"crypto/sha1"
"encoding/base64"
"encoding/binary"
"encoding/hex"
"encoding/xml"
"errors"
"fmt"
"io"
"sort"
"strings"
)
// =============================================================
// 微信被动消息:签名校验 / 安全模式 AES 加解密 / XML 解析与回复构造
// 规范参考:https://developers.weixin.qq.com/doc/offiaccount/Message_Management/Message_encryption_and_decryption_instructions.html
// =============================================================
// pkcs7BlockSize 微信官方填充块大小(注意不是 16,是 32)
const pkcs7BlockSize = 32
// InboundMessage 接收到的消息/事件(已解密后的 XML 内容)
type InboundMessage struct {
XMLName xml.Name `xml:"xml"`
ToUserName string `xml:"ToUserName"`
FromUserName string `xml:"FromUserName"`
CreateTime int64 `xml:"CreateTime"`
MsgType string `xml:"MsgType"` // text / image / event ...
Content string `xml:"Content"`
Event string `xml:"Event"` // subscribe / unsubscribe / SCAN / CLICK ...
EventKey string `xml:"EventKey"` // qrscene_<scene> 或 <scene>
Ticket string `xml:"Ticket"`
MsgID int64 `xml:"MsgId"`
Encrypt string `xml:"Encrypt"` // 安全模式下外层 XML 携带的密文
}
// ParseInboundXML 解析(明文/已解密的)消息 XML
func ParseInboundXML(raw string) (*InboundMessage, error) {
var msg InboundMessage
if err := xml.Unmarshal([]byte(raw), &msg); err != nil {
return nil, fmt.Errorf("解析微信消息失败: %w", err)
}
return &msg, nil
}
// CheckSignature 明文模式签名校验:sha1(sort(token, timestamp, nonce))
func CheckSignature(token, timestamp, nonce, signature string) bool {
if token == "" || signature == "" {
return false
}
arr := []string{token, timestamp, nonce}
sort.Strings(arr)
sum := sha1.Sum([]byte(strings.Join(arr, "")))
return hex.EncodeToString(sum[:]) == signature
}
// CheckMsgSignature 安全/兼容模式签名校验:sha1(sort(token, timestamp, nonce, encrypt))
func CheckMsgSignature(token, timestamp, nonce, encrypt, msgSignature string) bool {
if token == "" || msgSignature == "" {
return false
}
arr := []string{token, timestamp, nonce, encrypt}
sort.Strings(arr)
sum := sha1.Sum([]byte(strings.Join(arr, "")))
return hex.EncodeToString(sum[:]) == msgSignature
}
// MsgSignature 计算安全模式消息签名(回复加密时使用)
func MsgSignature(token, timestamp, nonce, encrypt string) string {
arr := []string{token, timestamp, nonce, encrypt}
sort.Strings(arr)
sum := sha1.Sum([]byte(strings.Join(arr, "")))
return hex.EncodeToString(sum[:])
}
// ============================ AES 加解密 ============================
func decodeAESKey(encodingAESKey string) ([]byte, error) {
key := strings.TrimSpace(encodingAESKey)
if len(key) != 43 {
return nil, fmt.Errorf("EncodingAESKey 长度应为 43 位(当前 %d)", len(key))
}
return base64.StdEncoding.DecodeString(key + "=")
}
// DecryptMessage 安全模式消息解密,返回明文 XML;appID 非空时会校验一致性
func DecryptMessage(aesKey, appID, encryptBase64 string) (string, error) {
key, err := decodeAESKey(aesKey)
if err != nil {
return "", err
}
raw, err := base64.StdEncoding.DecodeString(encryptBase64)
if err != nil {
return "", errors.New("密文不是有效的 base64")
}
block, err := aes.NewCipher(key)
if err != nil {
return "", err
}
if len(raw) < aes.BlockSize || len(raw)%aes.BlockSize != 0 {
return "", errors.New("密文长度不合法")
}
iv := key[:aes.BlockSize]
plain := make([]byte, len(raw))
cipher.NewCBCDecrypter(block, iv).CryptBlocks(plain, raw)
plain, err = pkcs7Unpad(plain)
if err != nil {
return "", err
}
if len(plain) < 20 {
return "", errors.New("解密内容长度不合法")
}
msgLen := int(binary.BigEndian.Uint32(plain[16:20]))
if msgLen < 0 || 20+msgLen > len(plain) {
return "", errors.New("解密内容长度不合法")
}
msg := string(plain[20 : 20+msgLen])
gotAppID := strings.TrimSpace(string(plain[20+msgLen:]))
if appID != "" && gotAppID != appID {
return "", fmt.Errorf("AppID 校验失败(收到 %s)", gotAppID)
}
return msg, nil
}
// EncryptMessage 安全模式消息加密,返回 base64 密文
func EncryptMessage(aesKey, appID, msg string) (string, error) {
key, err := decodeAESKey(aesKey)
if err != nil {
return "", err
}
random := make([]byte, 16)
if _, err := io.ReadFull(rand.Reader, random); err != nil {
return "", err
}
msgBytes := []byte(msg)
lenBytes := make([]byte, 4)
binary.BigEndian.PutUint32(lenBytes, uint32(len(msgBytes)))
buf := make([]byte, 0, 16+4+len(msgBytes)+len(appID))
buf = append(buf, random...)
buf = append(buf, lenBytes...)
buf = append(buf, msgBytes...)
buf = append(buf, []byte(appID)...)
buf = pkcs7Pad(buf)
block, err := aes.NewCipher(key)
if err != nil {
return "", err
}
iv := key[:aes.BlockSize]
ciphertext := make([]byte, len(buf))
cipher.NewCBCEncrypter(block, iv).CryptBlocks(ciphertext, buf)
return base64.StdEncoding.EncodeToString(ciphertext), nil
}
func pkcs7Pad(data []byte) []byte {
pad := pkcs7BlockSize - len(data)%pkcs7BlockSize
if pad <= 0 {
pad = pkcs7BlockSize
}
out := make([]byte, 0, len(data)+pad)
out = append(out, data...)
for i := 0; i < pad; i++ {
out = append(out, byte(pad))
}
return out
}
func pkcs7Unpad(data []byte) ([]byte, error) {
if len(data) == 0 {
return nil, errors.New("空数据")
}
pad := int(data[len(data)-1])
if pad < 1 || pad > pkcs7BlockSize || pad > len(data) {
return nil, errors.New("PKCS7 填充不合法")
}
for i := len(data) - pad; i < len(data); i++ {
if int(data[i]) != pad {
return nil, errors.New("PKCS7 填充不合法")
}
}
return data[:len(data)-pad], nil
}
// ============================ 被动回复构造 ============================
// BuildTextReply 构造文本消息回复 XML(明文模式直接返回)
func BuildTextReply(toUser, fromUser, content string, timestamp int64) string {
esc := func(s string) string {
return strings.NewReplacer(
"]]>", "]] >",
).Replace(s)
}
return fmt.Sprintf(
`<xml><ToUserName><![CDATA[%s]]></ToUserName><FromUserName><![CDATA[%s]]></FromUserName><CreateTime>%d</CreateTime><MsgType><![CDATA[text]]></MsgType><Content><![CDATA[%s]]></Content></xml>`,
esc(toUser), esc(fromUser), timestamp, esc(content))
}
// BuildEncryptedReply 安全模式回复:对明文回复加密并组装外层 XML
func BuildEncryptedReply(cfg *Config, plainReply, timestamp, nonce string) (string, error) {
encrypt, err := EncryptMessage(cfg.AESKey, cfg.AppID, plainReply)
if err != nil {
return "", err
}
sig := MsgSignature(cfg.Token, timestamp, nonce, encrypt)
return fmt.Sprintf(
`<xml><Encrypt><![CDATA[%s]]></Encrypt><MsgSignature><![CDATA[%s]]></MsgSignature><TimeStamp>%s</TimeStamp><Nonce><![CDATA[%s]]></Nonce></xml>`,
encrypt, sig, timestamp, nonce), nil
}
+113
View File
@@ -0,0 +1,113 @@
package wechatmp
import (
"errors"
"time"
"server/models"
)
// =============================================================
// 消息推送(认证后可用):
// 未认证服务号微信不允许主动推送(模板消息/订阅通知/客服消息均受限),
// 本层只负责在认证后通过模板消息把提醒/公告发给已关注粉丝;
// 未认证时返回 ErrNotVerified,由上层给出明确提示。
// =============================================================
// ErrNotVerified 公众号未认证,无法主动推送
var ErrNotVerified = errors.New("公众号未认证:模板消息推送需认证后使用(当前仅支持关注与被动回复)")
// PushMessage 推送内容(对应「标题/内容/备注」通用模板字段)
type PushMessage struct {
Title string // first:首行说明
Content string // keyword1:主体内容
Remark string // remark:备注
URL string // 点击跳转地址(可选)
}
// PushToOpenID 向指定粉丝推送一条模板消息
func PushToOpenID(cfg *Config, openid string, msg PushMessage) error {
if cfg == nil {
return ErrNotConfigured
}
if !cfg.Verified {
return ErrNotVerified
}
if openid == "" {
return errors.New("openid 为空")
}
if msg.Title == "" {
msg.Title = "您有一条新的消息"
}
if len([]rune(msg.Title)) > 100 {
msg.Title = string([]rune(msg.Title)[:100])
}
if len([]rune(msg.Content)) > 200 {
msg.Content = string([]rune(msg.Content)[:200])
}
if len([]rune(msg.Remark)) > 100 {
msg.Remark = string([]rune(msg.Remark)[:100])
}
return SendTemplateMessage(cfg, openid, msg.URL, map[string]string{
"first": msg.Title,
"keyword1": msg.Content,
"remark": msg.Remark,
})
}
// PushScope 推送范围
type PushScope struct {
// BindType 为空表示全部已关注粉丝;platform_user / tenant_user 只推绑定对应类型账号的粉丝
BindType string
// BindID 非 0 时只推给该绑定账号(与 BindType 组合使用)
BindID uint64
// Tid 非 0 时只推给绑定该租户的用户(tenant_user 场景)
Tid uint64
}
// MaxBatchPush 单次批量推送上限
const MaxBatchPush = 500
// PushToFollowers 按范围批量推送,返回成功/失败数量。
// 仅推送给「已关注(subscribe=1)且未删除」的粉丝。
func PushToFollowers(cfg *Config, scope PushScope, msg PushMessage) (sent, failed int, err error) {
if cfg == nil {
return 0, 0, ErrNotConfigured
}
if !cfg.Verified {
return 0, 0, ErrNotVerified
}
qs := models.Orm.QueryTable(new(models.WechatMpFollower)).
Filter("subscribe", 1).
Filter("delete_time__isnull", true)
if scope.BindType != "" {
qs = qs.Filter("bind_type", scope.BindType)
}
if scope.BindID > 0 {
qs = qs.Filter("bind_id", scope.BindID)
}
if scope.Tid > 0 {
qs = qs.Filter("bind_tid", scope.Tid)
}
var rows []models.WechatMpFollower
if _, qerr := qs.OrderBy("-id").Limit(MaxBatchPush).All(&rows); qerr != nil {
return 0, 0, qerr
}
if len(rows) == 0 {
return 0, 0, errors.New("没有匹配的已关注粉丝")
}
for i := range rows {
if perr := PushToOpenID(cfg, rows[i].OpenID, msg); perr != nil {
failed++
continue
}
sent++
// 简单节流,避免触发微信频率限制
if i%20 == 19 {
time.Sleep(200 * time.Millisecond)
}
}
return sent, failed, nil
}