1113PM

controllers/component.go

@@ -2,11 +2,13 @@ package controllers
 
 import (
 	"encoding/base64"
+	"encoding/json"
 	"io/ioutil"
 	"net/http"
 	"wechat-conf/models"
 	"wechat-conf/models/model"
 	"wechat-conf/service/autoreply"
+	"wechat-conf/service/wechat"
 	"wechat-conf/utils"
 
 	"github.com/kinisky564477/wechat/component"
@@ -17,7 +19,8 @@ import (
 // WechatController 用户
 type WechatController struct {
 	BaseController
-	serv *autoreply.AutoreplyServ
+	serv       *autoreply.AutoreplyServ
+	wechatServ *wechat.WechatServ
 }
 
 // Constructor 初始化 Controller
@@ -25,6 +28,7 @@ type WechatController struct {
 // @Description 初始化 Controller, 系统自动调用
 func (c *WechatController) Constructor() {
 	c.serv = autoreply.NewAutoreplyServ(c.Context)
+	c.wechatServ = wechat.NewWechatServ(c.Context)
 }
 
 const (
@@ -40,12 +44,7 @@ func (c *WechatController) ComponentPush() {
 	defer r.Body.Close()
 	con, _ := ioutil.ReadAll(r.Body)
 
-	// timestamp := c.GetString("timestamp")
-	// nonce := c.GetString("nonce")
-
-	// token := "testtoken"
 	EncodingAESKey := "key"
-	// appid := "appid"
 
 	AESKey, err := base64.StdEncoding.DecodeString(EncodingAESKey + "=")
 	if err != nil {
@@ -56,15 +55,18 @@ func (c *WechatController) ComponentPush() {
 	val, err := xp.Parse(string(con))
 	if err != nil {
 		utils.LogError("xml解析失败：", err)
+		c.ResponseRaw([]byte(""))
 	}
 
 	msgbyte, err := utils.AesDecrypt([]byte(val["Encrypt"]), AESKey)
 	if err != nil {
 		utils.LogError("解密失败：", err)
+		c.ResponseRaw([]byte(""))
 	}
 	msg, err := xp.Parse(string(msgbyte))
 	if err != nil {
 		utils.LogError("msgxml解析失败：", err)
+		c.ResponseRaw([]byte(""))
 	}
 	switch msg["InfoType"] {
 	case INFOTYPE_TICKET:
@@ -77,9 +79,20 @@ func (c *WechatController) ComponentPush() {
 			"appid":              msg["AuthorizerAppid"],
 			"authorization_code": msg["AuthorizationCode"],
 		}
-
-		wxclient := utils.WechatInit(cert)
+		wxclient := utils.WechatInit(cert, c.wechatServ.UpdateToken)
 		utils.AppendWxClient(wxclient)
+		// 插入数据库
+		mjson, _ := json.Marshal(msg)
+		var conf = model.SysWechatConf{
+			Appid:             msg["AuthorizerAppid"],
+			AuthorizationCode: msg["AuthorizationCode"],
+			AuthorizationInfo: string(mjson),
+		}
+		err := c.wechatServ.SaveWechatConf(conf)
+		if err != nil {
+			utils.LogError("保存微信授权信息失败：", err)
+			c.ResponseRaw([]byte(""))
+		}
 		c.ResponseRaw([]byte("success"))
 		break
 	case INFOTYPE_UPDATEAUTHORIZED:

models/wechat/wechat.go

@@ -0,0 +1,44 @@
+package wechat
+
+import (
+	"jcjy/demo/models"
+	"wechat-conf/models/model"
+	"wechat-conf/utils"
+
+	"github.com/go-xorm/xorm"
+)
+
+// WechatDAO 当前数据库操作对象
+type WechatDAO struct {
+	ctx *utils.Context
+	db  *xorm.Session
+}
+
+// NewWechatDAO New Inst
+func NewWechatDAO(ctx *utils.Context) *WechatDAO {
+	return &WechatDAO{
+		ctx: ctx,
+		db:  ctx.DB,
+	}
+}
+
+// GetWechatConfByAppID 根据appid获取配置信息
+func (m *WechatDAO) GetWechatConfByAppID(appid string) (*model.SysWechatConf, error) {
+	var conf model.SysWechatConf
+	_, err := m.db.Where("appid=?", appid).And("status>?", models.STATUS_DEL).Get(&conf)
+	return &conf, err
+}
+
+// SaveWechatConf 保存微信配置信息
+func (m *WechatDAO) SaveWechatConf(conf *model.SysWechatConf) error {
+	conf.ConfId = utils.GetGUID()
+	conf.Status = models.STATUS_READY
+	_, err := m.db.Insert(conf)
+	return err
+}
+
+// UpdateWechatConf 更新微信配置信息
+func (m *WechatDAO) UpdateWechatConf(conf model.SysWechatConf, cols []string) error {
+	_, err := m.db.Cols(cols...).Where("conf_id = ?", conf.ConfId).Update(conf)
+	return err
+}

service/wechat/wechat.go

@@ -0,0 +1,67 @@
+package wechat
+
+import (
+	"wechat-conf/models/model"
+	"wechat-conf/models/wechat"
+	"wechat-conf/utils"
+)
+
+// WechatServ 用户
+type WechatServ struct {
+	ctx *utils.Context
+	dao *wechat.WechatDAO
+}
+
+// NewWechatServ 初始化
+func NewWechatServ(ctx *utils.Context) *WechatServ {
+	return &WechatServ{
+		ctx: ctx,
+		dao: wechat.NewWechatDAO(ctx),
+	}
+}
+
+// SaveWechatConf 保存微信配置
+func (s *WechatServ) SaveWechatConf(conf model.SysWechatConf) error {
+	wechatConf, err := s.dao.GetWechatConfByAppID(conf.Appid)
+	if err != nil {
+		utils.LogError("根据appid获取微信配置失败：", err)
+		return err
+	}
+	if wechatConf != nil && wechatConf.ConfId != "" {
+		// 修改
+		conf.ConfId = wechatConf.ConfId
+		var cols = []string{
+			"refresh_token",
+			"token",
+		}
+		if conf.AuthorizationInfo != "" {
+			cols = append(cols, "authorization_info")
+		}
+		err := s.dao.UpdateWechatConf(conf, cols)
+		if err != nil {
+			utils.LogError("修改微信配置失败：", err)
+			return err
+		}
+	} else {
+		// 新增
+		err := s.dao.SaveWechatConf(&conf)
+		if err != nil {
+			utils.LogError("新增微信配置失败：", err)
+			return err
+		}
+	}
+	return nil
+}
+
+// UpdateToken 更新微信token
+func (s *WechatServ) UpdateToken(token map[string]string) {
+	var conf = model.SysWechatConf{
+		Token:        token["token"],
+		RefreshToken: token["refreshToken"],
+		Appid:        token["appid"],
+	}
+	err := s.SaveWechatConf(conf)
+	if err != nil {
+		utils.LogError("更新微信token失败：", err)
+	}
+}

utils/wechat.go

@@ -31,13 +31,14 @@ func RefreshComponentTicket(ticket string) {
 }
 
 // WechatInit 微信初始化
-func WechatInit(cert map[string]string) *component.WxClient {
+func WechatInit(cert map[string]string, reflashToken func(map[string]string)) *component.WxClient {
 	var wechatClient *component.WxClient
 
 	wechatClient = component.NewWxClient(
 		cert,
 		Component.GetToken,
 		Component.GetCertificate,
+		reflashToken,
 	)
 
 	return wechatClient