From 76953109904102e91eae7462b3e73f5d03ae6b2a Mon Sep 17 00:00:00 2001
From: Erez Rokah <erezrokah@users.noreply.github.com>
Date: Fri, 15 Mar 2024 18:04:13 +0100
Subject: [PATCH] feat: Support sync test connection API tokens (#135)

Similar to https://github.com/cloudquery/cloudquery-api-go/pull/109

Testing a connection is very similar to running a sync job, just with a different CLI command and permissions.
It needs an API token to be able to download plugins and update the status of the test (it won't have access to update usage and progress line sync run API tokens)
---
 auth/token.go | 22 ++++++++++++++--------
 1 file changed, 14 insertions(+), 8 deletions(-)

diff --git a/auth/token.go b/auth/token.go
index 2fd26b5..6276ad0 100644
--- a/auth/token.go
+++ b/auth/token.go
@@ -14,12 +14,13 @@ import (
 )
 
 const (
-	FirebaseAPIKey         = "AIzaSyCxsrwjABEF-dWLzUqmwiL-ct02cnG9GCs"
-	TokenBaseURL           = "https://securetoken.googleapis.com"
-	EnvVarCloudQueryAPIKey = "CLOUDQUERY_API_KEY"
-	ExpiryBuffer           = 60 * time.Second
-	tokenFilePath          = "cloudquery/token"
-	syncRunAPIKeyPrefix    = "cqsr_"
+	FirebaseAPIKey                 = "AIzaSyCxsrwjABEF-dWLzUqmwiL-ct02cnG9GCs"
+	TokenBaseURL                   = "https://securetoken.googleapis.com"
+	EnvVarCloudQueryAPIKey         = "CLOUDQUERY_API_KEY"
+	ExpiryBuffer                   = 60 * time.Second
+	tokenFilePath                  = "cloudquery/token"
+	syncRunAPIKeyPrefix            = "cqsr_"
+	syncTestConnectionAPIKeyPrefix = "cqstc_"
 )
 
 type tokenResponse struct {
@@ -39,6 +40,7 @@ const (
 	BearerToken
 	APIKey
 	SyncRunAPIKey
+	SyncTestConnectionAPIKey
 )
 
 var UndefinedToken = Token{Type: Undefined, Value: ""}
@@ -108,10 +110,14 @@ func (tc *TokenClient) GetToken() (Token, error) {
 // GetTokenType returns the type of token that will be returned by GetToken
 func (tc *TokenClient) GetTokenType() TokenType {
 	if token := os.Getenv(EnvVarCloudQueryAPIKey); token != "" {
-		if strings.HasPrefix(token, syncRunAPIKeyPrefix) {
+		switch {
+		case strings.HasPrefix(token, syncRunAPIKeyPrefix):
 			return SyncRunAPIKey
+		case strings.HasPrefix(token, syncTestConnectionAPIKeyPrefix):
+			return SyncTestConnectionAPIKey
+		default:
+			return APIKey
 		}
-		return APIKey
 	}
 	return BearerToken
 }