feat🆕 (Invoke-GitCheckout): support for searching branch before checkout

capdiem · Jun 27, 2023 · 26f59d6 · 26f59d6
1 parent 901f85c
commit 26f59d6
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/PSGitUtils/PSGitUtils.psd1 b/PSGitUtils/PSGitUtils.psd1
@@ -12,7 +12,7 @@
   RootModule        = 'PSGitUtils.psm1'
 
   # Version number of this module.
-  ModuleVersion     = '1.14.2'
+  ModuleVersion     = '1.15.0'
 
   # Supported PSEditions
   # CompatiblePSEditions = @()

diff --git a/PSGitUtils/PSGitUtils.psm1 b/PSGitUtils/PSGitUtils.psm1
@@ -254,14 +254,19 @@ function Get-OptionsForChoosingLocalBranch {
 function Get-OptionsForChoosingLocalOrOriginBranch {
   param (
       [string]$title = 'Choose branch',
-      [string]$message = 'Please choose a branch'
+      [string]$message = 'Please choose a branch',
+      [string]$pattern
   )
   [System.Management.Automation.Host.ChoiceDescription[]]$branchOptions = @()
   [string[]]$localBranches = Get-GitLocalBranches
   [string[]]$originBranches = Get-GitOriginBranches -onlyName
   [string[]]$branches = $localBranches + $originBranches | Select-Object -Unique
   [char[]]$existChars = @('q')
 
+  if ($pattern) {
+    $branches = $branches | where { $_ -match $pattern }
+  }
+
   $charBranchDict = @() # char:branch key:value
 
   for ($i = 0; $i -lt $branches.Count; $i++) {
@@ -324,6 +329,15 @@ function Invoke-GitCheckout {
       git checkout $branch
     }
   }
+  # Invoke-GitCheckout -s key
+  elseif ($args.Count -eq 2 -and $args[0] -eq "-s") {
+    [string]$pattern = $args[1]
+    $branch = Get-OptionsForChoosingLocalOrOriginBranch "Switch branch" "Please choose a branch to switch" $pattern
+
+    if ($branch) {
+      git checkout $branch
+    }
+  }
   else {
     git checkout $args
   }