mirror of
https://github.com/aykhans/slash-e.git
synced 2025-06-15 04:17:50 +00:00
chore: add feature matrix
This commit is contained in:
21
server/service/license/feature_matrix.go
Normal file
21
server/service/license/feature_matrix.go
Normal file
@ -0,0 +1,21 @@
|
||||
package license
|
||||
|
||||
type FeatureType string
|
||||
|
||||
const (
|
||||
// Accounts.
|
||||
|
||||
// FeatureTypeUnlimitedAccounts allows the user to create unlimited accounts.
|
||||
FeatureTypeUnlimitedAccounts FeatureType = "unlimited_accounts"
|
||||
|
||||
// Customization.
|
||||
|
||||
// FeatureTypeCustomStyle allows the user to customize the style.
|
||||
FeatureTypeCustomeStyle FeatureType = "custom_style"
|
||||
)
|
||||
|
||||
// FeatureMatrix is a matrix of features in [Free, Pro].
|
||||
var FeatureMatrix = map[FeatureType][2]bool{
|
||||
FeatureTypeUnlimitedAccounts: {false, true},
|
||||
FeatureTypeCustomeStyle: {false, true},
|
||||
}
|
@ -13,9 +13,10 @@ import (
|
||||
)
|
||||
|
||||
type LicenseService struct {
|
||||
Profile *profile.Profile
|
||||
Store *store.Store
|
||||
CachedSubscription *apiv2pb.Subscription
|
||||
Profile *profile.Profile
|
||||
Store *store.Store
|
||||
|
||||
cachedSubscription *apiv2pb.Subscription
|
||||
}
|
||||
|
||||
// NewLicenseService creates a new LicenseService.
|
||||
@ -23,7 +24,7 @@ func NewLicenseService(profile *profile.Profile, store *store.Store) *LicenseSer
|
||||
return &LicenseService{
|
||||
Profile: profile,
|
||||
Store: store,
|
||||
CachedSubscription: &apiv2pb.Subscription{
|
||||
cachedSubscription: &apiv2pb.Subscription{
|
||||
Plan: apiv2pb.PlanType_FREE,
|
||||
},
|
||||
}
|
||||
@ -66,7 +67,7 @@ func (s *LicenseService) LoadSubscription(ctx context.Context) (*apiv2pb.Subscri
|
||||
}
|
||||
subscription.StartedTime = timestamppb.New(startedTime)
|
||||
}
|
||||
s.CachedSubscription = subscription
|
||||
s.cachedSubscription = subscription
|
||||
return subscription, nil
|
||||
}
|
||||
|
||||
@ -92,3 +93,11 @@ func (s *LicenseService) UpdateSubscription(ctx context.Context, licenseKey stri
|
||||
}
|
||||
return s.LoadSubscription(ctx)
|
||||
}
|
||||
|
||||
func (s *LicenseService) IsFeatureEnabled(feature FeatureType) bool {
|
||||
matrix, ok := FeatureMatrix[feature]
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
return matrix[s.cachedSubscription.Plan-1]
|
||||
}
|
||||
|
Reference in New Issue
Block a user