wip: add expiration warning

rodrimaia · rodrimaia · commit ba7c486b4545 · 2023-04-20T20:43:58.000Z
diff --git a/enterprise/coderd/license/license.go b/enterprise/coderd/license/license.go
@@ -70,6 +70,13 @@ func Entitlements(
 			// LicenseExpires we must be in grace period.
 			entitlement = codersdk.EntitlementGracePeriod
 		}
+
+		// add warning if license is expiring soon
+		if claims.LicenseExpires.Time.Sub(now) < 30*24*time.Hour {
+			entitlements.Warnings = append(entitlements.Warnings, fmt.Sprintf(
+				"Your license expires in %d days.", int(claims.LicenseExpires.Time.Sub(now).Hours()/24)))
+		}
+
 		for featureName, featureValue := range claims.Features {
 			// Can this be negative?
 			if featureValue <= 0 {
diff --git a/enterprise/coderd/license/license_test.go b/enterprise/coderd/license/license_test.go
@@ -102,6 +102,34 @@ func TestEntitlements(t *testing.T) {
 			fmt.Sprintf("%s is enabled but your license for this feature is expired.", codersdk.FeatureAuditLog.Humanize()),
 		)
 	})
+	t.Run("Expiration warning", func(t *testing.T) {
+		t.Parallel()
+		db := dbfake.New()
+		db.InsertLicense(context.Background(), database.InsertLicenseParams{
+			JWT: coderdenttest.GenerateLicense(t, coderdenttest.LicenseOptions{
+				Features: license.Features{
+					codersdk.FeatureUserLimit: 100,
+					codersdk.FeatureAuditLog:  1,
+				},
+
+				GraceAt:   time.Now().Add(-time.Hour),
+				ExpiresAt: time.Now().AddDate(0, 0, 2),
+			}),
+			Exp: time.Now().Add(time.Hour),
+		})
+
+		entitlements, err := license.Entitlements(context.Background(), db, slog.Logger{}, 1, 1, coderdenttest.Keys, all)
+
+		require.NoError(t, err)
+		require.True(t, entitlements.HasLicense)
+		require.False(t, entitlements.Trial)
+
+		require.Equal(t, codersdk.EntitlementGracePeriod, entitlements.Features[codersdk.FeatureAuditLog].Entitlement)
+		require.Contains(
+			t, entitlements.Warnings,
+			"Your license expires in 2 days.",
+		)
+	})
 	t.Run("SingleLicenseNotEntitled", func(t *testing.T) {
 		t.Parallel()
 		db := dbfake.New()
