fix(keymap.helpers): improve _toggle* helper functions

Author: charliie-dev

lua/keymap/completion.lua

@@ -72,6 +72,18 @@ function M.lsp(buf)
 			:with_silent()
 			:with_buffer(buf)
 			:with_desc("lsp: Show outgoing calls"),
+		["n|<leader>lv"] = map_callback(function()
+				_toggle_diagnostic()
+			end)
+			:with_noremap()
+			:with_silent()
+			:with_desc("lsp: Toggle virtual text display of current buffer"),
+		["n|<leader>lh"] = map_callback(function()
+				_toggle_inlayhint()
+			end)
+			:with_noremap()
+			:with_silent()
+			:with_desc("lsp: Toggle inlay hints dispaly of current buffer"),
 	}
 	bind.nvim_load_mapping(map)
 

lua/keymap/editor.lua

@@ -63,20 +63,6 @@ local mappings = {
 			:with_desc("file: chmod +x current file"),
 		["n|<C-p>"] = map_cmd("<Nop>"):with_noremap():with_silent():with_desc("Disable native cmp"),
 		["n|<C-n>"] = map_cmd("<Nop>"):with_noremap():with_silent():with_desc("Disable native cmp"),
-
-		-- Builtins: Lsp
-		["n|<leader>td"] = map_callback(function()
-				_toggle_diagnostic()
-			end)
-			:with_noremap()
-			:with_silent()
-			:with_desc("edit: Toggle virtual text display of current buffer"),
-		["n|<leader>th"] = map_callback(function()
-				_toggle_inlayhint()
-			end)
-			:with_noremap()
-			:with_silent()
-			:with_desc("edit: Toggle inlay hints dispaly of current buffer"),
 	},
 	plugins = {
 		-- Plugin: persisted.nvim

lua/keymap/helpers.lua

@@ -38,6 +38,30 @@ _G._telescope_collections = function(picker_type)
 		:find()
 end
 
+_G._toggle_inlayhint = function()
+	local is_enabled = vim.lsp.inlay_hint.is_enabled()
+
+	vim.lsp.inlay_hint.enable(not is_enabled)
+	vim.notify(
+		(is_enabled and "Inlay hint disabled successfully" or "Inlay hint enabled successfully"),
+		vim.log.levels.INFO,
+		{ title = "LSP Inlay Hint" }
+	)
+end
+
+local _vt_enabled = require("core.settings").diagnostics_virtual_text
+_G._toggle_diagnostic = function()
+	if vim.diagnostic.is_enabled() then
+		_vt_enabled = not _vt_enabled
+		vim.diagnostic[_vt_enabled and "show" or "hide"]()
+		vim.notify(
+			(_vt_enabled and "Virtual text is now displayed" or "Virtual text is now hidden"),
+			vim.log.levels.INFO,
+			{ title = "LSP Diagnostic" }
+		)
+	end
+end
+
 _G._flash_esc_or_noh = function()
 	local flash_active, state = pcall(function()
 		return require("flash.plugins.char").state
@@ -49,31 +73,6 @@ _G._flash_esc_or_noh = function()
 	end
 end
 
-_G._toggle_inlayhint = function()
-	if vim.lsp.inlay_hint.is_enabled() then
-		vim.lsp.inlay_hint.enable(false)
-		vim.notify("Disable inlay hint successfully!", vim.log.levels.INFO, { title = "LSP Inlay Hint" })
-	else
-		vim.lsp.inlay_hint.enable(true)
-		vim.notify("Enable inlay hint successfully!", vim.log.levels.INFO, { title = "LSP Inlay Hint" })
-	end
-end
-
-local _diagnostic = 1
-_G._toggle_diagnostic = function()
-	if vim.diagnostic.is_enabled() then
-		if _diagnostic == 1 then
-			_diagnostic = 0
-			vim.diagnostic.hide()
-			vim.notify("Hide virtual text successfully!", vim.log.levels.INFO, { title = "LSP Diagnostic" })
-		else
-			_diagnostic = 1
-			vim.diagnostic.show()
-			vim.notify("Show virtual text successfully!", vim.log.levels.INFO, { title = "LSP Diagnostic" })
-		end
-	end
-end
-
 ---@param program string
 local function not_found_notify(program)
 	vim.notify(string.format("[%s] not found!", program), vim.log.levels.ERROR, { title = "toggleterm.nvim" })