From 997a3ea7820cf7b19a98ce1a4641d25302da5a05 Mon Sep 17 00:00:00 2001
From: Lucas Holten <lhc.holten@gmail.com>
Date: Wed, 13 Dec 2023 12:55:05 +0100
Subject: [PATCH] Choose primary label from inside the context

---
 src/handlers/graphical.rs | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/handlers/graphical.rs b/src/handlers/graphical.rs
index e9e84b8..9bba215 100644
--- a/src/handlers/graphical.rs
+++ b/src/handlers/graphical.rs
@@ -446,10 +446,16 @@ impl GraphicalReportHandler {
     ) -> fmt::Result {
         let (contents, lines) = self.get_lines(source, context.inner())?;
 
-        let primary_label = labels
-            .iter()
+        // only consider labels from the context as primary label
+        let ctx_labels = labels.iter().filter(|l| {
+            context.inner().offset() <= l.inner().offset()
+                && l.inner().offset() + l.inner().len()
+                    <= context.inner().offset() + context.inner().len()
+        });
+        let primary_label = ctx_labels
+            .clone()
             .find(|label| label.primary())
-            .or_else(|| labels.first());
+            .or_else(|| ctx_labels.clone().next());
 
         // sorting is your friend
         let labels = labels