Aug 19, 2024 · Aug 19, 2024 · Aug 19, 2024 · Aug 19, 2024 · Aug 19, 2024 · Aug 19, 2024
diff --git a/codersdk/deployment.go b/codersdk/deployment.go
 }[n]
 }

 // Enterprise returns true if the feature is an enterprise feature.
 func (n FeatureName) Enterprise() bool {
 switch n {
 // Add all features that should be excluded in the Enterprise feature set.
 case FeatureMultipleOrganizations, FeatureCustomRoles:
 return false
 default:
 return true
 }
 }

 // FeatureSet represents a grouping of features. Rather than manually
 // assigning features al-la-carte when making a license, a set can be specified.
 // Sets are dynamic in the sense a feature can be added to a set, granting the
 copy(enterpriseFeatures, FeatureNames)
 // Remove the selection
 enterpriseFeatures = slices.DeleteFunc(enterpriseFeatures, func(f FeatureName) bool {
 switch f {
 // Add all features that should be excluded in the Enterprise feature set.
 case FeatureMultipleOrganizations, FeatureCustomRoles:
 return true
 default:
 return false
 }
 return !f.Enterprise()
 })

 return enterpriseFeatures
diff --git a/enterprise/coderd/roles_test.go b/enterprise/coderd/roles_test.go

 // Verify functionality is lost
 _, err = owner.UpdateOrganizationRole(ctx, templateAdminCustom(first.OrganizationID))
 require.ErrorContains(t, err, "Custom Roles isan Enterprise feature")
 require.ErrorContains(t, err, "Custom Roles isa Premium feature")

 // Assign the custom template admin role
 tmplAdmin, _ := coderdtest.CreateAnotherUser(t, owner, first.OrganizationID, rbac.RoleIdentifier{Name: role.Name, OrganizationID: first.OrganizationID})
diff --git a/enterprise/coderd/templates.go b/enterprise/coderd/templates.go
 enabled := api.entitlements.Features[feat].Enabled
 api.entitlementsMu.RUnlock()
 if !enabled {
 licenseType := "a Premium"
 if feat.Enterprise() {
 licenseType = "an Enterprise"
 }
 httpapi.Write(r.Context(), rw, http.StatusForbidden, codersdk.Response{
 Message: fmt.Sprintf("%s isan Enterprisefeature. Contact sales!", feat.Humanize()),
 Message: fmt.Sprintf("%s is%sfeature. Contact sales!", feat.Humanize(), licenseType),
 })
 return
 }
Original file line number	Diff line number	Diff line change
Expand Up		@@ -128,6 +128,17 @@ func (n FeatureName) AlwaysEnable() bool {
		}[n]
		}

		// Enterprise returns true if the feature is an enterprise feature.
		func (n FeatureName) Enterprise() bool {
		switch n {
		// Add all features that should be excluded in the Enterprise feature set.
		case FeatureMultipleOrganizations, FeatureCustomRoles:
		return false
		default:
		return true
		}
		}

		// FeatureSet represents a grouping of features. Rather than manually
		// assigning features al-la-carte when making a license, a set can be specified.
		// Sets are dynamic in the sense a feature can be added to a set, granting the
Expand All		@@ -152,13 +163,7 @@ func (set FeatureSet) Features() []FeatureName {
		copy(enterpriseFeatures, FeatureNames)
		// Remove the selection
		enterpriseFeatures = slices.DeleteFunc(enterpriseFeatures, func(f FeatureName) bool {
		switch f {
		// Add all features that should be excluded in the Enterprise feature set.
		case FeatureMultipleOrganizations, FeatureCustomRoles:
		return true
		default:
		return false
		}
		return !f.Enterprise()
		})

		return enterpriseFeatures
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -125,7 +125,7 @@ func TestCustomOrganizationRole(t *testing.T) {

		// Verify functionality is lost
		_, err = owner.UpdateOrganizationRole(ctx, templateAdminCustom(first.OrganizationID))
		require.ErrorContains(t, err, "Custom Roles isan Enterprise feature")
		require.ErrorContains(t, err, "Custom Roles isa Premium feature")

		// Assign the custom template admin role
		tmplAdmin, _ := coderdtest.CreateAnotherUser(t, owner, first.OrganizationID, rbac.RoleIdentifier{Name: role.Name, OrganizationID: first.OrganizationID})
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -364,8 +364,12 @@ func (api *API) RequireFeatureMW(feat codersdk.FeatureName) func(http.Handler) h
		enabled := api.entitlements.Features[feat].Enabled
		api.entitlementsMu.RUnlock()
		if !enabled {
		licenseType := "a Premium"
		if feat.Enterprise() {
		licenseType = "an Enterprise"
		}
		httpapi.Write(r.Context(), rw, http.StatusForbidden, codersdk.Response{
		Message: fmt.Sprintf("%s isan Enterprisefeature. Contact sales!", feat.Humanize()),
		Message: fmt.Sprintf("%s is%sfeature. Contact sales!", feat.Humanize(), licenseType),
		})
		return
		}
Expand Down