Introduce AAD interactive auth mode (#1860)

2026-01-29 01:25:41 -05:00 · 2023-03-02 09:39:54 -08:00
parent 98e50c98fe
commit 187b6ecc14
47 changed files with 918 additions and 151 deletions
--- a/src/Microsoft.SqlTools.Authentication/Utility/Utils.cs
+++ b/src/Microsoft.SqlTools.Authentication/Utility/Utils.cs
@@ -0,0 +1,119 @@
+//
+// Copyright (c) Microsoft. All rights reserved.
+// Licensed under the MIT license. See LICENSE file in the project root for full license information.
+//
+
+using System.Net.Mail;
+using System.Runtime.InteropServices;
+using Microsoft.Identity.Client;
+using SqlToolsLogger = Microsoft.SqlTools.Utility.Logger;
+
+namespace Microsoft.SqlTools.Authentication.Utility
+{
+    internal sealed class Utils
+    {
+        /// <summary>
+        /// Validates provided <paramref name="userEmail"/> follows email format.
+        /// </summary>
+        /// <param name="useremail">Email address</param>
+        /// <returns>Whether email is in correct format.</returns>
+        public static bool isValidEmail(string userEmail)
+        {
+            try
+            {
+                new MailAddress(userEmail);
+                return true;
+            }
+            catch (FormatException)
+            {
+                return false;
+            }
+        }
+
+        /// <summary>
+        /// Builds directory path based on environment settings.
+        /// </summary>
+        /// <returns>Application directory path</returns>
+        /// <exception cref="Exception">When called on unsupported platform.</exception>
+        public static string BuildAppDirectoryPath()
+        {
+            var homedir = Environment.GetFolderPath(Environment.SpecialFolder.UserProfile);
+
+            // Windows
+            if (RuntimeInformation.IsOSPlatform(OSPlatform.Windows))
+            {
+                var appData = Environment.GetEnvironmentVariable("APPDATA");
+                var userProfile = Environment.GetEnvironmentVariable("USERPROFILE");
+                if (appData != null)
+                {
+                    return appData;
+                }
+                else if (userProfile != null)
+                {
+                    return string.Join(Environment.GetEnvironmentVariable("USERPROFILE"), "AppData", "Roaming");
+                }
+                else
+                {
+                    throw new Exception("Not able to find APPDATA or USERPROFILE");
+                }
+            }
+
+            // Mac
+            else if (RuntimeInformation.IsOSPlatform(OSPlatform.OSX))
+            {
+                return string.Join(homedir, "Library", "Application Support");
+            }
+
+            // Linux
+            else if (RuntimeInformation.IsOSPlatform(OSPlatform.Linux))
+            {
+                var xdgConfigHome = Environment.GetEnvironmentVariable("XDG_CONFIG_HOME");
+                if (xdgConfigHome != null)
+                {
+                    return xdgConfigHome;
+                }
+                else
+                {
+                    return string.Join(homedir, ".config");
+                }
+            }
+            else
+            {
+                throw new Exception("Platform not supported");
+            }
+        }
+
+        /// <summary>
+        /// Log callback handler used for MSAL Client applications.
+        /// </summary>
+        /// <param name="logLevel">Log level</param>
+        /// <param name="message">Log message</param>
+        /// <param name="pii">Whether message contains PII information.</param>
+        public static void MSALLogCallback(LogLevel logLevel, string message, bool pii)
+        {
+            switch (logLevel)
+            {
+                case LogLevel.Error:
+                    if (pii) SqlToolsLogger.Pii(message);
+                    else SqlToolsLogger.Error(message);
+                    break;
+                case LogLevel.Warning:
+                    if (pii) SqlToolsLogger.Pii(message);
+                    else SqlToolsLogger.Warning(message);
+                    break;
+                case LogLevel.Info:
+                    if (pii) SqlToolsLogger.Pii(message);
+                    else SqlToolsLogger.Information(message);
+                    break;
+                case LogLevel.Verbose:
+                    if (pii) SqlToolsLogger.Pii(message);
+                    else SqlToolsLogger.Verbose(message);
+                    break;
+                case LogLevel.Always:
+                    if (pii) SqlToolsLogger.Pii(message);
+                    else SqlToolsLogger.Critical(message);
+                    break;
+            }
+        }
+    }
+}