From 41cb710a7dff59a9bde126556be7f5a877c1dafd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kat=20March=C3=A1n?= <kzm@zkat.tech>
Date: Tue, 21 Sep 2021 21:59:57 -0700
Subject: [PATCH] fix(context): get labels/snippets working when using
 .context()

---
 src/eyreish/context.rs | 16 ++++++++++++++++
 src/eyreish/wrapper.rs |  8 ++++++++
 2 files changed, 24 insertions(+)

diff --git a/src/eyreish/context.rs b/src/eyreish/context.rs
index e2547a6..7673d8e 100644
--- 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
index e58ce4d..f3605e6 100644
--- 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 {