Support for PowerShell cmdlets in admin mode

.gitignore

@@ -329,3 +329,8 @@ ASALocalRun/
 
 # MFractors (Xamarin productivity tool) working folder 
 .mfractor/
+
+# Generated files from WinGetServer.idl
+**/WinGetServer/WinGetServer.h
+**/WinGetServer/WinGetServer_c.c
+**/WinGetServer/WinGetServer_s.c

src/AppInstallerCLIE2ETests/Constants.cs

@@ -17,6 +17,7 @@ public class Constants
         public const string MsiInstallerPathParameter = "MsiTestInstallerPath";
         public const string MsixInstallerPathParameter = "MsixTestInstallerPath";
         public const string PackageCertificatePathParameter = "PackageCertificatePath";
+        public const string PowerShellModulePathParameter = "PowerShellModulePath";
         public const string AppInstallerTestCert = "AppInstallerTest.cer";
         public const string AppInstallerTestCertThumbprint = "d03e7a688b388b1edde8476a627531c49db88017";
 
@@ -74,6 +75,14 @@ public class Constants
         public const string TestExeUninstallerFileName = "UninstallTestExe.bat";
         public const string TestExeUninstalledFileName = "TestExeUninstalled.txt";
 
+        // PowerShell Cmdlets
+        public const string FindCmdlet = "Find-WinGetPackage";
+        public const string GetCmdlet = "Get-WinGetPackage";
+        public const string GetSourceCmdlet = "Get-WinGetSource";
+        public const string InstallCmdlet = "Install-WinGetPackage";
+        public const string UninstallCmdlet = "Uninstall-WinGetPackage";
+        public const string UpdateCmdlet = "Update-WinGetPackage";
+
         // Locations
         public const string LocalAppData = "LocalAppData";
 

src/AppInstallerCLIE2ETests/PowerShell/PowerShellModule.cs

@@ -0,0 +1,91 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+namespace AppInstallerCLIE2ETests
+{
+    using NUnit.Framework;
+
+    /// <summary>
+    /// Basic E2E tests for verifying that behavior of the PowerShell module cmdlets.
+    /// </summary>
+    public class PowerShellModule
+    {
+        [OneTimeSetUp]
+        public void Setup()
+        {
+            // Add-WinGetPackage is a function and not a cmdlet that uses COM. Add source to WinGetDev directly to ensure test source exists.
+            TestCommon.RunAICLICommand("source add", $"-n {Constants.TestSourceName} {Constants.TestSourceUrl}");
+        }
+
+        [OneTimeTearDown]
+        public void TearDown()
+        {
+            // Remove-WinGetPackage is a function and not a cmdlet that uses COM. Remove source from WinGetDev directory to ensure clean state.
+            TestCommon.RunAICLICommand("source remove", $"-n {Constants.TestSourceName}");
+        }
+
+        [Test]
+        public void GetWinGetSource()
+        {
+            var getSourceResult = TestCommon.RunPowerShellCommandWithResult(Constants.GetSourceCmdlet, $"-Name {Constants.TestSourceName}");
+            Assert.IsTrue(getSourceResult.ExitCode == 0);
+            Assert.IsTrue(getSourceResult.StdOut.Contains($"{Constants.TestSourceName}"));
+        }
+
+        [Test]
+        public void FindWinGetPackage()
+        {
+            var result = TestCommon.RunPowerShellCommandWithResult(Constants.FindCmdlet, $"-Id {Constants.ExeInstallerPackageId}");
+            Assert.IsTrue(result.ExitCode == 0);
+            Assert.IsTrue(result.StdOut.Contains("TestExeInstaller"));
+        }
+
+        [Test]
+        public void GetWinGetPackage()
+        {
+            var installResult = TestCommon.RunPowerShellCommandWithResult(Constants.InstallCmdlet, $"-Id {Constants.MsiInstallerPackageId}");
+            var getResult = TestCommon.RunPowerShellCommandWithResult(Constants.GetCmdlet, $"-Id {Constants.MsiInstallerPackageId}");
+            var uninstallResult = TestCommon.RunPowerShellCommandWithResult(Constants.UninstallCmdlet, $"-Id {Constants.MsiInstallerPackageId}");
+
+            Assert.IsTrue(installResult.ExitCode == 0);
+            Assert.IsTrue(getResult.ExitCode == 0);
+            Assert.IsTrue(uninstallResult.ExitCode == 0);
+
+            Assert.IsTrue(!string.IsNullOrEmpty(installResult.StdOut));
+            Assert.IsTrue(getResult.StdOut.Contains("TestMsiInstaller"));
+            Assert.IsTrue(!string.IsNullOrEmpty(uninstallResult.StdOut));
+        }
+
+        [Test]
+        public void InstallWinGetPackage()
+        {
+            var installResult = TestCommon.RunPowerShellCommandWithResult(Constants.InstallCmdlet, $"-Id {Constants.ExeInstallerPackageId}");
+            var uninstallResult = TestCommon.RunPowerShellCommandWithResult(Constants.UninstallCmdlet, $"-Id {Constants.ExeInstallerPackageId}");
+
+            Assert.IsTrue(installResult.ExitCode == 0);
+            Assert.IsTrue(uninstallResult.ExitCode == 0);
+
+            Assert.IsTrue(!string.IsNullOrEmpty(installResult.StdOut));
+            Assert.IsTrue(!string.IsNullOrEmpty(uninstallResult.StdOut));
+        }
+
+        [Test]
+        public void UpdateWinGetPackage()
+        {
+            var installResult = TestCommon.RunPowerShellCommandWithResult(Constants.InstallCmdlet, $"-Id {Constants.ExeInstallerPackageId} -Version 1.0.0.0");
+            var updateResult = TestCommon.RunPowerShellCommandWithResult(Constants.UpdateCmdlet, $"-Id {Constants.ExeInstallerPackageId}");
+            var getResult = TestCommon.RunPowerShellCommandWithResult(Constants.GetCmdlet, $"-Id {Constants.ExeInstallerPackageId}");
+            var uninstallResult = TestCommon.RunPowerShellCommandWithResult(Constants.UninstallCmdlet, $"-Id {Constants.ExeInstallerPackageId}");
+
+            Assert.IsTrue(installResult.ExitCode == 0);
+            Assert.IsTrue(updateResult.ExitCode == 0);
+            Assert.IsTrue(getResult.ExitCode == 0);
+            Assert.IsTrue(uninstallResult.ExitCode == 0);
+
+            Assert.IsTrue(!string.IsNullOrEmpty(installResult.StdOut));
+            Assert.IsTrue(!string.IsNullOrEmpty(updateResult.StdOut));
+            Assert.IsTrue(getResult.StdOut.Contains("2.0.0.0"));
+            Assert.IsTrue(!string.IsNullOrEmpty(uninstallResult.StdOut));
+        }
+    }
+}

src/AppInstallerCLIE2ETests/README.md

@@ -56,7 +56,8 @@ Therefore to run the executable in the command line, simply change into the dire
 | StaticFileRootPath | Path to the set of static test files that will be served as the source for testing purposes. This path should be identical to the one provided to the LocalHostWebServer|
 | MsixTestInstallerPath | The MSIX (or APPX) Installer executable under test. |
 | ExeTestInstallerPath |The Exe Installer executable under test. |
-| PackageCertificatePath |Signing Certificate Path used to certify test index source package|
+| PackageCertificatePath | Signing Certificate Path used to certify test index source package |
+| PowerShellModulePath | Path to the PowerShell module manifest file under test |
 
 #### Example of Test.runsettings format:
 
@@ -72,6 +73,7 @@ Therefore to run the executable in the command line, simply change into the dire
 			<Parameter name="MsixTestInstallerPath" value="MsixTestInstaller.msix" />
 			<Parameter name="ExeTestInstallerPath" value="ExeTestInstaller.exe" />
 			<Parameter name="PackageCertificatePath" value="certificate.pfx"/>
+			<Parameter name="PowerShellModulePath" value="TestPowerShellModule.psd1" />
 		</TestRunParameters>
 	</RunSettings>
 
@@ -90,6 +92,7 @@ Make sure to replace **MSFT** with your own user name. Modifying this example wi
 			<Parameter name="MsixTestInstallerPath" value="C:\Users\MSFT\Temp\MsixTestInstaller.msix" />
 			<Parameter name="ExeTestInstallerPath" value="C:\Users\MSFT\source\repos\winget-cli\src\x64\Debug\AppInstallerTestExeInstaller\AppInstallerTestExeInstaller.exe" />
 			<Parameter name="PackageCertificatePath" value="C:\Users\MSFT\Temp\packageCertificate.pfx"/>
+			<Parameter name="PowerShellModulePath" value="C:\Users\MSFT\source\repos\winget-cli\src\x64\Debug\PowerShell\Microsoft.WinGet.Client.psd1" />
 		</TestRunParameters>
 	</RunSettings>
 

src/AppInstallerCLIE2ETests/SetUpFixture.cs

@@ -96,6 +96,11 @@ public void Setup()
                 TestCommon.PackageCertificatePath = TestContext.Parameters.Get(Constants.PackageCertificatePathParameter);
             }
 
+            if (TestContext.Parameters.Exists(Constants.PowerShellModulePathParameter))
+            {
+                TestCommon.PowerShellModulePath = TestContext.Parameters.Get(Constants.PowerShellModulePathParameter);
+            }
+
             ReadTestInstallerPaths();
 
             TestIndexSetup.GenerateTestDirectory();

src/AppInstallerCLIE2ETests/Test.runsettings

@@ -19,6 +19,7 @@
 	   MsixTestInstallerPath: The MSIX (or APPX) Installer executable under test.
 	   ExeTestInstallerPath: The Exe Installer executable under test.
 	   PackageCertificatePath: Signing Certificate Path used to certify Index Source Package
+	   PowerShellModulePath: Path to the PowerShell module manifest file under test
   -->
   <TestRunParameters>
 	  <Parameter name="PackagedContext" value="true" />
@@ -32,5 +33,6 @@
 	  <Parameter name="MsixTestInstallerPath" value="MsixTestInstaller.msix" />
 	  <Parameter name="ExeTestInstallerPath" value="ExeTestInstaller.exe" />
 	  <Parameter name="PackageCertificatePath" value="certificate.pfx"/>
+      <Parameter name="PowerShellModulePath" value="TestPowerShellModule.psd1" />
   </TestRunParameters>
 </RunSettings>

src/AppInstallerCLIE2ETests/TestCommon.cs

@@ -36,6 +36,8 @@ public class TestCommon
 
         public static string PackageCertificatePath { get; set; }
 
+        public static string PowerShellModulePath { get; set; }
+
         public static string SettingsJsonFilePath {
             get
             {
@@ -252,6 +254,11 @@ public static RunCommandResult RunCommandWithResult(string fileName, string args
             return result;
         }
 
+        public static RunCommandResult RunPowerShellCommandWithResult(string cmdlet, string args, int timeOut = 60000)
+        {
+            return RunCommandWithResult("pwsh", $"-Command ipmo {PowerShellModulePath}; {cmdlet} {args}", timeOut);
+        }
+
         public static string GetTestFile(string fileName)
         {
             return Path.Combine(TestContext.CurrentContext.TestDirectory, fileName);

src/AppInstallerCLIPackage/AppInstallerCLIPackage.wapproj

@@ -54,7 +54,7 @@
     <TargetPlatformVersion>10.0.22000.0</TargetPlatformVersion>
     <TargetPlatformMinVersion>10.0.17763.0</TargetPlatformMinVersion>
     <DefaultLanguage>en-US</DefaultLanguage>
-    <AppxPackageSigningEnabled>false</AppxPackageSigningEnabled>
+    <AppxPackageSigningEnabled>False</AppxPackageSigningEnabled>
     <EntryPointProjectUniqueName>..\AppInstallerCLI\AppInstallerCLI.vcxproj</EntryPointProjectUniqueName>
   </PropertyGroup>
   <ItemGroup>

src/AppInstallerCLIPackage/Package.appxmanifest

@@ -37,6 +37,16 @@
             <uap5:ExecutionAlias Alias="WinGetDev.exe" />
           </uap5:AppExecutionAlias>
         </uap5:Extension>
+      </Extensions>
+    </Application>
+    <Application Id="WinGetComServer" Executable="WinGetServer\WindowsPackageManagerServer.exe" EntryPoint="Windows.FullTrustApplication">
+      <uap:VisualElements DisplayName="WinGet Dev COM Server" Square150x150Logo="Images\Square150x150Logo.png" Square44x44Logo="Images\Square44x44Logo.png" Description="The WinGet COM server." BackgroundColor="#0078d7" AppListEntry="none" />
+      <Extensions>
+        <uap5:Extension Category="windows.appExecutionAlias">
+          <uap5:AppExecutionAlias>
+            <uap5:ExecutionAlias Alias="WindowsPackageManagerServerDev.exe" />
+          </uap5:AppExecutionAlias>
+        </uap5:Extension>          
         <com:Extension Category="windows.comServer">
           <com:ComServer>
             <com:ExeServer Executable="WinGetServer\WindowsPackageManagerServer.exe" DisplayName="Windows Package Manager Server" LaunchAndActivationPermission="O:SYG:SYD:(A;;11;;;WD)(A;;11;;;RC)(A;;11;;;AC)(A;;11;;;AN)S:P(ML;;NX;;;S-1-16-0)">

src/PowerShell/Microsoft.WinGet.Client/Commands/GetSourceCommand.cs

@@ -6,7 +6,6 @@
 
 namespace Microsoft.WinGet.Client.Commands
 {
-    using System;
     using System.Management.Automation;
     using Microsoft.Management.Deployment;
     using Microsoft.WinGet.Client.Common;

src/PowerShell/Microsoft.WinGet.Client/Common/BaseClientCommand.cs

@@ -29,9 +29,9 @@ static BaseClientCommand()
         public BaseClientCommand()
             : base()
         {
-            if (Utilities.ExecutingAsAdministrator)
+            if (Utilities.ExecutingAsSystem)
             {
-                throw new Exception(Utilities.ResourceManager.GetString("ExceptionAdministratorDisabled"));
+                throw new Exception(Utilities.ResourceManager.GetString("ExceptionSystemDisabled"));
             }
         }
 

src/PowerShell/Microsoft.WinGet.Client/Common/Utilities.cs

@@ -37,5 +37,18 @@ public static bool ExecutingAsAdministrator
                 return principal.IsInRole(WindowsBuiltInRole.Administrator);
             }
         }
+
+        /// <summary>
+        /// Gets a value indicating whether the current assembly is executing as a SYSTEM user.
+        /// </summary>
+        public static bool ExecutingAsSystem
+        {
+            get
+            {
+                WindowsIdentity identity = WindowsIdentity.GetCurrent();
+                WindowsPrincipal principal = new (identity);
+                return principal.IsInRole(WindowsBuiltInRole.SystemOperator);
+            }
+        }
     }
 }