Option to refresh a single PR, fixes #212

Author: Rachel Macfarlane

package.json

@@ -82,6 +82,11 @@
         "title": "Open Pull Request in GitHub",
         "category": "GitHub Pull Requests"
       },
+      {
+        "command": "pr.refreshPullRequest",
+        "title": "Refresh Pull Request",
+        "category": "GitHub Pull Requests"
+      },
       {
         "command": "pr.openFileInGitHub",
         "title": "Open File in GitHub",
@@ -152,6 +157,10 @@
           "command": "pr.openFileInGitHub",
           "when": "false"
         },
+        {
+          "command": "pr.refreshPullRequest",
+          "when": "false"
+        },
         {
           "command": "pr.deleteLocalBranch",
           "when": "false"
@@ -188,19 +197,24 @@
           "group": "pullrequest@1"
         },
         {
-          "command": "pr.openPullRequestInGitHub",
+          "command": "pr.refreshPullRequest",
           "when": "view == pr && viewItem =~ /pullrequest/",
           "group": "pullrequest@2"
         },
         {
-          "command": "pr.close",
+          "command": "pr.openPullRequestInGitHub",
           "when": "view == pr && viewItem =~ /pullrequest/",
           "group": "pullrequest@3"
         },
+        {
+          "command": "pr.close",
+          "when": "view == pr && viewItem =~ /pullrequest/",
+          "group": "pullrequest@4"
+        },
         {
           "command": "pr.deleteLocalBranch",
           "when": "view == pr && viewItem =~ /pullrequest:local:nonactive/",
-          "group": "pullrequest@4"
+          "group": "pullrequest@5"
         },
         {
           "command": "pr.openFileInGitHub",

src/view/prsTreeDataProvider.ts

@@ -44,6 +44,10 @@ export class PullRequestsTreeDataProvider implements vscode.TreeDataProvider<Tre
 		this._childrenDisposables = [];
 	}
 
+	async refresh(node?: TreeNode) {
+		return node ? this._onDidChangeTreeData.fire(node) : this._onDidChangeTreeData.fire();
+	}
+
 	getTreeItem(element: TreeNode): vscode.TreeItem {
 		return element.getTreeItem();
 	}

src/view/reviewManager.ts

@@ -21,7 +21,7 @@ import { FileChangeNode, RemoteFileChangeNode, fileChangeNodeFilter } from './tr
 import Logger from '../common/logger';
 import { PullRequestsTreeDataProvider } from './prsTreeDataProvider';
 import { IConfiguration } from '../authentication/configuration';
-import { providePRDocumentComments } from './treeNodes/pullRequestNode';
+import { providePRDocumentComments, PRNode } from './treeNodes/pullRequestNode';
 
 export class ReviewManager implements vscode.DecorationProvider {
 	private static _instance: ReviewManager;
@@ -70,6 +70,14 @@ export class ReviewManager implements vscode.DecorationProvider {
 			this.prFileChangesProvider.refresh();
 		}));
 
+		this._disposables.push(vscode.commands.registerCommand('pr.refreshPullRequest', (prNode: PRNode) => {
+			if (prNode.pullRequestModel.equals(this._prManager.activePullRequest)) {
+				this.updateComments();
+			}
+
+			this._prsTreeDataProvider.refresh(prNode);
+		}));
+
 		this._prsTreeDataProvider = new PullRequestsTreeDataProvider(this._configuration, _repository, _prManager);
 		this._disposables.push(this._prsTreeDataProvider);
 		this._disposables.push(vscode.window.registerDecorationProvider(this));