use super::error_kind::ErrorKind;
pub struct Error {
kind: ErrorKind,
position: Option<usize>,
}
impl Error {
pub fn new(kind: ErrorKind, position: usize) -> Error {
Error {
kind,
position: Some(position),
}
}
pub fn message_only(kind: ErrorKind) -> Error {
Error {
kind,
position: None,
}
}
pub fn prettify(self, input: &str) -> String {
if self.position.is_some() {
let (line_number, column_number) = self.get_line_column_info(input);
let option_line = input.split_terminator('\n').nth(line_number - 1);
let error_message: String = self.kind.into();
if let Some(line) = option_line {
let len = line_number.to_string().len();
format!(
"{} |\n{} | {}\n{} | {}^-- {}\n",
" ".repeat(len),
line_number,
line,
" ".repeat(len),
" ".repeat(column_number - 1),
error_message,
)
} else {
format!(
"An Error Occurred On Line {} And Column {}.\n{}",
line_number, column_number, error_message,
)
}
} else {
let error_message: String = self.kind.into();
format!("An Error Occurred.\n{}", error_message)
}
}
fn get_line_column_info(&self, input: &str) -> (usize, usize) {
let (mut line_number, mut column_number) = (1, 0);
input.chars().enumerate().find(|(idx, ch)| {
if ch == &'\n' {
line_number += 1;
column_number = 0;
} else {
column_number += 1;
}
idx == &(self.position.unwrap() - 1)
});
(line_number, column_number)
}
}