Use vim.b to store buffer state

m4xshen · Mar 21, 2023 · 531c68f · 531c68f
1 parent cb7875f
commit 531c68f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/lua/smartcolumn.lua b/lua/smartcolumn.lua
@@ -35,7 +35,6 @@ local function exceed(buf, win, min_colorcolumn)
    return not is_disabled() and max_column > min_colorcolumn
 end
 
-local buf_states = {}
 local function update()
    local buf_filetype = vim.api.nvim_buf_get_option(0, "filetype")
    local colorcolumns =
@@ -56,8 +55,8 @@ local function update()
       local buf = vim.api.nvim_win_get_buf(win)
       if buf == current_buf then
          local current_state = exceed(buf, win, min_colorcolumn)
-         if current_state ~= buf_states[buf] then
-            buf_states[buf] = current_state
+         if current_state ~= vim.b.prev_state then
+            vim.b.prev_state = current_state
             if current_state then
                if type(colorcolumns) == "table" then
                   vim.wo[win].colorcolumn = table.concat(colorcolumns, ",")