misc/nit_env.sh: add support for zsh and fix fallback creating a file

Signed-off-by: Alexis Laferrière <alexis.laf@xymus.net>

diff --git a/misc/nit_env.sh b/misc/nit_env.sh
index 4b5a122..78ad0a3 100644 (file)
--- a/misc/nit_env.sh
+++ b/misc/nit_env.sh
@@ -128,8 +128,11 @@ __nit_env_main() {
                elif [ -f "$HOME/.bash_profile" ]; then
                        echo "$str" >> "$HOME/.bash_profile"
                        echo "succesfuly registerd nit .bash_profile"
+               elif [ -f "$HOME/.zshrc" ]; then
+                       echo "$str" >> "$HOME/.zshrc"
+                       echo "successfuly registered nit in .zshrc"
                else
-                       echo "$str" >> "$HOME/.profile"
+                       echo "$str" > "$HOME/.profile"
                        echo "couldn't find .bashrc, .bash_profile, .profile. Created .profile and registered nit in it"
                fi
        fi