Authentication issue in AzurePowerShell task

Tasks/AzurePowerShellV5/AzurePowerShell.ps1

@@ -192,8 +192,9 @@ finally {
     if ($__vstsAzPSInlineScriptPath -and (Test-Path -LiteralPath $__vstsAzPSInlineScriptPath) ) {
         Remove-Item -LiteralPath $__vstsAzPSInlineScriptPath -ErrorAction 'SilentlyContinue'
     }
-
-    Import-Module $PSScriptRoot\ps_modules\VstsAzureHelpers_
+    . "$PSScriptRoot\Utility.ps1"
+    Import-Module "$PSScriptRoot\ps_modules\VstsAzureHelpers_"
     Remove-EndpointSecrets
-    Disconnect-AzureAndClearContext -ErrorAction SilentlyContinue
-}
+    Update-PSModulePathForHostedAgent
+    Disconnect-AzureAndClearContext -restrictContext 'True' -ErrorAction SilentlyContinue
+}

Tasks/AzurePowerShellV5/RemoveAzContext.ps1

@@ -0,0 +1,5 @@
+. "$PSScriptRoot/Utility.ps1"
+. "$PSScriptRoot/ps_modules/VstsAzureHelpers_/Utility.ps1"
+
+Update-PSModulePathForHostedAgentLinux
+Disconnect-AzureAndClearContext -restrictContext 'True'

Tasks/AzurePowerShellV5/azurepowershell.ts

@@ -12,6 +12,9 @@ function convertToNullIfUndefined<T>(arg: T): T|null {
 }
 
 async function run() {
+    let input_workingDirectory = tl.getPathInput('workingDirectory', /*required*/ true, /*check*/ true);
+    let tempDirectory = tl.getVariable('agent.tempDirectory');
+    tl.checkPath(tempDirectory, `${tempDirectory} (agent.tempDirectory)`);
     try {
         tl.setResourcePath(path.join(__dirname, 'task.json'));
 
@@ -35,7 +38,6 @@ async function run() {
         let customTargetAzurePs: string = convertToNullIfUndefined(tl.getInput('CustomTargetAzurePs', false));
         let serviceName = tl.getInput('ConnectedServiceNameARM',/*required*/true);
         let endpointObject= await new AzureRMEndpoint(serviceName).getEndpoint();
-        let input_workingDirectory = tl.getPathInput('workingDirectory', /*required*/ true, /*check*/ true);
         let isDebugEnabled = (process.env['SYSTEM_DEBUG'] || "").toLowerCase() === "true";
 
         // string constants
@@ -99,8 +101,6 @@ async function run() {
 
         // Write the script to disk.
         tl.assertAgent('2.115.0');
-        let tempDirectory = tl.getVariable('agent.tempDirectory');
-        tl.checkPath(tempDirectory, `${tempDirectory} (agent.tempDirectory)`);
         let filePath = path.join(tempDirectory, uuidV4() + '.ps1');
 
         await fs.writeFile(
@@ -160,6 +160,43 @@ async function run() {
     catch (err) {
         tl.setResult(tl.TaskResult.Failed, err.message || 'run() failed');
     }
+    finally {
+        let RemoveScriptContent: string[] = [];
+        let RemoveAzContextPath = path.join(path.resolve(__dirname), 'RemoveAzContext.ps1');
+        let removeScripts = `${RemoveAzContextPath} -ErrorAction continue`
+        RemoveScriptContent.push(removeScripts);
+        const removeScriptFilePath = path.join(tempDirectory, uuidV4() + '.ps1');
+        try {
+            await fs.writeFile(
+                removeScriptFilePath,
+                '\ufeff' + RemoveScriptContent.join(os.EOL), // Prepend the Unicode BOM character.
+                { encoding: 'utf8' }, // Since UTF8 encoding is specified, node will
+                                            // encode the BOM into its UTF8 binary sequence.
+                function (err) {
+                    if (err) throw err;
+                });
+            const powershell = tl.tool(tl.which('pwsh') || tl.which('powershell') || tl.which('pwsh', true))
+                .arg('-NoLogo')
+                .arg('-NoProfile')
+                .arg('-NonInteractive')
+                .arg('-ExecutionPolicy')
+                .arg('Unrestricted')
+                .arg('-Command')
+                .arg(`. '${removeScriptFilePath.replace(/'/g, "''")}'`);
+
+            let options = <tr.IExecOptions>{
+                    cwd: input_workingDirectory,
+                    failOnStdErr: false,
+                    errStream: process.stdout, // Direct all output to STDOUT, otherwise the output may appear out
+                    outStream: process.stdout, // of order since Node buffers it's own STDOUT but not STDERR.
+                    ignoreReturnCode: true
+                };
+            await powershell.exec(options);
+        }
+        catch (err) {
+            tl.debug("Az-clearContext not completed due to an error");
+        }
+    }
 }
 
 run();

Tasks/AzurePowerShellV5/task.json

@@ -17,7 +17,7 @@
   "author": "Microsoft Corporation",
   "version": {
     "Major": 5,
-    "Minor": 234,
+    "Minor": 235,
     "Patch": 0
   },
   "releaseNotes": "Added support for Az Module and cross platform agents.",

Tasks/AzurePowerShellV5/task.loc.json

@@ -17,7 +17,7 @@
   "author": "Microsoft Corporation",
   "version": {
     "Major": 5,
-    "Minor": 234,
+    "Minor": 235,
     "Patch": 0
   },
   "releaseNotes": "ms-resource:loc.releaseNotes",

_generated/AzurePowerShellV5.versionmap.txt

@@ -1,2 +1,2 @@
-Default|5.234.0
-Node20_229_2|5.234.1
+Default|5.235.0
+Node20_229_2|5.235.1

_generated/AzurePowerShellV5/AzurePowerShell.ps1

@@ -192,8 +192,9 @@ finally {
     if ($__vstsAzPSInlineScriptPath -and (Test-Path -LiteralPath $__vstsAzPSInlineScriptPath) ) {
         Remove-Item -LiteralPath $__vstsAzPSInlineScriptPath -ErrorAction 'SilentlyContinue'
     }
-
-    Import-Module $PSScriptRoot\ps_modules\VstsAzureHelpers_
+    . "$PSScriptRoot\Utility.ps1"
+    Import-Module "$PSScriptRoot\ps_modules\VstsAzureHelpers_"
     Remove-EndpointSecrets
-    Disconnect-AzureAndClearContext -ErrorAction SilentlyContinue
-}
+    Update-PSModulePathForHostedAgent
+    Disconnect-AzureAndClearContext -restrictContext 'True' -ErrorAction SilentlyContinue
+}

_generated/AzurePowerShellV5/RemoveAzContext.ps1

@@ -0,0 +1,5 @@
+. "$PSScriptRoot/Utility.ps1"
+. "$PSScriptRoot/ps_modules/VstsAzureHelpers_/Utility.ps1"
+
+Update-PSModulePathForHostedAgentLinux
+Disconnect-AzureAndClearContext -restrictContext 'True'

_generated/AzurePowerShellV5/azurepowershell.ts

@@ -12,6 +12,9 @@ function convertToNullIfUndefined<T>(arg: T): T|null {
 }
 
 async function run() {
+    let input_workingDirectory = tl.getPathInput('workingDirectory', /*required*/ true, /*check*/ true);
+    let tempDirectory = tl.getVariable('agent.tempDirectory');
+    tl.checkPath(tempDirectory, `${tempDirectory} (agent.tempDirectory)`);
     try {
         tl.setResourcePath(path.join(__dirname, 'task.json'));
 
@@ -35,7 +38,6 @@ async function run() {
         let customTargetAzurePs: string = convertToNullIfUndefined(tl.getInput('CustomTargetAzurePs', false));
         let serviceName = tl.getInput('ConnectedServiceNameARM',/*required*/true);
         let endpointObject= await new AzureRMEndpoint(serviceName).getEndpoint();
-        let input_workingDirectory = tl.getPathInput('workingDirectory', /*required*/ true, /*check*/ true);
         let isDebugEnabled = (process.env['SYSTEM_DEBUG'] || "").toLowerCase() === "true";
 
         // string constants
@@ -99,8 +101,6 @@ async function run() {
 
         // Write the script to disk.
         tl.assertAgent('2.115.0');
-        let tempDirectory = tl.getVariable('agent.tempDirectory');
-        tl.checkPath(tempDirectory, `${tempDirectory} (agent.tempDirectory)`);
         let filePath = path.join(tempDirectory, uuidV4() + '.ps1');
 
         await fs.writeFile(
@@ -160,6 +160,43 @@ async function run() {
     catch (err) {
         tl.setResult(tl.TaskResult.Failed, err.message || 'run() failed');
     }
+    finally {
+        let RemoveScriptContent: string[] = [];
+        let RemoveAzContextPath = path.join(path.resolve(__dirname), 'RemoveAzContext.ps1');
+        let removeScripts = `${RemoveAzContextPath} -ErrorAction continue`
+        RemoveScriptContent.push(removeScripts);
+        const removeScriptFilePath = path.join(tempDirectory, uuidV4() + '.ps1');
+        try {
+            await fs.writeFile(
+                removeScriptFilePath,
+                '\ufeff' + RemoveScriptContent.join(os.EOL), // Prepend the Unicode BOM character.
+                { encoding: 'utf8' }, // Since UTF8 encoding is specified, node will
+                                            // encode the BOM into its UTF8 binary sequence.
+                function (err) {
+                    if (err) throw err;
+                });
+            const powershell = tl.tool(tl.which('pwsh') || tl.which('powershell') || tl.which('pwsh', true))
+                .arg('-NoLogo')
+                .arg('-NoProfile')
+                .arg('-NonInteractive')
+                .arg('-ExecutionPolicy')
+                .arg('Unrestricted')
+                .arg('-Command')
+                .arg(`. '${removeScriptFilePath.replace(/'/g, "''")}'`);
+
+            let options = <tr.IExecOptions>{
+                    cwd: input_workingDirectory,
+                    failOnStdErr: false,
+                    errStream: process.stdout, // Direct all output to STDOUT, otherwise the output may appear out
+                    outStream: process.stdout, // of order since Node buffers it's own STDOUT but not STDERR.
+                    ignoreReturnCode: true
+                };
+            await powershell.exec(options);
+        }
+        catch (err) {
+            tl.debug("Az-clearContext not completed due to an error");
+        }
+    }
 }
 
 run();

_generated/AzurePowerShellV5/task.json

@@ -17,7 +17,7 @@
   "author": "Microsoft Corporation",
   "version": {
     "Major": 5,
-    "Minor": 234,
+    "Minor": 235,
     "Patch": 0
   },
   "releaseNotes": "Added support for Az Module and cross platform agents.",
@@ -204,7 +204,7 @@
     "PS_ExitCode": "PowerShell exited with code '{0}'."
   },
   "_buildConfigMapping": {
-    "Default": "5.234.0",
-    "Node20_229_2": "5.234.1"
+    "Default": "5.235.0",
+    "Node20_229_2": "5.235.1"
   }
 }

_generated/AzurePowerShellV5/task.loc.json

@@ -17,7 +17,7 @@
   "author": "Microsoft Corporation",
   "version": {
     "Major": 5,
-    "Minor": 234,
+    "Minor": 235,
     "Patch": 0
   },
   "releaseNotes": "ms-resource:loc.releaseNotes",
@@ -204,7 +204,7 @@
     "PS_ExitCode": "ms-resource:loc.messages.PS_ExitCode"
   },
   "_buildConfigMapping": {
-    "Default": "5.234.0",
-    "Node20_229_2": "5.234.1"
+    "Default": "5.235.0",
+    "Node20_229_2": "5.235.1"
   }
 }

_generated/AzurePowerShellV5_Node20/AzurePowerShell.ps1

@@ -192,8 +192,9 @@ finally {
     if ($__vstsAzPSInlineScriptPath -and (Test-Path -LiteralPath $__vstsAzPSInlineScriptPath) ) {
         Remove-Item -LiteralPath $__vstsAzPSInlineScriptPath -ErrorAction 'SilentlyContinue'
     }
-
-    Import-Module $PSScriptRoot\ps_modules\VstsAzureHelpers_
+    . "$PSScriptRoot\Utility.ps1"
+    Import-Module "$PSScriptRoot\ps_modules\VstsAzureHelpers_"
     Remove-EndpointSecrets
-    Disconnect-AzureAndClearContext -ErrorAction SilentlyContinue
-}
+    Update-PSModulePathForHostedAgent
+    Disconnect-AzureAndClearContext -restrictContext 'True' -ErrorAction SilentlyContinue
+}

_generated/AzurePowerShellV5_Node20/RemoveAzContext.ps1

@@ -0,0 +1,5 @@
+. "$PSScriptRoot/Utility.ps1"
+. "$PSScriptRoot/ps_modules/VstsAzureHelpers_/Utility.ps1"
+
+Update-PSModulePathForHostedAgentLinux
+Disconnect-AzureAndClearContext -restrictContext 'True'

_generated/AzurePowerShellV5_Node20/azurepowershell.ts

@@ -12,6 +12,9 @@ function convertToNullIfUndefined<T>(arg: T): T|null {
 }
 
 async function run() {
+    let input_workingDirectory = tl.getPathInput('workingDirectory', /*required*/ true, /*check*/ true);
+    let tempDirectory = tl.getVariable('agent.tempDirectory');
+    tl.checkPath(tempDirectory, `${tempDirectory} (agent.tempDirectory)`);
     try {
         tl.setResourcePath(path.join(__dirname, 'task.json'));
 
@@ -35,7 +38,6 @@ async function run() {
         let customTargetAzurePs: string = convertToNullIfUndefined(tl.getInput('CustomTargetAzurePs', false));
         let serviceName = tl.getInput('ConnectedServiceNameARM',/*required*/true);
         let endpointObject= await new AzureRMEndpoint(serviceName).getEndpoint();
-        let input_workingDirectory = tl.getPathInput('workingDirectory', /*required*/ true, /*check*/ true);
         let isDebugEnabled = (process.env['SYSTEM_DEBUG'] || "").toLowerCase() === "true";
 
         // string constants
@@ -99,8 +101,6 @@ async function run() {
 
         // Write the script to disk.
         tl.assertAgent('2.115.0');
-        let tempDirectory = tl.getVariable('agent.tempDirectory');
-        tl.checkPath(tempDirectory, `${tempDirectory} (agent.tempDirectory)`);
         let filePath = path.join(tempDirectory, uuidV4() + '.ps1');
 
         await fs.writeFile(
@@ -160,6 +160,43 @@ async function run() {
     catch (err) {
         tl.setResult(tl.TaskResult.Failed, err.message || 'run() failed');
     }
+    finally {
+        let RemoveScriptContent: string[] = [];
+        let RemoveAzContextPath = path.join(path.resolve(__dirname), 'RemoveAzContext.ps1');
+        let removeScripts = `${RemoveAzContextPath} -ErrorAction continue`
+        RemoveScriptContent.push(removeScripts);
+        const removeScriptFilePath = path.join(tempDirectory, uuidV4() + '.ps1');
+        try {
+            await fs.writeFile(
+                removeScriptFilePath,
+                '\ufeff' + RemoveScriptContent.join(os.EOL), // Prepend the Unicode BOM character.
+                { encoding: 'utf8' }, // Since UTF8 encoding is specified, node will
+                                            // encode the BOM into its UTF8 binary sequence.
+                function (err) {
+                    if (err) throw err;
+                });
+            const powershell = tl.tool(tl.which('pwsh') || tl.which('powershell') || tl.which('pwsh', true))
+                .arg('-NoLogo')
+                .arg('-NoProfile')
+                .arg('-NonInteractive')
+                .arg('-ExecutionPolicy')
+                .arg('Unrestricted')
+                .arg('-Command')
+                .arg(`. '${removeScriptFilePath.replace(/'/g, "''")}'`);
+
+            let options = <tr.IExecOptions>{
+                    cwd: input_workingDirectory,
+                    failOnStdErr: false,
+                    errStream: process.stdout, // Direct all output to STDOUT, otherwise the output may appear out
+                    outStream: process.stdout, // of order since Node buffers it's own STDOUT but not STDERR.
+                    ignoreReturnCode: true
+                };
+            await powershell.exec(options);
+        }
+        catch (err) {
+            tl.debug("Az-clearContext not completed due to an error");
+        }
+    }
 }
 
 run();

_generated/AzurePowerShellV5_Node20/task.json

@@ -17,7 +17,7 @@
   "author": "Microsoft Corporation",
   "version": {
     "Major": 5,
-    "Minor": 234,
+    "Minor": 235,
     "Patch": 1
   },
   "releaseNotes": "Added support for Az Module and cross platform agents.",
@@ -208,7 +208,7 @@
     "PS_ExitCode": "PowerShell exited with code '{0}'."
   },
   "_buildConfigMapping": {
-    "Default": "5.234.0",
-    "Node20_229_2": "5.234.1"
+    "Default": "5.235.0",
+    "Node20_229_2": "5.235.1"
   }
 }

_generated/AzurePowerShellV5_Node20/task.loc.json

@@ -17,7 +17,7 @@
   "author": "Microsoft Corporation",
   "version": {
     "Major": 5,
-    "Minor": 234,
+    "Minor": 235,
     "Patch": 1
   },
   "releaseNotes": "ms-resource:loc.releaseNotes",
@@ -208,7 +208,7 @@
     "PS_ExitCode": "ms-resource:loc.messages.PS_ExitCode"
   },
   "_buildConfigMapping": {
-    "Default": "5.234.0",
-    "Node20_229_2": "5.234.1"
+    "Default": "5.235.0",
+    "Node20_229_2": "5.235.1"
   }
 }