verb filtering by file extension

Fix #508

1 files changed, 20 insertions, 0 deletions

diff --git a/src/app/selection.rs b/src/app/selection.rs
index eebd791..f435dcb 100644
--- a/src/app/selection.rs
+++ b/src/app/selection.rs
@@ -124,4 +124,24 @@ impl<'a> SelInfo<'a> {
             _ => None,
         }
     }
+    pub fn extension(&self) -> Option<&str> {
+        match self {
+            SelInfo::None => None,
+            SelInfo::One(sel) => sel.path.extension().and_then(|e| e.to_str()),
+            SelInfo::More(stage) => {
+                let common_extension = stage.paths()[0]
+                    .extension().and_then(|e| e.to_str());
+                if common_extension.is_none() {
+                    return None;
+                }
+                for path in stage.paths().iter().skip(1) {
+                    let extension = path.extension().and_then(|e| e.to_str());
+                    if extension != common_extension {
+                        return None;
+                    }
+                }
+                common_extension
+            }
+        }
+    }
 }