Merge branch 'develop' of git.ailuowan.com:zzw/lone-services into develop

pkg/wecom/auth/token.go

@@ -0,0 +1,77 @@
+package auth
+
+import (
+	"context"
+	"crypto/md5"
+	"encoding/hex"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"net/http"
+	"time"
+
+	"lone-services/pkg/redis"
+	"lone-services/pkg/utils"
+)
+
+type Client struct {
+	CorpId     string
+	CorpSecret string
+}
+
+type tokenResponse struct {
+	ErrCode     int    `json:"errcode"`
+	ErrMsg      string `json:"errmsg"`
+	AccessToken string `json:"access_token"`
+	ExpiresIn   int    `json:"expires_in"`
+}
+
+func NewClient(corpId, corpSecret string) *Client {
+	return &Client{
+		CorpId:     corpId,
+		CorpSecret: corpSecret,
+	}
+}
+
+func (c *Client) AccessToken(ctx context.Context) (string, error) {
+	secretMd5 := md5.Sum([]byte(c.CorpSecret))
+	key := fmt.Sprintf("wecom:access_token:%s:%s", c.CorpId, hex.EncodeToString(secretMd5[:])[:8])
+
+	if redis.Client != nil {
+		token, err := redis.Client.Get(ctx, key).Result()
+		if err == nil && token != utils.StringEmpty {
+			return token, nil
+		}
+	}
+
+	url := fmt.Sprintf(
+		"https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid=%s&corpsecret=%s",
+		c.CorpId,
+		c.CorpSecret,
+	)
+
+	req, err := http.NewRequestWithContext(ctx, http.MethodGet, url, nil)
+	if err != nil {
+		return utils.StringEmpty, err
+	}
+
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return utils.StringEmpty, err
+	}
+	defer resp.Body.Close()
+
+	var result tokenResponse
+	if err := json.NewDecoder(resp.Body).Decode(&result); err != nil {
+		return utils.StringEmpty, err
+	}
+	if result.ErrCode != utils.NumberZero {
+		return utils.StringEmpty, errors.New(result.ErrMsg)
+	}
+
+	expire := time.Duration(result.ExpiresIn-300) * time.Second
+	if redis.Client != nil {
+		_ = redis.Client.Set(ctx, key, result.AccessToken, expire).Err()
+	}
+	return result.AccessToken, nil
+}

pkg/wecom/user/client.go

@@ -0,0 +1,33 @@
+package user
+
+import "lone-services/pkg/wecom/auth"
+
+type Client struct {
+	auth *auth.Client
+}
+
+func NewClient(authClient *auth.Client) *Client {
+	return &Client{auth: authClient}
+}
+
+type CreateRequest struct {
+	UserId        string `json:"userid"`
+	Name          string `json:"name"`
+	Mobile        string `json:"mobile"`
+	AvatarMediaID string `json:"avatar_mediaid,omitempty"`
+	Enable        int    `json:"enable"`
+	Department    []int  `json:"department"`
+}
+
+type UpdateRequest struct {
+	UserId        string `json:"userid"`
+	Name          string `json:"name"`
+	Mobile        string `json:"mobile"`
+	AvatarMediaID string `json:"avatar_mediaid,omitempty"`
+	Enable        int    `json:"enable"`
+}
+
+type apiResponse struct {
+	ErrCode int    `json:"errcode"`
+	ErrMsg  string `json:"errmsg"`
+}

pkg/wecom/user/create.go

@@ -0,0 +1,50 @@
+package user
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"net/http"
+
+	"lone-services/pkg/utils"
+)
+
+func (c *Client) Create(ctx context.Context, reqData CreateRequest) error {
+	token, err := c.auth.AccessToken(ctx)
+	if err != nil {
+		return err
+	}
+
+	url := fmt.Sprintf(
+		"https://qyapi.weixin.qq.com/cgi-bin/user/create?access_token=%s",
+		token,
+	)
+
+	body, err := json.Marshal(reqData)
+	if err != nil {
+		return err
+	}
+
+	req, err := http.NewRequestWithContext(ctx, http.MethodPost, url, bytes.NewBuffer(body))
+	if err != nil {
+		return err
+	}
+	req.Header.Set("Content-Type", "application/json")
+
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	var result apiResponse
+	if err := json.NewDecoder(resp.Body).Decode(&result); err != nil {
+		return err
+	}
+	if result.ErrCode != utils.NumberZero {
+		return errors.New(result.ErrMsg)
+	}
+	return nil
+}

pkg/wecom/user/delete.go

@@ -0,0 +1,45 @@
+package user
+
+import (
+	"context"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"net/http"
+	"net/url"
+
+	"lone-services/pkg/utils"
+)
+
+func (c *Client) Delete(ctx context.Context, userId string) error {
+	token, err := c.auth.AccessToken(ctx)
+	if err != nil {
+		return err
+	}
+
+	apiURL := fmt.Sprintf(
+		"https://qyapi.weixin.qq.com/cgi-bin/user/delete?access_token=%s&userid=%s",
+		token,
+		url.QueryEscape(userId),
+	)
+
+	req, err := http.NewRequestWithContext(ctx, http.MethodGet, apiURL, nil)
+	if err != nil {
+		return err
+	}
+
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	var result apiResponse
+	if err := json.NewDecoder(resp.Body).Decode(&result); err != nil {
+		return err
+	}
+	if result.ErrCode != utils.NumberZero {
+		return errors.New(result.ErrMsg)
+	}
+	return nil
+}

pkg/wecom/user/update.go

@@ -0,0 +1,50 @@
+package user
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"net/http"
+
+	"lone-services/pkg/utils"
+)
+
+func (c *Client) Update(ctx context.Context, reqData UpdateRequest) error {
+	token, err := c.auth.AccessToken(ctx)
+	if err != nil {
+		return err
+	}
+
+	url := fmt.Sprintf(
+		"https://qyapi.weixin.qq.com/cgi-bin/user/update?access_token=%s",
+		token,
+	)
+
+	body, err := json.Marshal(reqData)
+	if err != nil {
+		return err
+	}
+
+	req, err := http.NewRequestWithContext(ctx, http.MethodPost, url, bytes.NewBuffer(body))
+	if err != nil {
+		return err
+	}
+	req.Header.Set("Content-Type", "application/json")
+
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	var result apiResponse
+	if err := json.NewDecoder(resp.Body).Decode(&result); err != nil {
+		return err
+	}
+	if result.ErrCode != utils.NumberZero {
+		return errors.New(result.ErrMsg)
+	}
+	return nil
+}