From 08d8379bbee4b201f2a1092ec64d1b03c2259f60 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Matej=20Jane=C5=BEi=C4=8D?= <janezic.mj@gmail.com>
Date: Sat, 20 May 2023 00:49:12 +0200
Subject: [PATCH] cmp: change border for floats

---
 lua/plugins/cmp.lua | 27 +++++++++++++++------------
 lua/utils.lua       | 16 ++++++++++++++++
 2 files changed, 31 insertions(+), 12 deletions(-)
 create mode 100644 lua/utils.lua

diff --git a/lua/plugins/cmp.lua b/lua/plugins/cmp.lua
index 77572e0..76082be 100644
--- a/lua/plugins/cmp.lua
+++ b/lua/plugins/cmp.lua
@@ -8,21 +8,15 @@ function leave_snippet()
   end
 end
 
-local border = {
-  { '╭', 'CmpBorder' },
-  { '─', 'CmpBorder' },
-  { '╮', 'CmpBorder' },
-  { '│', 'CmpBorder' },
-  { '╯', 'CmpBorder' },
-  { '─', 'CmpBorder' },
-  { '╰', 'CmpBorder' },
-  { '│', 'CmpBorder' },
-}
-
 vim.api.nvim_command [[
     autocmd ModeChanged * lua leave_snippet()
 ]]
 
+vim.cmd 'highlight! link CmpPmenu         Pmenu'
+vim.cmd 'highlight! link CmpPmenuBorder   Pmenu'
+vim.cmd 'highlight! CmpPmenu         guibg=#282828'
+vim.cmd 'highlight! CmpPmenuBorder   guifg=#615750'
+
 return {
   {
     'hrsh7th/nvim-cmp',
@@ -35,9 +29,12 @@ return {
       -- nvim-cmp setup
       local cmp = require 'cmp'
       local luasnip = require 'luasnip'
+      local border = require('utils').nvim_open_win_border
 
       luasnip.config.setup {}
 
+      local highlight_opts = 'Normal:CmpPmenu,FloatBorder:CmpPmenuBorder,CursorLine:PmenuSel,Search:None'
+
       cmp.setup {
         snippet = {
           expand = function(args)
@@ -78,7 +75,13 @@ return {
         },
         window = {
           completion = {
-            border = border,
+            border = border 'CmpBorder',
+            side_padding = 1,
+            winhighlight = highlight_opts,
+          },
+          documentation = {
+            border = border 'CmpDocBorder',
+            winhighlight = highlight_opts,
           },
         },
       }
diff --git a/lua/utils.lua b/lua/utils.lua
new file mode 100644
index 0000000..f49e6a1
--- /dev/null
+++ b/lua/utils.lua
@@ -0,0 +1,16 @@
+local M = {}
+
+M.nvim_open_win_border = function(hl_name)
+  return {
+    { '┌', hl_name },
+    { '─', hl_name },
+    { '┐', hl_name },
+    { '│', hl_name },
+    { '┘', hl_name },
+    { '─', hl_name },
+    { '└', hl_name },
+    { '│', hl_name },
+  }
+end
+
+return M