From 6652d4d55571db0ebd6cb636c523cb8a87785c27 Mon Sep 17 00:00:00 2001
From: Julian Fricker <402357+TC72@users.noreply.github.com>
Date: Thu, 4 Nov 2021 06:22:09 +0000
Subject: [PATCH] feat: Full width select (#1403)

Can be changed back by setting
telescope.setup {
  defaults = {
    hl_result_eol = false,
  }
}
---
 lua/telescope/config.lua             | 10 ++++++++++
 lua/telescope/pickers/highlights.lua |  9 ++++++++-
 2 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/lua/telescope/config.lua b/lua/telescope/config.lua
index c7d151e..dc01d3b 100644
--- a/lua/telescope/config.lua
+++ b/lua/telescope/config.lua
@@ -320,6 +320,16 @@ append(
   Default: function that shows current count / all]]
 )
 
+append(
+  "hl_result_eol",
+  true,
+  [[
+  Changes if the highlight for the selected item in the results
+  window is always the full width of the window
+
+  Default: true]]
+)
+
 append(
   "dynamic_preview_title",
   false,
diff --git a/lua/telescope/pickers/highlights.lua b/lua/telescope/pickers/highlights.lua
index e7b99e1..6d89b78 100644
--- a/lua/telescope/pickers/highlights.lua
+++ b/lua/telescope/pickers/highlights.lua
@@ -1,5 +1,6 @@
 local a = vim.api
 local log = require "telescope.log"
+local conf = require("telescope.config").values
 
 local highlights = {}
 
@@ -69,7 +70,13 @@ function Highlighter:hi_selection(row, caret)
   a.nvim_buf_clear_namespace(results_bufnr, ns_telescope_selection, 0, -1)
   a.nvim_buf_add_highlight(results_bufnr, ns_telescope_selection, "TelescopeSelectionCaret", row, 0, #caret)
 
-  a.nvim_buf_add_highlight(results_bufnr, ns_telescope_selection, "TelescopeSelection", row, #caret, -1)
+  a.nvim_buf_set_extmark(
+    results_bufnr,
+    ns_telescope_selection,
+    row,
+    #caret,
+    { end_line = row + 1, hl_eol = conf.hl_result_eol, hl_group = "TelescopeSelection" }
+  )
 end
 
 function Highlighter:hi_multiselect(row, is_selected)