feat(vagrant): Configure when the module is shown (#2314)

* feat(vagrant): Configure when the module is shown

This makes it possible to configure when the vagrant module is shown based on the contents of a directory.

* fix documentation

Co-authored-by: Thomas O'Donnell <andytom@users.noreply.github.com>

Co-authored-by: Thomas O'Donnell <andytom@users.noreply.github.com>

docs/config/README.md

@@ -2528,16 +2528,19 @@ show_always = true
 ## Vagrant
 
 The `vagrant` module shows the currently installed version of Vagrant.
-The module will be shown if any of the following conditions are met:
+By default the module will be shown if any of the following conditions are met:
 
 - The current directory contains a `Vagrantfile` file
 
 ### Options
 
 | Option              | Default                              | Description                                         |
-| ---------- | ---------------------------------- | --------------------------------------------------- |
+| ------------------- | ------------------------------------ | --------------------------------------------------- |
 | `format`            | `"via [$symbol($version )]($style)"` | The format for the module.                          |
 | `symbol`            | `"⍱ "`                               | A format string representing the symbol of Vagrant. |
+| `detect_extensions` | `[]`                                 | Which extensions should trigger this module.        |
+| `detect_files`      | `["Vagrantfile"]`                    | Which filenames should trigger this module.         |
+| `detect_folders`    | `[]`                                 | Which folders should trigger this module.           |
 | `style`             | `"cyan bold"`                        | The style for the module.                           |
 | `disabled`          | `false`                              | Disables the `Vagrant` module.                      |
 

src/configs/vagrant.rs

@@ -8,6 +8,9 @@ pub struct VagrantConfig<'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 VagrantConfig<'a> {
@@ -17,6 +20,9 @@ impl<'a> RootModuleConfig<'a> for VagrantConfig<'a> {
             symbol: "⍱ ",
             style: "cyan bold",
             disabled: false,
+            detect_extensions: vec![],
+            detect_files: vec!["Vagrantfile"],
+            detect_folders: vec![],
         }
     }
 }

src/modules/vagrant.rs

@@ -4,21 +4,21 @@ use crate::configs::vagrant::VagrantConfig;
 use crate::formatter::StringFormatter;
 
 /// Creates a module with the current Vagrant version
-///
-/// Will display the Vagrant version if any of the following criteria are met:
-///     - Current directory contains a `Vagrantfile` file
 pub fn module<'a>(context: &'a Context) -> Option<Module<'a>> {
+    let mut module = context.new_module("vagrant");
+    let config = VagrantConfig::try_load(module.config);
+
     let is_vagrant_project = context
         .try_begin_scan()?
-        .set_files(&["Vagrantfile"])
+        .set_files(&config.detect_files)
+        .set_extensions(&config.detect_extensions)
+        .set_folders(&config.detect_folders)
         .is_match();
 
     if !is_vagrant_project {
         return None;
     }
 
-    let mut module = context.new_module("vagrant");
-    let config = VagrantConfig::try_load(module.config);
     let parsed = StringFormatter::new(config.format).and_then(|formatter| {
         formatter
             .map_meta(|var, _| match var {