diff --git a/src/Core/Models/Table/ISubscriber.cs b/src/Core/Models/Table/ISubscriber.cs index fd22d7d099..cd611a3484 100644 --- a/src/Core/Models/Table/ISubscriber.cs +++ b/src/Core/Models/Table/ISubscriber.cs @@ -15,5 +15,6 @@ namespace Bit.Core.Models.Table string BraintreeCustomerIdPrefix(); string BraintreeIdField(); string GatewayIdField(); + bool IsUser(); } } diff --git a/src/Core/Models/Table/Organization.cs b/src/Core/Models/Table/Organization.cs index 054e3b4572..01404fa958 100644 --- a/src/Core/Models/Table/Organization.cs +++ b/src/Core/Models/Table/Organization.cs @@ -78,6 +78,11 @@ namespace Bit.Core.Models.Table return "organizationId"; } + public bool IsUser() + { + return false; + } + public long StorageBytesRemaining() { if(!MaxStorageGb.HasValue) diff --git a/src/Core/Models/Table/User.cs b/src/Core/Models/Table/User.cs index d2ef6de91a..570927132f 100644 --- a/src/Core/Models/Table/User.cs +++ b/src/Core/Models/Table/User.cs @@ -73,6 +73,11 @@ namespace Bit.Core.Models.Table return "userId"; } + public bool IsUser() + { + return true; + } + public Dictionary GetTwoFactorProviders() { if(string.IsNullOrWhiteSpace(TwoFactorProviders))