|
@@ -7,6 +7,7 @@ package google
|
|
import (
|
|
import (
|
|
"crypto/rsa"
|
|
"crypto/rsa"
|
|
"fmt"
|
|
"fmt"
|
|
|
|
+ "strings"
|
|
"time"
|
|
"time"
|
|
|
|
|
|
"golang.org/x/oauth2"
|
|
"golang.org/x/oauth2"
|
|
@@ -24,6 +25,28 @@ import (
|
|
// optimization supported by a few Google services.
|
|
// optimization supported by a few Google services.
|
|
// Unless you know otherwise, you should use JWTConfigFromJSON instead.
|
|
// Unless you know otherwise, you should use JWTConfigFromJSON instead.
|
|
func JWTAccessTokenSourceFromJSON(jsonKey []byte, audience string) (oauth2.TokenSource, error) {
|
|
func JWTAccessTokenSourceFromJSON(jsonKey []byte, audience string) (oauth2.TokenSource, error) {
|
|
|
|
+ return newJWTSource(jsonKey, audience, nil)
|
|
|
|
+}
|
|
|
|
+
|
|
|
|
+// JWTAccessTokenSourceWithScope uses a Google Developers service account JSON
|
|
|
|
+// key file to read the credentials that authorize and authenticate the
|
|
|
|
+// requests, and returns a TokenSource that does not use any OAuth2 flow but
|
|
|
|
+// instead creates a JWT and sends that as the access token.
|
|
|
|
+// The scope is typically a list of URLs that specifies the scope of the
|
|
|
|
+// credentials.
|
|
|
|
+//
|
|
|
|
+// Note that this is not a standard OAuth flow, but rather an
|
|
|
|
+// optimization supported by a few Google services.
|
|
|
|
+// Unless you know otherwise, you should use JWTConfigFromJSON instead.
|
|
|
|
+func JWTAccessTokenSourceWithScope(jsonKey []byte, scope ...string) (oauth2.TokenSource, error) {
|
|
|
|
+ return newJWTSource(jsonKey, "", scope)
|
|
|
|
+}
|
|
|
|
+
|
|
|
|
+func newJWTSource(jsonKey []byte, audience string, scopes []string) (oauth2.TokenSource, error) {
|
|
|
|
+ if len(scopes) == 0 && audience == "" {
|
|
|
|
+ return nil, fmt.Errorf("google: missing scope/audience for JWT access token")
|
|
|
|
+ }
|
|
|
|
+
|
|
cfg, err := JWTConfigFromJSON(jsonKey)
|
|
cfg, err := JWTConfigFromJSON(jsonKey)
|
|
if err != nil {
|
|
if err != nil {
|
|
return nil, fmt.Errorf("google: could not parse JSON key: %v", err)
|
|
return nil, fmt.Errorf("google: could not parse JSON key: %v", err)
|
|
@@ -35,6 +58,7 @@ func JWTAccessTokenSourceFromJSON(jsonKey []byte, audience string) (oauth2.Token
|
|
ts := &jwtAccessTokenSource{
|
|
ts := &jwtAccessTokenSource{
|
|
email: cfg.Email,
|
|
email: cfg.Email,
|
|
audience: audience,
|
|
audience: audience,
|
|
|
|
+ scopes: scopes,
|
|
pk: pk,
|
|
pk: pk,
|
|
pkID: cfg.PrivateKeyID,
|
|
pkID: cfg.PrivateKeyID,
|
|
}
|
|
}
|
|
@@ -47,6 +71,7 @@ func JWTAccessTokenSourceFromJSON(jsonKey []byte, audience string) (oauth2.Token
|
|
|
|
|
|
type jwtAccessTokenSource struct {
|
|
type jwtAccessTokenSource struct {
|
|
email, audience string
|
|
email, audience string
|
|
|
|
+ scopes []string
|
|
pk *rsa.PrivateKey
|
|
pk *rsa.PrivateKey
|
|
pkID string
|
|
pkID string
|
|
}
|
|
}
|
|
@@ -54,12 +79,14 @@ type jwtAccessTokenSource struct {
|
|
func (ts *jwtAccessTokenSource) Token() (*oauth2.Token, error) {
|
|
func (ts *jwtAccessTokenSource) Token() (*oauth2.Token, error) {
|
|
iat := time.Now()
|
|
iat := time.Now()
|
|
exp := iat.Add(time.Hour)
|
|
exp := iat.Add(time.Hour)
|
|
|
|
+ scope := strings.Join(ts.scopes, " ")
|
|
cs := &jws.ClaimSet{
|
|
cs := &jws.ClaimSet{
|
|
- Iss: ts.email,
|
|
|
|
- Sub: ts.email,
|
|
|
|
- Aud: ts.audience,
|
|
|
|
- Iat: iat.Unix(),
|
|
|
|
- Exp: exp.Unix(),
|
|
|
|
|
|
+ Iss: ts.email,
|
|
|
|
+ Sub: ts.email,
|
|
|
|
+ Aud: ts.audience,
|
|
|
|
+ Scope: scope,
|
|
|
|
+ Iat: iat.Unix(),
|
|
|
|
+ Exp: exp.Unix(),
|
|
}
|
|
}
|
|
hdr := &jws.Header{
|
|
hdr := &jws.Header{
|
|
Algorithm: "RS256",
|
|
Algorithm: "RS256",
|