feat: add toString code action

s1n7ax · s1n7ax · commit 4b1e1bd52061 · 2024-07-15T00:32:28.000+05:30
diff --git a/lua/java/api/generate.lua b/lua/java/api/generate.lua
@@ -1,15 +1,15 @@
+local runner = require('async.runner')
+
 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 instance = require('java.utils.instance_factory')
 	local get_error_handler = require('java.handlers.error')
 	local ui = require('java.utils.ui')
 
-	return async(function()
-			local jdtls = JdtlsClient(get_client())
+	return runner(function()
+			local jdtls = instance.jdtls_client()
 			local status = jdtls:java_check_constructors_status(params.params)
 
 			if not status or not status.constructors then
@@ -52,4 +52,49 @@ function M.generate_constructor(params)
 		.run()
 end
 
+---@param params nvim.CodeActionParamsResponse
+function M.generate_to_string(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_to_string_status(params.params)
+
+			if status.exists then
+				local prompt = string.format(
+					'Method "toString()" already exists in the Class %s. Do you want to replace the implementation?',
+					status.type
+				)
+				local choice = ui.select(prompt, { 'Replace', 'Cancel' })
+
+				if choice ~= 'Replace' then
+					return
+				end
+			end
+
+			local fields = ui.multi_select(
+				'Select the fields to include in the toString() method.',
+				status.fields,
+				function(field)
+					return field.name
+				end
+			)
+
+			if not fields then
+				return
+			end
+
+			local edit = jdtls:java_generate_to_string({
+				context = params.params,
+				fields = fields,
+			})
+
+			vim.lsp.util.apply_workspace_edit(edit, 'utf-8')
+		end)
+		.catch(get_error_handler('Generating to string failed'))
+		.run()
+end
+
 return M
diff --git a/lua/java/commands/init.lua b/lua/java/commands/init.lua
@@ -1,8 +1,5 @@
 local log = require('java.utils.log')
 local notify = require('java-core.utils.notify')
-local ui = require('java.utils.ui')
-local get_error_handler = require('java.handlers.error')
-local runner = require('async.runner')
 
 local M = {}
 
@@ -124,6 +121,10 @@ M.handlers = {
 		require('java.api.generate').generate_constructor(params)
 	end,
 
+	[M.commands.GENERATE_TOSTRING_PROMPT] = function(_, params)
+		require('java.api.generate').generate_to_string(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)
diff --git a/lua/java/utils/instance_factory.lua b/lua/java/utils/instance_factory.lua
@@ -0,0 +1,25 @@
+local class = require('java-core.utils.class')
+
+local M = class()
+
+---@private
+---@return vim.lsp.Client
+local function get_client()
+	local clients = vim.lsp.get_clients({ name = 'jdtls' })
+
+	if #clients < 1 then
+		local message = string.format('No jdtls client found to instantiate class')
+		require('java-core.utils.notify').error(message)
+		require('java.utils.log').error(message)
+		error(message)
+	end
+
+	return clients[1]
+end
+
+---@return java-core.JdtlsClient
+function M.jdtls_client()
+	return require('java-core.ls.clients.jdtls-client')(get_client())
+end
+
+return M
