From 64ddcaaf0f56c02f484c47679a35c7c6e6ce5abe Mon Sep 17 00:00:00 2001 From: Kyle Spearrin Date: Tue, 17 Sep 2019 16:55:40 -0400 Subject: [PATCH] isUser function --- src/Core/Models/Table/ISubscriber.cs | 1 + src/Core/Models/Table/Organization.cs | 5 +++++ src/Core/Models/Table/User.cs | 5 +++++ 3 files changed, 11 insertions(+) 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))