Merge pull request #57 from teamviewer/Warnings_Fix

PSShouldProcess warning fix
teamviewer · Oct 17, 2023 · 6c0ba45 · 6c0ba45
2 parents 41a1bd5 + b231eaf
commit 6c0ba45
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/Cmdlets/Public/Remove-TeamViewerAssignment.ps1 b/Cmdlets/Public/Remove-TeamViewerAssignment.ps1
@@ -16,9 +16,11 @@ function Remove-TeamViewerAssignment {
             $cmd = 'teamviewer unassign'
             $FilePath = 'sudo'
         }
-        $process = Start-Process -FilePath $FilePath -ArgumentList $cmd -Wait -PassThru
-        $process.ExitCode | Resolve-AssignmentErrorCode
-        Set-Location $CurrentDirectory
+        if ($PSCmdlet.ShouldProcess($installationDirectory, 'Remove device assignment')) {
+            $process = Start-Process -FilePath $FilePath -ArgumentList $cmd -Wait -PassThru
+            $process.ExitCode | Resolve-AssignmentErrorCode
+            Set-Location $CurrentDirectory
+        }
     }
     else {
         Write-Output 'TeamViewer is not installed.'

diff --git a/Cmdlets/Public/Remove-TeamViewerCustomization.ps1 b/Cmdlets/Public/Remove-TeamViewerCustomization.ps1
@@ -8,8 +8,10 @@ function Remove-TeamViewerCustomization {
             $currentDirectory = Get-Location
             Set-Location $installationDirectory
             $cmd = 'customize --remove'
-            $process = Start-Process -FilePath TeamViewer.exe -ArgumentList $cmd -Wait -PassThru
-            $process.ExitCode | Resolve-CustomizationErrorCode
+            if ($PSCmdlet.ShouldProcess($installationDirectory, 'Remove Client Customization')) {
+                $process = Start-Process -FilePath TeamViewer.exe -ArgumentList $cmd -Wait -PassThru
+                $process.ExitCode | Resolve-CustomizationErrorCode
+            }
             Set-Location $currentDirectory
         }
         else {