Refactoring: make logic more fit for rearrangement

With functions, we have more leeway in what to call

libexec/pyenv-init

@@ -43,7 +43,29 @@ fi
 
 root="${0%/*}/.."
 
-if [ -z "$print" ]; then
+function main() {
+  case "$mode" in
+  "help")
+    help_
+    exit 1
+    ;;
+  "path")
+    print_path
+    exit 0
+    ;;
+  "print")
+    init_dirs
+    print_env
+    print_completion
+    print_shell_function
+    exit 0
+    ;;
+  esac
+  # should never get here
+  exit 2
+}
+
+function help_() {
   case "$shell" in
   bash )
     if [ -f "${HOME}/.bashrc" ] && [ ! -f "${HOME}/.bash_profile" ]; then
@@ -79,35 +101,40 @@ if [ -z "$print" ]; then
     esac
     echo
   } >&2
+}
 
-  exit 1
-fi
+function init_dirs() {
+  mkdir -p "${PYENV_ROOT}/"{shims,versions}
+}
 
-mkdir -p "${PYENV_ROOT}/"{shims,versions}
-
-case "$shell" in
-fish )
+function print_env() {
+  case "$shell" in
+  fish )
     echo "set -gx PATH '${PYENV_ROOT}/shims' \$PATH"
     echo "set -gx PYENV_SHELL $shell"
-;;
-* )
+    ;;
+  * )
     echo 'export PATH="'${PYENV_ROOT}'/shims:${PATH}"'
     echo "export PYENV_SHELL=$shell"
-;;
-esac
+    ;;
+  esac
+}
 
-completion="${root}/completions/pyenv.${shell}"
-if [ -r "$completion" ]; then
+function print_completion() {
+  completion="${root}/completions/pyenv.${shell}"
+  if [ -r "$completion" ]; then
     echo "source '$completion'"
-fi
+  fi
 
-if [ -z "$no_rehash" ]; then
+  if [ -z "$no_rehash" ]; then
     echo 'command pyenv rehash 2>/dev/null'
-fi
+  fi
+}
 
-commands=(`pyenv-commands --sh`)
-case "$shell" in
-fish )
+function print_shell_function() {
+  commands=(`pyenv-commands --sh`)
+  case "$shell" in
+  fish )
     cat <<EOS
 function pyenv
   set command \$argv[1]
@@ -122,23 +149,23 @@ function pyenv
 end
 EOS
     ;;
-ksh )
+  ksh )
     cat <<EOS
 function pyenv {
   typeset command
 EOS
     ;;
-* )
+  * )
     cat <<EOS
 pyenv() {
   local command
 EOS
     ;;
-esac
+  esac
 
-if [ "$shell" != "fish" ]; then
-IFS="|"
-cat <<EOS
+  if [ "$shell" != "fish" ]; then
+    IFS="|"
+    cat <<EOS
   command="\${1:-}"
   if [ "\$#" -gt 0 ]; then
     shift
@@ -146,10 +173,15 @@ cat <<EOS
 
   case "\$command" in
   ${commands[*]})
-    eval "\$(pyenv "sh-\$command" "\$@")";;
+    eval "\$(pyenv "sh-\$command" "\$@")"
+    ;;
   *)
-    command pyenv "\$command" "\$@";;
+    command pyenv "\$command" "\$@"
+    ;;
   esac
 }
 EOS
-fi
+  fi
+}
+
+main