From df52ca02c0da4ff25dff146cf8c6da49c21fc6c9 Mon Sep 17 00:00:00 2001
From: Peng Lyu <penn.lv@gmail.com>
Date: Thu, 5 Apr 2018 11:18:28 -0700
Subject: [PATCH] Mine PRs.

---
 extensions/git-extended/src/prProvider.ts | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/extensions/git-extended/src/prProvider.ts b/extensions/git-extended/src/prProvider.ts
index 0ffaf5f077a..a9d475e0ff3 100644
--- a/extensions/git-extended/src/prProvider.ts
+++ b/extensions/git-extended/src/prProvider.ts
@@ -194,12 +194,13 @@ export class PRProvider implements vscode.TreeDataProvider<PRGroup | PullRequest
 
 		if (element instanceof PRGroup) {
 			// fetch PRs
-			if (element.groupType === PRGroupType.RequestReview) {
+			if (element.groupType !== PRGroupType.All) {
 				let promises = this.repository.remotes.map(async remote => {
 					const octo = await this.crendentialStore.getOctokit(remote);
 					if (octo) {
+						const user = await octo.users.get();
 						const { data } = await octo.search.issues({
-							q: `is:open review-requested:rebornix type:pr repo:${remote.owner}/${remote.name}`
+							q: this.getPRFetchQuery(remote.owner, remote.name, user.data.login, element.groupType)
 						});
 						return data.items.map(item => new PullRequest(octo, remote.owner, remote.name, item));
 					}
@@ -262,6 +263,22 @@ export class PRProvider implements vscode.TreeDataProvider<PRGroup | PullRequest
 		}
 	}
 
+	getPRFetchQuery(owner: string, repo: string, user: string, type: PRGroupType) {
+		let filter = '';
+		switch (type) {
+			case PRGroupType.RequestReview:
+				filter = `review-requested:${user}`;
+				break;
+			case PRGroupType.RequestReview:
+				filter = `author:${user}`;
+				break;
+			default:
+				break;
+		}
+
+		return `is:open ${filter} type:pr repo:${owner}/${repo}`;
+	}
+
 	async provideComments(document: vscode.TextDocument): Promise<vscode.CommentThread[]> {
 		if (!this._comments) {
 			return [];
-- 
GitLab