feat: add generate constructor code action

s1n7ax · s1n7ax · commit ea5371bf0de9 · 2024-07-14T01:22:24.000+05:30
diff --git a/lua/java/api/build.lua b/lua/java/api/build.lua
@@ -7,15 +7,15 @@ local M = {}
 ---@param is_full_build? boolean
 ---@return number
 function M.full_build_workspace(is_full_build)
-	local JavaCoreJdtlsClient = require('java-core.ls.clients.jdtls-client')
+	local JdtlsClient = require('java-core.ls.clients.jdtls-client')
 	local jdtls = require('java.utils.jdtls2')
 	local buf_util = require('java.utils.buffer')
 	local notify = require('java-core.utils.notify')
 
 	is_full_build = type(is_full_build) == 'boolean' and is_full_build or true
 
 	return async(function()
-			JavaCoreJdtlsClient(jdtls()):java_build_workspace(
+			JdtlsClient(jdtls()):java_build_workspace(
 				is_full_build,
 				buf_util.get_curr_buf()
 			)
diff --git a/lua/java/api/generate.lua b/lua/java/api/generate.lua
@@ -0,0 +1,55 @@
+local M = {}
+
+---@param params nvim.CodeActionParamsResponse
+function M.generate_constructor(params)
+	local JdtlsClient = require('java-core.ls.clients.jdtls-client')
+	local async = require('java-core.utils.async').sync
+	local get_client = require('java.utils.jdtls2')
+	local get_error_handler = require('java.handlers.error')
+	local ui = require('java.utils.ui')
+
+	return async(function()
+			local jdtls = JdtlsClient(get_client())
+			local status = jdtls:java_check_constructors_status(params.params)
+
+			if not status or not status.constructors then
+				return
+			end
+
+			local selected_constructor = ui.select(
+				'Select super class constructor(s).',
+				status.constructors,
+				function(constructor)
+					return string.format(
+						'%s %s',
+						constructor.name,
+						table.concat(constructor.parameters, ', ')
+					)
+				end
+			)
+
+			if not selected_constructor then
+				return
+			end
+
+			local selected_fields = ui.multi_select(
+				'Select Fields:',
+				status.fields,
+				function(field)
+					return field.name
+				end
+			)
+
+			local edit = jdtls:java_generate_constructor({
+				context = params.params,
+				constructors = { selected_constructor },
+				fields = selected_fields or {},
+			})
+
+			vim.lsp.util.apply_workspace_edit(edit, 'utf-8')
+		end)
+		.catch(get_error_handler('Generating constructor failed'))
+		.run()
+end
+
+return M
diff --git a/lua/java/api/refactor.lua b/lua/java/api/refactor.lua
@@ -29,7 +29,6 @@ function M.convert_variable_to_field()
 	M.extract('convertVariableToField')
 end
 
----
 ---@param refactor_command jdtls.CodeActionCommand
 function M.extract(refactor_command)
 	return async(function()
diff --git a/lua/java/commands/init.lua b/lua/java/commands/init.lua
@@ -194,6 +194,13 @@ M.commands = {
 }
 
 M.handlers = {
+	---@param _ lsp.Command
+	---@param params nvim.CodeActionParamsResponse
+	[M.commands.GENERATE_CONSTRUCTORS_PROMPT] = function(_, params)
+		require('java.api.generate').generate_constructor(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)
 	end,
diff --git a/lua/java/utils/ui.lua b/lua/java/utils/ui.lua
@@ -1,6 +1,6 @@
-local async = require('java-core.utils.async')
-local await = async.wait
+local wait = require('async.waits.wait')
 local notify = require('java-core.utils.notify')
+local List = require('java-core.utils.list')
 
 local M = {}
 
@@ -11,16 +11,61 @@ local M = {}
 ---@param format_item? fun(item: T): string
 ---@return T
 function M.select(prompt, values, format_item)
-	return await(function(callback)
+	return wait(function(callback)
 		vim.ui.select(values, {
 			prompt = prompt,
 			format_item = format_item,
 		}, callback)
 	end)
 end
 
+function M.multi_select(prompt, values, format_item)
+	return wait(function(callback)
+		local wrapped_items = List:new(values):map(function(item, index)
+			return {
+				index = index,
+				is_selected = false,
+				value = item,
+			}
+		end)
+
+		local open_select
+
+		open_select = function()
+			vim.ui.select(wrapped_items, {
+				prompt = prompt,
+				format_item = function(item)
+					local prefix = item.is_selected and '* ' or ''
+					return prefix
+						.. (format_item and format_item(item.value) or item.value)
+				end,
+			}, function(selected)
+				if not selected then
+					local selected_items = wrapped_items
+						:filter(function(item)
+							return item.is_selected
+						end)
+						:map(function(item)
+							return item.value
+						end)
+
+					callback(#selected_items > 0 and selected_items or nil)
+					return
+				end
+
+				wrapped_items[selected.index].is_selected =
+					not wrapped_items[selected.index].is_selected
+
+				open_select()
+			end)
+		end
+
+		open_select()
+	end)
+end
+
 function M.input(prompt)
-	return await(function(callback)
+	return wait(function(callback)
 		vim.ui.input({
 			prompt = prompt,
 		}, callback)
