1 files changed, 41 insertions, 0 deletions
diff --git a/src/modules/hostname.rs b/src/modules/hostname.rs
new file mode 100644
index 000000000..d71f733fa
--- /dev/null
+++ b/src/modules/hostname.rs
@@ -0,0 +1,41 @@
+use ansi_term::Color;
+use std::env;
+
+use super::{Context, Module};
+use std::ffi::OsString;
+
+/// Creates a module with the system hostname
+///
+/// Will display the hostname if all of the following criteria are met:
+///     - hostname.disabled is absent or false
+///     - hostname.ssh_only is false OR the user is currently connected as an SSH session (`$SSH_CONNECTION`)
+pub fn module<'a>(context: &'a Context) -> Option<Module<'a>> {
+    let mut module = context.new_module("hostname");
+    let module_style = module
+        .config_value_style("style")
+        .unwrap_or_else(|| Color::Green.bold().dimmed());
+
+    let ssh_connection = env::var("SSH_CONNECTION").ok();
+    if module.config_value_bool("ssh_only").unwrap_or(true) && ssh_connection.is_none() {
+        return None;
+    }
+
+    let os_hostname: OsString = gethostname::gethostname();
+
+    let host = match os_hostname.into_string() {
+        Ok(host) => host,
+        Err(bad) => {
+            log::debug!("hostname is not valid UTF!\n{:?}", bad);
+            return None;
+        }
+    };
+
+    let prefix = module.config_value_str("prefix").unwrap_or("").to_owned();
+    let suffix = module.config_value_str("suffix").unwrap_or("").to_owned();
+
+    module.set_style(module_style);
+    module.new_segment("hostname", &format!("{}{}{}", prefix, host, suffix));
+    module.get_prefix().set_value("on ");
+
+    Some(module)
+}