color files with lscolors

Author: meain

Cargo.lock

@@ -27,6 +27,7 @@ time = "0.1.40"
 users = "0.8.0"
 chrono-humanize = "0.0.11"
 unicode-width = "0.1.5"
+lscolors = "0.5.0"
 
 [dependencies.clap]
 features = ["suggestions", "color", "wrap_help"]

Cargo.toml

@@ -1,4 +1,5 @@
 use ansi_term::{ANSIString, Colour, Style};
+use lscolors::{Indicator, LsColors};
 use std::collections::HashMap;
 
 #[allow(dead_code)]
@@ -62,6 +63,7 @@ pub enum Theme {
 
 pub struct Colors {
     colors: Option<HashMap<Elem, Colour>>,
+    lscolors: Option<LsColors>,
 }
 
 impl Colors {
@@ -70,15 +72,63 @@ impl Colors {
             Theme::NoColor => None,
             Theme::Default => Some(Self::get_light_theme_colour_map()),
         };
+        let lscolors = LsColors::from_env();
 
-        Self { colors }
+        Self { colors, lscolors }
     }
 
     pub fn colorize<'a>(&self, input: String, elem: &Elem) -> ColoredString<'a> {
         self.style(elem).paint(input)
     }
 
+    pub fn colorize_using_path<'a>(
+        &self,
+        input: String,
+        path: &str,
+        elem: &Elem,
+    ) -> ColoredString<'a> {
+        let style_from_path = self.style_from_path(path);
+        match style_from_path {
+            Some(style_from_path) => style_from_path.paint(input),
+            None => self.colorize(input, elem),
+        }
+    }
+
+    fn style_from_path(&self, path: &str) -> Option<Style> {
+        match &self.lscolors {
+            Some(lscolors) => {
+                let style = lscolors
+                    .style_for_path(path)
+                    .map(lscolors::Style::to_ansi_term_style);
+                if let Some(style) = style {
+                    Some(style)
+                } else {
+                    None
+                }
+            }
+            None => None,
+        }
+    }
+
     fn style(&self, elem: &Elem) -> Style {
+        match &self.lscolors {
+            Some(lscolors) => {
+                let indicator = self.get_indicator_from_elem(elem);
+                match indicator {
+                    Some(style) => {
+                        let style = lscolors.style_for_indicator(style);
+                        style
+                            .map(lscolors::Style::to_ansi_term_style)
+                            .unwrap_or_default()
+                    }
+                    None => self.style_default(elem),
+                }
+            }
+            None => self.style_default(elem),
+        }
+    }
+
+    fn style_default(&self, elem: &Elem) -> Style {
         if let Some(ref colors) = self.colors {
             let style_fg = Style::default().fg(colors[elem]);
             if elem.has_suid() {
@@ -91,6 +141,41 @@ impl Colors {
         }
     }
 
+    fn get_indicator_from_elem(&self, elem: &Elem) -> Option<Indicator> {
+        let indicator_string = match elem {
+            Elem::File { exec, uid } => {
+                if *uid {
+                    None
+                } else {
+                    if *exec {
+                        Some("ex")
+                    } else {
+                        Some("fi")
+                    }
+                }
+            }
+            Elem::Dir { uid } => {
+                if *uid {
+                    None
+                } else {
+                    Some("di")
+                }
+            }
+            Elem::SymLink => Some("ln"),
+            Elem::Pipe => Some("pi"),
+            Elem::Socket => Some("so"),
+            Elem::BlockDevice => Some("bd"),
+            Elem::CharDevice => Some("cd"),
+            Elem::BrokenSymLink => Some("or"),
+            _ => None,
+        };
+
+        match indicator_string {
+            Some(ids) => Indicator::from(ids),
+            None => None,
+        }
+    }
+
     // You can find the table for each color, code, and display at:
     //
     //https://jonasjacek.github.io/colors/

src/color.rs

@@ -13,6 +13,7 @@ extern crate clap;
 extern crate ansi_term;
 extern crate chrono_humanize;
 extern crate libc;
+extern crate lscolors;
 #[cfg(test)]
 extern crate tempdir;
 extern crate term_grid;

src/main.rs

@@ -7,6 +7,7 @@ use std::path::Path;
 #[derive(Debug, Eq)]
 pub struct Name {
     name: String,
+    path: String,
     extension: Option<String>,
     file_type: FileType,
 }
@@ -27,8 +28,11 @@ impl Name {
             );
         }
 
+        let path_string = path.to_string_lossy().to_string();
+
         Self {
             name,
+            path: path_string,
             extension,
             file_type,
         }
@@ -39,6 +43,7 @@ impl Name {
         let mut content = String::with_capacity(icon.len() + self.name.len() + 3 /* spaces */);
 
         content += icon.as_str();
+        content += &self.name;
 
         let elem = match self.file_type {
             FileType::CharDevice => Elem::CharDevice,
@@ -51,9 +56,7 @@ impl Name {
             },
         };
 
-        content += &self.name;
-
-        colors.colorize(content, &elem)
+        colors.colorize_using_path(content, &self.path, &elem)
     }
 
     pub fn name(&self) -> String {