tree-sitter/cli/src/error.rs
2019-10-17 11:00:31 -07:00

87 lines
2 KiB
Rust

use std::fmt::Write;
use std::io;
use tree_sitter::QueryError;
#[derive(Debug)]
pub struct Error(pub Vec<String>);
pub type Result<T> = std::result::Result<T, Error>;
impl Error {
pub fn grammar(message: &str) -> Self {
Error(vec![format!("Grammar error: {}", message)])
}
pub fn regex(message: &str) -> Self {
Error(vec![format!("Regex error: {}", message)])
}
pub fn undefined_symbol(name: &str) -> Self {
Error(vec![format!("Undefined symbol `{}`", name)])
}
pub fn new(message: String) -> Self {
Error(vec![message])
}
pub fn err<T>(message: String) -> Result<T> {
Err(Error::new(message))
}
pub fn wrap<E: Into<Self>, M: ToString, F: FnOnce() -> M>(
message_fn: F,
) -> impl FnOnce(E) -> Self {
|e| {
let mut result = e.into();
result.0.push(message_fn().to_string());
result
}
}
pub fn message(&self) -> String {
let mut result = self.0.last().unwrap().clone();
if self.0.len() > 1 {
result.push_str("\nDetails:\n");
for msg in self.0[0..self.0.len() - 1].iter().rev() {
writeln!(&mut result, " {}", msg).unwrap();
}
}
result
}
}
impl<'a> From<QueryError> for Error {
fn from(error: QueryError) -> Self {
Error::new(format!("{:?}", error))
}
}
impl<'a> From<tree_sitter_highlight::Error> for Error {
fn from(error: tree_sitter_highlight::Error) -> Self {
Error::new(format!("{:?}", error))
}
}
impl From<serde_json::Error> for Error {
fn from(error: serde_json::Error) -> Self {
Error::new(error.to_string())
}
}
impl From<io::Error> for Error {
fn from(error: io::Error) -> Self {
Error::new(error.to_string())
}
}
impl From<regex_syntax::ast::Error> for Error {
fn from(error: regex_syntax::ast::Error) -> Self {
Error::new(error.to_string())
}
}
impl From<String> for Error {
fn from(error: String) -> Self {
Error::new(error)
}
}