feat: add generate hash code and equals code action

s1n7ax · s1n7ax · commit 6a714fedc4a8 · 2024-07-15T01:20:32.000+05:30
diff --git a/lua/java/api/generate.lua b/lua/java/api/generate.lua
@@ -97,4 +97,63 @@ function M.generate_to_string(params)
 		.run()
 end
 
+---@param params nvim.CodeActionParamsResponse
+function M.generate_hash_code_and_equals(params)
+	local instance = require('java.utils.instance_factory')
+	local get_error_handler = require('java.handlers.error')
+	local ui = require('java.utils.ui')
+
+	runner(function()
+			local jdtls = instance.jdtls_client()
+			local status = jdtls:java_check_hash_code_equals_status(params.params)
+
+			if not status or not status.fields or #status.fields < 1 then
+				local message = string.format(
+					'The operation is not applicable to the type %s.',
+					status.type
+				)
+				require('java-core.utils.notify').warn(message)
+				return
+			end
+
+			local regenerate = false
+
+			if status.existingMethods and #status.existingMethods > 0 then
+				local prompt = string.format(
+					'Methods %s already exists in the Class %s. Do you want to regenerate the implementation?',
+					'Regenerate',
+					'Cancel'
+				)
+
+				local choice = ui.select(prompt, { 'Regenerate', 'Cancel' })
+
+				if choice == 'Regenerate' then
+					regenerate = true
+				end
+			end
+
+			local fields = ui.multi_select(
+				'Select the fields to include in the hashCode() and equals() methods.',
+				status.fields,
+				function(field)
+					return field.name
+				end
+			)
+
+			if not fields or #fields < 1 then
+				return
+			end
+
+			local edit = jdtls:java_generate_hash_code_equals({
+				context = params.params,
+				fields = fields,
+				regenerate = regenerate,
+			})
+
+			vim.lsp.util.apply_workspace_edit(edit, 'utf-8')
+		end)
+		.catch(get_error_handler('Generating hash code failed'))
+		.run()
+end
+
 return M
diff --git a/lua/java/commands/init.lua b/lua/java/commands/init.lua
@@ -125,6 +125,10 @@ M.handlers = {
 		require('java.api.generate').generate_to_string(params)
 	end,
 
+	[M.commands.HASHCODE_EQUALS_PROMPT] = function(_, params)
+		require('java.api.generate').generate_hash_code_and_equals(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)
