feat: adding delegate method generate code action

s1n7ax · s1n7ax · commit 94625466f502 · 2024-07-15T21:39:08.000+05:30
diff --git a/lua/java/api/generate.lua b/lua/java/api/generate.lua
@@ -156,4 +156,83 @@ function M.generate_hash_code_and_equals(params)
 		.run()
 end
 
+---@param params nvim.CodeActionParamsResponse
+function M.generate_delegate_mothods_prompt(params)
+	local instance = require('java.utils.instance_factory')
+	local get_error_handler = require('java.handlers.error')
+	local ui = require('java.utils.ui')
+	local List = require('java-core.utils.list')
+
+	runner(function()
+			local jdtls = instance.jdtls_client()
+			local status = jdtls:java_check_delegate_methods_status(params.params)
+
+			if
+				not status
+				or not status.delegateFields
+				or #status.delegateFields < 1
+			then
+				require('notify').warn(
+					'All delegatable methods are already implemented.'
+				)
+				return
+			end
+
+			local selected_delegate_field = ui.select(
+				'Select target to generate delegates for.',
+				status.delegateFields,
+				function(field)
+					return field.field.name .. ': ' .. field.field.type
+				end
+			)
+
+			if not selected_delegate_field then
+				return
+			end
+
+			if #selected_delegate_field.delegateMethods < 1 then
+				require('notify').warn(
+					'All delegatable methods are already implemented.'
+				)
+				return
+			end
+
+			local selected_delegate_methods = ui.multi_select(
+				'Select methods to generate delegates for.',
+				selected_delegate_field.delegateMethods,
+				function(method)
+					return string.format(
+						'%s.%s(%s)',
+						selected_delegate_field.field.name,
+						method.name,
+						table.concat(method.parameters, ', ')
+					)
+				end
+			)
+
+			if not selected_delegate_methods or #selected_delegate_methods < 1 then
+				return
+			end
+
+			local delegate_entries = List:new(selected_delegate_methods):map(
+				---@param method jdtls.MethodBinding
+				function(method)
+					return {
+						field = selected_delegate_field.field,
+						delegateMethod = method,
+					}
+				end
+			)
+
+			local edit = jdtls:java_generate_delegate_methods({
+				context = params.params,
+				delegateEntries = delegate_entries,
+			})
+
+			vim.lsp.util.apply_workspace_edit(edit, 'utf-8')
+		end)
+		.catch(get_error_handler('Generating delegate mothods failed'))
+		.run()
+end
+
 return M
diff --git a/lua/java/commands/init.lua b/lua/java/commands/init.lua
@@ -129,6 +129,10 @@ M.handlers = {
 		require('java.api.generate').generate_hash_code_and_equals(params)
 	end,
 
+	[M.commands.GENERATE_DELEGATE_METHODS_PROMPT] = function(_, params)
+		require('java.api.generate').generate_delegate_mothods_prompt(params)
+	end,
+
 	---@param is_full_build boolean
 	[M.commands.COMPILE_WORKSPACE] = function(is_full_build)
 		require('java.api.build').full_build_workspace(is_full_build)
