error.go

Documentation: golang.org/x/oauth2/google

     1  // Copyright 2022 The Go Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  package google
     6  
     7  import (
     8  	"errors"
     9  
    10  	"golang.org/x/oauth2"
    11  )
    12  
    13  // AuthenticationError indicates there was an error in the authentication flow.
    14  //
    15  // Use (*AuthenticationError).Temporary to check if the error can be retried.
    16  type AuthenticationError struct {
    17  	err *oauth2.RetrieveError
    18  }
    19  
    20  func newAuthenticationError(err error) error {
    21  	re := &oauth2.RetrieveError{}
    22  	if !errors.As(err, &re) {
    23  		return err
    24  	}
    25  	return &AuthenticationError{
    26  		err: re,
    27  	}
    28  }
    29  
    30  // Temporary indicates that the network error has one of the following status codes and may be retried: 500, 503, 408, or 429.
    31  func (e *AuthenticationError) Temporary() bool {
    32  	if e.err.Response == nil {
    33  		return false
    34  	}
    35  	sc := e.err.Response.StatusCode
    36  	return sc == 500 || sc == 503 || sc == 408 || sc == 429
    37  }
    38  
    39  func (e *AuthenticationError) Error() string {
    40  	return e.err.Error()
    41  }
    42  
    43  func (e *AuthenticationError) Unwrap() error {
    44  	return e.err
    45  }
    46  
    47  type errWrappingTokenSource struct {
    48  	src oauth2.TokenSource
    49  }
    50  
    51  func newErrWrappingTokenSource(ts oauth2.TokenSource) oauth2.TokenSource {
    52  	return &errWrappingTokenSource{src: ts}
    53  }
    54  
    55  // Token returns the current token if it's still valid, else will
    56  // refresh the current token (using r.Context for HTTP client
    57  // information) and return the new one.
    58  func (s *errWrappingTokenSource) Token() (*oauth2.Token, error) {
    59  	t, err := s.src.Token()
    60  	if err != nil {
    61  		return nil, newAuthenticationError(err)
    62  	}
    63  	return t, nil
    64  }
    65
View as plain text