1 files changed, 48 insertions, 0 deletions
diff --git a/src/modules/docker_context.rs b/src/modules/docker_context.rs
new file mode 100644
index 000000000..6cf0e2d94
--- /dev/null
+++ b/src/modules/docker_context.rs
@@ -0,0 +1,48 @@
+use dirs::home_dir;
+
+use super::{Context, Module, RootModuleConfig};
+
+use crate::configs::docker_context::DockerContextConfig;
+use crate::utils;
+
+const DOCKER_CONFIG_FILE: &str = ".docker/config.json";
+
+/// Creates a module with the currently active Docker context
+///
+/// Will display the Docker context if the following criteria are met:
+///     - There is a file named `$HOME/.docker/config.json`
+///     - The file is JSON and contains a field named `currentContext`
+///     - The value of `currentContext` is not `default`
+pub fn module<'a>(context: &'a Context) -> Option<Module<'a>> {
+    let mut module = context.new_module("docker_context");
+    let config: DockerContextConfig = DockerContextConfig::try_load(module.config);
+
+    if config.only_with_files
+        && !context
+            .try_begin_scan()?
+            .set_files(&["docker-compose.yml", "Dockerfile"])
+            .is_match()
+    {
+        return None;
+    }
+
+    let config_path = home_dir()?.join(DOCKER_CONFIG_FILE);
+    let json = utils::read_file(config_path).ok()?;
+    let parsed_json = serde_json::from_str(&json).ok()?;
+
+    match parsed_json {
+        serde_json::Value::Object(root) => {
+            let current_context = root.get("currentContext")?;
+            match current_context {
+                serde_json::Value::String(ctx) => {
+                    module.set_style(config.style);
+                    module.create_segment("symbol", &config.symbol);
+                    module.create_segment("context", &config.context.with_value(&ctx));
+                    Some(module)
+                }
+                _ => None,
+            }
+        }
+        _ => None,
+    }
+}