add message if trial is not started

Author: zeroshade

auth/credentials.go

@@ -274,6 +274,8 @@ func IsColumnarPrivateRegistry(u *url.URL) bool {
 
 const licenseURI = "https://heimdall.columnar.tech/trial_license"
 
+var ErrNoTrialLicense = errors.New("no trial license found")
+
 func FetchColumnarLicense(cred *Credential) error {
 	licensePath := filepath.Join(filepath.Dir(credPath), "columnar.lic")
 	_, err := os.Stat(licensePath)
@@ -298,7 +300,7 @@ func FetchColumnarLicense(cred *Credential) error {
 
 		_, ok := tk.Claims.(jwt.MapClaims)["urn:columnar:trial_start"]
 		if !ok {
-			return nil // not a trial account
+			return ErrNoTrialLicense
 		}
 		authToken = "Bearer " + cred.GetAuthToken()
 	}

cmd/dbc/auth.go

@@ -195,12 +195,11 @@ func (m loginModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 				return err
 			}
 
-			if msg.RegistryURL.Host == auth.DefaultOauthURI {
+			if auth.IsColumnarPrivateRegistry((*url.URL)(&msg.RegistryURL)) {
 				return auth.FetchColumnarLicense(&msg)
 			}
 			return nil
-		}, tea.Println("Authentication successful!"),
-			tea.Quit)
+		}, tea.Println("Authentication successful!"), tea.Quit)
 	}
 
 	base, cmd := m.baseModel.Update(msg)

cmd/dbc/main.go

@@ -15,6 +15,7 @@
 package main
 
 import (
+	"errors"
 	"fmt"
 	"os"
 	"slices"
@@ -23,6 +24,7 @@ import (
 	tea "github.com/charmbracelet/bubbletea"
 	"github.com/charmbracelet/lipgloss"
 	"github.com/columnar-tech/dbc"
+	"github.com/columnar-tech/dbc/auth"
 	"github.com/columnar-tech/dbc/cmd/dbc/completions"
 	"github.com/columnar-tech/dbc/config"
 	"github.com/mattn/go-isatty"
@@ -118,6 +120,9 @@ func (m baseModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 			return m, tea.Quit
 		}
 	case error:
+		if errors.Is(msg, auth.ErrNoTrialLicense) {
+			return m, tea.Println(errStyle.Render("Could not download license, trial not started"))
+		}
 		m.status = 1
 		return m, tea.Sequence(tea.Println("Error: ", msg.Error()), tea.Quit)
 	}