增加令牌模型、客户端模型

Author: LyricTian

const.go

@@ -24,8 +24,8 @@ func (rt ResponseType) String() string {
 type GrantType byte
 
 const (
-	// AuthorizationCode 授权码模式
-	AuthorizationCode GrantType = 1 << (iota + 1)
+	// AuthorizationCodeCredentials 授权码模式
+	AuthorizationCodeCredentials GrantType = 1 << (iota + 1)
 	// PasswordCredentials 密码模式
 	PasswordCredentials
 	// ClientCredentials 客户端模式

generate.go

@@ -13,12 +13,12 @@ type (
 	// AuthorizeTokenGenerate 授权令牌生成接口
 	AuthorizeTokenGenerate interface {
 		// 授权令牌
-		Token(data *TokenGenerateBasic) (string, error)
+		Token(data *TokenGenerateBasic) (token string, err error)
 	}
 
 	// TokenGenerate 令牌生成接口
 	TokenGenerate interface {
 		// 生成令牌
-		Token(data *TokenGenerateBasic, isGenRefresh bool) (string, string, error)
+		Token(data *TokenGenerateBasic, isGenRefresh bool) (token string, refresh string, err error)
 	}
 )

manage.go

@@ -8,7 +8,7 @@ type TokenGenerateRequest struct {
 	RedirectURI       string // 重定向URI
 	Scope             string // 授权范围
 	Code              string // 授权码(授权码模式使用)
-	IsGenerateRefresh bool   // 是否生成刷新令牌
+	IsGenerateRefresh bool   // 是否生成更新令牌
 }
 
 // Manager OAuth2授权管理接口
@@ -18,13 +18,13 @@ type Manager interface {
 	// tgr 生成令牌的请求参数
 	GenerateAuthToken(rt ResponseType, tgr *TokenGenerateRequest) (token string, err error)
 
-	// GenerateToken 生成访问令牌、刷新令牌
+	// GenerateToken 生成访问令牌、更新令牌
 	// rt 授权模式
 	// tgr 生成令牌的请求参数
 	GenerateToken(rt GrantType, tgr *TokenGenerateRequest) (token, refresh string, err error)
 
-	// RefreshToken 使用刷新令牌更新访问令牌
-	// refresh 刷新令牌
+	// RefreshToken 使用更新令牌更新访问令牌
+	// refresh 更新令牌
 	// scope 作用域
 	RefreshToken(refresh, scope string) (token string, err error)
 
@@ -35,6 +35,6 @@ type Manager interface {
 	// CheckToken 令牌检查，如果存在则返回令牌信息
 	CheckToken(token string) (ti TokenInfo, err error)
 
-	// CheckRefreshToken 访问令牌检查，如果存在则返回令牌信息
+	// CheckRefreshToken 更新令牌检查，如果存在则返回令牌信息
 	CheckRefreshToken(refresh string) (ti TokenInfo, err error)
 }

manage/manager.go

@@ -9,13 +9,28 @@ import (
 
 // Config 授权配置参数
 type Config struct {
-	TokenExpiresIn   time.Duration // 令牌有效期
-	RefreshExpiresIn time.Duration // 刷新令牌有效期
+	TokenExp   time.Duration // 令牌有效期
+	RefreshExp time.Duration // g令牌有效期
 }
 
 // NewManager 创建Manager的实例
 func NewManager() *Manager {
-	return nil
+	m := &Manager{
+		injector: inject.New(),
+	}
+	// 设定参数默认值
+
+	// 设定授权码的有效期为10分钟
+	m.SetRTConfig(oauth2.Code, &Config{TokenExp: time.Minute * 10})
+	// 设定简化模式授权令牌的有效期为1小时
+	m.SetRTConfig(oauth2.Token, &Config{TokenExp: time.Hour * 1})
+
+	// 设定授权码模式令牌的有效期为2小时,g令牌的有效期为3天
+	m.SetGTConfig(oauth2.PasswordCredentials, &Config{TokenExp: time.Hour * 2, RefreshExp: time.Hour * 24 * 3})
+
+	// 设定客户端模式令牌的有效期为1小时
+	m.SetGTConfig(oauth2.ClientCredentials, &Config{TokenExp: time.Hour * 2})
+	return m
 }
 
 // Manager OAuth2授权管理
@@ -149,7 +164,7 @@ func (m *Manager) GenerateAuthToken(rt oauth2.ResponseType, tgr *oauth2.TokenGen
 		ti.SetRedirectURI(tgr.RedirectURI)
 		ti.SetScope(tgr.Scope)
 		ti.SetTokenCreateAt(td.CreateAt)
-		ti.SetTokenExpiresIn(m.rtcfg[rt].TokenExpiresIn)
+		ti.SetTokenExpiresIn(m.rtcfg[rt].TokenExp)
 		ti.SetToken(tv)
 		err = stor.Create(ti)
 		if err != nil {
@@ -188,7 +203,7 @@ func (m *Manager) checkAuthToken(tgr *oauth2.TokenGenerateRequest) (err error) {
 // gt 授权模式
 // tgr 生成令牌的参数
 func (m *Manager) GenerateToken(gt oauth2.GrantType, tgr *oauth2.TokenGenerateRequest) (token, refresh string, err error) {
-	if gt == oauth2.AuthorizationCode {
+	if gt == oauth2.AuthorizationCodeCredentials {
 		err = m.checkAuthToken(tgr)
 		if err != nil {
 			return
@@ -217,11 +232,11 @@ func (m *Manager) GenerateToken(gt oauth2.GrantType, tgr *oauth2.TokenGenerateRe
 		ti.SetRedirectURI(tgr.RedirectURI)
 		ti.SetScope(tgr.Scope)
 		ti.SetTokenCreateAt(td.CreateAt)
-		ti.SetTokenExpiresIn(m.gtcfg[gt].TokenExpiresIn)
+		ti.SetTokenExpiresIn(m.gtcfg[gt].TokenExp)
 		ti.SetToken(tv)
 		if rv != "" {
 			ti.SetRefreshCreateAt(td.CreateAt)
-			ti.SetRefreshExpiresIn(m.gtcfg[gt].RefreshExpiresIn)
+			ti.SetRefreshExpiresIn(m.gtcfg[gt].RefreshExp)
 			ti.SetRefresh(rv)
 		}
 		err = stor.Create(ti)
@@ -236,7 +251,7 @@ func (m *Manager) GenerateToken(gt oauth2.GrantType, tgr *oauth2.TokenGenerateRe
 	return
 }
 
-// RefreshToken 刷新令牌
+// RefreshToken 更新访问令牌
 func (m *Manager) RefreshToken(refresh, scope string) (token string, err error) {
 	ti, err := m.CheckRefreshToken(refresh)
 	if err != nil {
@@ -305,7 +320,7 @@ func (m *Manager) CheckToken(token string) (info oauth2.TokenInfo, err error) {
 		} else if ti == nil {
 			err = ErrTokenInvalid
 			return
-		} else if ti.GetRefresh() != "" && ti.GetRefreshCreateAt().Add(ti.GetRefreshExpiresIn()).Before(ct) { // 检查刷新令牌是否过期
+		} else if ti.GetRefresh() != "" && ti.GetRefreshCreateAt().Add(ti.GetRefreshExpiresIn()).Before(ct) { // 检查g令牌是否过期
 			if verr := stor.ExpiredByRefresh(ti.GetRefresh()); verr != nil {
 				err = verr
 				return
@@ -327,7 +342,7 @@ func (m *Manager) CheckToken(token string) (info oauth2.TokenInfo, err error) {
 	return
 }
 
-// CheckRefreshToken 访问令牌检查
+// CheckRefreshToken 更新令牌检查
 func (m *Manager) CheckRefreshToken(refresh string) (info oauth2.TokenInfo, err error) {
 	if refresh == "" {
 		err = ErrRefreshInvalid

model.go

@@ -6,7 +6,7 @@ import "time"
 type (
 	// ClientInfo 客户端信息模型接口
 	ClientInfo interface {
-		// 客户端唯一标识
+		// 客户端ID
 		GetID() string
 		// 客户端秘钥
 		GetSecret() string
@@ -18,13 +18,13 @@ type (
 
 	// TokenInfo 令牌信息模型接口
 	TokenInfo interface {
-		// 客户端标识
+		// 客户端ID
 		GetClientID() string
-		// 设置客户端标识
+		// 设置客户端ID
 		SetClientID(string)
-		// 用户标识
+		// 用户ID
 		GetUserID() string
-		// 设置用户标识
+		// 设置用户ID
 		SetUserID(string)
 		// 重定向URI
 		GetRedirectURI() string
@@ -48,17 +48,17 @@ type (
 		// 设置令牌有效期
 		SetTokenExpiresIn(time.Duration)
 
-		// 刷新令牌
+		// 更新令牌
 		GetRefresh() string
-		// 设置刷新令牌
+		// 设置更新令牌
 		SetRefresh(string)
-		// 刷新令牌创建时间
+		// 更新令牌创建时间
 		GetRefreshCreateAt() time.Time
-		// 设置刷新令牌创建时间
+		// 设置更新令牌创建时间
 		SetRefreshCreateAt(time.Time)
-		// 刷新令牌有效期
+		// 更新令牌有效期
 		GetRefreshExpiresIn() time.Duration
-		// 设置刷新令牌有效期
+		// 设置更新令牌有效期
 		SetRefreshExpiresIn(time.Duration)
 	}
 )

models/client.go

@@ -0,0 +1,28 @@
+package models
+
+// Client 客户端信息
+type Client struct {
+	ClientID string `bson:"ClientID"` // 客户端ID
+	Secret   string `bson:"Secret"`   // 密钥
+	Domain   string `bson:"Domain"`   // 域名url
+}
+
+// GetID 客户端ID
+func (c *Client) GetID() string {
+	return c.ClientID
+}
+
+// GetSecret 客户端秘钥
+func (c *Client) GetSecret() string {
+	return c.Secret
+}
+
+// GetDomain 域名URL
+func (c *Client) GetDomain() string {
+	return c.Domain
+}
+
+// GetRetainData 自定义数据
+func (c *Client) GetRetainData() interface{} {
+	return nil
+}

models/token.go

@@ -0,0 +1,118 @@
+package models
+
+import "time"
+
+// Token 令牌信息
+type Token struct {
+	ID               int64         `bson:"_id"`              // 唯一标识
+	ClientID         string        `bson:"ClientID"`         // 客户端标识
+	UserID           string        `bson:"UserID"`           // 用户标识
+	RedirectURI      string        `bson:"RedirectURI"`      // 重定向URI
+	Scope            string        `bson:"Scope"`            // 权限范围
+	Token            string        `bson:"Token"`            // 令牌
+	TokenCreateAt    time.Time     `bson:"TokenCreateAt"`    // 令牌创建时间
+	TokenExpiresIn   time.Duration `bson:"TokenExpiresIn"`   // 令牌有效期
+	Refresh          string        `bson:"Refresh"`          // 更新令牌
+	RefreshCreateAt  time.Time     `bson:"RefreshCreateAt"`  // 更新令牌创建时间
+	RefreshExpiresIn time.Duration `bson:"RefreshExpiresIn"` // 更新令牌有效期
+}
+
+// GetClientID 客户端ID
+func (t *Token) GetClientID() string {
+	return t.ClientID
+}
+
+// SetClientID 设置客户端ID
+func (t *Token) SetClientID(clientID string) {
+	t.ClientID = clientID
+}
+
+// GetUserID 用户ID
+func (t *Token) GetUserID() string {
+	return t.UserID
+}
+
+// SetUserID 设置用户ID
+func (t *Token) SetUserID(userID string) {
+	t.UserID = userID
+}
+
+// GetRedirectURI 重定向URI
+func (t *Token) GetRedirectURI() string {
+	return t.RedirectURI
+}
+
+// SetRedirectURI 设置重定向URI
+func (t *Token) SetRedirectURI(redirectURI string) {
+	t.RedirectURI = redirectURI
+}
+
+// GetScope 权限范围
+func (t *Token) GetScope() string {
+	return t.Scope
+}
+
+// SetScope 设置权限范围
+func (t *Token) SetScope(scope string) {
+	t.Scope = scope
+}
+
+// GetToken 令牌
+func (t *Token) GetToken() string {
+	return t.Token
+}
+
+// SetToken 设置令牌
+func (t *Token) SetToken(token string) {
+	t.Token = token
+}
+
+// GetTokenCreateAt 令牌创建时间
+func (t *Token) GetTokenCreateAt() time.Time {
+	return t.TokenCreateAt
+}
+
+// SetTokenCreateAt 设置令牌创建时间
+func (t *Token) SetTokenCreateAt(createAt time.Time) {
+	t.TokenCreateAt = createAt
+}
+
+// GetTokenExpiresIn 令牌有效期
+func (t *Token) GetTokenExpiresIn() time.Duration {
+	return t.TokenExpiresIn
+}
+
+// SetTokenExpiresIn 设置令牌有效期
+func (t *Token) SetTokenExpiresIn(exp time.Duration) {
+	t.TokenExpiresIn = exp
+}
+
+// GetRefresh 更新令牌
+func (t *Token) GetRefresh() string {
+	return t.Refresh
+}
+
+// SetRefresh 设置更新令牌
+func (t *Token) SetRefresh(refresh string) {
+	t.Refresh = refresh
+}
+
+// GetRefreshCreateAt 更新令牌创建时间
+func (t *Token) GetRefreshCreateAt() time.Time {
+	return t.RefreshCreateAt
+}
+
+// SetRefreshCreateAt 设置更新令牌创建时间
+func (t *Token) SetRefreshCreateAt(createAt time.Time) {
+	t.RefreshCreateAt = createAt
+}
+
+// GetRefreshExpiresIn 更新令牌有效期
+func (t *Token) GetRefreshExpiresIn() time.Duration {
+	return t.RefreshExpiresIn
+}
+
+// SetRefreshExpiresIn 设置更新令牌有效期
+func (t *Token) SetRefreshExpiresIn(exp time.Duration) {
+	t.RefreshExpiresIn = exp
+}