package client

import (
	"encoding/json"
	"fmt"
	"io"
	"net/http"
)

const (
	UserAppID          = ""
	UserSecret         = ""
	DriverAppID        = ""
	DriverSecret       = ""
	Code2SessionAppURL = "https://api.weixin.qq.com/sns/oauth2/access_token"
	UserInfoAppURL     = "https://api.weixin.qq.com/sns/userinfo"

	Code2SessionJsURL = "https://api.weixin.qq.com/sns/jscode2session"

	SendMessageURL    = "https://api.weixin.qq.com/cgi-bin/message/subscribe/send?access_token=ACCESS_TOKEN"
	GetAccessTokenURL = "https://api.weixin.qq.com/cgi-bin/token"
)

func WeChatNewAuthJsReqMap(code string) map[string]string {
	return map[string]string{
		"appid":      UserAppID,
		"secret":     UserSecret,
		"js_code":    code,
		"grant_type": "authorization_code"}
}

func WeChatNewAuthAppReqMap(code string) map[string]string {
	return map[string]string{
		"appid":      UserAppID,
		"secret":     UserSecret,
		"code":       code,
		"grant_type": "authorization_code"}
}

func WeChatGetAccessTokenReq(_type int) map[string]string {
	switch _type {
	case 1: // 用户
		return map[string]string{
			"grant_type": "client_credential",
			"appid":      UserAppID,
			"secret":     UserSecret,
		}
	case 2: // 其他
		return map[string]string{
			"grant_type": "client_credential",
			"appid":      DriverAppID,
			"secret":     DriverSecret,
		}
	}
	return nil
}

// 小程序换取token的返回
type WeChatLoginJsRes struct {
	SessionKey string `json:"session_key"`
	UnionID    string `json:"unionid"`
	OpenID     string `json:"openid"`
	ErrCode    int32  `json:"errcode"`
	ErrMsg     string `json:"errmsg"`
}

// APP换取token的返回
type WeChatLoginAppRes struct {
	AccessToken  string `json:"access_token"`  // 接口调用凭证
	ExpiresIn    int32  `json:"expires_in"`    // access_token 接口调用凭证超时时间，单位（秒）
	RefreshToken string `json:"refresh_token"` // 用户刷新 access_token
	UnionID      string `json:"unionid"`       // 用户统一标识。针对一个微信开放平台账号下的应用，同一用户的 unionid 是唯一的
	OpenID       string `json:"openid"`        // 授权用户唯一标识
	Scope        string `json:"scope"`         // 用户授权的作用域，使用逗号（,）分隔
	ErrCode      int32  `json:"errcode"`
	ErrMsg       string `json:"errmsg"`
}

// APP获取用户信息
type WeChatUserInfoAppRes struct {
	UnionID  string `json:"unionid"`    // 用户统一标识。针对一个微信开放平台账号下的应用，同一用户的 unionid 是唯一的
	OpenID   string `json:"openid"`     // 授权用户唯一标识
	Nickname string `json:"nickname"`   // 普通用户昵称
	Sex      int32  `json:"sex"`        // 普通用户性别，1 为男性，2 为女性
	Province string `json:"province"`   // 普通用户个人资料填写的省份
	City     string `json:"city"`       // 普通用户个人资料填写的城市
	Country  string `json:"country"`    // 国家，如中国为 CN
	Avatar   string `json:"headimgurl"` // 用户头像，最后一个数值代表正方形头像大小（有 0、46、64、96、132 数值可选，0 代表 640*640 正方形头像），用户没有头像时该项为空
	ErrCode  int32  `json:"errcode"`
	ErrMsg   string `json:"errmsg"`
}

// 换取token所用的请求
func WeChatNewAuthReqMapUser(code string) map[string]string {
	return map[string]string{
		"appid":      UserAppID,
		"secret":     UserSecret,
		"js_code":    code,
		"grant_type": "authorization_code"}
}

// 小程序换取token的请求
func WeChatCode2JsSession(headers map[string]string, params map[string]string) (wxLoginBody WeChatLoginJsRes, e error) {
	var (
		client = NewClient()
		resp   *http.Response
		res    []byte
	)
	req, _ := http.NewRequest(http.MethodGet, Code2SessionJsURL, nil)
	for k, v := range headers {
		req.Header.Add(k, v)
	}
	q := req.URL.Query()
	for k, v := range params {
		q.Add(k, v)
	}
	req.URL.RawQuery = q.Encode()
	if resp, e = client.Do(req); e != nil {
		return
	}
	defer func(body io.ReadCloser) {
		_ = body.Close()
	}(resp.Body)
	if res, e = io.ReadAll(resp.Body); e != nil {
		return
	}
	if !success(resp.StatusCode) {
		e = fmt.Errorf("status code error: %s", resp.Status)
	}
	if e = json.Unmarshal(res, &wxLoginBody); e != nil {
		return
	}
	if wxLoginBody.ErrCode != 0 {
		e = fmt.Errorf("获取微信登录凭证错误，错误码: %d 错误信息：%s", wxLoginBody.ErrCode, wxLoginBody.ErrMsg)
		return
	}
	return
}

// APP换取token的请求
func WeChatCode2AppToken(headers map[string]string, params map[string]string) (wxLoginBody WeChatLoginAppRes, e error) {
	var (
		client = NewClient()
		resp   *http.Response
		res    []byte
	)
	req, _ := http.NewRequest(http.MethodGet, Code2SessionAppURL, nil)
	for k, v := range headers {
		req.Header.Add(k, v)
	}
	q := req.URL.Query()
	for k, v := range params {
		q.Add(k, v)
	}
	req.URL.RawQuery = q.Encode()
	if resp, e = client.Do(req); e != nil {
		return
	}
	defer func(body io.ReadCloser) {
		_ = body.Close()
	}(resp.Body)
	if res, e = io.ReadAll(resp.Body); e != nil {
		return
	}
	if !success(resp.StatusCode) {
		e = fmt.Errorf("status code error: %s", resp.Status)
	}
	if e = json.Unmarshal(res, &wxLoginBody); e != nil {
		return
	}
	if wxLoginBody.ErrCode != 0 {
		e = fmt.Errorf("获取微信登录凭证错误，错误码: %d 错误信息：%s", wxLoginBody.ErrCode, wxLoginBody.ErrMsg)
		return
	}
	return
}

// APP获取用户信息
func WeChatUserInfoApp(token, openId string) (wxUserInfoBody WeChatUserInfoAppRes, e error) {
	var (
		client = NewClient()
		resp   *http.Response
		res    []byte
	)
	req, _ := http.NewRequest(http.MethodGet, UserInfoAppURL, nil)

	q := req.URL.Query()
	q.Add("access_token", token)
	q.Add("openid", openId)
	req.URL.RawQuery = q.Encode()
	if resp, e = client.Do(req); e != nil {
		return
	}
	defer func(body io.ReadCloser) {
		_ = body.Close()
	}(resp.Body)
	if res, e = io.ReadAll(resp.Body); e != nil {
		return
	}
	if !success(resp.StatusCode) {
		e = fmt.Errorf("status code error: %s", resp.Status)
	}
	if e = json.Unmarshal(res, &wxUserInfoBody); e != nil {
		return
	}
	if wxUserInfoBody.ErrCode != 0 {
		e = fmt.Errorf("获取微信登录凭证错误，错误码: %d 错误信息：%s", wxUserInfoBody.ErrCode, wxUserInfoBody.ErrMsg)
		return
	}
	return
}

func success(code int) bool {
	return code >= 200 && code < 400
}