feat: support licenses that require telemetry

ammario · ammario · commit b6e4c51c4fb5 · 2023-02-13T20:59:30.000Z
diff --git a/codersdk/deployment.go b/codersdk/deployment.go
@@ -81,11 +81,12 @@ type Feature struct {
 }
 
 type Entitlements struct {
-	Features   map[FeatureName]Feature `json:"features"`
-	Warnings   []string                `json:"warnings"`
-	Errors     []string                `json:"errors"`
-	HasLicense bool                    `json:"has_license"`
-	Trial      bool                    `json:"trial"`
+	Features         map[FeatureName]Feature `json:"features"`
+	Warnings         []string                `json:"warnings"`
+	Errors           []string                `json:"errors"`
+	HasLicense       bool                    `json:"has_license"`
+	Trial            bool                    `json:"trial"`
+	RequireTelemetry bool                    `json:"require_telemetry"`
 
 	// DEPRECATED: use Experiments instead.
 	Experimental bool `json:"experimental"`
diff --git a/enterprise/coderd/coderd.go b/enterprise/coderd/coderd.go
@@ -5,6 +5,7 @@ import (
 	"crypto/ed25519"
 	"crypto/tls"
 	"crypto/x509"
+	"fmt"
 	"net/http"
 	"sync"
 	"time"
@@ -252,6 +253,11 @@ func (api *API) updateEntitlements(ctx context.Context) error {
 	if err != nil {
 		return err
 	}
+
+	if entitlements.RequireTelemetry && !api.DeploymentConfig.Telemetry.Enable.Value {
+		return xerrors.New("telemetry disabled, but a license requires telemetry")
+	}
+
 	entitlements.Experimental = api.DeploymentConfig.Experimental.Value || len(api.AGPL.Experiments) != 0
 
 	featureChanged := func(featureName codersdk.FeatureName) (changed bool, enabled bool) {
diff --git a/enterprise/coderd/license/license.go b/enterprise/coderd/license/license.go
@@ -98,6 +98,7 @@ func Entitlements(
 		if claims.AllFeatures {
 			allFeatures = true
 		}
+		entitlements.RequireTelemetry = entitlements.RequireTelemetry || claims.RequireTelemetry
 	}
 
 	if allFeatures {
@@ -224,13 +225,14 @@ type Claims struct {
 	// the end of the grace period (identical to LicenseExpires if there is no grace period).
 	// The reason we use the standard claim for the end of the grace period is that we want JWT
 	// processing libraries to consider the token "valid" until then.
-	LicenseExpires *jwt.NumericDate `json:"license_expires,omitempty"`
-	AccountType    string           `json:"account_type,omitempty"`
-	AccountID      string           `json:"account_id,omitempty"`
-	Trial          bool             `json:"trial"`
-	AllFeatures    bool             `json:"all_features"`
-	Version        uint64           `json:"version"`
-	Features       Features         `json:"features"`
+	LicenseExpires   *jwt.NumericDate `json:"license_expires,omitempty"`
+	AccountType      string           `json:"account_type,omitempty"`
+	AccountID        string           `json:"account_id,omitempty"`
+	Trial            bool             `json:"trial"`
+	AllFeatures      bool             `json:"all_features"`
+	Version          uint64           `json:"version"`
+	Features         Features         `json:"features"`
+	RequireTelemetry bool             `json:"require_telemetry,omitempty"`
 }
 
 // ParseRaw consumes a license and returns the claims.
