teo_parser/diagnostics/
printer.rs

1use std::borrow::Cow;
2use std::env;
3use std::process::exit;
4use colored::Colorize;
5use pathdiff::diff_paths;
6use std::fs::read_to_string;
7use std::iter::repeat;
8use crate::builtin::STD_TEO;
9use crate::diagnostics::diagnostics::{Diagnostics, DiagnosticsLog};
10
11pub fn print_diagnostics_and_exit(diagnostics: &Diagnostics, print_warnings: bool) {
12    print_diagnostics(diagnostics, print_warnings);
13    exit(1);
14}
15
16pub fn print_diagnostics(diagnostics: &Diagnostics, print_warnings: bool) {
17    if diagnostics.has_warnings() && print_warnings {
18        for log in diagnostics.warnings() {
19            print_diagnostics_log(log);
20        }
21    }
22    if diagnostics.has_errors() {
23        for log in diagnostics.errors() {
24            print_diagnostics_log(log);
25        }
26    }
27}
28
29fn print_diagnostics_log<T>(log: T) where T: DiagnosticsLog {
30    let source = log.source_path();
31    let current_dir = &env::current_dir().unwrap();
32    let filename = if source.starts_with("(builtin)") {
33        source.to_owned()
34    } else {
35        if let Some(path) = diff_paths(source, current_dir) {
36            let result = path.to_str().unwrap().to_owned();
37            if result.starts_with(".") {
38                result
39            } else {
40                if cfg!(windows) {
41                    ".\\".to_owned() + result.as_str()
42                } else {
43                    "./".to_owned() + result.as_str()
44                }
45            }
46        } else {
47            source.to_owned()
48        }
49    };
50    let title = if log.is_warning() {
51        "Warning".yellow().bold()
52    } else if log.is_error() {
53        "Error".red().bold()
54    } else {
55        "Unknown".yellow().bold()
56    };
57    let mut code = "".to_owned();
58    let file_content = if source.starts_with("(builtin)") {
59        Cow::Borrowed(STD_TEO)
60    } else {
61        Cow::Owned(read_to_string(source).unwrap())
62    };
63    let first_line_content: &str = file_content.as_ref().lines().nth(log.span().start_position.0 - 1).unwrap();
64    code += format!("{} {}\n", "|".blue().bold(), first_line_content).as_str();
65    if log.span().start_position.0 == log.span().end_position.0 {
66        let before_len = log.span().start_position.1 - 1;
67        let content_len = log.span().end_position.1 - log.span().start_position.1;
68        code += format!("{} {}{}\n", "|".blue().bold(), repeat(" ").take(before_len).collect::<String>(), repeat("^").take(content_len).collect::<String>().bright_blue()).as_str()
69    } else {
70        let before_len = log.span().start_position.1 - 1;
71        let content_len = first_line_content.len() - before_len;
72        code += format!("{} {}{}\n", "|".blue().bold(), repeat(" ").take(before_len).collect::<String>(), repeat("^").take(content_len).collect::<String>().bright_blue()).as_str()
73    }
74    if log.span().start_position.0 != log.span().end_position.0 {
75        if log.span().start_position.0 + 1 != log.span().end_position.0 {
76            code += format!("{} ...\n", "|".blue().bold()).as_str();
77        }
78        let last_line_content = file_content.lines().nth(log.span().end_position.0 - 1).unwrap();
79        code += format!("{} {}\n", "|".blue().bold(), last_line_content).as_str();
80        let len = log.span().end_position.1;
81        code += format!("{} {}\n", "|".blue().bold(), repeat("^").take(len).collect::<String>().bright_blue()).as_str();
82    }
83    println!("{}: {}:{}:{} - {}:{}\n{}{}", title, filename, log.span().start_position.0, log.span().start_position.1, log.span().end_position.0, log.span().end_position.1, code, log.message());
84}
teo_parser/diagnostics/printer.rs

teo_parser/diagnostics/
printer.rs
