From b5ef8632e2b18dd72551a0abd423a3a4ad853ee8 Mon Sep 17 00:00:00 2001
From: Wenxuan Zhang <wenxuangm@gmail.com>
Date: Tue, 14 Jan 2020 22:36:52 +0800
Subject: [PATCH] Refactor

Former-commit-id: fa3303b432fc71385c15f1e86877275b72084cc7
---
 forgit.plugin.fish | 5 +----
 forgit.plugin.zsh  | 7 ++-----
 2 files changed, 3 insertions(+), 9 deletions(-)

diff --git a/forgit.plugin.fish b/forgit.plugin.fish
index ffb08b8..8fe7a30 100755
--- a/forgit.plugin.fish
+++ b/forgit.plugin.fish
@@ -77,10 +77,7 @@ end
 function forgit::add 
     forgit::inside_work_tree || return 1
     # Add files if passed as arguments
-    if not count $argv > /dev/null
-        git add $argv
-        return
-    end
+    count $argv >/dev/null && git add "$argv" && return
 
     set changed (git config --get-color color.status.changed red)
     set unmerged (git config --get-color color.status.unmerged red)
diff --git a/forgit.plugin.zsh b/forgit.plugin.zsh
index 864e6ad..ec729ec 100755
--- a/forgit.plugin.zsh
+++ b/forgit.plugin.zsh
@@ -51,13 +51,10 @@ forgit::diff() {
 # git add selector
 forgit::add() {
     forgit::inside_work_tree || return 1
-    local changed unmerged untracked files opts
     # Add files if passed as arguments
-    [[ $# -ne 0 ]] && {
-        git add $@
-        return
-    }
+    [[ $# -ne 0 ]] && git add "$@" && return
 
+    local changed unmerged untracked files opts
     changed=$(git config --get-color color.status.changed red)
     unmerged=$(git config --get-color color.status.unmerged red)
     untracked=$(git config --get-color color.status.untracked red)