From 201f7d9aa45bb4d0ac8c156947c78d641fe0df5b Mon Sep 17 00:00:00 2001
From: Kyle Spearrin <kyle.spearrin@gmail.com>
Date: Fri, 1 Mar 2019 17:30:59 -0500
Subject: [PATCH] current context for organization clients

---
 src/Core/CurrentContext.cs | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/src/Core/CurrentContext.cs b/src/Core/CurrentContext.cs
index 977d56f7a0..a222214b0a 100644
--- a/src/Core/CurrentContext.cs
+++ b/src/Core/CurrentContext.cs
@@ -25,6 +25,7 @@ namespace Bit.Core
         public virtual string IpAddress { get; set; }
         public virtual List<CurrentContentOrganization> Organizations { get; set; }
         public virtual Guid? InstallationId { get; set; }
+        public virtual Guid? OrganizationId { get; set; }
 
         public void Build(HttpContext httpContext, GlobalSettings globalSettings)
         {
@@ -73,11 +74,21 @@ namespace Bit.Core
 
             var clientId = GetClaimValue(claimsDict, "client_id");
             var clientSubject = GetClaimValue(claimsDict, "client_sub");
-            if((clientId?.StartsWith("installation.") ?? false) && clientSubject != null)
+            if(clientSubject != null)
             {
-                if(Guid.TryParse(clientSubject, out var idGuid))
+                if(clientId?.StartsWith("installation.") ?? false)
                 {
-                    InstallationId = idGuid;
+                    if(Guid.TryParse(clientSubject, out var idGuid))
+                    {
+                        InstallationId = idGuid;
+                    }
+                }
+                else if(clientId?.StartsWith("organization.") ?? false)
+                {
+                    if(Guid.TryParse(clientSubject, out var idGuid))
+                    {
+                        OrganizationId = idGuid;
+                    }
                 }
             }