feat(nut): add tab.tablist.number

Author: MunifTanjim

lua/nougat/nut/tab/tablist/number.lua

@@ -0,0 +1,43 @@
+local core = require("nougat.core")
+local Item = require("nougat.item")
+
+-- re-used tables
+local o_label_opts = { tabnr = nil }
+local o_label_parts = {}
+
+---@param ctx nougat.nut.tab.tablist_ctx
+local function content(_, ctx)
+  o_label_opts.tabnr = ctx.tab.tabnr
+  local parts_len = core.add_label(tostring(ctx.tab.tabnr), o_label_opts, o_label_parts, 0)
+  return table.concat(o_label_parts, nil, 1, parts_len)
+end
+
+local hl = {}
+
+function hl.diagnostic()
+  return require("nougat.nut.tab.tablist.label").hl.diagnostic()
+end
+
+local mod = {
+  hl = hl,
+}
+
+---@param config nougat_item_config__nil
+function mod.create(config)
+  local item = Item({
+    priority = config.priority,
+    hidden = config.hidden,
+    hl = config.hl,
+    sep_left = config.sep_left,
+    prefix = config.prefix,
+    content = content,
+    suffix = config.suffix,
+    sep_right = config.sep_right,
+    on_click = config.on_click,
+    context = config.context,
+  })
+
+  return item
+end
+
+return mod

tests/nougat/nut/tab/tablist_spec.lua

@@ -33,6 +33,8 @@ describe("nut.tab.tablist", function()
       label = require("nougat.nut.tab.tablist.label").create,
       label_hl = require("nougat.nut.tab.tablist.label").hl,
       modified = require("nougat.nut.tab.tablist.modified").create,
+      number = require("nougat.nut.tab.tablist.number").create,
+      number_hl = require("nougat.nut.tab.tablist.number").hl,
     }
 
     tabs = {}
@@ -394,6 +396,64 @@ describe("nut.tab.tablist", function()
     end)
   end)
 
+  describe("number", function()
+    it("handles tab move", function()
+      vim.api.nvim_buf_set_name(tabs[1].bufnr, "_A_")
+      vim.api.nvim_buf_set_name(add_tab().bufnr, "_B_")
+      vim.api.nvim_buf_set_name(add_tab().bufnr, "_C_")
+
+      bar:add_item(tablist.tabs({
+        active_tab = {
+          content = {
+            tablist.number({}),
+          },
+        },
+        inactive_tab = {
+          content = {
+            tablist.number({}),
+          },
+        },
+      }))
+
+      t.eq(
+        bar:generate(ctx),
+        table.concat({
+          "%#bg_0a0b10_fg_c4c6cd_b#",
+          "%1T1%T",
+          "%#bg_0a0b10_fg_c4c6cd_#",
+          "%#bg_0a0b10_fg_c4c6cd_#",
+          "%2T2%T",
+          "%#bg_0a0b10_fg_c4c6cd_#",
+          "%#bg_0a0b10_fg_c4c6cd_#",
+          "%3T3%T",
+          "%#bg_0a0b10_fg_c4c6cd_#",
+        })
+      )
+
+      vim.api.nvim_set_current_tabpage(tabs[1].tabid)
+      vim.cmd.tabmove(2)
+
+      t.eq(
+        bar:generate(ctx),
+        table.concat({
+          "%#bg_0a0b10_fg_c4c6cd_#",
+          "%1T1%T",
+          "%#bg_0a0b10_fg_c4c6cd_#",
+          "%#bg_0a0b10_fg_c4c6cd_b#",
+          "%2T2%T",
+          "%#bg_0a0b10_fg_c4c6cd_#",
+          "%#bg_0a0b10_fg_c4c6cd_#",
+          "%3T3%T",
+          "%#bg_0a0b10_fg_c4c6cd_#",
+        })
+      )
+    end)
+
+    it(".hl.diagnostic()", function()
+      t.ref(tablist.number_hl.diagnostic(), tablist.label_hl.diagnostic())
+    end)
+  end)
+
   it("diagnostic_count", function()
     vim.api.nvim_buf_set_name(tabs[1].bufnr, "_A_")
     vim.api.nvim_buf_set_name(add_tab().bufnr, "_B_")