自定义菜单接口

controllers/mpwechat/menu_controller.go

@@ -3,7 +3,9 @@ package mpwechat
 import (
 	base_ctl "SCRM/controllers"
 	"SCRM/enums"
+	"SCRM/models"
 	"SCRM/service/wechat_service"
+	"time"
 
 	"encoding/json"
 
@@ -12,6 +14,7 @@ import (
 
 func MpMenusCtlRegistRouters() {
 	beego.Router("/api/mpwechat/menus", &MpMenusAPIController{}, "Get:GetMenus")
+	beego.Router("/api/mpwechat/savemenus", &MpMenusAPIController{}, "Put:SaveMenus")
 }
 
 type MpMenusAPIController struct {
@@ -22,7 +25,7 @@ func (c *MpMenusAPIController) GetMenus() {
 
 	adminUserInfo := c.GetAdminUserInfo()
 
-	buttons, err := wechat_service.GetMenusByOrgID(adminUserInfo.CurrentOrgId)
+	buttons, err := wechat_service.GetOrgMenusWithMsgs(adminUserInfo.CurrentOrgId)
 	if err != nil {
 		c.ServeFailJsonSend(enums.ErrorCodeDataException, "读取菜单配置信息失败:（"+err.Error()+"）")
 		return
@@ -38,6 +41,23 @@ func (c *MpMenusAPIController) GetMenus() {
 			c.ServeFailJsonSend(enums.ErrorCodeDataException, "解析菜单配置信息失败:（"+err.Error()+"）")
 			return
 		}
+
+		msgMap := make(map[string]string, 0)
+		for _, msg := range buttons.Messages {
+			msgMap[msg.MessageKeyName] = msg.MessageContent
+		}
+
+		for index, menu := range menus.Button {
+			if _, exist := msgMap[menu.Key]; exist {
+				menus.Button[index].Message = msgMap[menu.Key]
+			}
+			for sindex, sub := range menus.Button[index].SubButton {
+				if _, exist := msgMap[sub.Key]; exist {
+					menus.Button[index].SubButton[sindex].Message = msgMap[sub.Key]
+				}
+			}
+		}
+
 	}
 
 	returnData := make(map[string]interface{}, 0)
@@ -46,3 +66,125 @@ func (c *MpMenusAPIController) GetMenus() {
 	c.ServeSuccessJSON(returnData)
 	return
 }
+
+func (c *MpMenusAPIController) SaveMenus() {
+	adminUserInfo := c.GetAdminUserInfo()
+	authorization, err := wechat_service.GetAuthorizationByOrgID(adminUserInfo.CurrentOrgId)
+	if err != nil {
+		c.ServeFailJsonSend(enums.ErrorCodeDataException, "保存菜单失败:（"+err.Error()+"）")
+		return
+	}
+	if authorization == nil || authorization.AuthorizerStatus != 1 {
+		c.ServeFailJsonSend(enums.ErrorCodeDBUpdate, "保存菜单失败:（未授权）")
+		return
+	}
+
+	// 授权方公众号类型，0代表订阅号，1代表由历史老帐号升级后的订阅号，2代表服务号
+	// 授权方认证类型，-1代表未认证，0代表微信认证，1代表新浪微博认证，2代表腾讯微博认证，3代表已资质认证通过但还未通过名称认证，4代表已资质认证通过、还未通过名称认证，但通过了新浪微博认证，5代表已资质认证通过、还未通过名称认证，但通过了腾讯微博认证
+	if (authorization.AuthorizerServiceTypeInfo == 0 || authorization.AuthorizerServiceTypeInfo == 1) && authorization.AuthorizerVerifyTypeInfo == -1 {
+		c.ServeFailJsonSend(enums.ErrorCodeDBUpdate, "保存菜单失败:（未认证的订阅号没有自定义菜单的接口权限）")
+		return
+	}
+	var buttons wechat_service.SelfMenuInfo
+
+	err = json.Unmarshal(c.Ctx.Input.RequestBody, &buttons)
+	if err != nil {
+		c.ServeFailJsonSend(enums.ErrorCodeDBUpdate, "解析菜单失败:（"+err.Error()+"）")
+		return
+	}
+
+	clickMap := make(map[string]string, 0)
+	for pindex, button := range buttons.Button {
+		if button.Type == "click" && len(button.Key) > 0 && len(button.SubButton) == 0 {
+			clickMap[button.Key] = button.Message
+		}
+		if button.Type != "click" {
+			buttons.Button[pindex].Key = ""
+		}
+		if len(button.SubButton) > 0 {
+			for sindex, child := range button.SubButton {
+				if child.Type == "click" && len(child.Key) > 0 {
+					clickMap[child.Key] = child.Message
+				}
+				buttons.Button[pindex].SubButton[sindex].Message = ""
+
+				if child.Type != "click" {
+					buttons.Button[pindex].SubButton[sindex].Key = ""
+				}
+			}
+		}
+		buttons.Button[pindex].Message = ""
+	}
+	messages, err := wechat_service.GetOrgAllClickMessages(adminUserInfo.CurrentOrgId)
+	if err != nil {
+		c.ServeFailJsonSend(enums.ErrorCodeDataException, "保存菜单失败:（"+err.Error()+"）")
+		return
+	}
+	oldMsgs := make(map[string]models.AuthorizationMessageManagements)
+	for _, message := range messages {
+		if _, exist := clickMap[message.MessageKeyName]; exist {
+			msg := *message
+			oldMsgs[message.MessageKeyName] = msg
+		}
+	}
+
+	buttonFileds, err := json.Marshal(buttons)
+	if err != nil {
+		c.ServeFailJsonSend(enums.ErrorCodeDBUpdate, "解析菜单失败:（"+err.Error()+"）")
+		return
+	}
+
+	timeNow := time.Now().Unix()
+	var saveButton wechat_service.Buttons
+	saveButton.UserOrgID = adminUserInfo.CurrentOrgId
+	saveButton.UpdatedTime = timeNow
+	saveButton.ButtonStatus = 1
+	saveButton.SendStatus = 0
+	saveButton.ButtonFileds = string(buttonFileds)
+
+	button, err := wechat_service.GetMenusByOrgID(adminUserInfo.CurrentOrgId)
+	if err != nil {
+		c.ServeFailJsonSend(enums.ErrorCodeDataException, "保存菜单失败:（"+err.Error()+"）")
+		return
+	}
+	if button != nil {
+		saveButton.ID = button.ID
+		saveButton.CreatedTime = button.CreatedTime
+	}
+
+	saveButton.Messages = make([]*models.AuthorizationMessageManagements, 0)
+	for keyName, Content := range clickMap {
+		var msg models.AuthorizationMessageManagements
+		msg.MessageMsgType = "event"
+		msg.MessageMsgEvent = "click"
+		msg.MessageContent = Content
+		msg.UpdatedTime = timeNow
+		msg.MessageStatus = 1
+		msg.MessageKeyName = keyName
+		msg.UserOrgId = adminUserInfo.CurrentOrgId
+
+		if _, exist := oldMsgs[keyName]; exist {
+			msg.ID = oldMsgs[keyName].ID
+			msg.CreatedTime = oldMsgs[keyName].CreatedTime
+		} else {
+			msg.CreatedTime = timeNow
+		}
+		saveButton.Messages = append(saveButton.Messages, &msg)
+	}
+
+	err = wechat_service.SaveButtons(&saveButton)
+	if err != nil {
+		c.ServeFailJsonSend(enums.ErrorCodeDBUpdate, "保存菜单失败:（"+err.Error()+"）")
+		return
+	}
+	err = wechat_service.SendMpWechatMenus(authorization.AuthorizerAccessToken, buttonFileds)
+	if err != nil {
+		c.ServeFailJsonSend(enums.ErrorCodeDBUpdate, "保存菜单成功，但发布菜单到公众号失败:（"+err.Error()+"）")
+		return
+	}
+
+	returnData := make(map[string]interface{}, 0)
+	returnData["buttons"] = buttons
+	c.ServeSuccessJSON(returnData)
+	return
+}

jobcron/open_wechat_job.go

@@ -24,6 +24,13 @@ func init() {
 			RequestComponentAccessToken()
 		}()
 	})
+
+	openWechatCron.AddFunc("@every 1h30m", func() {
+		go func() {
+			RequestMpWechatAccessToken()
+		}()
+	})
+
 }
 
 func StartOpenWechatCron() {
@@ -114,3 +121,80 @@ func RequestComponentAccessToken() {
 
 	utils.InfoLog("Request ComponentAccessToken Finish.")
 }
+
+type ReqRefreshTokenStruct struct {
+	ComponentAppid         string `json:"component_appid"`
+	AuthorizerAppid        string `json:"authorizer_appid"`
+	AuthorizerRefreshToken string `json:"authorizer_refresh_token"`
+}
+
+func RequestMpWechatAccessToken() {
+	utils.InfoLog("Request MpWechatAccessToken start...")
+	mps, err := wechat_service.GetAuthorizations()
+	if err != nil {
+		utils.ErrorLog("get mps error: %s", err)
+		return
+	}
+
+	redisClient := service.RedisClient()
+	defer redisClient.Close()
+
+	componentAccessToken, err := redisClient.Get("sgj_patient:component_access_token").Result()
+	if err != nil {
+		utils.ErrorLog("get component_access_token error: %s", err)
+		return
+	}
+
+	for _, item := range mps {
+		var ReqRefreshToken ReqRefreshTokenStruct
+		ReqRefreshToken.ComponentAppid = appID
+		ReqRefreshToken.AuthorizerAppid = item.AuthorizerAppid
+		ReqRefreshToken.AuthorizerRefreshToken = item.AuthorizerRefreshToken
+
+		uri := fmt.Sprintf("%s?component_access_token=%s", "https://api.weixin.qq.com/cgi-bin/component/api_authorizer_token", componentAccessToken)
+		var responseBytes []byte
+		responseBytes, err = util.PostJSON(uri, ReqRefreshToken)
+		if err != nil {
+			utils.ErrorLog(" error: %s", err)
+			return
+		}
+
+		var res wechat_service.MPResult
+		err = json.Unmarshal(responseBytes, &res)
+		if err != nil {
+			utils.ErrorLog(" error: %s", err)
+			return
+		}
+
+		if res.ErrCode > 0 {
+			utils.ErrorLog(" error: %s", res.ErrMsg)
+			return
+		}
+
+		var accessToken wechat_service.RefreshToken
+		err = json.Unmarshal(responseBytes, &accessToken)
+		if err != nil {
+			utils.ErrorLog("Unmarshal error: %s", err)
+			continue
+		}
+
+		info := *item
+		info.AuthorizerAccessToken = accessToken.AuthorizerAccessToken
+		info.AuthorizerRefreshToken = accessToken.AuthorizerRefreshToken
+		info.UpdatedTime = time.Now().Unix()
+		err = wechat_service.SaveAuthorizationInfo(&info)
+		if err != nil {
+			utils.ErrorLog("Save error: %s", err)
+			continue
+		}
+
+		atKey := fmt.Sprintf("Patient:AccessToken[%s]", item.AuthorizerAppid)
+		_, err = redisClient.Set(atKey, accessToken.AuthorizerAccessToken, 0).Result()
+		if err != nil {
+			utils.ErrorLog("Save redis error: %s", err)
+			continue
+		}
+	}
+	utils.InfoLog("Request MpWechatAccessToken END")
+
+}

main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"SCRM/jobcron"
 	_ "SCRM/routers"
 	"SCRM/service"
 
@@ -18,10 +19,10 @@ func main() {
 	//微信开放平台
 	go func() {
 		// jobcron.RequestComponentAccessToken()
-		// jobcron.ReqRefreshToken()
+		// jobcron.RequestMpWechatAccessToken()
 		// jobcron.ReqJsapiTicket()
 	}()
-	// jobcron.StartOpenWechatCron()
+	jobcron.StartOpenWechatCron()
 
 	beego.Run()
 }

service/wechat_service/authorization_service.go

@@ -69,45 +69,7 @@ func SaveAuthorizationInfo(authorization *models.PatientAuthorizations) (err err
 	return
 }
 
-//GetTextReplyMessagesByKey 通过关键词搜索 user_org_id（机构ID）取信息
-func GetTextReplyMessagesByKey(userOrgID int64, keywrods string) (messages []*models.AuthorizationMessageManagements, err error) {
-	if len(keywrods) == 0 {
-		return
-	}
-
-	likekey := "%" + keywrods + "%"
-	err = service.PatientReadDB().Where("user_org_id=? AND message_msg_type='text' AND message_status=1 AND ((message_key_name LIKE ? AND message_key_type=1) OR (message_key_name = ? AND message_key_type=2))", userOrgID, likekey, keywrods).Order("id desc").Find(&messages).Error
+func GetAuthorizations() (mps []*models.PatientAuthorizations, err error) {
+	err = service.PatientReadDB().Where("authorizer_status = 1").Find(&mps).Error
 	return
 }
-
-//GetSubscribeReplyMessagesByOrgID 通过user_org_id（机构ID）取信息
-func GetSubscribeReplyMessagesByOrgID(orgID int64) (*models.AuthorizationMessageManagements, error) {
-
-	var message models.AuthorizationMessageManagements
-
-	err := service.PatientReadDB().Where("user_org_id=? AND message_msg_type='event' AND message_msg_event='subscribe' AND message_status=1", orgID).First(&message).Error
-	if err == gorm.ErrRecordNotFound {
-		return nil, nil
-	}
-	if err != nil {
-		return nil, err
-	}
-	return &message, nil
-
-}
-
-//GetClickButtonReplyMessagesByOrgID 通过user_org_id（机构ID）取信息
-func GetClickButtonReplyMessagesByOrgID(orgID int64, key string) (*models.AuthorizationMessageManagements, error) {
-
-	var message models.AuthorizationMessageManagements
-
-	err := service.PatientReadDB().Where("user_org_id=? AND message_msg_type='event' AND message_msg_event='click' AND message_key_name=? AND message_status=1", orgID, key).First(&message).Error
-	if err == gorm.ErrRecordNotFound {
-		return nil, nil
-	}
-	if err != nil {
-		return nil, err
-	}
-	return &message, nil
-
-}

service/wechat_service/menu_service.go

@@ -3,6 +3,7 @@ package wechat_service
 import (
 	"SCRM/models"
 	"SCRM/service"
+	"time"
 
 	"github.com/jinzhu/gorm"
 )
@@ -19,3 +20,44 @@ func GetMenusByOrgID(orgID int64) (*models.AuthorizationButtons, error) {
 	}
 	return &button, nil
 }
+
+func GetOrgMenusWithMsgs(orgID int64) (*Buttons, error) {
+	var button Buttons
+	err := service.PatientReadDB().Where("user_org_id=?", orgID).Order("id desc").Preload("Messages", "user_org_id=? AND message_msg_type='event' AND message_msg_event='click' AND message_status=1", orgID).First(&button).Error
+	if err == gorm.ErrRecordNotFound {
+		return nil, nil
+	}
+	if err != nil {
+		return nil, err
+	}
+	return &button, nil
+}
+
+type Buttons struct {
+	ID           int64  `gorm:"column:id" json:"id"`
+	UserOrgID    int64  `gorm:"column:user_org_id" json:"user_org_id"`
+	ButtonFileds string `gorm:"column:button_fileds" json:"button_fileds"`
+	CreatedTime  int64  `gorm:"column:created_time" json:"created_time"`
+	UpdatedTime  int64  `gorm:"column:updated_time" json:"updated_time"`
+	ButtonStatus int64  `gorm:"column:button_status" json:"button_status"`
+	SendStatus   int64  `gorm:"column:send_status" json:"send_status"`
+
+	Messages []*models.AuthorizationMessageManagements `gorm:"ForeignKey:user_org_id;association_foreignkey:user_org_id" json:"messages"`
+}
+
+func (Buttons) TableName() string {
+	return "sgj_patient_authorization_buttons"
+}
+
+func SaveButtons(menu *Buttons) (err error) {
+	tx := service.PatientWriteDB().Begin()
+
+	err = tx.Model(&models.AuthorizationMessageManagements{}).Where("user_org_id=? AND message_msg_type='event' AND message_msg_event='click'", menu.UserOrgID).Update(map[string]interface{}{"MessageStatus": 2, "UpdatedTime": time.Now().Unix()}).Error
+	if err != nil {
+		tx.Rollback()
+		return
+	}
+	err = tx.Save(menu).Error
+	tx.Commit()
+	return
+}

service/wechat_service/messages_service.go

@@ -0,0 +1,53 @@
+package wechat_service
+
+import (
+	"SCRM/models"
+	"SCRM/service"
+
+	"github.com/jinzhu/gorm"
+)
+
+//GetTextReplyMessagesByKey 通过关键词搜索 user_org_id（机构ID）取信息
+func GetTextReplyMessagesByKey(userOrgID int64, keywrods string) (messages []*models.AuthorizationMessageManagements, err error) {
+	if len(keywrods) == 0 {
+		return
+	}
+
+	likekey := "%" + keywrods + "%"
+	err = service.PatientReadDB().Where("user_org_id=? AND message_msg_type='text' AND message_status=1 AND ((message_key_name LIKE ? AND message_key_type=1) OR (message_key_name = ? AND message_key_type=2))", userOrgID, likekey, keywrods).Order("id desc").Find(&messages).Error
+	return
+}
+
+//GetSubscribeReplyMessagesByOrgID 通过user_org_id（机构ID）取信息
+func GetSubscribeReplyMessagesByOrgID(orgID int64) (*models.AuthorizationMessageManagements, error) {
+
+	var message models.AuthorizationMessageManagements
+
+	err := service.PatientReadDB().Where("user_org_id=? AND message_msg_type='event' AND message_msg_event='subscribe' AND message_status=1", orgID).First(&message).Error
+	if err == gorm.ErrRecordNotFound {
+		return nil, nil
+	}
+	if err != nil {
+		return nil, err
+	}
+	return &message, nil
+
+}
+
+//GetClickButtonReplyMessagesByOrgID 通过user_org_id（机构ID）取信息
+func GetClickButtonReplyMessagesByOrgID(orgID int64, key string) (*models.AuthorizationMessageManagements, error) {
+	var message models.AuthorizationMessageManagements
+	err := service.PatientReadDB().Where("user_org_id=? AND message_msg_type='event' AND message_msg_event='click' AND message_key_name=? AND message_status=1", orgID, key).First(&message).Error
+	if err == gorm.ErrRecordNotFound {
+		return nil, nil
+	}
+	if err != nil {
+		return nil, err
+	}
+	return &message, nil
+}
+
+func GetOrgAllClickMessages(orgID int64) (messages []*models.AuthorizationMessageManagements, err error) {
+	err = service.PatientReadDB().Where("user_org_id=? AND message_msg_type='event' AND message_msg_event='click'", orgID).Find(&messages).Error
+	return
+}

service/wechat_service/open_wechat_model.go

@@ -122,6 +122,7 @@ type AuthorizerInfo struct {
 		QrcodeURL       string                 `json:"qrcode_url"`
 		PrincipalName   string                 `json:"principal_name"`
 		BusinessInfo    BusinessInfo           `json:"business_info"`
+		FuncInfo        []FuncInfoItems        `json:"func_info"`
 	} `json:"authorizer_info"`
 }
 
@@ -150,12 +151,19 @@ type RRreqMenu struct {
 
 //SelfMenuButton 自定义菜单配置详情
 type SelfMenuButton struct {
-	Type      string           `json:"type"`
 	Name      string           `json:"name"`
-	Key       string           `json:"key"`
+	Type      string           `json:"type,omitempty"`
+	Key       string           `json:"key,omitempty"`
+	AppID     string           `json:"appid,omitempty"`
+	PagePath  string           `json:"pagepath,omitempty"`
+	Message   string           `json:"message,omitempty"`
 	URL       string           `json:"url,omitempty"`
-	Value     string           `json:"value,omitempty"`
-	Act       string           `json:"act,omitempty"`
 	MediaId   string           `json:"media_id,omitempty"`
-	SubButton []SelfMenuButton `json:"sub_button"`
+	SubButton []SelfMenuButton `json:"sub_button,omitempty"`
+}
+
+type RefreshToken struct {
+	AuthorizerAccessToken  string `json:"authorizer_access_token"`
+	ExpiresIn              int64  `json:"expires_in"`
+	AuthorizerRefreshToken string `json:"authorizer_refresh_token"`
 }

service/wechat_service/open_wechat_service.go

@@ -25,6 +25,7 @@ import (
 
 	"github.com/astaxie/beego"
 	"github.com/astaxie/beego/context"
+	"github.com/astaxie/beego/httplib"
 	"github.com/silenceper/wechat/util"
 )
 
@@ -434,3 +435,61 @@ func ComponentAPIGetAuthorizerInfo(AuthorizerAppid string, ComponentAccessToken
 	resp.Body.Close()
 	return body, nil
 }
+
+func PostJSON(uri string, jsonData []byte) ([]byte, error) {
+	// jsonData, err := json.Marshal(obj)
+	// if err != nil {
+	// 	return nil, err
+	// }
+
+	jsonData = bytes.Replace(jsonData, []byte("\\u003c"), []byte("<"), -1)
+	jsonData = bytes.Replace(jsonData, []byte("\\u003e"), []byte(">"), -1)
+	jsonData = bytes.Replace(jsonData, []byte("\\u0026"), []byte("&"), -1)
+
+	body := bytes.NewBuffer(jsonData)
+	response, err := http.Post(uri, "application/json;charset=utf-8", body)
+	if err != nil {
+		return nil, err
+	}
+	defer response.Body.Close()
+
+	if response.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf("http get error : uri=%v , statusCode=%v", uri, response.StatusCode)
+	}
+	return ioutil.ReadAll(response.Body)
+}
+
+func SendMpWechatMenus(AuthorizerAccessToken string, jsonData []byte) (err error) {
+	//读取响应
+	body, err := PostJSON("https://api.weixin.qq.com/cgi-bin/menu/create?access_token="+AuthorizerAccessToken, jsonData)
+	if err != nil {
+		return
+	}
+	var result MPResult
+	err = json.Unmarshal([]byte(body), &result)
+	if err != nil {
+		return
+	}
+	if result.ErrCode > 0 {
+		utils.ErrorLog("MPResult ERROR:%v", result)
+		err = errors.New(result.ErrMsg)
+		return
+	}
+	return
+
+}
+
+func DeleteMpWechatMenus(AuthorizerAccessToken string) (bool, error) {
+	uri := fmt.Sprintf("https://api.weixin.qq.com/cgi-bin/menu/delete?access_token=%s", AuthorizerAccessToken)
+
+	b := httplib.Get(uri)
+	var result MPResult
+	err := b.ToJSON(&result)
+	if err != nil {
+		return false, err
+	}
+	if result.ErrCode > 0 {
+		return false, errors.New(result.ErrMsg)
+	}
+	return true, nil
+}