基本

FileTypeに対してautocmdをしかける形で、require("tsnode-marker").set_automarkを実行します。

たとえば、Markdownファイルのコードブロック中のコードの背景色を変更するなら、以下のように書けます。

vim.api.nvim_create_autocmd("FileType", {
  group = vim.api.nvim_create_augroup("tsnode-marker-markdown", {}),
  pattern = "markdown",
  callback = function(ctx)
    require("tsnode-marker").set_automark(ctx.buf, {
      target = { "code_fence_content" }, -- list of target node types
      hl_group = "CursorLine", -- highlight group
    })
  end,
})

targetに指定している{ "code_fence_content" }が背景色を変更する範囲です。 teesitterによるキャプチャの名前をリストで複数記述できます。

指定したい範囲のキャプチャ名が分からない場合はNeovim 0.9以降に導入された:InspectTreeを使ってみましょう。以下のように、キャプチャとその範囲を閲覧できます。

(section) ; [1:1 - 3]
 (fenced_code_block) ; [1:1 - 3]
  (fenced_code_block_delimiter) ; [1:1 - 4:0]
  (info_string) ; [1:5 - 9:0]
   (language) ; [1:5 - 9:0]
  (block_continuation) ; [2:1 - 1:1]
  (code_fence_content) ; [2:1 - 1:2]
   (command) ; [2:1 - 5:1]
    name: (command_name) ; [2:1 - 5:1]
     (word) ; [2:1 - 5:1]
   (block_continuation) ; [3:1 - 1:2]
  (fenced_code_block_delimiter) ; [3:1 - 4:2]

発展

キャプチャ名による素朴なハイライト判定以外にも、ユーザー定義関数による複雑な判定も可能です。

これにより、関数中に定義された関数をハイライトするといった、複雑な操作を実現できます。

---ノードが関数定義か判定する関数
---@param node tsnode
---@return bool
local function is_def(node)
  return vim.tbl_contains({
    "func_literal",
    "function_declaration",
    "function_definition",
    "method_declaration",
    "method_definition",
  }, node:type())
end

---ノードが関数定義中の関数定義か判定する関数
---@param _ bufnr
---@param node tsnode
---@return bool
local function is_nested_def(_, node)
  if not is_def(node) then
    return false
  end
  local parent = node:parent()
  while parent do
    if is_def(parent) then
      return true
    end
    parent = parent:parent()
  end
  return false
end

-- autocmd
vim.api.nvim_create_autocmd("FileType", {
  group = vim.api.nvim_create_augroup("tsnode-marker-nested-func", {}),
  pattern = { "lua", "python", "go" },
  callback = function(ctx)
    require("tsnode-marker").set_automark(ctx.buf, {
      target = is_nested_def,
      hl_group = "CursorLine",
    })
  end,
})