feat: Add method to return token type (#75)

Whilst the token itself now contains the token type, it is useful to be able to determine the type of tokens the token client will return without fetching a token first. An example user of this functionality will be the usage client, which needs to determine if an API call is needed to fetch the team name when the authentication is using an API key.

Refs: cloudquery/cloudquery-issues#893

Author: mnorbury

auth/token.go

@@ -66,8 +66,8 @@ func NewTokenClient() *TokenClient {
 // GetToken returns the ID token
 // If CLOUDQUERY_API_KEY is set, it returns that value, otherwise it returns an ID token generated from the refresh token.
 func (tc *TokenClient) GetToken() (Token, error) {
-	if token := os.Getenv(EnvVarCloudQueryAPIKey); token != "" {
-		return Token{Type: APIKey, Value: token}, nil
+	if tc.GetTokenType() == APIKey {
+		return Token{Type: APIKey, Value: os.Getenv(EnvVarCloudQueryAPIKey)}, nil
 	}
 
 	// If the token is not expired, return it
@@ -98,6 +98,14 @@ func (tc *TokenClient) GetToken() (Token, error) {
 	return Token{Type: BearerToken, Value: tc.idToken}, nil
 }
 
+// GetTokenType returns the type of token that will be returned by GetToken
+func (tc *TokenClient) GetTokenType() TokenType {
+	if token := os.Getenv(EnvVarCloudQueryAPIKey); token != "" {
+		return APIKey
+	}
+	return BearerToken
+}
+
 func (tc *TokenClient) generateToken(refreshToken string) (*tokenResponse, error) {
 	data := url.Values{}
 	data.Set("grant_type", "refresh_token")

auth/token_test.go

@@ -9,6 +9,7 @@ import (
 	"testing"
 	"time"
 
+	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 )
 
@@ -103,6 +104,20 @@ func TestTokenClient_GetToken_LongExpiry(t *testing.T) {
 	require.Equal(t, Token{Type: BearerToken, Value: "my_id_token_0"}, token, "expected to reuse token")
 }
 
+func TestTokenClient_BearerTokenType(t *testing.T) {
+	tc := NewTokenClient()
+
+	assert.Equal(t, BearerToken, tc.GetTokenType())
+}
+
+func TestTokenClient_APIKeyTokenType(t *testing.T) {
+	t.Setenv(EnvVarCloudQueryAPIKey, "my_token")
+
+	tc := NewTokenClient()
+
+	assert.Equal(t, APIKey, tc.GetTokenType())
+}
+
 func overrideEnvironmentVariable(t *testing.T, key, value string) func() {
 	originalValue := os.Getenv(key)
 	resetFn := func() {