feat: add extract variable command (#5)

Author: s1n7ax

lua/java-refactor/lsp-refactor-commands.lua

@@ -0,0 +1,41 @@
+local ui = require('java.utils.ui')
+
+local M = {
+	---@class java-refactor.RenameAction
+	---@field length number
+	---@field offset number
+	---@field uri string
+
+	---Rename a given item
+	---@param arguments java-refactor.RenameAction[]
+	['java.action.rename'] = function(arguments)
+		for _, rename in ipairs(arguments) do
+			local buffer = vim.uri_to_bufnr(rename.uri)
+			local line
+
+			vim.api.nvim_buf_call(buffer, function()
+				line = vim.fn.byte2line(rename.offset)
+			end)
+
+			local start_char = rename.offset - vim.fn.line2byte(line) + 1
+			local end_char = start_char + rename.length
+
+			local name = ui.input('Variable Name')
+
+			if not name then
+				return
+			end
+
+			vim.api.nvim_buf_set_text(
+				buffer,
+				line - 1,
+				start_char,
+				line - 1,
+				end_char,
+				{ name }
+			)
+		end
+	end,
+}
+
+return M

lua/java-refactor/refactor-commands.lua

@@ -0,0 +1,59 @@
+local class = require('java-core.utils.class')
+local notify = require('java-core.utils.notify')
+local lsp_refactor_commands = require('java-refactor.lsp-refactor-commands')
+
+local JdtlsClient = require('java-core.ls.clients.jdtls-client')
+
+---@class java-refactor.ClientCommands
+---@field client lsp.Client
+local RefactorCommands = class()
+
+function RefactorCommands:_init(client)
+	self.client = client
+	self.jdtls_client = JdtlsClient(client)
+end
+
+function RefactorCommands:extract_variable()
+	local context = vim.lsp.util.make_range_params(0)
+	context.context = {}
+	context.context.diagnostics = vim.lsp.diagnostic.get_line_diagnostics(0)
+
+	local buffer = vim.api.nvim_get_current_buf()
+
+	local selection =
+		self.jdtls_client:java_infer_selection('extractVariable', context, buffer)
+
+	local edit = self.jdtls_client:java_get_refactor_edit(
+		'extractVariable',
+		context,
+		{},
+		selection,
+		buffer
+	)
+
+	vim.lsp.util.apply_workspace_edit(edit.edit, 'utf-8')
+
+	RefactorCommands.run_lsp_client_command(
+		edit.command.command,
+		edit.command.arguments
+	)
+end
+
+function RefactorCommands.run_lsp_client_command(command_name, arguments)
+	local command
+
+	if lsp_refactor_commands[command_name] then
+		command = lsp_refactor_commands[command_name]
+	else
+		command = vim.lsp.commands[command_name]
+	end
+
+	if not command then
+		notify.error('Command "' .. command_name .. '" is not supported')
+		return
+	end
+
+	command(arguments)
+end
+
+return RefactorCommands