aykhans/slash-e
1
0
Fork 0
mirror of https://github.com/aykhans/slash-e.git synced 2025-07-04 20:28:00 +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

20
server/route/api/v1/subscription_service.go
View File

@ -9,22 +9,26 @@ import (
v1pb "github.com/yourselfhosted/slash/proto/gen/api/v1"
)
func (s *APIV1Service) GetSubscription(ctx context.Context, _ *v1pb.GetSubscriptionRequest) (*v1pb.GetSubscriptionResponse, error) {
func (s *APIV1Service) GetSubscription(ctx context.Context, _ *v1pb.GetSubscriptionRequest) (*v1pb.Subscription, error) {
subscription, err := s.LicenseService.LoadSubscription(ctx)
if err != nil {
return nil, status.Errorf(codes.Internal, "failed to load subscription: %v", err)
}
return &v1pb.GetSubscriptionResponse{
Subscription: subscription,
}, nil
return subscription, nil
}
func (s *APIV1Service) UpdateSubscription(ctx context.Context, request *v1pb.UpdateSubscriptionRequest) (*v1pb.UpdateSubscriptionResponse, error) {
func (s *APIV1Service) UpdateSubscription(ctx context.Context, request *v1pb.UpdateSubscriptionRequest) (*v1pb.Subscription, error) {
subscription, err := s.LicenseService.UpdateSubscription(ctx, request.LicenseKey)
if err != nil {
return nil, status.Errorf(codes.Internal, "failed to load subscription: %v", err)
}
return &v1pb.UpdateSubscriptionResponse{
Subscription: subscription,
}, nil
return subscription, nil
}
func (s *APIV1Service) DeleteSubscription(ctx context.Context, _ *v1pb.DeleteSubscriptionRequest) (*v1pb.Subscription, error) {
subscription, err := s.LicenseService.UpdateSubscription(ctx, "")
if err != nil {
return nil, status.Errorf(codes.Internal, "failed to load subscription: %v", err)
}
return subscription, nil
}

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 {