osdb/src/error_display.rs

use crate::{command::CommandParseError, tokens::ScanError};
use ariadne::{Color, Label, Report, ReportKind, Source};

pub trait OSDBError {
    fn display(&self, file: &str, input: &str);
}

impl OSDBError for CommandParseError {
    fn display(&self, file: &str, input: &str) {
        if let CommandParseError::Scan(x) = self {
            x.display(file, input);
        } else {
            Report::build(ReportKind::Error, (file, 0..input.len() - 1))
                .with_message(self.message())
                .with_label(Label::new((file, 0..input.len() - 1)).with_color(Color::Red))
                .finish()
                .print((file, Source::from(input)))
                .unwrap();
        }
    }
}

impl OSDBError for ScanError {
    fn display(&self, file: &str, input: &str) {
        let location = (file, Into::<std::ops::Range<usize>>::into(&self.location));
        Report::build(ReportKind::Error, location.clone())
            .with_message(format!("{self}"))
            .with_label(Label::new(location).with_color(Color::Red))
            .finish()
            .print((file, Source::from(input)))
            .unwrap();
    }
}