1
0
Эх сурвалжийг харах

Highlight compiled files if their source is present

Compiled files with a source present (such as code.o when code.c is present) are now classified as temporary files. If the source isn't present, they're highlighted in a kind of drab colour (using the new 255-colour ability, yay!)

The code does do exists() checks on the filesystem when it could be possible to compare the files to the list of files we got from the call to readdir(), but it doesn't.
Ben S 11 жил өмнө
parent
commit
cce20d5953
1 өөрчлөгдсөн 35 нэмэгдсэн , 1 устгасан
  1. 35 1
      file.rs

+ 35 - 1
file.rs

@@ -69,6 +69,21 @@ impl<'a> File<'a> {
     fn is_tmpfile(&self) -> bool {
         self.name.ends_with("~") || (self.name.starts_with("#") && self.name.ends_with("#"))
     }
+    
+    fn with_extension(&self, newext: &'static str) -> String {
+        format!("{}.{}", self.path.filestem_str().unwrap(), newext)
+    }
+    
+    fn get_source_files(&self) -> Vec<String> {
+        match self.ext {
+            Some("class") => vec![self.with_extension("java")],  // Java
+            Some("elc") => vec![self.name.chop()],  // Emacs Lisp
+            Some("hi") => vec![self.with_extension("hs")],  // Haskell
+            Some("o") => vec![self.with_extension("c"), self.with_extension("cpp")],  // C, C++
+            Some("pyc") => vec![self.name.chop()],  // Python
+            _ => vec![],
+        }
+    }
 
     pub fn display(&self, column: &Column) -> String {
         match *column {
@@ -134,7 +149,16 @@ impl<'a> File<'a> {
             Red.normal()
         }
         else {
-            Plain
+            let source_files = self.get_source_files();
+            if source_files.len() == 0 {
+                Plain
+            }
+            else if source_files.iter().any(|filename| Path::new(format!("{}/{}", self.path.dirname_str().unwrap(), filename)).exists()) {
+                Fixed(244).normal()
+            }
+            else {
+                Fixed(137).normal()
+            }
         }
     }
 
@@ -165,3 +189,13 @@ impl<'a> File<'a> {
         }
     }
 }
+
+trait Chop {
+    fn chop(&self) -> String;
+}
+
+impl<'a> Chop for &'a str {
+    fn chop(&self) -> String {
+        self.slice_to(self.len() - 1).to_string()
+    }
+}