Refactor: line numbers

1 files changed, 10 insertions, 6 deletions

diff --git a/src/paint.rs b/src/paint.rs
index b3abd004..f829534e 100644
--- a/src/paint.rs
+++ b/src/paint.rs
@@ -190,12 +190,16 @@ impl<'a> Painter<'a> {
 
             let mut handled_prefix = false;
             let mut ansi_strings = Vec::new();
-            if config.line_numbers && line_numbers.is_some() {
-                ansi_strings.extend(line_numbers::format_and_paint_line_numbers(
-                    line_numbers_data,
-                    line_numbers,
-                    config,
-                ))
+            if config.line_numbers {
+                if let (Some(line_numbers), Some(line_numbers_data)) =
+                    (line_numbers, line_numbers_data)
+                {
+                    ansi_strings.extend(line_numbers::format_and_paint_line_numbers(
+                        line_numbers_data,
+                        *line_numbers,
+                        config,
+                    ))
+                }
             }
             for (section_style, mut text) in superimpose_style_sections(
                 syntax_sections,