How can I use ast-gerp with snacks picker?

[Flamme13]

How can I use ast-grep (https://github.com/ast-grep/ast-grep) for searching?

[drowning-cat]

Technically you can use whatever you want. This is a simplified version of the actual grep finder:

With grep

vim.keymap.set('n', '<space>sg', function()
  Snacks.picker.pick {
    notify = false,
    format = 'file',
    show_empty = true,
    live = true,
    supports_live = true,
    finder = function(opts, ctx)
      local cmd = 'rg'
      local args = {
        '--color=never',
        '--line-number',
        '--column',
        '--max-columns=500',
        '-g=!.git',
      }
      vim.list_extend(args, opts.args or {}) ---@diagnostic disable-line
      local pattern, pargs = Snacks.picker.util.parse(ctx.filter.search)
      vim.list_extend(args, pargs)
      table.insert(args, '--')
      table.insert(args, pattern)
      return require('snacks.picker.source.proc').proc({
        opts,
        {
          cmd = cmd,
          args = args,
          transform = function(item)
            local file, line, col, text = item.text:match '^(.+):(%d+):(%d+):(.*)$'
            if not file then
              return false
            else
              item.line = text
              item.file = file
              item.pos = { tonumber(line), tonumber(col) - 1 }
            end
          end,
        },
      }, ctx)
    end,
  }
end)

---

And here is the corresponding ast-grep picker derived from that:

With ast-grep (bad parsing)

vim.keymap.set('n', '<space>sg', function()
  Snacks.picker.pick {
    format = 'file',
    show_empty = true,
    live = true,
    finder = function(opts, ctx)
      local cmd = 'ast-grep'
      local args = {
        'run',
        '--context=0',
        '--heading=never',
      }
      vim.list_extend(args, opts.args or {}) ---@diagnostic disable-line
      local pattern, pargs = Snacks.picker.util.parse(ctx.filter.search)
      vim.list_extend(args, pargs)
      table.insert(args, string.format('--pattern=%s', pattern))
      return require('snacks.picker.source.proc').proc({
        opts,
        {
          cmd = cmd,
          args = args,
          transform = function(item)
            local file, line, text = item.text:match '^(.+):(%d+):(.*)$'
            if not file then
              return false
            else
              item.line = text
              item.file = file
              item.pos = { tonumber(line), 0 }
            end
          end,
        },
      }, ctx)
    end,
  }
end)

With ast-grep

vim.keymap.set('n', '<space>sg', function()
  Snacks.picker.pick {
    format = 'file',
    notify = false, -- Also prevents error when searching with additional arguments
    show_empty = true,
    live = true,
    supports_live = true,
    -- hidden = true,
    -- ignored = true,
    ---@param opts snacks.picker.grep.Config
    finder = function(opts, ctx)
      local cmd = 'ast-grep'
      local args = { 'run', '--color=never', '--json=stream' }
      if vim.fn.has 'win32' == 1 then
        cmd = 'sg'
      end
      if opts.hidden then
        table.insert(args, '--no-ignore=hidden')
      end
      if opts.ignored then
        table.insert(args, '--no-ignore=vcs')
      end
      local pattern, pargs = Snacks.picker.util.parse(ctx.filter.search)
      table.insert(args, string.format('--pattern=%s', pattern))
      vim.list_extend(args, pargs)
      return require('snacks.picker.source.proc').proc({
        opts,
        {
          cmd = cmd,
          args = args,
          transform = function(item)
            local entry = vim.json.decode(item.text)
            if vim.tbl_isempty(entry) then
              return false
            else
              local start = entry.range.start
              item.cwd = svim.fs.normalize(opts and opts.cwd or vim.uv.cwd() or '.') or nil
              item.file = entry.file
              item.line = entry.text
              item.pos = { tonumber(start.line) + 1, tonumber(start.column) }
            end
          end,
        },
      }, ctx)
    end,
  }
end)

[alcroito]

That seems to work fine, but the ctrl-g action from a regular picker like grep doesn't trigger. Is that because the custom picker doesn't inherit any of the default picker keys -> action bindings? How would one add them back?

[alcroito]

Looks like it needed the supports_live option set.

[Flamme13]

Thank you!

[Flamme13]

Is it possible to configure the output so that only one line is displayed per match, even if the found function spans multiple lines?

[drowning-cat]

Where?

[Flamme13]

My apologies, that was likely my oversight. When I searched for a React component using the query <ComponentName, the system returned every line where that component appears.

[drowning-cat]

I've updated the code, see #1966 (comment).

[Flamme13]

Wow! Now it's working amazingly. You are really a pro. Thank you so much!!!