only download license for trial_started accounts

Author: zeroshade

auth/credentials.go

@@ -27,6 +27,7 @@ import (
 	"slices"
 	"sync"
 
+	"github.com/golang-jwt/jwt/v5"
 	"github.com/pelletier/go-toml/v2"
 )
 
@@ -267,7 +268,7 @@ func UpdateCreds() error {
 	})
 }
 
-var licenseURI = "https://dbc-cf-api.columnar.workers.dev/trial_license"
+const licenseURI = "https://dbc-cf-api.columnar.workers.dev/trial_license"
 
 func FetchColumnarLicense(cred *Credential) error {
 	licensePath := filepath.Join(filepath.Dir(credPath), "columnar.lic")
@@ -280,19 +281,29 @@ func FetchColumnarLicense(cred *Credential) error {
 		return err
 	}
 
-	req, err := http.NewRequest(http.MethodGet, licenseURI, nil)
-	if err != nil {
-		return err
-	}
-
 	var authToken string
 	switch cred.Type {
 	case TypeApiKey:
 		authToken = cred.ApiKey
 	case TypeToken:
+		p := jwt.NewParser()
+		tk, err := p.Parse(cred.GetAuthToken(), nil)
+		if err != nil && !errors.Is(err, jwt.ErrTokenUnverifiable) {
+			return fmt.Errorf("failed to parse oauth token: %w", err)
+		}
+
+		_, ok := tk.Claims.(jwt.MapClaims)["urn:columnar:trial_start"]
+		if !ok {
+			return nil // not a trial account
+		}
 		authToken = "Bearer " + cred.GetAuthToken()
 	}
 
+	req, err := http.NewRequest(http.MethodGet, licenseURI, nil)
+	if err != nil {
+		return err
+	}
+
 	req.Header.Add("authorization", authToken)
 	resp, err := http.DefaultClient.Do(req)
 	if err != nil {

go.mod

@@ -52,6 +52,7 @@ require (
 	github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f // indirect
 	github.com/go-faster/errors v0.7.1 // indirect
 	github.com/go-faster/jx v1.1.0 // indirect
+	github.com/golang-jwt/jwt/v5 v5.3.0 // indirect
 	github.com/lucasb-eyer/go-colorful v1.2.0 // indirect
 	github.com/mattn/go-localereader v0.0.1 // indirect
 	github.com/mattn/go-runewidth v0.0.16 // indirect

go.sum

@@ -51,6 +51,8 @@ github.com/go-faster/jx v1.1.0 h1:ZsW3wD+snOdmTDy9eIVgQdjUpXRRV4rqW8NS3t+20bg=
 github.com/go-faster/jx v1.1.0/go.mod h1:vKDNikrKoyUmpzaJ0OkIkRQClNHFX/nF3dnTJZb3skg=
 github.com/go-faster/yaml v0.4.6 h1:lOK/EhI04gCpPgPhgt0bChS6bvw7G3WwI8xxVe0sw9I=
 github.com/go-faster/yaml v0.4.6/go.mod h1:390dRIvV4zbnO7qC9FGo6YYutc+wyyUSHBgbXL52eXk=
+github.com/golang-jwt/jwt/v5 v5.3.0 h1:pv4AsKCKKZuqlgs5sUmn4x8UlGa0kEVt/puTpKx9vvo=
+github.com/golang-jwt/jwt/v5 v5.3.0/go.mod h1:fxCRLWMO43lRc8nhHWY6LGqRcf+1gQWArsqaEUEa5bE=
 github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
 github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/kr/pretty v0.2.1 h1:Fmg33tUaq4/8ym9TJN1x7sLJnHVwhP33CNkpYV/7rwI=