forked from Mirrors/oauth2
Introduction of cache interface and a file cacher.
This commit is contained in:
parent
643fd905db
commit
227bfbf02f
|
@ -0,0 +1,58 @@
|
||||||
|
// Copyright 2014 Google Inc. All Rights Reserved.
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package oauth2
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"io/ioutil"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Cache represents a token cacher.
|
||||||
|
type Cache interface {
|
||||||
|
// Read reads a cache token from the specified file.
|
||||||
|
Read() (token *Token, err error)
|
||||||
|
// Write writes a token to the specified file.
|
||||||
|
Write(token *Token) (err error)
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewFileCache creates a new file cache.
|
||||||
|
func NewFileCache(filename string) *FileCache {
|
||||||
|
return &FileCache{filename: filename}
|
||||||
|
}
|
||||||
|
|
||||||
|
// FileCache represents a file based token cacher.
|
||||||
|
type FileCache struct {
|
||||||
|
filename string
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read reads a cache token from the specified file.
|
||||||
|
func (f *FileCache) Read() (token *Token, err error) {
|
||||||
|
data, err := ioutil.ReadFile(f.filename)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
token = &Token{}
|
||||||
|
err = json.Unmarshal(data, &token)
|
||||||
|
return token, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Write writes a token to the specified file.
|
||||||
|
func (f *FileCache) Write(token *Token) error {
|
||||||
|
data, err := json.Marshal(token)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return ioutil.WriteFile(f.filename, data, 0644)
|
||||||
|
}
|
6
jwt.go
6
jwt.go
|
@ -66,6 +66,7 @@ type JWTConfig struct {
|
||||||
opts *JWTOptions
|
opts *JWTOptions
|
||||||
aud string
|
aud string
|
||||||
signature []byte
|
signature []byte
|
||||||
|
cache Cache
|
||||||
}
|
}
|
||||||
|
|
||||||
// Options returns JWT options.
|
// Options returns JWT options.
|
||||||
|
@ -153,3 +154,8 @@ func (c *JWTConfig) FetchToken(existing *Token) (token *Token, err error) {
|
||||||
token.Expiry = time.Now().Add(time.Duration(b.ExpiresIn) * time.Second)
|
token.Expiry = time.Now().Add(time.Duration(b.ExpiresIn) * time.Second)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Cache returns a cache if specified, otherwise nil.
|
||||||
|
func (c *JWTConfig) Cache() Cache {
|
||||||
|
return c.cache
|
||||||
|
}
|
||||||
|
|
|
@ -74,6 +74,7 @@ type TokenFetcher interface {
|
||||||
// If the implementation doesn't know how to retrieve a new token,
|
// If the implementation doesn't know how to retrieve a new token,
|
||||||
// it returns an error.
|
// it returns an error.
|
||||||
FetchToken(existing *Token) (*Token, error)
|
FetchToken(existing *Token) (*Token, error)
|
||||||
|
Cache() Cache
|
||||||
}
|
}
|
||||||
|
|
||||||
// Options represents options to provide OAuth 2.0 client credentials
|
// Options represents options to provide OAuth 2.0 client credentials
|
||||||
|
@ -138,6 +139,8 @@ type Config struct {
|
||||||
authURL string
|
authURL string
|
||||||
// TokenURL is the URL used to retrieve OAuth tokens.
|
// TokenURL is the URL used to retrieve OAuth tokens.
|
||||||
tokenURL string
|
tokenURL string
|
||||||
|
|
||||||
|
cache Cache
|
||||||
}
|
}
|
||||||
|
|
||||||
// Options returns options.
|
// Options returns options.
|
||||||
|
@ -224,6 +227,11 @@ func (c *Config) FetchToken(existing *Token) (*Token, error) {
|
||||||
return existing, err
|
return existing, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Cache returns a cache if specified, otherwise nil.
|
||||||
|
func (c *Config) Cache() Cache {
|
||||||
|
return c.cache
|
||||||
|
}
|
||||||
|
|
||||||
// Checks if all required configuration fields have non-zero values.
|
// Checks if all required configuration fields have non-zero values.
|
||||||
func (c *Config) validate() error {
|
func (c *Config) validate() error {
|
||||||
if c.opts.ClientID == "" {
|
if c.opts.ClientID == "" {
|
||||||
|
|
Loading…
Reference in New Issue