use core::cmp::min;
use lalrpop_util::ParseError;
pub type SyntaxError<'a, T> = ParseError<usize, T, &'a str>;
fn make_error(line: &str, unexpected: &str, line_number: usize, column_number: usize) -> String {
let underline = format!(
"{}^{}",
" ".repeat(column_number),
"-".repeat(unexpected.len() - 1)
);
format!(
"{WS} |
{line_number} | {line}
{WS} | {underline}
{WS} |
{WS} = unexpected `{unexpected}`",
WS = " ".repeat(line_number.to_string().len()),
line_number = line_number,
line = line,
underline = underline,
unexpected = unexpected
)
}
fn get_line(script: &str, location: usize) -> (usize, String, usize) {
let line_number = script[..min(location + 1, script.len())].lines().count();
let line = match script.lines().nth(line_number - 1) {
Some(line) => line,
None => {
if let Some(line) = script.lines().last() {
line
} else {
""
}
}
};
let mut column = {
let mut current_column = 0;
for ch in script[..location].chars() {
if ch == '\n' {
current_column = 0;
} else {
current_column += 1;
}
}
current_column
};
let trimmed_line = line.trim_start();
column -= (line.len() - trimmed_line.len()) as i32;
(line_number, String::from(trimmed_line), column as usize)
}
pub fn format_error<T: core::fmt::Debug>(script: &str, err: SyntaxError<T>) -> String {
match err {
SyntaxError::InvalidToken { location } => {
let (line_number, line, column) = get_line(script, location);
make_error(
&line,
&(script.as_bytes()[location] as char).to_string(),
line_number,
column,
)
}
SyntaxError::UnrecognizedEOF { location, .. } => {
let (line_number, line, _) = get_line(script, location);
make_error(&line, "EOF", line_number, line.len())
}
SyntaxError::UnrecognizedToken { token, .. } => {
let start = token.0;
let end = token.2;
let (line_number, line, column) = get_line(script, start);
let unexpected = &script[start..end];
make_error(&line, unexpected, line_number, column)
}
SyntaxError::ExtraToken { token } => {
let start = token.0;
let end = token.2;
let (line_number, line, column) = get_line(script, start);
let unexpected = &script[start..end];
make_error(&line, unexpected, line_number, column)
}
SyntaxError::User { error } => format!(
" |\n? | {}\n | {}\n |\n = unexpected compiling error",
error,
format!("^{}", "-".repeat(error.len() - 1))
),
}
}