enterprise/common/qyweixin/qyweixin.go

package qyweixin

import (
	"encoding/json"
	butil "enterprise/base/util"
	"errors"
	"fmt"
	log "github.com/sirupsen/logrus"
	"github.com/smbrave/goutil"
	"github.com/spf13/cast"
	"gorm.io/gorm/utils"
	"time"
)

var (
	urlGetToken       = "https://qyapi.weixin.qq.com/cgi-bin/gettoken"
	urlGetCheckinRlue = "https://qyapi.weixin.qq.com/cgi-bin/checkin/getcorpcheckinoption"
	urlGetCheckinData = "https://qyapi.weixin.qq.com/cgi-bin/checkin/getcheckindata"
)

type QyWeixin struct {
	corpId      string
	secret      string
	token       string
	tokenExpire int64
}

func NewQyWeixin(corpId, secret string) *QyWeixin {
	return &QyWeixin{
		corpId:      corpId,
		secret:      secret,
		tokenExpire: 0,
	}
}

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

	reqUrl := fmt.Sprintf("%s?corpid=%s&corpsecret=%s", urlGetToken, q.corpId, q.secret)
	rspBody, err := butil.HttpGet(reqUrl, nil)
	if err != nil {
		log.Errorf("http url[%s] error :%s", reqUrl, err.Error())
		return err
	}
	result := make(map[string]interface{})
	if err := json.Unmarshal(rspBody, &result); err != nil {
		log.Errorf("http url[%s] result[%s] error :%s", reqUrl, string(rspBody), err.Error())
		return err
	}
	if cast.ToInt(result["errcode"]) != 0 {
		log.Errorf("http url[%s] result[%s] error ", reqUrl, string(rspBody))
		return errors.New(string(rspBody))
	}

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

func (q *QyWeixin) GetCheckinEmployee(groupIds []string) ([]string, error) {
	if err := q.refreshToken(); err != nil {
		return nil, err
	}

	reqUrl := fmt.Sprintf("%s?access_token=%s", urlGetCheckinRlue, q.token)
	rspBody, err := butil.HttpPostJson(reqUrl, nil, []byte("{}"))
	if err != nil {
		return nil, err
	}
	result := make(map[string]interface{})
	if err := json.Unmarshal(rspBody, &result); err != nil {
		log.Errorf("http url[%s] result[%s] error :%s", reqUrl, string(rspBody), err.Error())
		return nil, err
	}
	if cast.ToInt(result["errcode"]) != 0 {
		log.Errorf("http url[%s] result[%s] error ", reqUrl, string(rspBody))
		return nil, errors.New(string(rspBody))
	}

	resultUser := make([]string, 0)
	groups := cast.ToSlice(result["group"])
	for _, group := range groups {
		g := cast.ToStringMap(group)
		if !utils.Contains(groupIds, cast.ToString(g["groupid"])) {
			continue
		}
		ranges := cast.ToStringMap(g["range"])
		userid := cast.ToStringSlice(ranges["userid"])
		resultUser = append(resultUser, userid...)
	}
	return resultUser, nil
}

func (q *QyWeixin) GetCheckinData(day, userId string) (*UserCheckIn, error) {
	if err := q.refreshToken(); err != nil {
		return nil, err
	}
	dayTime, _ := time.ParseInLocation("2006-01-02", day, time.Local)

	reqData := make(map[string]interface{})
	reqData["opencheckindatatype"] = 1
	reqData["starttime"] = dayTime.Unix()
	reqData["endtime"] = dayTime.Unix() + 86400
	reqData["useridlist"] = []string{userId}
	reqUrl := fmt.Sprintf("%s?access_token=%s", urlGetCheckinData, q.token)
	rspBody, err := butil.HttpPostJson(reqUrl, nil, []byte(goutil.EncodeJSON(reqData)))
	if err != nil {
		return nil, err
	}
	result := make(map[string]interface{})
	if err := json.Unmarshal(rspBody, &result); err != nil {
		log.Errorf("http url[%s] result[%s] error :%s", reqUrl, string(rspBody), err.Error())
		return nil, err
	}
	if cast.ToInt(result["errcode"]) != 0 {
		log.Errorf("http url[%s] result[%s] error ", reqUrl, string(rspBody))
		return nil, errors.New(string(rspBody))
	}

	checkindatas := cast.ToSlice(result["checkindata"])
	userData := new(UserCheckIn)
	userData.UserId = userId
	for _, checkdata := range checkindatas {
		c := cast.ToStringMap(checkdata)
		checkinType := cast.ToString(c["checkin_type"])
		if checkinType == "上班打卡" {
			userData.StartTime = cast.ToInt64(c["checkin_time"])
		} else if checkinType == "下班打卡" {
			userData.EndTime = cast.ToInt64(c["checkin_time"])
		}
	}
	if userData.EndTime == 0 && userData.StartTime == 0 {
		return nil, nil
	}
	return userData, nil
}