gosdk/qyweixin/app.go

package qyweixin

import (
	"bytes"
	"crypto/sha1"
	"encoding/hex"
	"encoding/json"
	"fmt"
	"github.com/gin-gonic/gin"
	log "github.com/sirupsen/logrus"
	"github.com/smbrave/goutil"
	"github.com/spf13/cast"
	"gitlab.batiao8.com/open/gosdk/util"
	"gitlab.batiao8.com/open/gosdk/wechat"
	"gitlab.batiao8.com/open/gosdk/wechat/cache"
	"gitlab.batiao8.com/open/gosdk/wechat/message"
	wutil "gitlab.batiao8.com/open/gosdk/wechat/util"
	"io"
	"io/ioutil"
	"mime/multipart"
	"net/http"
	"os"
	"path/filepath"
	"strings"
	"time"
)

var (
	wechatCache cache.Cache = cache.NewMemory()
)
var (
	urlQiyeSend = "https://qyapi.weixin.qq.com/cgi-bin/message/send"
)

type MessageRequest struct {
	ToUser  string `json:"touser"`
	ToParty string `json:"toparty"`
	ToTag   string `json:"totag"`
	MstType string `json:"msgtype"`
	AgentId string `json:"agentid"`

	Text struct {
		Content string `json:"content"`
	} `json:"text"`

	Image struct {
		MediaID string `json:"media_id"`
	} `json:"image"`

	Video struct {
		MediaID string `json:"media_id"`
		Title   string `json:"title"`
		Desc    string `json:"description"`
	} `json:"video"`
}

type JsapiConfig struct {
	Corpid    string `json:"corpid"`
	Agent     string `json:"agent"`
	Signature string `json:"signature"`
	Noncestr  string `json:"noncestr"`
	Timestamp string `json:"timestamp"`
}

type BaseResponse struct {
	ErrCode int    `json:"errcode"`
	ErrMsg  string `json:"errmsg"`
}

type AppConfig struct {
	Corpid string
	Secret string
	Agent  string
	Token  string
	AesKey string
	Replay func(message.MixMessage) *message.Reply
}

type App struct {
	tokenExpire int64
	token       string
	config      *AppConfig
}

type UserInfo struct {
	UserId   string
	RealName string
}

func NewApp(cfg *AppConfig) *App {
	return &App{
		config: cfg,
	}
}

func (q *App) GetToken() string {
	if time.Now().Unix() <= q.tokenExpire-600 {
		return q.token
	}
	q.refreshToken()
	return q.token
}

func (q *App) GetResult(rspBody []byte) (map[string]interface{}, error) {
	result := make(map[string]interface{})
	if err := json.Unmarshal(rspBody, &result); err != nil {
		log.Errorf("result[%s] error :%s", string(rspBody), err.Error())
		return nil, err
	}
	if cast.ToInt(result["errcode"]) != 0 {
		log.Errorf("result[%s] error ", string(rspBody))
		return nil, fmt.Errorf("%d:%s", cast.ToInt(result["errcode"]), cast.ToString(result["errmsg"]))
	}
	return result, nil
}

func (q *App) GetOpenid(userid string) (string, error) {
	if err := q.refreshToken(); err != nil {
		return "", err
	}
	reqUrl := fmt.Sprintf("https://qyapi.weixin.qq.com/cgi-bin/user/convert_to_openid?access_token=%s", q.GetToken())
	rspBody, err := util.HttpPostJson(reqUrl, nil, []byte(fmt.Sprintf(`{"userid" : "%s"}`, userid)))
	if err != nil {
		log.Errorf("httpPost url[%s] error :%s", reqUrl, err.Error())
		return "", err
	}
	result, err := q.GetResult(rspBody)
	if err != nil {
		return "", err
	}

	return cast.ToString(result["openid"]), nil
}

func (q *App) GetUserInfo(userid string) (*UserInfo, error) {
	if err := q.refreshToken(); err != nil {
		return nil, err
	}
	reqUrl := fmt.Sprintf("https://qyapi.weixin.qq.com/cgi-bin/user/get?access_token=%s&userid=%s", q.GetToken(), userid)
	rspBody, err := util.HttpGet(reqUrl, nil)
	if err != nil {
		log.Errorf("httpPost url[%s] error :%s", reqUrl, err.Error())
		return nil, err
	}
	result, err := q.GetResult(rspBody)
	if err != nil {
		return nil, err
	}

	userInfo := new(UserInfo)
	userInfo.UserId = userid
	userInfo.RealName = cast.ToString(result["name"])

	return userInfo, nil
}

func (q *App) GetDepartmentUserid(departmentId int) ([]string, error) {
	if err := q.refreshToken(); err != nil {
		return nil, err
	}
	reqUrl := fmt.Sprintf("https://qyapi.weixin.qq.com/cgi-bin/user/list?access_token=%s&department_id=%d", q.GetToken(), departmentId)
	rspBody, err := util.HttpGet(reqUrl, nil)
	if err != nil {
		log.Errorf("httpPost url[%s] error :%s", reqUrl, err.Error())
		return nil, err
	}
	result, err := q.GetResult(rspBody)
	if err != nil {
		return nil, err
	}

	userids := make([]string, 0)
	userlist := cast.ToSlice(result["userlist"])
	for _, u := range userlist {
		user := cast.ToStringMap(u)
		userids = append(userids, cast.ToString(user["userid"]))
	}
	return userids, nil
}

func (a *App) SendText(receiver []string, content string) error {
	reqUrl := fmt.Sprintf("%s?access_token=%s", urlQiyeSend, a.GetToken())
	req := new(MessageRequest)
	req.MstType = "text"
	req.AgentId = a.config.Agent
	req.Text.Content = content
	req.ToUser = strings.Join(receiver, "|")

	data, _ := json.Marshal(req)
	result, err := util.HttpPostJson(reqUrl, nil, data)
	if err != nil {
		return err
	}

	var rsp BaseResponse
	err = json.Unmarshal([]byte(result), &rsp)
	if err != nil {
		return err
	}

	if rsp.ErrCode != 0 {
		return fmt.Errorf("%d:%s", rsp.ErrCode, rsp.ErrMsg)
	}
	return nil
}

func (a *App) SendImage(receiver []string, meidiaId string) error {

	url := fmt.Sprintf("%s?access_token=%s", urlQiyeSend, a.GetToken())
	req := new(MessageRequest)
	req.MstType = "image"
	req.AgentId = a.config.Agent
	req.ToUser = strings.Join(receiver, "|")
	req.Image.MediaID = meidiaId

	data, _ := json.Marshal(req)
	result, err := util.HttpPostJson(url, nil, data)
	if err != nil {
		return err
	}

	var rsp BaseResponse
	err = json.Unmarshal([]byte(result), &rsp)
	if err != nil {
		return err
	}
	if rsp.ErrCode != 0 {
		return fmt.Errorf("%d:%s", rsp.ErrCode, rsp.ErrMsg)
	}

	return nil
}

func (a *App) SendVideo(receiver []string, meidiaId, title, desc string) error {

	url := fmt.Sprintf("%s?access_token=%s", urlQiyeSend, a.GetToken())
	req := new(MessageRequest)
	req.MstType = "video"
	req.AgentId = a.config.Agent
	req.ToUser = strings.Join(receiver, "|")
	req.Video.MediaID = meidiaId
	req.Video.Title = title
	req.Video.Desc = desc
	data, _ := json.Marshal(req)
	result, err := util.HttpPostJson(url, nil, data)
	if err != nil {
		return err
	}

	var rsp BaseResponse
	err = json.Unmarshal([]byte(result), &rsp)
	if err != nil {
		return err
	}
	if rsp.ErrCode != 0 {
		return fmt.Errorf("%d:%s", rsp.ErrCode, rsp.ErrMsg)
	}

	return nil
}

func (a *App) Upload(path, kind string) (string, error) {

	fh, err := os.Open(path)
	if err != nil {
		return "", err
	}
	defer fh.Close()
	bodyBuf := &bytes.Buffer{}
	bodyWriter := multipart.NewWriter(bodyBuf)

	fileWriter, err := bodyWriter.CreateFormFile(kind, filepath.Base(path))
	if err != nil {
		return "", err
	}

	_, err = io.Copy(fileWriter, fh)
	if err != nil {
		return "", err
	}
	bodyWriter.Close()

	url := fmt.Sprintf("https://qyapi.weixin.qq.com/cgi-bin/media/upload?access_token=%s&type=%s", a.GetToken(), kind)
	req, err := http.NewRequest("POST", url, bodyBuf)
	req.Header.Add("Content-Type", bodyWriter.FormDataContentType())
	urlQuery := req.URL.Query()
	if err != nil {
		return "", err
	}

	req.URL.RawQuery = urlQuery.Encode()
	client := http.Client{}
	res, err := client.Do(req)
	if err != nil {
		return "", err
	}
	defer res.Body.Close()

	jsonbody, err := ioutil.ReadAll(res.Body)
	if err != nil {
		return "", err
	}

	result := make(map[string]interface{})
	if err := json.Unmarshal(jsonbody, &result); err != nil {
		return "", err
	}

	if cast.ToInt(result["errcode"]) != 0 {
		return "", fmt.Errorf("%d:%s", cast.ToInt(result["errcode"]), cast.ToString(result["errmsg"]))
	}
	return cast.ToString(result["media_id"]), nil
}

func (q *App) GetJsapiConfig(url string) (*JsapiConfig, error) {
	ticket, err := q.getJsapiTicket()
	if err != nil {
		return nil, err
	}
	cfg := new(JsapiConfig)
	noncestr := goutil.RandomStr(32)
	timstamp := cast.ToString(time.Now().Unix())
	str := "jsapi_ticket=" + cast.ToString(ticket) + "&noncestr=" + noncestr + "&timestamp=" + timstamp + "&url=" + url
	h := sha1.New()
	h.Write([]byte(str))
	signature := hex.EncodeToString(h.Sum(nil))
	cfg.Signature = signature
	cfg.Timestamp = timstamp
	cfg.Noncestr = noncestr
	cfg.Agent = q.config.Agent
	cfg.Corpid = q.config.Corpid
	return cfg, nil
}

func (q *App) Callback(ctx *gin.Context) {

	//配置微信参数
	wechatConfig := &wechat.Config{
		AppID:          q.config.Corpid,
		AppSecret:      q.config.Secret,
		Token:          q.config.Token,
		EncodingAESKey: q.config.AesKey,
		Cache:          wechatCache,
	}

	// 首次配置
	if strings.ToUpper(ctx.Request.Method) == http.MethodGet {
		sign := wutil.Signature(ctx.Query("timestamp"), ctx.Query("echostr"),
			ctx.Query("nonce"), wechatConfig.Token)
		if sign != ctx.Query("msg_signature") {
			log.Errorf("sign error forcheck config")
			return
		}

		_, resp, err := wutil.DecryptMsg(wechatConfig.AppID, ctx.Query("echostr"), wechatConfig.EncodingAESKey)
		if err != nil {
			log.Errorf("DecryptMsg failed! error:%s", err.Error())
			return
		}
		ctx.Data(http.StatusOK, "Content-type: text/plain", resp)
		return
	}

	// 2.响应消息
	wc := wechat.NewWechat(wechatConfig)
	ctx.Request.URL.RawQuery += "&encrypt_type=aes"
	server := wc.GetServer(ctx.Request, ctx.Writer)

	server.SetMessageHandler(q.config.Replay)

	server.SetDebug(true)
	err := server.Serve()
	if err != nil {
		log.Errorf("qiye weixin Service err:%s", err.Error())
		return
	}
	err = server.Send()
	if err != nil {
		log.Errorf("qiye weixin Send err:%s", err.Error())
		return
	}
}

func (q *App) getJsapiTicket() (string, error) {

	resp, err := http.Get("https://qyapi.weixin.qq.com/cgi-bin/ticket/get?access_token=" + q.GetToken() + "&type=agent_config")
	if err != nil {
		return "", err
	}
	defer resp.Body.Close()
	body, err := io.ReadAll(resp.Body)
	if err != nil {
		return "", err
	}
	jsonData := make(map[string]interface{})
	json.Unmarshal(body, &jsonData)
	return cast.ToString(jsonData["ticket"]), nil
}

func (q *App) refreshToken() error {
	if time.Now().Unix() <= q.tokenExpire-600 {
		return nil
	}

	reqUrl := fmt.Sprintf("https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid=%s&corpsecret=%s", q.config.Corpid, q.config.Secret)
	rspBody, err := util.HttpGet(reqUrl, nil)
	if err != nil {

		return err
	}
	result, err := q.GetResult(rspBody)
	if err != nil {
		return err
	}

	q.token = cast.ToString(result["access_token"])
	q.tokenExpire = time.Now().Unix() + cast.ToInt64(result["expires_in"])
	return nil
}