Add Pull Request for BitBucket and GitLab (#52)

Author: johnpaularthur

package.json

@@ -26,7 +26,8 @@
     },
     "activationEvents": [
         "onCommand:extension.openInGitHub",
-        "onCommand:extension.copyGitHubLinkToClipboard"
+        "onCommand:extension.copyGitHubLinkToClipboard",
+        "onCommand:extension.openPrGitProvider"
     ],
     "main": "./src/extension",
     "contributes": {
@@ -38,6 +39,10 @@
             {
                 "command": "extension.copyGitHubLinkToClipboard",
                 "title": "Copy GitHub link to clipboard"
+            },
+            {
+                "command": "extension.openPrGitProvider",
+                "title": "Open Pull Request"
             }
         ],
         "configuration": {
@@ -63,6 +68,11 @@
                 "command": "extension.copyGitHubLinkToClipboard",
                 "key": "ctrl+l c",
                 "mac": "ctrl+l c"
+            },
+            {
+                "command": "extension.openPrGitProvider",
+                "key": "ctrl+l p",
+                "mac": "ctrl+l p"
             }
         ],
         "menus": {
@@ -72,6 +82,9 @@
                 },
                 {
                     "command": "extension.copyGitHubLinkToClipboard"
+                },
+                {
+                    "command": "extension.openPrGitProvider"
                 }
             ]
         }

src/extension.js

@@ -17,7 +17,7 @@ var findParentDir = require('find-parent-dir');
 
 const gitProvider = require('./gitProvider');
 
-function getGitProviderLink(cb, fileFsPath, line) {
+function getGitProviderLink(cb, fileFsPath, line, pr) {
     var cwd = workspace.rootPath;
     var repoDir = findParentDir.sync(workspace.rootPath, '.git') || cwd;
 
@@ -44,11 +44,19 @@ function getGitProviderLink(cb, fileFsPath, line) {
                 subdir = repoRelativePath + subdir;
             }
 
-            cb(provider.webUrl(branch, subdir, line));
+            if (pr){
+                cb(provider.prUrl(branch));
+            }
+            else {
+                cb(provider.webUrl(branch, subdir, line));
+            }
+            
         });
     });
 }
 
+
+
 function getGitProviderLinkForFile(fileFsPath, cb) {
     getGitProviderLink(cb, fileFsPath);
 }
@@ -66,8 +74,16 @@ function getGitProviderLinkForRepo(cb) {
     getGitProviderLink(cb);
 }
 
-function branchOnCallingContext(args, cb) {
-    if (args && args.fsPath) {
+function getGitProviderPullRequest(cb) {
+    getGitProviderLink(cb, undefined, undefined, true);
+}
+
+
+function branchOnCallingContext(args, cb, pr) {
+    if (pr) {
+        getGitProviderPullRequest(cb);
+    }
+    else if (args && args.fsPath) {
         getGitProviderLinkForFile(args.fsPath, cb);
     }
     else if (Window.activeTextEditor) {
@@ -78,6 +94,7 @@ function branchOnCallingContext(args, cb) {
     }
 }
 
+
 function openInGitProvider(args) {
     branchOnCallingContext(args, open);
 }
@@ -86,10 +103,15 @@ function copyGitProviderLinkToClipboard(args) {
     branchOnCallingContext(args, copy);
 }
 
+function openPrGitProvider(args) {
+    branchOnCallingContext(args, open, true);
+}
+
 //TODO: rename openInGitHub to openInGitProvider
 function activate(context) {
     context.subscriptions.push(commands.registerCommand('extension.openInGitHub', openInGitProvider));
     context.subscriptions.push(commands.registerCommand('extension.copyGitHubLinkToClipboard', copyGitProviderLinkToClipboard));
+    context.subscriptions.push(commands.registerCommand('extension.openPrGitProvider', openPrGitProvider));
 }
 
 exports.activate = activate;

src/gitProvider.js

@@ -24,6 +24,9 @@ class BaseProvider {
     webUrl(branch, filePath, line) {
         return '';
     }
+    prUrl(branch) {
+        return '';
+    }    
 }
 
 class GitHub extends BaseProvider {
@@ -33,12 +36,18 @@ class GitHub extends BaseProvider {
         }
         return `${this.baseUrl}/tree/${branch}`;
     }
+    prUrl(branch){
+        return `${this.baseUrl}/pull/new/${branch}`;
+    }    
 }
 
 class Bitbucket extends BaseProvider {
     webUrl(branch, filePath, line) {
         return `${this.baseUrl}/src/${branch}` + (filePath ? `${filePath}` : '') + (line ? `#cl-${line}` : '');
     }
+    prUrl(branch){
+        return `${this.baseUrl}/pull-requests/new?source=${branch}`;
+    }
 }
 
 class GitLab extends GitHub {