1 files changed, 49 insertions, 0 deletions
diff --git a/src/modules/nodejs.rs b/src/modules/nodejs.rs
new file mode 100644
index 000000000..b30f2f531
--- /dev/null
+++ b/src/modules/nodejs.rs
@@ -0,0 +1,49 @@
+use ansi_term::Color;
+use std::process::Command;
+
+use super::{Context, Module};
+
+/// Creates a module with the current Node.js version
+///
+/// Will display the Node.js version if any of the following criteria are met:
+///     - Current directory contains a `.js` file
+///     - Current directory contains a `package.json` file
+///     - Current directory contains a `node_modules` directory
+pub fn module<'a>(context: &'a Context) -> Option<Module<'a>> {
+    let is_js_project = context
+        .try_begin_scan()?
+        .set_files(&["package.json"])
+        .set_extensions(&["js"])
+        .set_folders(&["node_modules"])
+        .is_match();
+
+    if !is_js_project {
+        return None;
+    }
+
+    match get_node_version() {
+        Some(node_version) => {
+            const NODE_CHAR: &str = "⬢ ";
+
+            let mut module = context.new_module("nodejs");
+            let module_style = module
+                .config_value_style("style")
+                .unwrap_or_else(|| Color::Green.bold());
+            module.set_style(module_style);
+
+            let formatted_version = node_version.trim();
+            module.new_segment("symbol", NODE_CHAR);
+            module.new_segment("version", formatted_version);
+
+            Some(module)
+        }
+        None => None,
+    }
+}
+
+fn get_node_version() -> Option<String> {
+    match Command::new("node").arg("--version").output() {
+        Ok(output) => Some(String::from_utf8(output.stdout).unwrap()),
+        Err(_) => None,
+    }
+}