From 7f5621320e14a23844097b5d74e56a882bc24b1b Mon Sep 17 00:00:00 2001
From: 1549469775 <1549469775@qq.com>
Date: Wed, 13 Nov 2019 15:36:35 +0800
Subject: [PATCH] 'a'

---
 app.js | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/app.js b/app.js
index 899fd97..5d66537 100644
--- a/app.js
+++ b/app.js
@@ -14,6 +14,27 @@ const haveGit = fs.existsSync('./.git');
  * 2. 添加GIT源
  * 3. 设置默认源以供一键上传
  */
+function getBranch() {
+  return new Promise((resolve, reject) => {
+    exec('git branch', function (error, stdout, stderr) {
+      if (error) {
+        console.log(error);
+        reject(error);
+        return;
+      }
+      let branches = stdout.split('\n').slice(0, stdout.split('\n').length - 1);
+      branches = branches.filter(v => {
+        if (v.indexOf('*') != -1) {
+          return true;
+        } else {
+          return false;
+        }
+      })
+      let currentBranch = branches[0].slice(2);
+      resolve(currentBranch);
+    });
+  })
+}
 (async () => {
   if (haveGit) {
     exec('git remote', function (error, stdout, stderr) {
@@ -32,9 +53,10 @@ const haveGit = fs.existsSync('./.git');
       }]).then(async answers => {
         if (answers.all) {
           // 所有源提交
+          let branch = await getBranch();
           for (let i = 0; i < origin.length; i++) {
             const o = origin[i];
-            await git_util.all('a', o);
+            await git_util.all('a', o, branch);
           }
         } else {
           inquirer.prompt([{