Always expanding the path given in the treefmt.toml (#107)

* add cabal-fmt formatter

* always expand the path provided in treefmt.toml

* adding */ format in expanding path for string

* adding unit test for expand_if_path function

* Update src/lib.rs

* revert treefmt.toml

Co-authored-by: Jonas Chevalier <zimbatm@zimbatm.com>

devshell.toml

@@ -17,6 +17,7 @@ packages = [
   "elmPackages.elm-format",
   "go",
   "haskellPackages.ormolu",
+  "haskellPackages.cabal-fmt",
   "nixpkgs-fmt",
   "nodePackages.prettier",
   "python3.pkgs.black",

examples/haskell/Foo.hs

@@ -0,0 +1,4 @@
+module Foo where
+
+foo :: IO ()
+foo = putStrLn "Hello, Riyan!"

examples/haskell/Nested/Foo.hs

@@ -0,0 +1,4 @@
+module Nested.Foo where
+
+foo :: IO ()
+foo = putStrLn "Hello, Riyan!"

examples/haskell/treefmt.toml

@@ -0,0 +1,10 @@
+[formatter.haskell]
+command = "ormolu"
+options = [
+    "--ghc-opt", "-XBangPatterns",
+    "--ghc-opt", "-XPatternSynonyms",
+    "--ghc-opt", "-XTypeApplications",
+    "--mode", "inplace",
+    "--check-idempotence",
+]
+includes = ["Foo.hs"]

src/lib.rs

@@ -57,6 +57,82 @@ pub fn expand_if_path(str: String, reference: &Path) -> String {
             .to_string_lossy()
             .to_string()
     } else {
-        str
+        format!("*/{}",str)
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use std::{path::Path};
+    use globset::{GlobBuilder, GlobSetBuilder};
+    use super::expand_if_path;
+    
+    
+    #[test]
+    fn test_expand_if_path_single_pattern() {
+        let path = vec![
+            "/Foo.hs",
+            "/nested/Foo.hs",
+            "/nested/nested_again/Foo.hs",
+            "/different_folder/Foo.hs",
+            "/nested/different_folder/Foo.hs",
+        ];
+        let pattern = "Foo.hs";
+        let mut sum = GlobSetBuilder::new();
+        let tree_root = Path::new("/");
+        let pat = expand_if_path(pattern.to_string(), &tree_root);
+        let glob = GlobBuilder::new(&pat).build().unwrap();
+        sum.add(glob);
+        let result = sum.build().unwrap();
+        let test = path.clone().into_iter().filter(|p| result.is_match(p)).collect::<Vec<&str>>();
+
+        assert_eq!(path, test);
+    }
+
+    #[test]
+    fn test_expand_if_path_wildcard() {
+        let path = vec![
+            "/Foo.hs",
+            "/nested/Foo.hs",
+            "/nested/nested_again/Foo.hs",
+            "/different_folder/Foo.hs",
+            "/nested/different_folder/Foo.hs",
+            "/nested/Bar.hs",
+            "/nested/different_folder/Bar.hs"
+        ];
+        
+        let pattern = "*.hs";
+        let mut sum = GlobSetBuilder::new();
+        let tree_root = Path::new("/");
+        let pat = expand_if_path(pattern.to_string(), &tree_root);
+        let glob = GlobBuilder::new(&pat).build().unwrap();
+        sum.add(glob);
+        let result = sum.build().unwrap();
+        let test = path.clone().into_iter().filter(|p| result.is_match(p)).collect::<Vec<&str>>();
+
+        assert_eq!(path, test);
+    }
+    
+    #[test]
+    fn test_expand_if_path_single_pattern_with_slash() {
+        let path = vec![
+            "/Foo.hs",
+            "/nested/Foo.hs",
+            "/nested/nested/Foo.hs",
+            "/different_folder/Foo.hs",
+            "/nested/different_folder/Foo.hs",
+            "/nested/Bar.hs",
+            "/nested/different_folder/Bar.hs"
+        ];
+        let pattern = "nested/Foo.hs";
+        let mut sum = GlobSetBuilder::new();
+        let tree_root = Path::new("/");
+        let pat = expand_if_path(pattern.to_string(), &tree_root);
+        let glob = GlobBuilder::new(&pat).build().unwrap();
+        sum.add(glob);
+        let result = sum.build().unwrap();
+        let test = path.into_iter().filter(|p| result.is_match(p)).collect::<Vec<&str>>();
+        
+        assert_eq!(vec!["/nested/Foo.hs"], test);
     }
 }