diff options
Diffstat (limited to 'lua/telescope/pickers.lua')
| -rw-r--r-- | lua/telescope/pickers.lua | 83 |
1 files changed, 68 insertions, 15 deletions
diff --git a/lua/telescope/pickers.lua b/lua/telescope/pickers.lua index a8624b0..f9f64df 100644 --- a/lua/telescope/pickers.lua +++ b/lua/telescope/pickers.lua @@ -3,6 +3,7 @@ local popup = require('popup') local mappings = require('telescope.mappings') local state = require('telescope.state') +local utils = require('telescope.utils') local pickers = {} @@ -19,8 +20,15 @@ function Picker:new(opts) end -function Picker:find(finder) - local prompt_string = 'Find File' +function Picker:find(opts) + opts = opts or {} + + local finder = opts.finder + assert(finder, "Finder is required to do picking") + + local sorter = opts.sorter + + local prompt_string = opts.prompt -- Create three windows: -- 1. Prompt window -- 2. Options window @@ -75,26 +83,71 @@ function Picker:find(finder) -- 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] + local on_lines = function(_, _, _, first_line, last_line) + local prompt = 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, {}) + + -- Create a closure that has all the data we need + -- We pass a function called "newResult" to get_results + -- get_results calles "newResult" every time it gets a new result + -- picker then (if available) calls sorter + -- and then appropriately places new result in the buffer. + + local line_scores = {} + + -- TODO: We need to fix the sorting + -- TODO: We should provide a simple fuzzy matcher in Lua for people + -- TODO: We should get all the stuff on the bottom line directly, not floating around + -- TODO: We need to handle huge lists in a good way, cause currently we'll just put too much stuff in the buffer + -- TODO: Stop having things crash if we have an error. + finder(prompt, function(line) + if sorter then + local sort_score = sorter:score(prompt, line) + if sort_score == -1 then + return + end + + -- { 7, 3, 1, 1 } + -- 2 + for row, row_score in utils.reversed_ipairs(line_scores) do + if row_score > sort_score then + -- Insert line at row + vim.api.nvim_buf_set_lines(results_bufnr, row, row, false, { + string.format("%s // %s %s", line, sort_score, row) + }) + + -- Insert current score in the table + table.insert(line_scores, row + 1, sort_score) + + -- All done :) + return + end + end + + -- Worst score so far, so add to end + vim.api.nvim_buf_set_lines(results_bufnr, -1, -1, false, {line}) + table.insert(line_scores, sort_score) + else + -- Just always append to the end of the buffer if this is all you got. + vim.api.nvim_buf_set_lines(results_bufnr, -1, -1, false, {line}) + end + end) + -- local results = finder:get_results(results_win, results_bufnr, line) + end - vim.api.nvim_buf_set_lines(results_bufnr, 0, -1, false, {}) - local results = finder:get_results(results_win, results_bufnr, line) - end), + -- Call this once to pre-populate if it makes sense + vim.schedule_wrap(on_lines(nil, nil, nil, 0, 1)) + + -- Register attach + vim.api.nvim_buf_attach(prompt_bufnr, true, { + on_lines = vim.schedule_wrap(on_lines), 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( |