Any way to make a hydra for Tree-sitter incremental selection?

[Hubro]

I have Tree-sitter incremental selection mapped to <M-8>/<M-9> which is a bit cumbersome. This sucks, because it's a super nice feature. I want to make a hydra for it to encourage me to use it more often.

I want to initialize selection on <Leader>k, then decrease/increase the selection with j/k.

Here's where I'm at:

hydra({
  name = "Tree-sitter incremental selection",
  mode = { "n", "v", "x" },
  body = "<Leader>",
  heads = {
    { "j", "<M-8>", { desc = "Decrease selection", mode = "x", } },
    { "k", "<M-9>", { desc = "Increase selection", mode = "x", } },
    { "k", "<M-9>", { desc = "Initiate selection", mode = "n", } },
  },
  config = {
    desc = "Tree-sitter incremental selection",
  }
})

It doesn't do anything, when I press <Leader>k the hydra doesn't even activate.

If I change mode to just n, the hydra successfully activates, but pressing j/k doesn't do anything.

Any idea how to make this work?

[Hubro]

After perusing the source code of nvim-treesitter, I found the underlying functions for my <M-8>/<M-9> keybinds. After a bit of experimentation I got it working:

local ts_select = require("nvim-treesitter.incremental_selection")

local ts_select_hydra = hydra({
  name = "Tree-sitter incremental selection",
  mode = "x",
  heads = {
    { "j",     ts_select.node_decremental, { desc = "Decrease selection", mode = "x", } },
    { "k",     ts_select.node_incremental, { desc = "Increase selection", mode = "x", } },
    { "<Esc>", "<Esc>",                    { desc = "Exit", mode = "x", exit_before = true } },
  },
  config = {
    desc = "Tree-sitter incremental selection",
  }
})

vim.keymap.set({ "n", "x" }, "<Leader>k", function()
  ts_select.init_selection()
  ts_select_hydra:activate()
end, { desc = "Tree-sitter incremental selection" })

Maybe this is possible to achieve without a separate keybind for activating the hydra, but I couldn't figure that out.