Refactor builtin (#287)

lua/telescope/builtin/files.lua

@@ -0,0 +1,248 @@
+local actions = require('telescope.actions')
+local finders = require('telescope.finders')
+local make_entry = require('telescope.make_entry')
+local pickers = require('telescope.pickers')
+local previewers = require('telescope.previewers')
+local utils = require('telescope.utils')
+
+local conf = require('telescope.config').values
+
+local flatten = vim.tbl_flatten
+
+local files = {}
+
+files.live_grep = function(opts)
+  local live_grepper = finders.new_job(function(prompt)
+      -- TODO: Probably could add some options for smart case and whatever else rg offers.
+
+      if not prompt or prompt == "" then
+        return nil
+      end
+
+      return flatten { conf.vimgrep_arguments, prompt }
+    end,
+    opts.entry_maker or make_entry.gen_from_vimgrep(opts),
+    opts.max_results
+  )
+
+  pickers.new(opts, {
+    prompt_title = 'Live Grep',
+    finder = live_grepper,
+    previewer = previewers.vimgrep.new(opts),
+    sorter = conf.generic_sorter(opts),
+  }):find()
+end
+
+-- Special keys:
+--  opts.search -- the string to search.
+files.grep_string = function(opts)
+  -- TODO: This should probably check your visual selection as well, if you've got one
+  local search = opts.search or vim.fn.expand("<cword>")
+
+  opts.entry_maker = opts.entry_maker or make_entry.gen_from_vimgrep(opts)
+  opts.word_match = opts.word_match or nil
+
+  pickers.new(opts, {
+    prompt_title = 'Find Word',
+    finder = finders.new_oneshot_job(
+      flatten { conf.vimgrep_arguments, opts.word_match, search},
+      opts
+    ),
+    previewer = previewers.vimgrep.new(opts),
+    sorter = conf.generic_sorter(opts),
+  }):find()
+end
+
+-- TODO: Maybe just change this to `find`.
+--          Support `find` and maybe let people do other stuff with it as well.
+files.find_files = function(opts)
+  local find_command = opts.find_command
+
+  if not find_command then
+    if 1 == vim.fn.executable("fd") then
+      find_command = { 'fd', '--type', 'f' }
+    elseif 1 == vim.fn.executable("fdfind") then
+      find_command = { 'fdfind', '--type', 'f' }
+    elseif 1 == vim.fn.executable("rg") then
+      find_command = { 'rg', '--files' }
+    elseif 1 == vim.fn.executable("find") then
+      find_command = { 'find', '.', '-type', 'f' }
+    end
+  end
+
+  if not find_command then
+    print("You need to install either find, fd, or rg. You can also submit a PR to add support for another file finder :)")
+    return
+  end
+
+  if opts.cwd then
+    opts.cwd = vim.fn.expand(opts.cwd)
+  end
+
+  opts.entry_maker = opts.entry_maker or make_entry.gen_from_file(opts)
+
+  pickers.new(opts, {
+    prompt_title = 'Find Files',
+    finder = finders.new_oneshot_job(
+      find_command,
+      opts
+    ),
+    previewer = previewers.cat.new(opts),
+    sorter = conf.file_sorter(opts),
+  }):find()
+end
+
+local function prepare_match(entry, kind)
+  local entries = {}
+
+  if entry.node then
+      entry["kind"] = kind
+      table.insert(entries, entry)
+  else
+    for name, item in pairs(entry) do
+        vim.list_extend(entries, prepare_match(item, name))
+    end
+  end
+
+  return entries
+end
+
+files.treesitter = function(opts)
+  opts.show_line = utils.get_default(opts.show_line, true)
+
+  local has_nvim_treesitter, _ = pcall(require, 'nvim-treesitter')
+  if not has_nvim_treesitter then
+    print('You need to install nvim-treesitter')
+    return
+  end
+
+  local parsers = require('nvim-treesitter.parsers')
+  if not parsers.has_parser() then
+    print('No parser for the current buffer')
+    return
+  end
+
+  local ts_locals = require('nvim-treesitter.locals')
+  local bufnr = opts.bufnr or vim.api.nvim_get_current_buf()
+
+  local results = {}
+  for _, definitions in ipairs(ts_locals.get_definitions(bufnr)) do
+    local entries = prepare_match(definitions)
+    for _, entry in ipairs(entries) do
+      table.insert(results, entry)
+    end
+  end
+
+  if vim.tbl_isempty(results) then
+    return
+  end
+
+  pickers.new(opts, {
+    prompt_title = 'Treesitter Symbols',
+    finder    = finders.new_table {
+      results = results,
+      entry_maker = make_entry.gen_from_treesitter(opts)
+    },
+    previewer = previewers.vimgrep.new(opts),
+    sorter = conf.generic_sorter(opts),
+  }):find()
+end
+
+files.current_buffer_fuzzy_find = function(opts)
+  local lines = vim.api.nvim_buf_get_lines(0, 0, -1, false)
+  local lines_with_numbers = {}
+  for k, v in ipairs(lines) do
+    table.insert(lines_with_numbers, {k, v})
+  end
+
+  local bufnr = vim.api.nvim_get_current_buf()
+
+  pickers.new(opts, {
+    prompt_title = 'Current Buffer Fuzzy',
+    finder = finders.new_table {
+      results = lines_with_numbers,
+      entry_maker = function(enumerated_line)
+        return {
+          bufnr = bufnr,
+          display = enumerated_line[2],
+          ordinal = enumerated_line[2],
+
+          lnum = enumerated_line[1],
+        }
+      end
+    },
+    sorter = conf.generic_sorter(opts),
+    attach_mappings = function()
+      actions._goto_file_selection:enhance {
+        post = function()
+          local selection = actions.get_selected_entry()
+          vim.api.nvim_win_set_cursor(0, {selection.lnum, 0})
+        end,
+      }
+
+      return true
+    end
+  }):find()
+end
+
+files.tags = function(opts)
+  local ctags_file = opts.ctags_file or 'tags'
+
+  if not vim.loop.fs_open(vim.fn.expand(ctags_file), "r", 438) then
+    print('Tags file does not exists. Create one with ctags -R')
+    return
+  end
+
+  local fd = assert(vim.loop.fs_open(vim.fn.expand(ctags_file), "r", 438))
+  local stat = assert(vim.loop.fs_fstat(fd))
+  local data = assert(vim.loop.fs_read(fd, stat.size, 0))
+  assert(vim.loop.fs_close(fd))
+
+  local results = vim.split(data, '\n')
+
+  pickers.new(opts,{
+    prompt = 'Tags',
+    finder = finders.new_table {
+      results = results,
+      entry_maker = make_entry.gen_from_ctags(opts),
+    },
+    previewer = previewers.ctags.new(opts),
+    sorter = conf.generic_sorter(opts),
+    attach_mappings = function()
+      actions._goto_file_selection:enhance {
+        post = function()
+          local selection = actions.get_selected_entry()
+
+          local scode = string.gsub(selection.scode, '[$]$', '')
+          scode = string.gsub(scode, [[\\]], [[\]])
+          scode = string.gsub(scode, [[\/]], [[/]])
+          scode = string.gsub(scode, '[*]', [[\*]])
+
+          vim.cmd('norm! gg')
+          vim.fn.search(scode)
+          vim.cmd('norm! zz')
+        end,
+      }
+      return true
+    end
+  }):find()
+end
+
+files.current_buffer_tags = function(opts)
+  return files.tags(vim.tbl_extend("force", {only_current_file = true, hide_filename = true}, opts))
+end
+
+
+local function apply_checks(mod)
+  for k, v in pairs(mod) do
+    mod[k] = function(opts)
+      opts = opts or {}
+
+      v(opts)
+    end
+  end
+
+  return mod
+end
+
+return apply_checks(files)