fix(context): get labels/snippets working when using .context()

zkat · Sep 22, 2021 · e029657 · e029657
1 parent 9d07dc5
commit e029657
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/src/eyreish/context.rs b/src/eyreish/context.rs
@@ -144,6 +144,14 @@ where
     fn labels<'a>(&'a self) -> Option<Box<dyn Iterator<Item = LabeledSpan> + 'a>> {
         self.error.labels()
     }
+
+    fn source_code(&self) -> Option<&dyn crate::SourceCode> {
+        self.error.source_code()
+    }
+
+    fn related<'a>(&'a self) -> Option<Box<dyn Iterator<Item = &'a dyn Diagnostic> + 'a>> {
+        self.error.related()
+    }
 }
 
 impl<D> Diagnostic for ContextError<D, Report>
@@ -169,6 +177,14 @@ where
     fn labels<'a>(&'a self) -> Option<Box<dyn Iterator<Item = LabeledSpan> + 'a>> {
         self.error.inner.diagnostic().labels()
     }
+
+    fn source_code(&self) -> Option<&dyn crate::SourceCode> {
+        self.error.source_code()
+    }
+
+    fn related<'a>(&'a self) -> Option<Box<dyn Iterator<Item = &'a dyn Diagnostic> + 'a>> {
+        self.error.related()
+    }
 }
 
 struct Quoted<D>(D);

diff --git a/src/eyreish/wrapper.rs b/src/eyreish/wrapper.rs
@@ -94,6 +94,14 @@ impl Diagnostic for BoxedError {
     fn labels<'a>(&'a self) -> Option<Box<dyn Iterator<Item = LabeledSpan> + 'a>> {
         self.0.labels()
     }
+
+    fn source_code(&self) -> Option<&dyn miette::SourceCode> {
+        self.0.source_code()
+    }
+
+    fn related<'a>(&'a self) -> Option<Box<dyn Iterator<Item = &'a dyn Diagnostic> + 'a>> {
+        self.0.related()
+    }
 }
 
 impl Debug for BoxedError {