Quote tab completion items with PS special chars.

Fix #293
dahlbyk · Feb 12, 2017 · 4fb926e · 4fb926e
1 parent ca7db50
commit 4fb926e
Show file tree

Hide file tree

Showing 2 changed files with 80 additions and 12 deletions.
diff --git a/src/GitTabExpansion.ps1 b/src/GitTabExpansion.ps1
@@ -51,6 +51,15 @@ catch {
     Write-Debug "Search for 'flow' in 'git help' output failed with error: $_"
 }
 
+filter quoteStringWithSpecialChars {
+    if ($_ -and ($_ -match '\s+|#|@|\$|;|\{|\}|\(|\)')) {
+        "'" + $_ + "'"
+    }
+    else {
+        $_
+    }
+}
+
 function script:gitCommands($filter, $includeAliases) {
     $cmdList = @()
     if (-not $global:GitTabSettings.AllCommands) {
@@ -86,7 +95,8 @@ function script:gitBranches($filter, $includeHEAD = $false, $prefix = '') {
 
     $branches |
         Where-Object { $_ -ne '(no branch)' -and $_ -like "$filter*" } |
-        ForEach-Object { $prefix + $_ }
+        ForEach-Object { $prefix + $_ } |
+        quoteStringWithSpecialChars
 }
 
 function script:gitRemoteUniqueBranches($filter) {
@@ -95,27 +105,31 @@ function script:gitRemoteUniqueBranches($filter) {
         Group-Object -NoElement |
         Where-Object { $_.Count -eq 1 } |
         Select-Object -ExpandProperty Name |
-        Where-Object { $_ -like "$filter*" }
+        Where-Object { $_ -like "$filter*" } |
+        quoteStringWithSpecialChars
 }
 
 function script:gitTags($filter, $prefix = '') {
     git tag |
         Where-Object { $_ -like "$filter*" } |
-        ForEach-Object { $prefix + $_ }
+        ForEach-Object { $prefix + $_ } |
+        quoteStringWithSpecialChars
 }
 
 function script:gitFeatures($filter, $command){
     $featurePrefix = git config --local --get "gitflow.prefix.$command"
     $branches = @(git branch --no-color | ForEach-Object { if ($_ -match "^\*?\s*$featurePrefix(?<ref>.*)") { $matches['ref'] } })
     $branches |
         Where-Object { $_ -ne '(no branch)' -and $_ -like "$filter*" } |
-        ForEach-Object { $prefix + $_ }
+        ForEach-Object { $prefix + $_ } |
+        quoteStringWithSpecialChars
 }
 
 function script:gitRemoteBranches($remote, $ref, $filter, $prefix = '') {
     git branch --no-color -r |
         Where-Object { $_ -like "  $remote/$filter*" } |
-        ForEach-Object { $prefix + $ref + ($_ -replace "  $remote/","") }
+        ForEach-Object { $prefix + $ref + ($_ -replace "  $remote/","") } |
+        quoteStringWithSpecialChars
 }
 
 function script:gitStashes($filter) {
@@ -133,7 +147,7 @@ function script:gitTfsShelvesets($filter) {
 function script:gitFiles($filter, $files) {
     $files | Sort-Object |
         Where-Object { $_ -like "$filter*" } |
-        ForEach-Object { if ($_ -like '* *') { "'$_'" } else { $_ } }
+        quoteStringWithSpecialChars
 }
 
 function script:gitIndex($GitStatus, $filter) {

diff --git a/test/TabExpansion.Tests.ps1 b/test/TabExpansion.Tests.ps1
@@ -127,7 +127,7 @@ Describe 'TabExpansion Tests' {
             $temp = [System.IO.Path]::GetTempPath()
             $repoPath = Join-Path $temp ([IO.Path]::GetRandomFileName())
 
-            git init $repoPath
+            git.exe init $repoPath
             Set-Location $repoPath
         }
         AfterEach {
@@ -137,16 +137,70 @@ Describe 'TabExpansion Tests' {
             }
         }
         It 'Tab completes non-ASCII file name' {
-            git config core.quotepath true # Problematic (default) config
+            git.exe config core.quotepath true # Problematic (default) config
 
             $fileName = "posh$([char]8226)git.txt"
-            New-Item $fileName
+            New-Item $fileName -ItemType File
 
-            $GitStatus = & $module Get-GitStatus
-
-            $result = & $module GitTabExpansionInternal 'git add ' $GitStatus
+            $gitStatus = & $module Get-GitStatus
 
+            $result = & $module GitTabExpansionInternal 'git add ' $gitStatus
             $result | Should BeExactly $fileName
         }
     }
+
+    Context 'PowerShell Special Chars Tests' {
+        BeforeAll {
+            [System.Diagnostics.CodeAnalysis.SuppressMessageAttribute('PSUseDeclaredVarsMoreThanAssigments', '')]
+            $origPath = Get-Location
+            $temp = [System.IO.Path]::GetTempPath()
+            $repoPath = Join-Path $temp ([IO.Path]::GetRandomFileName())
+
+            git.exe init $repoPath
+            Set-Location $repoPath
+
+            'readme' | Out-File .\README.md -Encoding ascii
+            git.exe add .\README.md
+            git.exe commit -m "initial commit."
+        }
+        AfterAll {
+            Set-Location $origPath
+            if (Test-Path $repoPath) {
+                Remove-Item $repoPath -Recurse -Force
+            }
+        }
+        AfterEach {
+            Set-Location $repoPath
+            git.exe reset HEAD --hard
+            git.exe checkout master 2>$null
+        }
+        It 'Tab completes branch name with special char as quoted' {
+            git.exe branch '#develop' 2>$null
+
+            $result = & $module GitTabExpansionInternal 'git checkout #'
+            $result | Should BeExactly "'#develop'"
+        }
+        It 'Tab completes git feature branch name with special char as quoted' {
+            git.exe branch '#develop' 2>$null
+
+            $result = & $module GitTabExpansionInternal 'git flow feature list #'
+            $result | Should BeExactly "'#develop'"
+        }
+        It 'Tab completes a tag name with special char as quoted' {
+            $tag = "v1.0.0;abcdef"
+            git.exe tag $tag
+
+            $result = & $module GitTabExpansionInternal 'git show v1'
+            $result | Should BeExactly "'$tag'"
+        }
+        It 'Tab completes add file in working dir with special char as quoted' {
+            $filename = 'foo{bar} (x86).txt';
+            New-Item $filename -ItemType File
+
+            $gitStatus = & $module Get-GitStatus
+
+            $result = & $module GitTabExpansionInternal 'git add ' $gitStatus
+            $result | Should BeExactly "'$filename'"
+        }
+    }
 }