Allow specifying multiple values for line_length_guide in config

Signed-off-by: Christoph Heiss <[email protected]>

Author: christoph-heiss

src/commands/selection.rs

@@ -86,9 +86,9 @@ pub fn justify(app: &mut Application) -> Result {
     let range = sel_to_range(app)?;
     let buffer = app.workspace.current_buffer.as_mut().unwrap();
 
-    let limit = match app.preferences.borrow().line_length_guide() {
-        Some(n) => n,
-        None => bail!("Justification requires a line_length_guide."),
+    let limit = match app.preferences.borrow().line_length_guides()[..] {
+        [first, ..] => first,
+        [] => bail!("Justification requires a line_length_guide."),
     };
 
     buffer.start_operation_group();

src/models/application/preferences/mod.rs

@@ -207,24 +207,29 @@ impl Preferences {
             })
     }
 
-    pub fn line_length_guide(&self) -> Option<usize> {
+    pub fn line_length_guides(&self) -> Vec<usize> {
         self.data
             .as_ref()
-            .and_then(|data| match data[LINE_LENGTH_GUIDE_KEY] {
-                Yaml::Integer(line_length) => Some(line_length as usize),
+            .map(|data| match data[LINE_LENGTH_GUIDE_KEY] {
+                Yaml::Integer(line_length) => vec![line_length as usize],
                 Yaml::Boolean(line_length_guide) => {
                     let default = self.default[LINE_LENGTH_GUIDE_KEY]
                         .as_i64()
                         .expect("Couldn't find default line length guide setting!");
 
                     if line_length_guide {
-                        Some(default as usize)
+                        vec![default as usize]
                     } else {
-                        None
+                        vec![]
                     }
                 }
-                _ => None,
+                Yaml::Array(ref guides) => guides
+                    .into_iter()
+                    .filter_map(|value| value.as_i64().map(|v| v as usize))
+                    .collect(),
+                _ => vec![],
             })
+            .unwrap_or_default()
     }
 
     pub fn line_wrapping(&self) -> bool {
@@ -584,23 +589,31 @@ mod tests {
         let data = YamlLoader::load_from_str("line_length_guide: 100").unwrap();
         let preferences = Preferences::new(data.into_iter().nth(0));
 
-        assert_eq!(preferences.line_length_guide(), Some(100));
+        assert_eq!(preferences.line_length_guides(), vec![100]);
+    }
+
+    #[test]
+    fn preferences_returns_user_defined_multiple_line_length_guides() {
+        let data = YamlLoader::load_from_str("line_length_guide: [80, 100, 120]").unwrap();
+        let preferences = Preferences::new(data.into_iter().nth(0));
+
+        assert_eq!(preferences.line_length_guides(), vec![80, 100, 120]);
     }
 
     #[test]
     fn preferences_returns_user_disabled_line_length_guide() {
         let data = YamlLoader::load_from_str("line_length_guide: false").unwrap();
         let preferences = Preferences::new(data.into_iter().nth(0));
 
-        assert_eq!(preferences.line_length_guide(), None);
+        assert_eq!(preferences.line_length_guides(), Vec::<usize>::new());
     }
 
     #[test]
     fn preferences_returns_user_default_line_length_guide() {
         let data = YamlLoader::load_from_str("line_length_guide: true").unwrap();
         let preferences = Preferences::new(data.into_iter().nth(0));
 
-        assert_eq!(preferences.line_length_guide(), Some(80));
+        assert_eq!(preferences.line_length_guides(), vec![80]);
     }
 
     #[test]

src/view/buffer/renderer.rs

@@ -85,10 +85,10 @@ impl<'a, 'p> BufferRenderer<'a, 'p> {
 
     fn print_rest_of_line(&mut self) {
         let on_cursor_line = self.on_cursor_line();
-        let guide_offset = self.length_guide_offset();
+        let guide_offsets = self.length_guide_offsets();
 
         for offset in self.screen_position.offset..self.terminal.width() {
-            let colors = if on_cursor_line || guide_offset.map(|go| go == offset).unwrap_or(false) {
+            let colors = if on_cursor_line || guide_offsets.contains(&offset) {
                 Colors::Focused
             } else {
                 Colors::Default
@@ -106,10 +106,12 @@ impl<'a, 'p> BufferRenderer<'a, 'p> {
         }
     }
 
-    fn length_guide_offset(&self) -> Option<usize> {
+    fn length_guide_offsets(&self) -> Vec<usize> {
         self.preferences
-            .line_length_guide()
+            .line_length_guides()
+            .into_iter()
             .map(|offset| self.gutter_width + offset)
+            .collect()
     }
 
     fn advance_to_next_line(&mut self) {