Get simple rg example to work

1 files changed, 26 insertions, 327 deletions

diff --git a/lua/telescope/init.lua b/lua/telescope/init.lua
index fc2b037..76abb5a 100644
--- a/lua/telescope/init.lua
+++ b/lua/telescope/init.lua
@@ -1,338 +1,37 @@
-package.loaded['popup'] = nil
-package.loaded['popup.border'] = nil
-package.loaded['popup.init'] = nil
-
 -- TODO: Debounce preview window maybe
+-- TODO: Make filters
+--          "fzf --filter"
+--           jobstart() -> | fzf --filter "input on prompt"
+
+local finders = require('telescope.finders')
+local pickers = require('telescope.pickers')
+local previewers = require('telescope.previewers')
+local state = require('telescope.state')
+
+local telescope = {
+  finders = finders,
+  pickers = pickers,
+  previewers = previewers,
+  state = state,
+}
 
-local a = vim.api
-
-local popup = require('popup')
-
-local telescope = {}
-
-local ns_telescope = a.nvim_create_namespace('telescope')
-local ns_telescope_highlight = a.nvim_create_namespace('telescope_highlight')
-
-local Finder = {}
-Finder.__index = Finder
-
-function Finder:new(fn_command)
-  -- TODO: Add config for:
-  --        - cwd
-
-  -- TODO:
-  -- - `types`
-  --    job
-  --    pipe
-  --        vim.loop.new_pipe (stdin / stdout). stdout => filter pipe
-  --        rg huge_search | fzf --filter prompt_is > buffer. buffer could do stuff do w/ preview callback
-  --    string
-  --    list
-  --    ...
-  return setmetatable({
-    fn_command = fn_command,
-    job_id = -1,
-  }, Finder)
-end
-
-function Finder:display_results(win, bufnr, prompt)
-  if self.job_id > 0 then
-    -- Make sure we kill old jobs.
-    vim.fn.jobstop(self.job_id)
-  end
-
-  self.job_id = vim.fn.jobstart(self.fn_command(prompt), {
-    -- TODO: Decide if we want this or don't want this.
-    stdout_buffered = true,
-    on_stdout = function(_, data, _)
-      a.nvim_buf_set_lines(bufnr, -1, -1, false, data)
-    end,
-
-    on_exit = function()
-      -- TODO: Add possibility to easily highlight prompt within buffer
-      -- without having to do weird stuff and with it actually working...
-      if false then
-        vim.fn.matchadd("Type", "\\<" .. prompt .. "\\>", 1, -1, {window = win})
-      end
-    end,
-  })
-
-  --[[
-  local function get_rg_results(bufnr, search_string)
-    local start_time = vim.fn.reltime()
-
-    vim.fn.jobstart(string.format('rg %s', search_string), {
-      cwd = '/home/tj/build/neovim',
-
-      on_stdout = function(job_id, data, event)
-        vim.api.nvim_buf_set_lines(bufnr, -1, -1, false, data)
-      end,
-
-      on_exit = function()
-        print("Finished in: ", vim.fn.reltimestr(vim.fn.reltime(start_time)))
-      end,
-
-      stdout_buffer = true,
-    })
-  end
-  --]]
-end
-
-local Previewer = {}
-Previewer.__index = Previewer
-
-function Previewer:new(fn)
-  return setmetatable({
-    fn = fn,
-  }, Previewer)
-end
-
-local picker_finders = {}
-
-local Picker = {}
-Picker.__index = Picker
-
-function Picker:new(opts)
-  opts = opts or {}
-  return setmetatable({
-    filter = opts.filter,
-    previewer = opts.previewer,
-    maps = opts.maps,
-  }, Picker)
-end
-
-local hack_status = {}
-
-local previewers = {}
-
-function Picker:find(finder)
-  local prompt_string = 'MUNITER > '
-  -- Create three windows:
-  -- 1. Prompt window
-  -- 2. Options window
-  -- 3. Preview window
-
-  local width = 100
-  local col = 10
-  local prompt_line = 50
-
-  local result_height = 25
-  local prompt_height = 1
-
-  -- TODO: Add back the borders after fixing some stuff in popup.nvim
-  local results_win = popup.create('', {
-    height = result_height,
-    minheight = result_height,
-    width = width,
-    line = prompt_line - 2 - result_height,
-    col = col,
-    -- border = {},
-    enter = false,
-  })
-  local results_bufnr = a.nvim_win_get_buf(results_win)
-  picker_finders[results_bufnr] = finder
-
-  local preview_win = popup.create('', {
-    height = result_height + prompt_height + 4,
-    minheight = result_height + prompt_height + 4,
-    width = 100,
-    line = prompt_line - 2 - result_height,
-    col = col + width + 2,
-    -- border = {},
-    enter = false,
-    highlight = false,
-  })
-  local preview_bufnr = a.nvim_win_get_buf(preview_win)
-
-  -- TODO: For some reason, highlighting is kind of weird on these windows.
-  --        It may actually be my colorscheme tho...
-  a.nvim_win_set_option(preview_win, 'winhl', '')
-
-  -- TODO: We need to center this and make it prettier...
-  local prompt_win = popup.create('', {
-    height = prompt_height,
-    width = width,
-    line = prompt_line,
-    col = col,
-    border = {},
-  })
-  local prompt_bufnr = a.nvim_win_get_buf(prompt_win)
-
-  a.nvim_buf_set_option(prompt_bufnr, 'buftype', 'prompt')
-  vim.fn.prompt_setprompt(prompt_bufnr, prompt_string)
-
-  -- TODO: Please use the cool autocmds once you get off your lazy bottom and finish the PR ;)
-  local autocmd_string = string.format(
-    [[  autocmd TextChanged,TextChangedI <buffer> :lua __TelescopeOnChange(%s, "%s", %s, %s)]],
-    prompt_bufnr,
-    prompt_string,
-    results_bufnr,
-    results_win)
-
-  local buf_close = string.format(
-    [[  autocmd BufLeave <buffer> :bd! %s ]],
-    prompt_bufnr)
-
-  local results_close = string.format(
-    [[ autocmd BufLeave <buffer> :call nvim_win_close(%s, v:true) ]],
-    results_win)
-
-  local preview_close = string.format(
-    [[ autocmd BufLeave <buffer> :call nvim_win_close(%s, v:true) ]],
-    preview_win)
-
-  vim.cmd([[augroup PickerCommands]])
-  vim.cmd([[  au!]])
-  vim.cmd(    autocmd_string)
-  vim.cmd(    buf_close)
-  vim.cmd(    results_close)
-  vim.cmd(    preview_close)
-  vim.cmd([[augroup END]])
-
-  previewers[prompt_bufnr] = self.previewer
-
-  -- TODO: Clear this hack status stuff when closing
-  hack_status[prompt_bufnr] = {
-    prompt_bufnr = prompt_bufnr,
-    prompt_win = prompt_win,
-    results_bufnr = results_bufnr,
-    results_win = results_win,
-    preview_bufnr = preview_bufnr,
-    preview_win = preview_win,
-  }
-
-  local function default_mapper(map_key, table_key)
-    a.nvim_buf_set_keymap(
-      prompt_bufnr,
-      'i',
-      map_key,
-      string.format(
-        [[<C-O>:lua __TelescopeMapping(%s, %s, '%s')<CR>]],
-        prompt_bufnr,
-        results_bufnr,
-        table_key
-        ),
-      {
-        silent = true,
-      }
-    )
-  end
-
-  default_mapper('<c-n>', 'control-n')
-  default_mapper('<c-p>', 'control-p')
-  default_mapper('<CR>', 'enter')
-
-  vim.cmd [[startinsert]]
-end
-
--- TODO: All lower case mappings
-local telescope_selections = {}
-
-local function update_current_selection(prompt_bufnr, results_bufnr, row)
-  a.nvim_buf_clear_namespace(results_bufnr, ns_telescope_highlight, 0, -1)
-  a.nvim_buf_add_highlight(
-    results_bufnr,
-    ns_telescope_highlight,
-    'Error',
-    row,
-    0,
-    -1
-  )
-
-  telescope_selections[prompt_bufnr] = row
-
-  if previewers[prompt_bufnr] then
-    vim.g.got_here = true
-    local status = hack_status[prompt_bufnr]
-    previewers[prompt_bufnr].fn(
-      status.preview_win,
-      status.preview_bufnr,
-      status.results_bufnr,
-      row
-    )
-  end
-end
-
-local mappings = {}
-
--- TODO: Refactor this to use shared code.
--- TODO: Move from top to bottom, etc.
--- TODO: It seems like doing this brings us back to the beginning of the prompt, which is not great.
-mappings["control-n"] = function(prompt_bufnr, results_bufnr)
-  if telescope_selections[prompt_bufnr] == nil then
-    telescope_selections[prompt_bufnr] = 0
-  end
-
-  local row = telescope_selections[prompt_bufnr] + 1
-  update_current_selection(prompt_bufnr, results_bufnr, row)
-end
-
-mappings["control-p"] = function(prompt_bufnr, results_bufnr)
-  if telescope_selections[prompt_bufnr] == nil then
-    telescope_selections[prompt_bufnr] = 0
-  end
-
-  local row = telescope_selections[prompt_bufnr] - 1
-  update_current_selection(prompt_bufnr, results_bufnr, row)
-end
-
-mappings["enter"] = function(prompt_bufnr, results_bufnr)
-  local extmark = a.nvim_buf_get_extmarks(
-    results_bufnr,
-    ns_telescope_highlight,
-    0,
-    -1,
-    {}
-  )
+function __TelescopeOnLeave(prompt_bufnr)
+  local status = state.get_status(prompt_bufnr)
+  local picker = status.picker
 
-  print(vim.inspect(extmark))
-end
-
-function __TelescopeMapping(prompt_bufnr, results_bufnr, characters)
-  if mappings[characters] then
-    mappings[characters](prompt_bufnr, results_bufnr)
-  end
+  picker:close_windows(status)
 end
 
 -- TODO: Probably could attach this with nvim_buf_attach, and then I don't have to do the ugly global function stuff
-function __TelescopeOnChange(bufnr, prompt, results_bufnr, results_win)
-  local line = vim.api.nvim_buf_get_lines(bufnr, 0, -1, false)[1]
+function __TelescopeOnChange(prompt_bufnr, prompt, results_bufnr, results_win)
+  local line = vim.api.nvim_buf_get_lines(prompt_bufnr, 0, -1, false)[1]
   local prompt_input = string.sub(line, #prompt + 1)
-  print(string.format("|%s|", prompt_input))
-
-  local finder = picker_finders[results_bufnr]
-  a.nvim_buf_set_lines(results_bufnr, 0, -1, false, {})
-  finder:display_results(results_win, results_bufnr, prompt_input)
-end
-
--- Uhh, finder should probably just GET the results
--- and then update some table.
--- When updating the table, we should call filter on those items
--- and then only display ones that pass the filter
-local f = Finder:new(function(prompt)
-  return string.format('rg %s', prompt)
-end)
 
-local p = Picker:new {
-  previewer = Previewer:new(function(preview_win, preview_bufnr, results_bufnr, row)
-    local line = a.nvim_buf_get_lines(results_bufnr, row, row + 1, false)[1]
-    local file_name = vim.split(line, ":")[1]
+  local status = state.get_status(prompt_bufnr)
+  local finder = status.finder
 
-    -- print(file_name)
-    -- vim.fn.termopen(
-    --   string.format("bat --color=always --style=grid %s"),
-    local file_bufnr = vim.fn.bufnr(file_name, true)
-    -- TODO: We should probably call something like this because we're not always getting highlight and all that stuff.
-    -- api.nvim_command('doautocmd filetypedetect BufRead ' .. vim.fn.fnameescape(filename))
-    a.nvim_win_set_buf(preview_win, file_bufnr)
-  end)
-}
-p:find(f)
-
--- TODO: Make filters
--- "fzf --filter"
---  jobstart() -> | fzf --filter "input on prompt"
--- function Filter:new(command)
--- end
+  vim.api.nvim_buf_set_lines(results_bufnr, 0, -1, false, {})
+  local results = finder:get_results(results_win, results_bufnr, prompt_input)
+end
 
 return telescope