From 7047ef642d72e048f8592d72466769aa4737f4f9 Mon Sep 17 00:00:00 2001
From: kanade <3136520963@qq.com>
Date: Tue, 27 Apr 2021 14:52:55 +0800
Subject: [PATCH] init

---
 README.md  |  1 +
 go.mod     |  3 +++
 request.go | 37 +++++++++++++++++++++++++++
 wechat.go  | 75 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 4 files changed, 116 insertions(+)
 create mode 100644 README.md
 create mode 100644 go.mod
 create mode 100644 request.go
 create mode 100644 wechat.go

diff --git a/README.md b/README.md
new file mode 100644
index 0000000..6a9c3fd
--- /dev/null
+++ b/README.md
@@ -0,0 +1 @@
+# 微信sdk工具
\ No newline at end of file
diff --git a/go.mod b/go.mod
new file mode 100644
index 0000000..4fb8ffe
--- /dev/null
+++ b/go.mod
@@ -0,0 +1,3 @@
+module git.oa00.com/go/wechat
+
+go 1.16
diff --git a/request.go b/request.go
new file mode 100644
index 0000000..ab68008
--- /dev/null
+++ b/request.go
@@ -0,0 +1,37 @@
+package wechat
+
+import (
+	"io"
+	"net/http"
+	"strings"
+)
+
+const (
+	post = "POST"
+	get  = "GET"
+)
+
+var client = &http.Client{}
+
+// @Title 请求
+func request(method, url, data string, headers ...map[string]string) ([]byte, error) {
+	reqest, err := http.NewRequest(method, url, strings.NewReader(data))
+	if err != nil {
+		return nil, err
+	}
+	if len(headers) > 0 {
+		for key, value := range headers[0] {
+			reqest.Header.Add(key, value)
+		}
+	}
+	response, err := client.Do(reqest)
+	if err != nil {
+		return nil, err
+	}
+	defer response.Body.Close()
+	result, err := io.ReadAll(response.Body)
+	if err != nil {
+		return nil, err
+	}
+	return result, nil
+}
diff --git a/wechat.go b/wechat.go
new file mode 100644
index 0000000..45c6a97
--- /dev/null
+++ b/wechat.go
@@ -0,0 +1,75 @@
+package wechat
+
+import (
+	"encoding/json"
+	"fmt"
+)
+
+var Wechat = &wechat{}
+
+type wechat struct {
+	appid  string
+	secret string
+}
+
+type AuthAccessToken struct {
+	AccessToken  string `json:"access_token"`
+	ExpiresIn    int64  `json:"expires_in"`
+	Openid       string `json:"openid"`
+	RefreshToken string `json:"refresh_token"`
+	Scope        string `json:"scope"`
+}
+
+// InitWechat @Title 初始化
+func InitWechat(appid, secret string) *wechat {
+	Wechat.appid = appid
+	Wechat.secret = secret
+	return Wechat
+}
+
+// GetOpenId @Title 获取openId
+func (w *wechat) GetOpenId(code string) string {
+	accessToken, err := w.GetAccessToken(code)
+	if err != nil {
+		return ""
+	}
+	return accessToken.Openid
+}
+
+// GetAccessToken @Title 获取微信验证结构体
+func (w *wechat) GetAccessToken(code string) (*AuthAccessToken, error) {
+	res, err := request(get, fmt.Sprintf("https://api.weixin.qq.com/sns/oauth2/access_token?appid=%v&secret=%v&code=%v&grant_type=authorization_code", w.appid, w.secret, code), "")
+	if err != nil {
+		return nil, err
+	}
+	result := AuthAccessToken{}
+	if err := json.Unmarshal(res, &result); err != nil {
+		return nil, err
+	}
+	return &result, nil
+}
+
+type UserInfo struct {
+	City       string   `json:"city"`
+	Country    string   `json:"country"`
+	Headimgurl string   `json:"headimgurl"`
+	Nickname   string   `json:"nickname"`
+	Openid     string   `json:"openid"`
+	Privilege  []string `json:"privilege"`
+	Province   string   `json:"province"`
+	Sex        uint     `json:"sex"`
+	Unionid    string   `json:"unionid"`
+}
+
+// GetUserInfo @Title 获取微信登录会员信息
+func (w *wechat) GetUserInfo(accessToken, openId string) (*UserInfo, error) {
+	res, err := request(get, fmt.Sprintf("https://api.weixin.qq.com/sns/userinfo?access_token=%s&openid=%s&lang=zh_CN", accessToken, openId), "")
+	if err != nil {
+		return nil, err
+	}
+	result := UserInfo{}
+	if err := json.Unmarshal(res, &result); err != nil {
+		return nil, err
+	}
+	return &result, nil
+}