Get simple rg example to work

This commit is contained in:
TJ DeVries
2020-07-15 17:53:30 -04:00
parent 053417dd92
commit c6f0142fc6
12 changed files with 580 additions and 339 deletions

180
lua/telescope/pickers.lua Normal file
View File

@@ -0,0 +1,180 @@
local a = vim.api
local popup = require('popup')
local mappings = require('telescope.mappings')
local state = require('telescope.state')
local pickers = {}
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
function Picker:find(finder)
local prompt_string = 'Find File'
-- 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, results_opts = 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)
local preview_win, preview_opts = popup.create('', {
height = result_height + prompt_height + 2,
minheight = result_height + prompt_height + 2,
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', 'Normal:Normal')
-- TODO: We need to center this and make it prettier...
local prompt_win, prompt_opts = popup.create('', {
height = prompt_height,
width = width,
line = prompt_line,
col = col,
border = {},
title = prompt_string
})
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)
vim.api.nvim_buf_attach(prompt_bufnr, true, {
on_lines = vim.schedule_wrap(function(_, _, _, first_line, last_line)
local line = vim.api.nvim_buf_get_lines(prompt_bufnr, first_line, last_line, false)[1]
vim.api.nvim_buf_set_lines(results_bufnr, 0, -1, false, {})
local results = finder:get_results(results_win, results_bufnr, line)
end),
on_detach = function(...)
-- print("DETACH:", ...)
end,
})
-- -- 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,
-- '',
-- results_bufnr,
-- results_win)
-- TODO: Use WinLeave as well?
local on_buf_leave = string.format(
[[ autocmd BufLeave <buffer> ++nested ++once :lua __TelescopeOnLeave(%s)]],
prompt_bufnr)
vim.cmd([[augroup PickerInsert]])
vim.cmd([[ au!]])
vim.cmd( on_buf_leave)
vim.cmd([[augroup END]])
state.set_status(prompt_bufnr, {
prompt_bufnr = prompt_bufnr,
prompt_win = prompt_win,
prompt_border_win = prompt_opts.border.win_id,
results_bufnr = results_bufnr,
results_win = results_win,
results_border_win = results_opts.border.win_id,
preview_bufnr = preview_bufnr,
preview_win = preview_win,
preview_border_win = preview_opts.border.win_id,
picker = self,
previewer = self.previewer,
finder = finder,
})
-- print(vim.inspect(state.get_status(prompt_bufnr)))
mappings.set_keymap(prompt_bufnr, results_bufnr)
vim.cmd [[startinsert]]
end
function Picker:close_windows(status)
-- vim.fn['popup#close_win'](state.prompt_win)
-- vim.fn['popup#close_win'](state.results_win)
-- vim.fn['popup#close_win'](state.preview_win)
local prompt_win = status.prompt_win
local results_win = status.results_win
local preview_win = status.preview_win
local prompt_border_win = status.prompt_border_win
local results_border_win = status.results_border_win
local preview_border_win = status.preview_border_win
local function del_win(name, win_id, force)
local file = io.open("/home/tj/test.txt", "a")
file:write(string.format("Closing.... %s %s\n", name, win_id))
local ok = pcall(vim.api.nvim_win_close, win_id, force)
file:write(string.format("OK: %s\n", ok))
file:write("...Done\n\n")
file:close()
end
del_win("prompt_win", prompt_win, true)
del_win("results_win", results_win, true)
del_win("preview_win", preview_win, true)
del_win("prompt_border_win", prompt_border_win, true)
del_win("results_border_win", results_border_win, true)
del_win("preview_border_win", preview_border_win, true)
-- vim.cmd(string.format("bdelete! %s", status.prompt_bufnr))
-- Major hack?? Why do I have to od this.
-- Probably because we're currently IN the buffer.
-- Should wait to do this until after we're done.
vim.defer_fn(function()
del_win("prompt_win", prompt_win, true)
end, 10)
state.clear_status(status.prompt_bufnr)
end
pickers.new = function(...)
return Picker:new(...)
end
return pickers