feat: 兼容 ftplugin 配置启动

JavaHello · JavaHello · commit db7fe5c4b2a9 · 2024-12-31T16:34:53.000+08:00
diff --git a/lua/spring_boot.lua b/lua/spring_boot.lua
@@ -1,13 +1,15 @@
+local jdt_ext_jars = {
+  "io.projectreactor.reactor-core.jar",
+  "org.reactivestreams.reactive-streams.jar",
+  "jdt-ls-commons.jar",
+  "jdt-ls-extension.jar",
+  "sts-gradle-tooling.jar",
+}
 vim.g.spring_boot = {
+  autocmd = true,
   jdt_extensions_path = nil,
   -- https://github.com/spring-projects/sts4/blob/7d3d91ecfa6087ae2d0e0f595da61ce8f52fed96/vscode-extensions/vscode-spring-boot/package.json#L33
-  jdt_extensions_jars = {
-    "io.projectreactor.reactor-core.jar",
-    "org.reactivestreams.reactive-streams.jar",
-    "jdt-ls-commons.jar",
-    "jdt-ls-extension.jar",
-    "sts-gradle-tooling.jar",
-  },
+  jdt_extensions_jars = jdt_ext_jars,
 }
 
 local M = {}
@@ -54,7 +56,7 @@ end
 M.java_extensions = function()
   local bundles = {}
   local function bundle_jar(path)
-    for _, jar in ipairs(vim.g.spring_boot.jdt_extensions_jars) do
+    for _, jar in ipairs(vim.g.spring_boot.jdt_extensions_jars or jdt_ext_jars) do
       if vim.endswith(path, jar) then
         return true
       end
diff --git a/lua/spring_boot/config.lua b/lua/spring_boot/config.lua
@@ -20,7 +20,7 @@ local M = {
 
 local function init()
   local spring_boot = require("spring_boot")
-  M = vim.tbl_deep_extend("keep", spring_boot._config, M)
+  M = vim.tbl_deep_extend("keep", spring_boot._config or {}, M)
   spring_boot._config = nil
 end
 init()
diff --git a/lua/spring_boot/launch.lua b/lua/spring_boot/launch.lua
@@ -13,9 +13,10 @@ local root_dir = function()
   return vim.loop.cwd()
 end
 
-local bootls_path = function()
-  if config.ls_path then
-    return config.ls_path
+---@param opts bootls.Config
+local bootls_path = function(opts)
+  if opts.ls_path then
+    return opts.ls_path
   end
   local bls = vscode.find_one("/vmware.vscode-spring-boot-*/language-server")
   if bls then
@@ -27,25 +28,27 @@ M.enable_classpath_listening = function()
   util.boot_execute_command("sts.vscode-spring-boot.enableClasspathListening", { true })
 end
 
-local logfile = function(rt_dir)
+---@param opts bootls.Config
+local logfile = function(opts, rt_dir)
   local lf
-  if config.log_file ~= nil then
-    if type(config.log_file) == "function" then
-      lf = config.log_file(rt_dir)
-    elseif type(config.log_file) == "string" then
-      lf = config.log_file == "" and nil or config.log_file
+  if opts.log_file ~= nil then
+    if type(opts.log_file) == "function" then
+      lf = opts.log_file(rt_dir)
+    elseif type(opts.log_file) == "string" then
+      lf = opts.log_file == "" and nil or opts.log_file
     end
   end
   return lf or "/dev/null"
 end
 
-local function bootls_cmd(rt_dir, java_cmd)
-  local boot_path = bootls_path()
+---@param opts bootls.Config
+local function bootls_cmd(opts, rt_dir, java_cmd)
+  local boot_path = bootls_path(opts)
   if not boot_path then
     vim.notify("Spring Boot LS is not installed", vim.log.levels.WARN)
     return
   end
-  if config.exploded_ls_jar_data then
+  if opts.exploded_ls_jar_data then
     local boot_classpath = {}
     table.insert(boot_classpath, boot_path .. "/BOOT-INF/classes")
     table.insert(boot_classpath, boot_path .. "/BOOT-INF/lib/*")
@@ -58,7 +61,7 @@ local function bootls_cmd(rt_dir, java_cmd)
       "-cp",
       table.concat(boot_classpath, util.is_win and ";" or ":"),
       "-Dsts.lsp.client=vscode",
-      "-Dsts.log.file=" .. logfile(rt_dir),
+      "-Dsts.log.file=" .. logfile(opts, rt_dir),
       "-Dspring.config.location=file:" .. boot_path .. "/BOOT-INF/classes/application.properties",
       -- "-Dlogging.level.org.springframework=DEBUG",
       "org.springframework.ide.vscode.boot.app.BootLanguageServerBootApp",
@@ -76,15 +79,15 @@ local function bootls_cmd(rt_dir, java_cmd)
       "-Xmx1G",
       "-XX:+UseZGC",
       "-Dsts.lsp.client=vscode",
-      "-Dsts.log.file=" .. logfile(rt_dir),
+      "-Dsts.log.file=" .. logfile(opts, rt_dir),
       "-jar",
       server_jar[1],
     }
     return cmd
   end
 end
 
-local ls_config = {
+M.ls_config = {
   name = "spring-boot",
   filetypes = { "java", "yaml", "jproperties" },
   -- root_dir = root_dir,
@@ -112,63 +115,80 @@ local ls_config = {
     return filetype
   end,
 }
-classpath.register_classpath_service(ls_config)
-java_data.register_java_data_service(ls_config)
+classpath.register_classpath_service(M.ls_config)
+java_data.register_java_data_service(M.ls_config)
 
 vim.lsp.commands["vscode-spring-boot.ls.start"] = function(_, _, _)
   M.enable_classpath_listening()
   return {}
 end
 
-ls_config.handlers["sts/highlight"] = function() end
-ls_config.handlers["sts/moveCursor"] = function(err, result, ctx, config)
+M.ls_config.handlers["sts/highlight"] = function() end
+M.ls_config.handlers["sts/moveCursor"] = function(err, result, ctx, config)
   -- TODO: move cursor
   return { applied = true }
 end
-
-M.setup = function(_)
-  ls_config = vim.tbl_deep_extend("keep", ls_config, config.server)
-  local capabilities = ls_config.capabilities or vim.lsp.protocol.make_client_capabilities()
+M._update_config = false
+M.update_config = function(opts)
+  if M._update_config then
+    return
+  end
+  M._update_config = true
+  M.ls_config = vim.tbl_deep_extend("keep", M.ls_config, opts.server)
+  local capabilities = M.ls_config.capabilities or vim.lsp.protocol.make_client_capabilities()
   capabilities.workspace = {
     executeCommand = { value = true },
   }
-  ls_config.capabilities = capabilities
-  if not ls_config.root_dir then
-    ls_config.root_dir = root_dir()
+  M.ls_config.capabilities = capabilities
+  if not M.ls_config.root_dir then
+    M.ls_config.root_dir = root_dir()
   end
-  ls_config.cmd = (ls_config.cmd and #ls_config.cmd > 0) and ls_config.cmd
-    or bootls_cmd(ls_config.root_dir, config.java_cmd)
-  if not ls_config.cmd then
+  M.ls_config.cmd = (M.ls_config.cmd and #M.ls_config.cmd > 0) and M.ls_config.cmd
+    or bootls_cmd(opts, M.ls_config.root_dir, opts.java_cmd)
+  if not M.ls_config.cmd then
     return
   end
-  ls_config.init_options.workspaceFolders = ls_config.root_dir
-  local on_init = ls_config.on_init
-  ls_config.on_init = function(client, ctx)
+  M.ls_config.init_options.workspaceFolders = M.ls_config.root_dir
+  local on_init = M.ls_config.on_init
+  M.ls_config.on_init = function(client, ctx)
     util.boot_ls_init(client, ctx)
     if on_init then
       on_init(client, ctx)
     end
   end
-  local group = vim.api.nvim_create_augroup("spring_boot_ls", { clear = true })
-  vim.api.nvim_create_autocmd({ "FileType" }, {
-    group = group,
-    pattern = { "java", "yaml", "jproperties" },
-    desc = "Spring Boot Language Server",
-    callback = function(e)
-      if e.file == "java" and vim.bo[e.buf].buftype == "nofile" then
-        return
-      end
-      if vim.endswith(e.file, "pom.xml") then
-        return
-      end
-      if vim.endswith(e.file, ".yaml") or vim.endswith(e.file, ".yml") then
-        if not util.is_application_yml_file(e.file) then
-          return
-        end
-      end
-      vim.lsp.start(ls_config)
-    end,
-  })
+end
+
+M.setup = function(_)
+  M.update_config(config)
+  if vim.g.spring_boot.autocmd then
+    local group = vim.api.nvim_create_augroup("spring_boot_ls", { clear = true })
+    vim.api.nvim_create_autocmd({ "FileType" }, {
+      group = group,
+      pattern = { "java", "yaml", "jproperties" },
+      desc = "Spring Boot Language Server",
+      callback = function(_)
+        M.start(M.ls_config)
+      end,
+    })
+  end
+end
+M.start = function(opts)
+  local buf = vim.api.nvim_get_current_buf()
+  local filename = vim.uri_from_bufnr(buf)
+  if vim.endswith(filename, "pom.xml") then
+    return
+  end
+  if vim.endswith(filename, ".yaml") or vim.endswith(filename, ".yml") then
+    if not util.is_application_yml_file(filename) then
+      return
+    end
+  end
+  if "jproperties" == vim.bo[buf].filetype then
+    if not util.is_application_properties_file(filename) then
+      return
+    end
+  end
+  vim.lsp.start(opts)
 end
 
 -- 参考资料
