refactor: Extract calculation to get_lineno_width method

chengr4 · chengr4 · commit 43a1dc0a29e2 · 2024-08-23T11:43:49.000+08:00
diff --git a/src/renderer/display_list.rs b/src/renderer/display_list.rs
@@ -72,23 +72,7 @@ impl<'a> fmt::Debug for DisplayList<'a> {
 
 impl<'a> Display for DisplayList<'a> {
     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
-        let max_lineno = self.body.iter().fold(None, |max, set| {
-            set.display_lines.iter().fold(max, |max, line| match line {
-                DisplayLine::Source { lineno, .. } => match (max, lineno) {
-                    (Some(max), Some(lineno)) => Some(cmp::max(max, *lineno)),
-                    (Some(max), None) => Some(max),
-                    (None, Some(lineno)) => Some(*lineno),
-                    (None, None) => None,
-                },
-                _ => max,
-            })
-        });
-        let lineno_width = match max_lineno {
-            None => 0,
-            Some(0) => 1,
-            Some(_max) if self.anonymized_line_numbers => ANONYMIZED_LINE_NUM.len(),
-            Some(max) => (max as f64).log10().floor() as usize + 1,
-        };
+        let lineno_width = self.get_lineno_width();
 
         let multiline_depth = self.body.iter().fold(0, |max, set| {
             set.display_lines.iter().fold(max, |max2, line| match line {
@@ -152,6 +136,27 @@ impl<'a> DisplayList<'a> {
         }
         Ok(())
     }
+
+    fn get_lineno_width(&self) -> usize {
+        let max_lineno = self.body.iter().fold(None, |max, set| {
+            set.display_lines.iter().fold(max, |max, line| match line {
+                DisplayLine::Source { lineno, .. } => match (max, lineno) {
+                    (Some(max), Some(lineno)) => Some(cmp::max(max, *lineno)),
+                    (Some(max), None) => Some(max),
+                    (None, Some(lineno)) => Some(*lineno),
+                    (None, None) => None,
+                },
+                _ => max,
+            })
+        });
+        let lineno_width = match max_lineno {
+            None => 0,
+            Some(0) => 1,
+            Some(_max) if self.anonymized_line_numbers => ANONYMIZED_LINE_NUM.len(),
+            Some(max) => (max as f64).log10().floor() as usize + 1,
+        };
+        lineno_width
+    }
 }
 
 #[derive(Debug, PartialEq)]
