From 9f86b8a4c57918521bcd34ac0431eaf4516d9ebb Mon Sep 17 00:00:00 2001
From: Adin Klotz <adinklotz@gmail.com>
Date: Thu, 20 Aug 2020 18:51:09 -0400
Subject: [PATCH] Zsh: Filter PRs in `gh` completion

e.g. `gh pr reopen` shows closed PRs, since that's what it probably wants,
and `gh pr view` shows all, not just open, PRs
---
 src/_gh | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/src/_gh b/src/_gh
index d8fd380..73b1bb4 100644
--- a/src/_gh
+++ b/src/_gh
@@ -368,20 +368,20 @@ function _gh_pr {
 }
 
 function _pr_helper {
-  local options=$(gh pr list | awk -F '\t' '{print $1 "\\:" "\"" $2 " (" $3 ")" "\""}' | tr '\n' ' ')
+  local options=$(gh pr list "$@" | awk -F '\t' '{print $1 "\\:" "\"" $2 " (" $3 ")" "\""}' | tr '\n' ' ')
   echo $options
 }
 
 function _gh_pr_checkout {
   _arguments -C \
-    "1:PR:(($(_pr_helper)))" \
+    "1:PR:(($(_pr_helper --state all)))" \
     '--help[Show help for command]' \
     '(-R --repo)'{-R,--repo}'[Select another repository using the `OWNER/REPO` format]:'
 }
 
 function _gh_pr_close {
   _arguments -C \
-    "1:PR:(($(_pr_helper)))" \
+    "1:PR:(($(_pr_helper --state open)))" \
     '--help[Show help for command]' \
     '(-R --repo)'{-R,--repo}'[Select another repository using the `OWNER/REPO` format]:'
 }
@@ -405,7 +405,7 @@ function _gh_pr_create {
 
 function _gh_pr_diff {
   _arguments -C \
-    "1:PR:(($(_pr_helper)))" \
+    "1:PR:(($(_pr_helper --state all)))" \
     '(-c --color)'{-c,--color}'[Whether or not to output color: {always|never|auto}]:' \
     '--help[Show help for command]' \
     '(-R --repo)'{-R,--repo}'[Select another repository using the `OWNER/REPO` format]:'
@@ -425,7 +425,7 @@ function _gh_pr_list {
 
 function _gh_pr_merge {
   _arguments -C \
-    "1:PR:(($(_pr_helper)))" \
+    "1:PR:(($(_pr_helper --state open )))" \
     '(-d --delete-branch)'{-d,--delete-branch}'[Delete the local and remote branch after merge]' \
     '(-m --merge)'{-m,--merge}'[Merge the commits with the base branch]' \
     '(-r --rebase)'{-r,--rebase}'[Rebase the commits onto the base branch]' \
@@ -436,21 +436,21 @@ function _gh_pr_merge {
 
 function _gh_pr_ready {
   _arguments -C \
-    "1:PR:(($(_pr_helper)))" \
+    "1:PR:(($(_pr_helper --state open)))" \
     '--help[Show help for command]' \
     '(-R --repo)'{-R,--repo}'[Select another repository using the `OWNER/REPO` format]:'
 }
 
 function _gh_pr_reopen {
   _arguments -C \
-    "1:PR:(($(_pr_helper)))" \
+    "1:PR:(($(_pr_helper --state closed)))" \
     '--help[Show help for command]' \
     '(-R --repo)'{-R,--repo}'[Select another repository using the `OWNER/REPO` format]:'
 }
 
 function _gh_pr_review {
   _arguments -C \
-    "1:PR:(($(_pr_helper)))" \
+    "1:PR:(($(_pr_helper --state open)))" \
     '(-a --approve)'{-a,--approve}'[Approve pull request]' \
     '(-b --body)'{-b,--body}'[Specify the body of a review]:' \
     '(-c --comment)'{-c,--comment}'[Comment on a pull request]' \
@@ -467,7 +467,7 @@ function _gh_pr_status {
 
 function _gh_pr_view {
   _arguments -C \
-    "1:PR:(($(_pr_helper)))" \
+    "1:PR:(($(_pr_helper --state all)))" \
     '(-w --web)'{-w,--web}'[Open a pull request in the browser]' \
     '--help[Show help for command]' \
     '(-R --repo)'{-R,--repo}'[Select another repository using the `OWNER/REPO` format]:'