Add branch name

Author: mawi12345

lib/index.js

@@ -3,6 +3,7 @@ var runGitCommand = require('./helpers/run-git-command')
 
 var COMMITHASH_COMMAND = 'rev-parse HEAD'
 var VERSION_COMMAND = 'describe --always'
+var BRANCH_COMMAND = 'rev-parse --abbrev-ref HEAD'
 
 function GitRevisionPlugin (options) {
   options = options || {}
@@ -15,6 +16,9 @@ function GitRevisionPlugin (options) {
   this.versionCommand = options.versionCommand ||
     VERSION_COMMAND + (options.lightweightTags ? ' --tags' : '')
 
+  this.branchCommand = options.branchCommand ||
+    BRANCH_COMMAND
+
   if (options.versionCommand && options.lightweightTags) {
     throw new Error('lightweightTags can\'t be used together versionCommand')
   }
@@ -36,6 +40,14 @@ GitRevisionPlugin.prototype.apply = function (compiler) {
     /\[git-revision-version\]/gi,
     'VERSION'
   )
+
+  buildFile(
+    compiler,
+    this.gitWorkTree,
+    this.branchCommand,
+    /\[git-revision-branch\]/gi,
+    'BRANCH'
+  )
 }
 
 GitRevisionPlugin.prototype.commithash = function () {
@@ -52,4 +64,11 @@ GitRevisionPlugin.prototype.version = function () {
   )
 }
 
+GitRevisionPlugin.prototype.branch = function () {
+  return runGitCommand(
+    this.gitWorkTree,
+    this.branchCommand
+  )
+}
+
 module.exports = GitRevisionPlugin