From 8113a6c982d5fe0832245c526c413d3b7925cbf4 Mon Sep 17 00:00:00 2001
From: Daniil Tsivinsky <daniil@tsivinsky.com>
Date: Tue, 15 Feb 2022 19:05:51 +0300
Subject: [PATCH] complete copy_diagnostic_message function

---
 config/nvim/lua/user/utils.lua | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/config/nvim/lua/user/utils.lua b/config/nvim/lua/user/utils.lua
index 2833a6d..347880c 100644
--- a/config/nvim/lua/user/utils.lua
+++ b/config/nvim/lua/user/utils.lua
@@ -88,9 +88,20 @@ M.copy_diagnostic_message = function()
     return
   end
 
-  -- TODO: print all messages available and prompt which one to yank
+  local message = ""
 
-  local message = diagnostics[1].message
+  if #diagnostics == 1 then
+    message = diagnostics[1].message
+  elseif #diagnostics > 1 then
+    local d = {}
+    for _, diagnostic in ipairs(diagnostics) do
+      table.insert(d, diagnostic.message)
+    end
+
+    vim.ui.select(d, { prompt = "Pick a diagnostic to yank" }, function(item)
+      message = item
+    end)
+  end
 
   M.yank(message)