nvim-java · s1n7ax · Dec 4, 2023 · Dec 4, 2023 · Dec 4, 2023
diff --git a/doc/server-capabilities.md b/doc/server-capabilities.md
diff --git a/lua/java-core.lua b/lua/java-core.lua
diff --git a/lua/java-core/ls/servers/jdtls/init.lua b/lua/java-core/ls/servers/jdtls/init.lua
@@ -1,6 +1,7 @@
 local config = require('java-core.ls.servers.jdtls.config')
 local log = require('java-core.utils.log')
 local mason = require('java-core.utils.mason')
+local mason_reg = require('mason-registry')
 local path = require('java-core.utils.path')
 local plugins = require('java-core.ls.servers.jdtls.plugins')
 local util = require('lspconfig.util')
@@ -13,6 +14,7 @@ local M = {}
 ---Ex:- { 'pom.xml', 'build.gradle', '.git' }
 ---@field jdtls_plugins string[] list of jdtls plugins to load on start up
 ---Ex:- { 'java-test', 'java-debug-adapter' }
+---@field use_mason_jdk boolean whether to use mason jdk to load jdtls or not
 
 ---Returns a configuration for jdtls that you can pass into the setup of nvim-lspconfig
 ---@param opts JavaCoreGetConfigOptions
@@ -36,6 +38,21 @@ function M.get_config(opts)
 		'-javaagent:' .. lombok_path,
 	}
 
+	if opts.use_mason_jdk then
+		local jdk = mason_reg.get_package('openjdk-17')
+
+		if jdk:is_installed() then
+			local java_home =
+				vim.fn.glob(path.join(jdk:get_install_path(), '/jdk-17*'))
+			local java_bin = path.join(java_home, '/bin')
+
+			base_config.cmd_env = {
+				['PATH'] = vim.fn.getenv('PATH') .. ':' .. java_bin,
+				['JAVA_HOME'] = java_home,
+			}
+		end
+	end
+
 	base_config.root_dir = M.get_root_finder(opts.root_markers)
 	base_config.init_options.bundles = plugin_paths
 	base_config.init_options.workspace = workspace.get_default_workspace()