1115PM

conf/log.conf

@@ -1,4 +1,4 @@
 [common]
-filename="E:\\GoProjects\\src\\spaceofcheng\\services\\log\\common.log"
+filename="E:\\GoProject\\src\\wechat-conf\\log\\common.log"
 # log level "emergency", "alert", "critical", "error", "warning", "notice", "info", "debug"
 level="debug"

conf/wechat.conf

@@ -0,0 +1,3 @@
+[component]
+appid = test
+aeskey = aeskey

controllers/component.go

@@ -11,6 +11,8 @@ import (
 	"wechat-conf/service/wechat"
 	"wechat-conf/utils"
 
+	"github.com/astaxie/beego"
+
 	"github.com/kinisky564477/wechat/component"
 
 	"github.com/zjxpcyc/wechat/core"
@@ -40,40 +42,54 @@ const (
 
 // ComponentPush 第三方平台推送
 func (c *WechatController) ComponentPush() {
+	utils.LogError("推送开始")
 	r := c.Ctx.Request
 	defer r.Body.Close()
 	con, _ := ioutil.ReadAll(r.Body)
+	beego.Error(string(con))
+	// 解析xml
+	xp := &core.XMLParse{}
+	resMsg, err := xp.Parse(string(con))
+	if err != nil {
+		utils.LogError("xml解析失败：", err)
+		c.ResponseRaw([]byte("success"))
+	}
+	utils.LogError("解析xml成功:", resMsg)
+	encrypt := resMsg["Encrypt"]
+	beego.Error(encrypt)
 	conf, err := c.wechatServ.GetComponentInfo()
 	if err != nil || conf == nil || conf.Appid == "" {
 		utils.LogError("读取微信配置文件失败")
-		c.ResponseRaw([]byte(""))
+		c.ResponseRaw([]byte("success"))
 	}
+	beego.Error(conf.Aeskey)
 	EncodingAESKey := conf.Aeskey
 
 	AESKey, err := base64.StdEncoding.DecodeString(EncodingAESKey + "=")
 	if err != nil {
 		utils.LogError("DecodeString失败：", err)
-		c.ResponseRaw([]byte(""))
+		c.ResponseRaw([]byte("success"))
 	}
-	xp := &core.XMLParse{}
-	// 解析xml
-	val, err := xp.Parse(string(con))
+	beego.Error(AESKey)
+	EncryptVal, err := base64.StdEncoding.DecodeString(encrypt)
 	if err != nil {
-		utils.LogError("xml解析失败：", err)
-		c.ResponseRaw([]byte(""))
+		utils.LogError("密文解析失败：", err)
+		c.ResponseRaw([]byte("success"))
 	}
-
-	msgbyte, err := utils.AesDecrypt([]byte(val["Encrypt"]), AESKey)
+	msgbyte, err := utils.AesDecrypt(EncryptVal, AESKey)
 	if err != nil {
 		utils.LogError("解密失败：", err)
-		c.ResponseRaw([]byte(""))
+		c.ResponseRaw([]byte("success"))
 	}
+	beego.Error("解密数据：", string(msgbyte))
+	beego.Error("解密成功")
+	// 解析xml
 	msg, err := xp.Parse(string(msgbyte))
 	if err != nil {
-		utils.LogError("msgxml解析失败：", err)
-		c.ResponseRaw([]byte(""))
+		utils.LogError("xml解析失败：", err)
+		c.ResponseRaw([]byte("success"))
 	}
-	utils.LogError("接收微信推送消息：", msg)
+	beego.Error("解析xml成功:", msg)
 	switch msg["InfoType"] {
 	case INFOTYPE_TICKET:
 		// 更新ticket
@@ -102,9 +118,8 @@ func (c *WechatController) ComponentPush() {
 		err := c.wechatServ.SaveWechatConf(conf)
 		if err != nil {
 			utils.LogError("保存微信授权信息失败：", err)
-			c.ResponseRaw([]byte(""))
+			c.ResponseRaw([]byte("success"))
 		}
-		c.ResponseRaw([]byte("success"))
 		break
 	case INFOTYPE_UPDATEAUTHORIZED:
 		// 授权更新
@@ -113,6 +128,7 @@ func (c *WechatController) ComponentPush() {
 		// 取消授权
 		break
 	}
+	c.ResponseRaw([]byte("success"))
 }
 
 // WechatInfo 微信接入

controllers/config.go

@@ -34,10 +34,4 @@ func (c *BaseController) initConfig() {
 	if aliConf != nil {
 		c.Configer[AliYunConf] = aliConf
 	}
-
-	// 微信
-	wechatConf, _ := utils.GetConfiger("conf/wechat.conf")
-	if wechatConf != nil {
-		c.Configer[WeChatConf] = wechatConf
-	}
 }

models/model/sys_component_conf.go

@@ -1,9 +1,13 @@
 package model
 
+import "time"
+
 type SysComponentConf struct {
-	Wxid   string `xorm:"VARCHAR(20)"`
-	Appid  string `xorm:"VARCHAR(50)"`
-	Aeskey string `xorm:"VARCHAR(100)"`
-	Secret string `xorm:"VARCHAR(100)"`
-	Ticket string `xorm:"VARCHAR(100)"`
+	Wxid        string    `xorm:"VARCHAR(20)"`
+	Appid       string    `xorm:"VARCHAR(50)"`
+	Aeskey      string    `xorm:"VARCHAR(100)"`
+	Secret      string    `xorm:"VARCHAR(100)"`
+	Ticket      string    `xorm:"VARCHAR(100)"`
+	Token       string    `xorm:"VARCHAR(100)"`
+	ExpiresDate time.Time `xorm:"DATETIME"`
 }

models/model/sys_wechat_conf.go

@@ -14,4 +14,5 @@ type SysWechatConf struct {
 	Status            int       `xorm:"SMALLINT(6)"`
 	AuthorizationInfo string    `xorm:"TEXT"`
 	BindDate          time.Time `xorm:"DATETIME"`
+	ExpiresDate       time.Time `xorm:"DATETIME"`
 }

models/wechat.go

@@ -1,8 +1,11 @@
 package models
 
 import (
+	"time"
 	"wechat-conf/models/model"
 	"wechat-conf/utils"
+
+	"github.com/astaxie/beego"
 )
 
 // ComponentInit 初始化第三方
@@ -19,14 +22,52 @@ func ComponentInit() {
 		utils.LogError("初始化第三方数据失败，数据为空")
 		return
 	}
+	d := "0"
+	if !conf.ExpiresDate.IsZero() {
+		now := time.Now()
+		d = string(conf.ExpiresDate.Sub(now))
+	}
+
 	var cert = map[string]string{
 		"appid":                 conf.Appid,
 		"aeskey":                conf.Aeskey,
 		"secret":                conf.Secret,
 		"componentVerifyTicket": conf.Ticket,
+		"token":                 conf.Token,
+		"expires_in":            d,
 	}
 
-	utils.ComponentInit(cert)
+	utils.ComponentInit(cert, UpdateComponentToken)
+}
+
+// UpdateComponentToken 更新第三方token
+func UpdateComponentToken(token map[string]interface{}) {
+	beego.Error("更新第三方token：", token)
+	if token["accessToken"].(string) == "" {
+		return
+	}
+
+	var conf = model.SysComponentConf{
+		Token: token["accessToken"].(string),
+	}
+
+	expires := token["expire_in"].(int)
+	beego.Error(expires)
+	if expires > 0 {
+		expriesTime := time.Now().Local().Add(time.Duration(expires))
+		conf.ExpiresDate = expriesTime
+		sql := `update sys_component_conf set token=?,expires_date=?`
+		_, err := DBEngine.Exec(sql, conf.Token, conf.ExpiresDate)
+		if err != nil {
+			utils.LogError("更新第三方token失败：", err)
+		}
+	} else {
+		sql := `update sys_component_conf set token=?`
+		_, err := DBEngine.Exec(sql, conf.Token, conf.ExpiresDate)
+		if err != nil {
+			utils.LogError("更新第三方token失败：", err)
+		}
+	}
 }
 
 // InitWechat 初始化微信

utils/aes.go

@@ -4,6 +4,8 @@ import (
 	"bytes"
 	"crypto/aes"
 	"crypto/cipher"
+
+	"github.com/astaxie/beego"
 )
 
 // PKCS7Padding 填充序列的每个字节都填paddingSize
@@ -38,11 +40,14 @@ func AesEncrypt(origData, key []byte) ([]byte, error) {
 func AesDecrypt(crypted, key []byte) ([]byte, error) {
 	block, err := aes.NewCipher(key)
 	if err != nil {
+		beego.Error(err)
 		return nil, err
 	}
 	blockSize := block.BlockSize()
 	blockMode := cipher.NewCBCDecrypter(block, key[:blockSize])
 	origData := make([]byte, len(crypted))
+	beego.Error(origData)
+	beego.Error(crypted)
 	blockMode.CryptBlocks(origData, crypted)
 	origData = PKCS7UnPadding(origData)
 	return origData, nil

utils/wechat.go

@@ -8,8 +8,8 @@ import (
 var Component *component.ComponentClient
 
 // ComponentInit 第三方初始化
-func ComponentInit(cert map[string]string) {
-	Component = component.NewComponentClient(cert)
+func ComponentInit(cert map[string]string, updateToken func(map[string]interface{})) {
+	Component = component.NewComponentClient(cert, updateToken)
 	LogError(Component)
 }