[FEATURE] Enable zsh integration

pkg/integration/bash.go

@@ -1,68 +1,7 @@
 package integration
 
 var bashSource = `
-# Be careful! This runs in the user shell.
-
-# Mask the command dad with this shell function
-# This let us mutate the current shell
-dad() {
-    # Prepare a file to pass the finalize actions
-    local finalizer_file
-    finalizer_file="$(mktemp /tmp/dad-finalize-XXXXXX)"
-
-    # Run the actual command
-    env DAD_FINALIZER_FILE=$finalizer_file dad $@
-    return_code=$?
-
-    # Perform finalizers
-    local fin
-    while read -r fin; do
-        [ -n "${DAD_DEBUG:-}" ] && echo "DAD_DEBUG: finalizer: ${fin}"
-
-        case "${fin}" in
-            cd:*)
-                cd "${fin//cd:/}"
-                ;;
-            setenv:*)
-                export "${fin//setenv:/}"
-                ;;
-            *)
-                ;;
-        esac
-    done < "${finalizer_file}"
-    rm -f "${finalizer_file}"
-
-    return ${return_code}
-}
-
-__dad_prompt_command() {
-    # In shell hook mode, the command will use stderr to print in the console
-    # and stdout to mutate the shell (like activating a Python virtualenv)
-
-    # Fail fast if no dad executable is reachable
-    which dad > /dev/null || return
-
-    local hook_eval
-    hook_eval="$(command dad --shell-hook)"
-    [ -n "${DAD_DEBUG:-}" ] && echo -e "DAD_DEBUG: Hook eval:\n${hook_eval}\n---"
-    eval "${hook_eval}"
-}
-
 if [[ ! "${PROMPT_COMMAND:-}" == *__dad_prompt_command* ]]; then
   PROMPT_COMMAND="__dad_prompt_command; ${PROMPT_COMMAND:-}"
 fi
-
-dad-enable-debug() {
-    export DAD_DEBUG=1
-    echo "DAD_DEBUG: enabled"
-}
-
-dad-disable-debug() {
-    unset DAD_DEBUG
-    echo "DAD_DEBUG: disable"
-}
-
-if [[ -n "${DAD_DEBUG:-}" ]]; then
-    echo "DAD_DEBUG: Dad is now enabled..."
-fi
 `

pkg/integration/common.go

@@ -0,0 +1,64 @@
+package integration
+
+var shellSource = `
+# Be careful! This runs in the user shell.
+
+# Mask the command dad with this shell function
+# This let us mutate the current shell
+dad() {
+    # Prepare a file to pass the finalize actions
+    local finalizer_file
+    finalizer_file="$(mktemp /tmp/dad-finalize-XXXXXX)"
+
+    # Run the actual command
+    env DAD_FINALIZER_FILE=$finalizer_file dad $@
+    return_code=$?
+
+    # Perform finalizers
+    local fin
+    while read -r fin; do
+        [ -n "${DAD_DEBUG:-}" ] && echo "DAD_DEBUG: finalizer: ${fin}"
+
+        case "${fin}" in
+            cd:*)
+                cd "${fin//cd:/}"
+                ;;
+            setenv:*)
+                export "${fin//setenv:/}"
+                ;;
+            *)
+                ;;
+        esac
+    done < "${finalizer_file}"
+    rm -f "${finalizer_file}"
+
+    return ${return_code}
+}
+
+__dad_prompt_command() {
+    # In shell hook mode, the command will use stderr to print in the console
+    # and stdout to mutate the shell (like activating a Python virtualenv)
+
+    # Fail fast if no dad executable is reachable
+    which dad > /dev/null || return
+
+    local hook_eval
+    hook_eval="$(command dad --shell-hook)"
+    [ -n "${DAD_DEBUG:-}" ] && echo -e "DAD_DEBUG: Hook eval:\n${hook_eval}\n---"
+    eval "${hook_eval}"
+}
+
+dad-enable-debug() {
+    export DAD_DEBUG=1
+    echo "DAD_DEBUG: enabled"
+}
+
+dad-disable-debug() {
+    unset DAD_DEBUG
+    echo "DAD_DEBUG: disable"
+}
+
+if [[ -n "${DAD_DEBUG:-}" ]]; then
+    echo "DAD_DEBUG: Dad is now enabled..."
+fi
+`

pkg/integration/integration.go

@@ -3,12 +3,23 @@ package integration
 import (
 	"fmt"
 	"os"
+	"strings"
 
 	color "github.com/logrusorgru/aurora"
 )
 
 func Print() {
-	fmt.Println(bashSource)
+	var currentShell = os.Getenv("SHELL")
+
+	fmt.Println(shellSource)
+
+	if strings.HasSuffix(currentShell, "bash") {
+		fmt.Println(bashSource)
+	} else if strings.HasSuffix(currentShell, "zsh") {
+		fmt.Println(zshSource)
+	} else {
+		fmt.Println(color.Brown("Your shell is not supported"))
+	}
 }
 
 func AddFinalizerCd(path string) error {

pkg/integration/zsh.go

@@ -0,0 +1,6 @@
+package integration
+
+var zshSource = `
+prmptcmd() { $("__dad_prompt_command") }
+precmd_functions=(prmptcmd)
+`