client.go

package gosdk

import (
	"errors"
	"strconv"
	"time"

	"github.com/imroc/req"
)

const VOICE_AUTH_URL = "https://openapi.baidu.com/oauth/2.0/token"

// Authorizer 用于设置access_token
// 可以通过RESTFul api的方式从百度方获取
// 有效期为一个月，可以存至数据库中然后从数据库中获取
type Authorizer interface {
	Authorize(*Client) error
}

type Client struct {
	ClientID     string
	ClientSecret string
	AccessToken  string
	expireAt     time.Time
	Authorizer   Authorizer
}

type AuthResponse struct {
	AccessToken      string `json:"access_token"`  //要获取的Access Token
	ExpireIn         string `json:"expire_in"`     //Access Token的有效期(秒为单位，一般为1个月)；
	RefreshToken     string `json:"refresh_token"` //以下参数忽略，暂时不用
	Scope            string `json:"scope"`
	SessionKey       string `json:"session_key"`
	SessionSecret    string `json:"session_secret"`
	ERROR            string `json:"error"`             //错误码；关于错误码的详细信息请参考鉴权认证错误码(http://ai.baidu.com/docs#/Auth/top)
	ErrorDescription string `json:"error_description"` //错误描述信息，帮助理解和解决发生的错误。
}

type DefaultAuthorizer struct{}

func (da DefaultAuthorizer) Authorize(client *Client) error {
	resp, err := req.Post(VOICE_AUTH_URL, req.Param{
		"grant_type":    "client_credentials",
		"client_id":     client.ClientID,
		"client_secret": client.ClientSecret,
	})
	if err != nil {
		return err
	}
	authresponse := new(AuthResponse)
	if err := resp.ToJSON(authresponse); err != nil {
		return err
	}
	if authresponse.ERROR != "" || authresponse.AccessToken == "" {
		return errors.New("授权失败:" + authresponse.ErrorDescription)
	}

	client.AccessToken = authresponse.AccessToken
	expireIn, _ := strconv.Atoi(authresponse.ExpireIn)
	// 记录AccessToken 过期时间，过期时间前60秒就触发刷新
	client.expireAt = time.Now().Add(time.Second * time.Duration(expireIn-60))
	return nil
}

func (client *Client) Auth() error {
	if client.AccessToken != "" && time.Now().Before(client.expireAt) {
		return nil
	}

	if err := client.Authorizer.Authorize(client); err != nil {
		return err
	}
	return nil
}

func (client *Client) SetAuther(auth Authorizer) {
	client.Authorizer = auth
}

func NewClient(ApiKey, secretKey string) *Client {
	return &Client{
		ClientID:     ApiKey,
		ClientSecret: secretKey,
		Authorizer:   DefaultAuthorizer{},
	}
}