osdb/src/error_display.rs

use crate::{
    command::CommandParseError, meta_commands::MetaCommandParseError,
    statements::StatementParseError,
};
use ariadne::{Color, Label, Report, ReportKind, Source};

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

impl OSDBError for MetaCommandParseError {
    fn display(&self, file: &str, input: &str) {
        Report::build(ReportKind::Error, (file, 0..input.len() - 1))
            .with_message(format!("{self}"))
            .with_label(
                Label::new((file, 0..input.len() - 1))
                    .with_color(Color::Red)
                    .with_message("here"),
            )
            .finish()
            .print((file, Source::from(input)))
            .unwrap();
    }
}

impl OSDBError for StatementParseError {
    fn display(&self, file: &str, input: &str) {
        Report::build(ReportKind::Error, (file, 0..input.len() - 1))
            .with_message(format!("{self}"))
            .with_label(
                Label::new((file, 0..input.len() - 1))
                    .with_color(Color::Red)
                    .with_message("here"),
            )
            .finish()
            .print((file, Source::from(input)))
            .unwrap();
    }
}

impl OSDBError for CommandParseError {
    fn display(&self, file: &str, input: &str) {
        match self {
            CommandParseError::MetaCommand(err) => err.display(file, input),
            CommandParseError::Statement(err) => err.display(file, input),
        }
    }
}