From efdd17e22b235c159c0e1755ba7d06514a4d912c Mon Sep 17 00:00:00 2001
From: Matthew Leong <mattleong91@gmail.com>
Date: Wed, 30 Aug 2023 21:41:22 -0700
Subject: [PATCH] feat: simplify toggle

---
 lua/cosmic/lsp/mappings.lua | 10 +++++++++-
 lua/cosmic/utils/lsp.lua    |  4 ++--
 2 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/lua/cosmic/lsp/mappings.lua b/lua/cosmic/lsp/mappings.lua
index 1ae2a03..cc54005 100644
--- a/lua/cosmic/lsp/mappings.lua
+++ b/lua/cosmic/lsp/mappings.lua
@@ -1,7 +1,15 @@
 local utils = require('cosmic.utils')
 local lsp_utils = require('cosmic.utils.lsp')
+local user_config = require('cosmic.core.user')
 local M = {}
 
+local function toggle_inlay_hints(bufnr, enabled)
+  return function()
+    enabled = not enabled
+    vim.lsp.inlay_hint(bufnr, enabled)
+  end
+end
+
 -- Mappings.
 function M.init(client, bufnr)
   local buf_map = utils.create_buf_map(bufnr, {
@@ -35,7 +43,7 @@ function M.init(client, bufnr)
 
   -- inlay hints
   if client.supports_method('textDocument/inlayHint') then
-    buf_map('n', '<leader>lh', lsp_utils.toggle_inlay_hints(bufnr), { desc = 'Toggle inlay hints for buffer' })
+    buf_map('n', '<leader>lh', lsp_utils.toggle_inlay_hints(), { desc = 'Toggle inlay hints for buffer' })
   end
 
   -- code actions
diff --git a/lua/cosmic/utils/lsp.lua b/lua/cosmic/utils/lsp.lua
index add0b3b..5e71053 100644
--- a/lua/cosmic/utils/lsp.lua
+++ b/lua/cosmic/utils/lsp.lua
@@ -69,11 +69,11 @@ function M.get_lsp_status_str()
   return client_str
 end
 
-function M.toggle_inlay_hints(bufnr)
+function M.toggle_inlay_hints()
   local enabled = user_config.lsp.inlay_hint
   return function()
     enabled = not enabled
-    vim.lsp.inlay_hint(bufnr, enabled)
+    vim.lsp.inlay_hint(vim.api.nvim_get_current_buf() or 0, enabled)
   end
 end