Factor out ansi_style_for_path helper

Cargo.toml

@@ -11,6 +11,7 @@ name = "as-tree"
 path = "src/main.rs"
 
 [dependencies]
+ansi_term = "0.12"
 lscolors = "0.7"
 
 [raze]

src/BUILD

@@ -4,6 +4,7 @@ rust_binary(
     name = "as-tree",
     srcs = ["main.rs"],
     deps = [
+        "//third_party/cargo:ansi_term",
         "//third_party/cargo:lscolors",
     ],
     visibility = ["//visibility:public"],

src/main.rs

@@ -1,3 +1,4 @@
+extern crate ansi_term;
 extern crate lscolors;
 
 use std::collections::BTreeMap;
@@ -16,6 +17,13 @@ pub struct PathTrie {
     trie: BTreeMap<PathBuf, PathTrie>,
 }
 
+fn ansi_style_for_path(lscolors: &LsColors, path: &Path) -> ansi_term::Style {
+    lscolors
+        .style_for_path(&path)
+        .map(Style::to_ansi_term_style)
+        .unwrap_or_default()
+}
+
 impl PathTrie {
     pub fn insert(&mut self, path: &Path) {
         let mut cur = self;
@@ -40,10 +48,7 @@ impl PathTrie {
 
         for (idx, (path, it)) in self.trie.iter().enumerate() {
             let current_path = parent_path.join(path);
-            let ansi_style = lscolors
-                .style_for_path(&current_path)
-                .map(Style::to_ansi_term_style)
-                .unwrap_or_default();
+            let ansi_style = ansi_style_for_path(&lscolors, &current_path);
             if idx != self.trie.len() - 1 {
                 write!(
                     out,
@@ -76,20 +81,14 @@ impl fmt::Display for PathTrie {
         let lscolors = LsColors::from_env().unwrap_or_default();
 
         if let Some((path, it)) = self.trie.iter().next() {
-            let ansi_style = lscolors
-                .style_for_path(path)
-                .map(Style::to_ansi_term_style)
-                .unwrap_or_default();
+            let ansi_style = ansi_style_for_path(&lscolors, path);
             if path.is_absolute() || path == &PathBuf::from(".") {
                 write!(out, "{}\n", ansi_style.paint(path.to_string_lossy()))?;
                 return it._fmt(out, "", &lscolors, path.to_owned());
             }
         }
 
-        let ansi_style = lscolors
-            .style_for_path(".")
-            .map(Style::to_ansi_term_style)
-            .unwrap_or_default();
+        let ansi_style = ansi_style_for_path(&lscolors, Path::new("."));
         write!(out, "{}\n", ansi_style.paint("."))?;
         self._fmt(out, "", &lscolors, PathBuf::from("."))
     }

third_party/cargo/BUILD

@@ -8,6 +8,10 @@ package(default_visibility = ["//visibility:public"])
 licenses([
   "notice" # See individual crates for specific licenses
 ])
+alias(
+    name = "ansi_term",
+    actual = "@raze__ansi_term__0_12_1//:ansi_term",
+)
 alias(
     name = "lscolors",
     actual = "@raze__lscolors__0_7_0//:lscolors",