aboutsummaryrefslogtreecommitdiff
path: root/runtime/lua/vim/filetype/options.lua
blob: 2a28b5a8e3a6459e9cd93451dc5a8a8d76a75b0b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
local api = vim.api

local M = {}

local function get_ftplugin_runtime(filetype)
  local files = api.nvim__get_runtime({
    string.format('ftplugin/%s.vim', filetype),
    string.format('ftplugin/%s_*.vim', filetype),
    string.format('ftplugin/%s/*.vim', filetype),
    string.format('ftplugin/%s.lua', filetype),
    string.format('ftplugin/%s_*.lua', filetype),
    string.format('ftplugin/%s/*.lua', filetype),
  }, true, {}) --[[@as string[] ]]

  local r = {} ---@type string[]
  for _, f in ipairs(files) do
    -- VIMRUNTIME should be static so shouldn't need to worry about it changing
    if not vim.startswith(f, vim.env.VIMRUNTIME) then
      r[#r + 1] = f
    end
  end
  return r
end

-- Keep track of ftplugin files
local ftplugin_cache = {} ---@type table<string,table<string,integer>>

-- Keep track of total number of FileType autocmds
local ft_autocmd_num ---@type integer?

-- Keep track of filetype options
local ft_option_cache = {} ---@type table<string,table<string,any>>

--- @param path string
--- @return integer
local function hash(path)
  local mtime0 = vim.uv.fs_stat(path).mtime
  return mtime0.sec * 1000000000 + mtime0.nsec
end

--- Only update the cache on changes to the number of FileType autocmds
--- and changes to any ftplugin/ file. This isn't guaranteed to catch everything
--- but should be good enough.
--- @param filetype string
local function update_ft_option_cache(filetype)
  local new_ftautos = #api.nvim_get_autocmds({ event = 'FileType' })
  if new_ftautos ~= ft_autocmd_num then
    -- invalidate
    ft_option_cache[filetype] = nil
    ft_autocmd_num = new_ftautos
  end

  local files = get_ftplugin_runtime(filetype)

  ftplugin_cache[filetype] = ftplugin_cache[filetype] or {}

  if #files ~= #vim.tbl_keys(ftplugin_cache[filetype]) then
    -- invalidate
    ft_option_cache[filetype] = nil
    ftplugin_cache[filetype] = {}
  end

  for _, f in ipairs(files) do
    local mtime = hash(f)
    if ftplugin_cache[filetype][f] ~= mtime then
      -- invalidate
      ft_option_cache[filetype] = nil
      ftplugin_cache[filetype][f] = mtime
    end
  end
end

--- @private
--- @param filetype string Filetype
--- @param option string Option name
--- @return string|integer|boolean
function M.get_option(filetype, option)
  update_ft_option_cache(filetype)

  if not ft_option_cache[filetype] or not ft_option_cache[filetype][option] then
    ft_option_cache[filetype] = ft_option_cache[filetype] or {}
    ft_option_cache[filetype][option] = api.nvim_get_option_value(option, { filetype = filetype })
  end

  return ft_option_cache[filetype][option]
end

return M