solidhunter_lib/types/
file_diags.rs

1use super::LintDiag;
2use serde::{Deserialize, Serialize};
3use std::fmt;
4
5#[derive(Clone, Debug, Serialize, Deserialize)]
6pub struct FileDiags {
7    #[serde(skip_serializing_if = "String::is_empty", rename = "sourceFileContent")]
8    pub source_file_content: String,
9    pub diags: Vec<LintDiag>,
10}
11
12impl FileDiags {
13    pub fn new(source_file_content: String, diags: Vec<LintDiag>) -> Self {
14        FileDiags {
15            source_file_content,
16            diags,
17        }
18    }
19
20    fn format_highlighted_lines(&self, idx: usize) -> String {
21        let mut formatted = "   |\n".to_string();
22        let diag = &self.diags[idx];
23        let first_line = self
24            .source_file_content
25            .lines()
26            .nth(diag.range.start.line - 1)
27            .unwrap();
28        let trimmed_first_line = first_line.trim_start();
29        let max_offset = first_line.len() - trimmed_first_line.len();
30
31        for line_nb in diag.range.start.line..diag.range.end.line + 1 {
32            let line = self.source_file_content.lines().nth(line_nb - 1).unwrap();
33            let (trimmed_line, offset) = try_trim_max_offset(line, max_offset);
34            let mut higlight_length = trimmed_line.len();
35
36            if diag.range.start.line == diag.range.end.line {
37                higlight_length = diag.range.end.character - diag.range.start.character;
38            } else if line_nb == diag.range.start.line {
39                higlight_length = trimmed_line.len() - (diag.range.start.character - offset);
40            } else if line_nb == diag.range.end.line {
41                higlight_length = trimmed_line.len() - (diag.range.end.character - offset) + 1;
42            }
43
44            formatted = format!(
45                "{}{}{}|    {}\n   |    {}{}\n",
46                formatted,
47                line_nb,
48                compute_format_line_padding(line_nb),
49                trimmed_line,
50                " ".repeat(if line_nb == diag.range.start.line {
51                    diag.range.start.character - offset
52                } else {
53                    0
54                }),
55                "^".repeat(higlight_length)
56            );
57        }
58        formatted
59    }
60}
61
62impl fmt::Display for FileDiags {
63    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
64        for (idx, diag) in self.diags.iter().enumerate() {
65            write!(f, "{}\n{}", diag, self.format_highlighted_lines(idx))?;
66        }
67        Ok(())
68    }
69}
70
71fn compute_format_line_padding(line: usize) -> String {
72    let padding: String;
73    if line > 99 {
74        padding = " ".repeat(0);
75    } else if line > 9 {
76        padding = " ".to_string();
77    } else {
78        padding = " ".repeat(2);
79    }
80    padding
81}
82
83fn try_trim_max_offset(line: &str, max_offset: usize) -> (&str, usize) {
84    let mut offset: usize = 0;
85
86    for (i, c) in line.chars().enumerate() {
87        if i >= max_offset {
88            break;
89        }
90        if c.is_whitespace() {
91            offset += 1;
92        }
93    }
94    (&line[offset..], offset)
95}
solidhunter_lib/types/file_diags.rs

solidhunter_lib/types/
file_diags.rs
