From 27b3f5e9c8bdfbd63ec06cd0a3472f80ba8d3752 Mon Sep 17 00:00:00 2001
From: bnewbold <bnewbold@robocracy.org>
Date: Tue, 29 Nov 2016 20:45:55 -0800
Subject: small improvements to CLI tools

---
 src/lib.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'src/lib.rs')

diff --git a/src/lib.rs b/src/lib.rs
index 2a098ee..508eeec 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -108,12 +108,12 @@ pub fn load_model_entry(p: &Path) -> Result<ModelEntry,String> {
     })
 }
 
-// TODO: have this check for model.modelica etc
 pub fn search_models(p: &Path) -> Vec<String> {
     if fs::metadata(p).unwrap().is_dir() {
         fs::read_dir(p).unwrap()
                        .map(|x| x.unwrap())
                        .filter(|x| x.metadata().unwrap().is_dir())
+                       .filter(|x| x.path().join("model.modelica").exists())
                        .map(|x| x.path().to_string_lossy().to_string())
                        .collect()
     } else {
-- 
cgit v1.2.3