diff options
Diffstat (limited to 'src/modules/vagrant.rs')
-rw-r--r-- | src/modules/vagrant.rs | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/src/modules/vagrant.rs b/src/modules/vagrant.rs index 99f8bd6a0..06ebef75b 100644 --- a/src/modules/vagrant.rs +++ b/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 { |