feat: autogen metadata table(s) (#159)

Co-authored-by: William Boman <william@redwill.se>

1 files changed, 79 insertions, 0 deletions

diff --git a/scripts/autogen_metadata.lua b/scripts/autogen_metadata.lua
new file mode 100644
index 00000000..5bee63f7
--- /dev/null
+++ b/scripts/autogen_metadata.lua
@@ -0,0 +1,79 @@
+local uv = vim.loop
+local Path = require "nvim-lsp-installer.path"
+local Data = require "nvim-lsp-installer.data"
+
+local coalesce = Data.coalesce
+
+package.loaded["nvim-lsp-installer.servers"] = nil
+package.loaded["nvim-lsp-installer.fs"] = nil
+local servers = require "nvim-lsp-installer.servers"
+
+local generated_dir = Path.concat { vim.fn.getcwd(), "lua", "nvim-lsp-installer", "_generated" }
+
+print("Creating directory " .. generated_dir)
+vim.fn.mkdir(generated_dir, "p")
+
+for _, file in ipairs(vim.fn.glob(generated_dir .. "*", 1, 1)) do
+    print("Deleting " .. file)
+    vim.fn.delete(file)
+end
+
+local function write_file(path, txt, flag)
+    uv.fs_open(path, flag, 438, function(open_err, fd)
+        assert(not open_err, open_err)
+        uv.fs_write(fd, txt, -1, function(write_err)
+            assert(not write_err, write_err)
+            uv.fs_close(fd, function(close_err)
+                assert(not close_err, close_err)
+            end)
+        end)
+    end)
+end
+
+local function get_supported_filetypes(server)
+    local configs = require "lspconfig/configs"
+    local lspconfig_server_ok = pcall(require, ("lspconfig/" .. server.name))
+    if not lspconfig_server_ok then
+        -- This is expected behavior for servers that does not exist in lspconfig.
+        print(("Unable to import lspconfig/%s, continuing..."):format(server.name))
+    end
+    local default_options = server:get_default_options()
+    local filetypes = coalesce(
+        -- nvim-lsp-installer options has precedence
+        default_options.filetypes,
+        lspconfig_server_ok and configs[server.name].document_config.default_config.filetypes,
+        {}
+    )
+    -- it's probably still not safe to do this in runtime, but just in case
+    package.loaded["lspconfig/configs"] = nil
+    return filetypes
+end
+
+local function generate_metadata_table()
+    local metadata = {}
+
+    local function create_metadata_entry(server)
+        return { filetypes = get_supported_filetypes(server) }
+    end
+
+    local available_servers = servers.get_available_servers()
+    for _, server in pairs(available_servers) do
+        metadata[server.name] = create_metadata_entry(server)
+    end
+    print(string.format("found [%s] configurations", #vim.tbl_keys(metadata)))
+
+    return metadata
+end
+
+local mt = generate_metadata_table()
+
+local metadata_file_lua = Path.concat { generated_dir, "metadata.lua" }
+write_file(
+    metadata_file_lua,
+    table.concat({
+        "-- THIS FILE IS GENERATED. DO NOT EDIT MANUALLY.",
+        "-- stylua: ignore start",
+        "return " .. vim.inspect(mt),
+    }, "\n"),
+    "w"
+)