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

chore: tweak plans

Browse Source
This commit is contained in:
Steven
2024-08-16 21:39:14 +08:00
parent 0be4d8c906
commit faa6fcf31c
10 changed files with 163 additions and 100 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
server/route/api/v1/collection_service.go
View File

@ -108,15 +108,14 @@ func (s *APIV1Service) CreateCollection(ctx context.Context, request *v1pb.Creat
return nil, status.Errorf(codes.InvalidArgument, "name and title are required")
}
if !s.LicenseService.IsFeatureEnabled(license.FeatureTypeUnlimitedAccounts) {
collections, err := s.Store.ListCollections(ctx, &store.FindCollection{
VisibilityList: []store.Visibility{store.VisibilityWorkspace, store.VisibilityPublic},
})
if !s.LicenseService.IsFeatureEnabled(license.FeatureTypeUnlimitedCollections) {
collections, err := s.Store.ListCollections(ctx, &store.FindCollection{})
if err != nil {
return nil, status.Errorf(codes.Internal, "failed to get collection list, err: %v", err)
}
if len(collections) >= 5 {
return nil, status.Errorf(codes.PermissionDenied, "Maximum number of collections reached")
collectionsLimit := int(s.LicenseService.GetSubscription().CollectionsLimit)
if len(collections) >= collectionsLimit {
return nil, status.Errorf(codes.PermissionDenied, "Maximum number of collections %d reached", collectionsLimit)
}
}

11
server/route/api/v1/shortcut_service.go
View File

@ -132,6 +132,17 @@ func (s *APIV1Service) CreateShortcut(ctx context.Context, request *v1pb.CreateS
return nil, status.Errorf(codes.InvalidArgument, "name and link are required")
}
if !s.LicenseService.IsFeatureEnabled(license.FeatureTypeUnlimitedShortcuts) {
shortcuts, err := s.Store.ListShortcuts(ctx, &store.FindShortcut{})
if err != nil {
return nil, status.Errorf(codes.Internal, "failed to get shortcut list, err: %v", err)
}
shortcutsLimit := int(s.LicenseService.GetSubscription().ShortcutsLimit)
if len(shortcuts) >= shortcutsLimit {
return nil, status.Errorf(codes.PermissionDenied, "Maximum number of shortcuts %d reached", shortcutsLimit)
}
}
user, err := getCurrentUser(ctx, s.Store)
if err != nil {
return nil, status.Errorf(codes.Unauthenticated, "failed to get current user: %v", err)