Add interactive logon for PowerShell Core (#169)

* Add MSAL packages

* Add interactive auth support for PSCore

* Update release notes

+semver: minor

* Update client ID

* Refactor code for readability

* Update release notes

* Add Debug config as task default

* Change parameter to Enum

* Fix typo

* Add new aliases

* Improve folder handling

* Add preliminary support for sourcegen

* Add test

* Fix test

* Update tests

* Add validation against empty strings

* Improve path handling

* Replace string values with Enum

* Update release notes

* Replace Path to prevent sep char from being replaced

* Fix item path handling

* Update tests

* Remove redundant call to Data service

* Fix bug in recursive folder creation

* Update release notes

Author: igoravl

.vscode/tasks.json

@@ -50,6 +50,8 @@
                 "-NoProfile",
                 "-File",
                 "${workspaceRoot}/build.ps1",
+                "-Configuration",
+                "Debug",
                 "-Targets",
                 "Rebuild",
                 "-SkipTests"

CSharp/TfsCmdlets.Common/Attributes.cs

@@ -23,6 +23,17 @@ public CmdletControllerAttribute(Type dataType) : base(typeof(IController))
         }
     }
 
+    [AttributeUsage(System.AttributeTargets.Class, Inherited = true, AllowMultiple = false)]
+    public class ModelAttribute: Attribute
+    {
+        public Type DataType { get; }
+
+        public ModelAttribute(Type dataType)
+        {
+            DataType = dataType;
+        }
+    }
+
     // [AttributeUsage(System.AttributeTargets.Class, Inherited = false, AllowMultiple = false)]
     // public sealed class DesktopOnlyAttribute : Attribute
     // {

CSharp/TfsCmdlets.Common/Enums.cs

@@ -205,4 +205,25 @@ public enum WorkItemLinkType
         ArtifactLink = 1 << 31,
     }
 
+    /// <summary>
+    /// Work Item Query Item Scope
+    /// </summary>
+    [Flags]
+    public enum QueryItemScope
+    {
+        /// <summary>
+        /// Personal Scope ("My Queries" folder)
+        /// </summary>
+        Personal = 1,
+
+        /// <summary>
+        /// Shared Scope ("Shared Queries" folder)
+        /// </summary>
+        Shared = 2,
+
+        /// <summary>
+        /// Both scopes (all folders)
+        /// </summary>
+        Both = 3
+    }
 }

CSharp/TfsCmdlets.Common/Models/WorkItem/Query/QueryItem.cs

@@ -0,0 +1,10 @@
+using Microsoft.TeamFoundation.WorkItemTracking.WebApi.Models;
+
+namespace TfsCmdlets.Models.WorkItem.Query
+{
+    [Model(typeof(QueryHierarchyItem))]
+    partial class QueryItem
+    {
+
+    }
+}

CSharp/TfsCmdlets.Common/Services/IInteractiveAuthentication.cs

@@ -0,0 +1,7 @@
+namespace TfsCmdlets.Services
+{
+    public interface IInteractiveAuthentication
+    {
+        string GetToken(Uri uri);                                                                                                       
+    }
+}

CSharp/TfsCmdlets.Common/Services/Impl/InteractiveAuthenticationImpl.cs

@@ -0,0 +1,63 @@
+using Microsoft.Identity.Client;
+using Microsoft.Identity.Client.Desktop;
+using System.Reflection;
+using System.Threading;
+using System.Threading.Tasks;
+
+namespace TfsCmdlets.Services.Impl
+{
+    [Export(typeof(IInteractiveAuthentication)), Shared]
+    public class InteractiveAuthenticationImpl : IInteractiveAuthentication
+    {
+        private const string CLIENT_ID = "9f44d9a2-86ef-4794-b2b2-f9038a2628e0";
+        private const string SCOPE_ID = "499b84ac-1321-427f-aa17-267ca6975798/user_impersonation";
+
+        public string GetToken(Uri uri)
+        {
+            var authResult = SignInUserAndGetTokenUsingMSAL(new[] { SCOPE_ID }).GetAwaiter().GetResult();
+            var authHeader = authResult.AccessToken;
+
+            return authHeader;
+        }
+
+        /// <summary>
+        /// Sign-in user using MSAL and obtain an access token for Azure DevOps
+        /// </summary>
+        /// <param name="scopes"></param>
+        /// <returns>AuthenticationResult</returns>
+        private static async Task<AuthenticationResult> SignInUserAndGetTokenUsingMSAL(string[] scopes)
+        {
+            var application = PublicClientApplicationBuilder
+                .CreateWithApplicationOptions(new PublicClientApplicationOptions
+                {
+                    AadAuthorityAudience = AadAuthorityAudience.AzureAdAndPersonalMicrosoftAccount,
+                    ClientId = CLIENT_ID,
+                    ClientName = "TfsCmdlets.InteractiveAuth",
+                    ClientVersion = Assembly.GetExecutingAssembly().GetName().Version?.ToString()
+                })
+                .WithDesktopFeatures()
+                .WithDefaultRedirectUri()
+                .Build();
+
+            try
+            {
+                var accounts = (await application.GetAccountsAsync()).ToList();
+
+                return await application
+                    .AcquireTokenSilent(scopes, accounts.FirstOrDefault())
+                    .ExecuteAsync();
+            }
+            catch (MsalUiRequiredException ex)
+            {
+                var cts = new CancellationTokenSource();
+                cts.CancelAfter(60000);
+
+                return await application
+                    .AcquireTokenInteractive(scopes)
+                    .WithPrompt(Prompt.SelectAccount)
+                    .WithClaims(ex.Claims)
+                    .ExecuteAsync(cts.Token);
+            }
+        }
+    }
+}

CSharp/TfsCmdlets.Common/TfsCmdlets.Common.csproj

@@ -15,6 +15,11 @@
     <CompilerGeneratedFilesOutputPath>obj/Generated</CompilerGeneratedFilesOutputPath>
   </PropertyGroup>
 
+  <ItemGroup>
+    <!-- Source Generators -->
+    <ProjectReference Include="../TfsCmdlets.SourceGenerators/TfsCmdlets.SourceGenerators.csproj" OutputItemType="Analyzer" ReferenceOutputAssembly="false" />
+  </ItemGroup>
+
   <ItemGroup>
     <PackageReference Include="Microsoft.TeamFoundationServer.Client" Version="16.*-*" />
     <PackageReference Include="Microsoft.VisualStudio.Services.InteractiveClient" Version="16.*-*" />
@@ -28,6 +33,8 @@
     <PackageReference Include="System.Composition" Version="5.0.1" />
     <PackageReference Include="System.Composition.Hosting" Version="5.0.1" />
     <PackageReference Include="Microsoft.Win32.Registry" Version="5.*-*" />
+    <PackageReference Include="Microsoft.Identity.Client" Version="4.*" />
+    <PackageReference Include="Microsoft.Identity.Client.Desktop" Version="4.*" />
   </ItemGroup>
 
   <ItemGroup Condition="'$(TargetFramework)' == 'netcoreapp3.1'">
@@ -39,6 +46,12 @@
     <Compile Remove="**/*-core.cs" />
   </ItemGroup>
 
+  <ItemGroup>
+    <Compile Remove="Generated\**" />
+    <EmbeddedResource Remove="Generated\**" />
+    <None Remove="Generated\**" />
+  </ItemGroup>
+
   <ItemGroup>
     <AssemblyAttribute Include="System.Runtime.CompilerServices.InternalsVisibleToAttribute">
       <_Parameter1>TfsCmdlets</_Parameter1>

CSharp/TfsCmdlets.SourceGenerators/Generators/Cmdlets/Generator.cs

@@ -5,6 +5,6 @@ namespace TfsCmdlets.SourceGenerators.Generators.Cmdlets
     [Generator]
     public class CmdletGenerator : BaseGenerator<Filter, TypeProcessor>
     {
-        protected override string GeneratorName => "CmdletGenerator";
+        protected override string GeneratorName => nameof(CmdletGenerator);
     }
 }

CSharp/TfsCmdlets.SourceGenerators/Generators/Controllers/Generator.cs

@@ -3,8 +3,8 @@
 namespace TfsCmdlets.SourceGenerators.Generators.Controllers
 {
     [Generator]
-    public class CmdletGenerator : BaseGenerator<Filter, TypeProcessor>
+    public class ControllerGenerator : BaseGenerator<Filter, TypeProcessor>
     {
-        protected override string GeneratorName => "ControllerGenerator";
+        protected override string GeneratorName => nameof(ControllerGenerator);
     }
 }

CSharp/TfsCmdlets.SourceGenerators/Generators/Models/Filter.cs

@@ -4,11 +4,6 @@ namespace TfsCmdlets.SourceGenerators.Generators.Models
 {
     public class Filter : BaseFilter
     {
-        public override bool ShouldProcessType(INamedTypeSymbol type)
-        {
-            var baseClass = type.BaseType;
-
-            return baseClass != null && baseClass.FullName().StartsWith("TfsCmdlets.Models.ModelBase");
-        }
+        public override bool ShouldProcessType(INamedTypeSymbol type) => type.HasAttribute("ModelAttribute");
     }
 }