From ffc2a0adb145b39b8d3e93cee9d2e7c82aae0f85 Mon Sep 17 00:00:00 2001
From: InfiniteCoder <nayka.0.lobach.01@gmail.com>
Date: Fri, 2 Feb 2024 19:23:55 +0300
Subject: [PATCH] Implement with_label & with_note

---
 codespan-reporting/src/diagnostic.rs | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/codespan-reporting/src/diagnostic.rs b/codespan-reporting/src/diagnostic.rs
index 8e3abf4f..93e8fcd8 100644
--- a/codespan-reporting/src/diagnostic.rs
+++ b/codespan-reporting/src/diagnostic.rs
@@ -177,12 +177,24 @@ impl<FileId> Diagnostic<FileId> {
         self
     }
 
+    /// Add a label to the diagnostic.
+    pub fn with_label(mut self, label: Label<FileId>) -> Diagnostic<FileId> {
+        self.labels.push(label);
+        self
+    }
+
     /// Add some labels to the diagnostic.
     pub fn with_labels(mut self, mut labels: Vec<Label<FileId>>) -> Diagnostic<FileId> {
         self.labels.append(&mut labels);
         self
     }
 
+    /// Add a note to the diagnostic.
+    pub fn with_note(mut self, note: impl ToString) -> Diagnostic<FileId> {
+        self.notes.push(note.to_string());
+        self
+    }
+
     /// Add some notes to the diagnostic.
     pub fn with_notes(mut self, mut notes: Vec<String>) -> Diagnostic<FileId> {
         self.notes.append(&mut notes);