nhyatt/bitwarden
1
0
Fork 0
mirror of https://github.com/bitwarden/server.git synced 2025-07-04 09:32:48 -05:00
Code Activity

More CanAccessPremium checks

Browse Source
This commit is contained in:
Kyle Spearrin
2018-08-28 17:40:08 -04:00
parent c41a1e0936
commit cf73b168ee
16 changed files with 90 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/Core/Services/Implementations/CipherService.cs
View File

@ -24,6 +24,7 @@ namespace Bit.Core.Services
private readonly IPushNotificationService _pushService;
private readonly IAttachmentStorageService _attachmentStorageService;
private readonly IEventService _eventService;
private readonly IUserService _userService;
public CipherService(
ICipherRepository cipherRepository,
@ -35,7 +36,8 @@ namespace Bit.Core.Services
ICollectionCipherRepository collectionCipherRepository,
IPushNotificationService pushService,
IAttachmentStorageService attachmentStorageService,
IEventService eventService)
IEventService eventService,
IUserService userService)
{
_cipherRepository = cipherRepository;
_folderRepository = folderRepository;
@ -47,6 +49,7 @@ namespace Bit.Core.Services
_pushService = pushService;
_attachmentStorageService = attachmentStorageService;
_eventService = eventService;
_userService = userService;
}
public async Task SaveAsync(Cipher cipher, Guid savingUserId, bool orgAdmin = false)
@ -125,9 +128,9 @@ namespace Bit.Core.Services
if(cipher.UserId.HasValue)
{
var user = await _userRepository.GetByIdAsync(cipher.UserId.Value);
if(!user.Premium)
if(!(await _userService.CanAccessPremium(user)))
{
throw new BadRequestException("You must be a premium user to use attachments.");
throw new BadRequestException("You must have premium status to use attachments.");
}
storageBytesRemaining = user.StorageBytesRemaining();