Adding "pull/push current branch" commands -properly this time

singingwolfboy · Feb 10, 2012 · 3ddedf9 · 3ddedf9
1 parent f2bea18
commit 3ddedf9
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/Default.sublime-commands b/Default.sublime-commands
@@ -83,10 +83,18 @@
         "caption": "Git: Pull",
         "command": "git_pull"
     }
+    ,{
+        "caption": "Git: Pull Current Branch",
+        "command": "git_pull_current_branch"
+    }
     ,{
         "caption": "Git: Push",
         "command": "git_push"
     }
+    ,{
+        "caption": "Git: Push Current Branch",
+        "command": "git_push_current_branch"
+    }
     ,{
         "caption": "Git: Show Current File",
         "command": "git_show"

diff --git a/git.py b/git.py
@@ -721,11 +721,29 @@ def run(self):
         self.run_command(['git', 'pull'], callback=self.panel)
 
 
+class GitPullCurrentBranchCommand(GitWindowCommand):
+    def run(self):
+        self.run_command(['git', 'describe', '--contains',  '--all', 'HEAD'], callback=self.pull_current_branch_done)
+
+    def pull_current_branch_done(self, result):
+        result = result.strip()
+        self.run_command(['git', 'pull', 'origin', result], callback=self.panel)
+
+
 class GitPushCommand(GitWindowCommand):
     def run(self):
         self.run_command(['git', 'push'], callback=self.panel)
 
 
+class GitPushCurrentBranchCommand(GitWindowCommand):
+    def run(self):
+        self.run_command(['git', 'describe', '--contains',  '--all', 'HEAD'], callback=self.push_current_branch_done)
+
+    def push_current_branch_done(self, result):
+        result = result.strip()
+        self.run_command(['git', 'push', 'origin', result], callback=self.panel)
+
+
 class GitCustomCommand(GitTextCommand):
     may_change_files = True