diff options
author | David Knaack <davidkna@users.noreply.github.com> | 2021-02-21 18:01:17 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-02-21 18:01:17 +0100 |
commit | a499f30157bb231d8f39557cdf504e7c094b68aa (patch) | |
tree | ea07751f592443d1852a39e21c0b4f1bec857152 /src | |
parent | 9ba82e8d921f8d0e7ff48f1f063635ca5971e49d (diff) |
feat(rust): Configure when the module is shown (#2350)
This makes it possible to configure when the rust module is shown
based on the contents of a directory.
Diffstat (limited to 'src')
-rw-r--r-- | src/configs/rust.rs | 6 | ||||
-rw-r--r-- | src/modules/rust.rs | 14 |
2 files changed, 12 insertions, 8 deletions
diff --git a/src/configs/rust.rs b/src/configs/rust.rs index 974c5ede0..1b5657c33 100644 --- a/src/configs/rust.rs +++ b/src/configs/rust.rs @@ -8,6 +8,9 @@ pub struct RustConfig<'a> { pub symbol: &'a str, pub style: &'a str, pub disabled: bool, + pub detect_extensions: Vec<&'a str>, + pub detect_files: Vec<&'a str>, + pub detect_folders: Vec<&'a str>, } impl<'a> RootModuleConfig<'a> for RustConfig<'a> { @@ -17,6 +20,9 @@ impl<'a> RootModuleConfig<'a> for RustConfig<'a> { symbol: "🦀 ", style: "bold red", disabled: false, + detect_extensions: vec!["rs"], + detect_files: vec!["Cargo.toml"], + detect_folders: vec![], } } } diff --git a/src/modules/rust.rs b/src/modules/rust.rs index d33fe747b..3f10ee190 100644 --- a/src/modules/rust.rs +++ b/src/modules/rust.rs @@ -10,23 +10,21 @@ use crate::configs::rust::RustConfig; use crate::formatter::StringFormatter; /// Creates a module with the current Rust version -/// -/// Will display the Rust version if any of the following criteria are met: -/// - Current directory contains a file with a `.rs` extension -/// - Current directory contains a `Cargo.toml` file pub fn module<'a>(context: &'a Context) -> Option<Module<'a>> { + let mut module = context.new_module("rust"); + let config = RustConfig::try_load(module.config); + let is_rs_project = context .try_begin_scan()? - .set_files(&["Cargo.toml"]) - .set_extensions(&["rs"]) + .set_files(&config.detect_files) + .set_extensions(&config.detect_extensions) + .set_folders(&config.detect_folders) .is_match(); if !is_rs_project { return None; } - let mut module = context.new_module("rust"); - let config = RustConfig::try_load(module.config); let parsed = StringFormatter::new(config.format).and_then(|formatter| { formatter .map_meta(|var, _| match var { |