aykhans/slash-e
1
0
Fork 0
mirror of https://github.com/aykhans/slash-e.git synced 2025-07-05 12:42:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: update subscription service

Browse Source
This commit is contained in:
Steven
2024-08-16 21:02:51 +08:00
parent 63ebd6f8ea
commit 0be4d8c906
9 changed files with 310 additions and 245 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
server/service/license/license.go
View File

@ -66,32 +66,44 @@ func (s *LicenseService) LoadSubscription(ctx context.Context) (*v1pb.Subscripti
if result == nil {
return subscription, nil
}
subscription.Plan = result.Plan
subscription.ExpiresTime = timestamppb.New(result.ExpiresTime)
subscription.Seats = int32(result.Seats)
for _, feature := range result.Features {
subscription.Features = append(subscription.Features, feature.String())
}
s.cachedSubscription = subscription
return subscription, nil
}
func (s *LicenseService) UpdateSubscription(ctx context.Context, licenseKey string) (*v1pb.Subscription, error) {
if licenseKey == "" {
return nil, errors.New("license key is required")
if licenseKey != "" {
result, err := validateLicenseKey(licenseKey)
if err != nil {
return nil, errors.Wrap(err, "failed to validate license key")
}
if result == nil {
return nil, errors.New("invalid license key")
}
}
result, err := validateLicenseKey(licenseKey)
if err := s.UpdateLicenseKey(ctx, licenseKey); err != nil {
return nil, errors.Wrap(err, "failed to update license key")
}
subscription, err := s.LoadSubscription(ctx)
if err != nil {
return nil, errors.Wrap(err, "failed to validate license key")
}
if result == nil {
return nil, errors.New("invalid license key")
return nil, errors.Wrap(err, "failed to load subscription")
}
s.cachedSubscription = subscription
return subscription, nil
}
func (s *LicenseService) UpdateLicenseKey(ctx context.Context, licenseKey string) error {
workspaceSettingGeneral, err := s.Store.GetWorkspaceSetting(ctx, &store.FindWorkspaceSetting{
Key: storepb.WorkspaceSettingKey_WORKSPACE_SETTING_GENERAL,
})
if err != nil {
return nil, errors.Wrap(err, "failed to get workspace setting")
return errors.Wrap(err, "failed to get workspace setting")
}
if workspaceSettingGeneral == nil || workspaceSettingGeneral.GetGeneral() == nil {
workspaceSettingGeneral = &storepb.WorkspaceSetting{
@ -107,9 +119,9 @@ func (s *LicenseService) UpdateSubscription(ctx context.Context, licenseKey stri
}
_, err = s.Store.UpsertWorkspaceSetting(ctx, workspaceSettingGeneral)
if err != nil {
return nil, errors.Wrap(err, "failed to upsert workspace setting")
return errors.Wrap(err, "failed to upsert workspace setting")
}
return s.LoadSubscription(ctx)
return nil
}
func (s *LicenseService) GetSubscription() *v1pb.Subscription {