Show current brach name in the main view

Author: golbin

controller/main.js

@@ -99,6 +99,8 @@ var main = {
   reload: function () {
     git.status();
 
+    view.branchbox.setText(git.getCurrentBranchName());
+
     main.setItems();
 
     main.moveToUnstaged(0);

model/git.js

@@ -6,6 +6,8 @@ var gitExec = 'git';
 
 function Git (path) {
   this.path = path;
+
+  this.loadBranches();
 }
 
 Git.prototype.clear = function () {
@@ -207,4 +209,27 @@ Git.prototype.diff = function (type, index) {
   return stdout.toString();
 };
 
+Git.prototype.loadBranches = function () {
+  var self = this;
+
+  var gitCommand = gitExec + ' branch --list';
+
+  var stdout = execSync(gitCommand);
+
+  this.branches = stdout.toString().split('\n')
+    .map(function (line, index) {
+      if (line.charAt(0) === '*') {
+        self.currentBranchIndex = index;
+      }
+
+      return line.slice(2);
+    });
+
+  return this.branches;
+};
+
+Git.prototype.getCurrentBranchName = function () {
+  return this.branches[this.currentBranchIndex];
+};
+
 module.exports = Git;

view/main.js

@@ -7,24 +7,27 @@ var screen = blessed.screen(styles.screen);
 
 // bind screen to child elems
 _.merge(styles, {
-  title   : {
+  title    : {
     staged  : {parent: screen},
     unstaged: {parent: screen}
   },
-  list    : {
+  list     : {
     staged  : {parent: screen},
     unstaged: {parent: screen}
   },
-  menubar1: {
+  branchbox: {
     parent: screen
   },
-  menubar2: {
+  menubar1 : {
     parent: screen
   },
-  loading : {
+  menubar2 : {
     parent: screen
   },
-  popup   : {
+  loading  : {
+    parent: screen
+  },
+  popup    : {
     parent: screen
   }
 });
@@ -42,6 +45,8 @@ var list = {
 list.staged.name   = "staged";
 list.unstaged.name = "unstaged";
 
+var branchbox = blessed.text(styles.branchbox);
+
 blessed.listbar(styles.menubar1);
 blessed.listbar(styles.menubar2);
 
@@ -50,9 +55,10 @@ var loading = blessed.loading(styles.loading);
 var popup = blessed.box(styles.popup);
 
 module.exports = {
-  screen : screen,
-  title  : title,
-  list   : list,
-  loading: loading,
-  popup  : popup
+  screen   : screen,
+  title    : title,
+  list     : list,
+  branchbox: branchbox,
+  loading  : loading,
+  popup    : popup
 };

view/style/main.json

@@ -44,7 +44,7 @@
             "align": "left",
             "tags": true,
             "width": "50%",
-            "height": "100%-5",
+            "height": "100%-6",
             "interactive": false,
             "keys": true,
             "style": {
@@ -64,7 +64,7 @@
             "align": "left",
             "tags": true,
             "width": "50%",
-            "height": "100%-5",
+            "height": "100%-6",
             "interactive": true,
             "keys": true,
             "style": {
@@ -77,6 +77,18 @@
             }
         }
     },
+    "branchbox": {
+        "align": "center",
+        "bottom": 3,
+        "height": 1,
+        "align": "right",
+        "padding": {
+            "right": 2
+        },
+        "style": {
+            "fg": "yellow"
+        }
+    },
     "menubar1": {
         "align": "center",
         "bottom": 1,