User management support classes (#1856)

* WIP * Fix nullable warnings in UserData class * WIP2 * WIP * Refresh database prototype classes * Fix some typos & merge issues * WIP * WIP * WIP * Additional updates * Remove unneded using
2026-01-21 01:25:42 -05:00 · 2023-02-08 18:02:08 -08:00
parent ee086e2067
commit 2ef5f0918a
24 changed files with 3584 additions and 1336 deletions
--- a/src/Microsoft.SqlTools.ServiceLayer/Security/UserActions.cs
+++ b/src/Microsoft.SqlTools.ServiceLayer/Security/UserActions.cs
@@ -0,0 +1,144 @@
+//
+// Copyright (c) Microsoft. All rights reserved.
+// Licensed under the MIT license. See LICENSE file in the project root for full license information.
+//
+
+using Microsoft.SqlServer.Management.Sdk.Sfc;
+using Microsoft.SqlServer.Management.Smo;
+using Microsoft.SqlTools.ServiceLayer.Management;
+using Microsoft.SqlTools.ServiceLayer.Security.Contracts;
+
+namespace Microsoft.SqlTools.ServiceLayer.Security
+{
+    internal class UserActions : ManagementActionBase
+    {
+#region Variables
+        //private UserPrototypeData userData;
+        private UserPrototype userPrototype;
+        private UserInfo user;
+        private ConfigAction configAction;
+#endregion
+
+#region Constructors / Dispose
+        /// <summary>
+        /// required when loading from Object Explorer context
+        /// </summary>
+        /// <param name="context"></param>
+        public UserActions(
+            CDataContainer context,
+            UserInfo user,
+            ConfigAction configAction)
+        {
+            this.DataContainer = context;
+            this.user = user;
+            this.configAction = configAction;
+
+            this.userPrototype = InitUserNew(context, user);
+        }
+
+        // /// <summary> 
+        // /// Clean up any resources being used.
+        // /// </summary>
+        // protected override void Dispose(bool disposing)
+        // {
+        //     base.Dispose(disposing);
+        // }
+
+#endregion
+
+        /// <summary>
+        /// called on background thread by the framework to execute the action
+        /// </summary>
+        /// <param name="node"></param>
+        public override void OnRunNow(object sender)
+        {
+            if (this.configAction == ConfigAction.Drop)
+            {
+                // if (this.credentialData.Credential != null)
+                // {
+                //     this.credentialData.Credential.DropIfExists();
+                // }
+            }
+            else
+            {
+                this.userPrototype.ApplyChanges();
+            }
+        }
+        
+        private UserPrototype InitUserNew(CDataContainer dataContainer, UserInfo user)
+        {
+            // this.DataContainer = context;          
+            // this.parentDbUrn = new Urn(this.DataContainer.ParentUrn); 
+            // this.objectUrn = new Urn(this.DataContainer.ObjectUrn);
+            ExhaustiveUserTypes currentUserType;
+            UserPrototypeFactory userPrototypeFactory = UserPrototypeFactory.GetInstance(dataContainer, user);
+
+            if (dataContainer.IsNewObject)
+            {
+                if (IsParentDatabaseContained(dataContainer.ParentUrn, dataContainer))
+                {
+                    currentUserType = ExhaustiveUserTypes.SqlUserWithPassword;
+                }
+                else
+                {
+                    currentUserType = ExhaustiveUserTypes.LoginMappedUser;
+                }
+            }
+            else
+            {
+                currentUserType = this.GetCurrentUserTypeForExistingUser(
+                    dataContainer.Server.GetSmoObject(dataContainer.ObjectUrn) as User);
+            }
+
+           UserPrototype currentUserPrototype = userPrototypeFactory.GetUserPrototype(currentUserType);
+           return currentUserPrototype;
+        }
+
+        private ExhaustiveUserTypes GetCurrentUserTypeForExistingUser(User user)
+        {
+            switch (user.UserType)
+            {
+                case UserType.SqlUser:
+                    if (user.IsSupportedProperty("AuthenticationType"))
+                    {
+                        if (user.AuthenticationType == AuthenticationType.Windows)
+                        {
+                            return ExhaustiveUserTypes.WindowsUser;                            
+                        }
+                        else if (user.AuthenticationType == AuthenticationType.Database)
+                        {
+                            return ExhaustiveUserTypes.SqlUserWithPassword;
+                        }
+                    }
+
+                    return ExhaustiveUserTypes.LoginMappedUser;
+                    
+                case UserType.NoLogin:
+                    return ExhaustiveUserTypes.SqlUserWithoutLogin;
+                    
+                case UserType.Certificate:
+                    return ExhaustiveUserTypes.CertificateMappedUser;
+                    
+                case UserType.AsymmetricKey:
+                    return ExhaustiveUserTypes.AsymmetricKeyMappedUser;
+                    
+                default:
+                    return ExhaustiveUserTypes.Unknown;
+            }
+        }
+
+        private bool IsParentDatabaseContained(Urn parentDbUrn, CDataContainer dataContainer)
+        {
+            string parentDbName = parentDbUrn.GetNameForType("Database");
+            Database parentDatabase = dataContainer.Server.Databases[parentDbName];
+
+            if (parentDatabase.IsSupportedProperty("ContainmentType")
+                && parentDatabase.ContainmentType == ContainmentType.Partial)
+            {
+                return true;
+            }
+
+            return false;
+        }
+    }
+}