From 1f0dc54605089d74fff9987c3f7e7495f00bece2 Mon Sep 17 00:00:00 2001
From: Robbie Trencheny <me@robbiet.us>
Date: Sat, 29 Aug 2015 22:50:16 -0700
Subject: [PATCH] Add io.js support to the node module so iojs/node can be
 output

---
 modules/node/README.md           |  7 +++++--
 modules/node/functions/node-info | 11 ++++++++++-
 2 files changed, 15 insertions(+), 3 deletions(-)

diff --git a/modules/node/README.md b/modules/node/README.md
index 5f0c3caf..54ddc2c0 100644
--- a/modules/node/README.md
+++ b/modules/node/README.md
@@ -21,13 +21,15 @@ Functions
 Theming
 -------
 
-To display the version number of the current Node.js version, define the
+To display the version number or fork of the current Node.js or io.js version, define the
 following style inside the `prompt_name_setup` function.
 
     # %v - Node.js version.
+    # %f - Node.js fork (either node or iojs).
     zstyle ':prezto:module:node:info:version' format 'version:%v'
+    zstyle ':prezto:module:node:info:version' format 'fork:%f'
 
-Then add `$node_info[version]` to either `$PROMPT` or `$RPROMPT` and call
+Then add `$node_info[version]` and/or `$node_info[fork]` to either `$PROMPT` or `$RPROMPT` and call
 `node-info` in `prompt_name_preexec` hook function.
 
 Authors
@@ -37,6 +39,7 @@ Authors
 
   - [Sorin Ionescu](https://github.com/sorin-ionescu)
   - [Zeh Rizzatti](https://github.com/zehrizzatti)
+  - [Robbie Trencheny](https://github.com/robbiet480)
 
 [1]: http://nodejs.org
 [2]: http://npmjs.org
diff --git a/modules/node/functions/node-info b/modules/node/functions/node-info
index 7742f904..ccc28b25 100644
--- a/modules/node/functions/node-info
+++ b/modules/node/functions/node-info
@@ -4,6 +4,7 @@
 #
 # Authors:
 #   Zeh Rizzatti <zehrizzatti@gmail.com>
+#   Robbie Trencheny <me@robbiet.us>
 #
 
 local version
@@ -14,11 +15,19 @@ unset node_info
 typeset -gA node_info
 
 if (( $+functions[nvm_version] )); then
-  version="${$(nvm_version)#v}"
+  version="${$(nvm_version)#iojs-#v}"
+  if nvm_is_iojs_version "$(nvm_version)"; then
+    fork="$(nvm_iojs_prefix)"
+  else
+    fork="$(nvm_node_prefix)"
+  fi
 fi
 
 if [[ "$version" != (none|) ]]; then
   zstyle -s ':prezto:module:node:info:version' format 'version_format'
   zformat -f version_formatted "$version_format" "v:$version"
+  zstyle -s ':prezto:module:node:info:fork' format 'fork_format'
+  zformat -f fork_formatted "$fork_format" "f:$fork"
   node_info[version]="$version_formatted"
+  node_info[fork]="$fork_formatted"
 fi