From 1c2b1e0d0d3c0956f74c3614b7b57ce0ba1609fb Mon Sep 17 00:00:00 2001 From: TJ DeVries Date: Sat, 29 Aug 2020 22:22:51 -0400 Subject: [PATCH] Try to fix thing for rocker --- lua/telescope/mappings.lua | 4 ++-- lua/telescope/pickers.lua | 6 +++++- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/lua/telescope/mappings.lua b/lua/telescope/mappings.lua index 6bbbac5..1847325 100644 --- a/lua/telescope/mappings.lua +++ b/lua/telescope/mappings.lua @@ -49,8 +49,8 @@ keymap["control-p"] = function(prompt_bufnr, _) end keymap["enter"] = function(prompt_bufnr, results_bufnr) - local picker = state.get_status(prompt_bufnr).picker - local entry = picker:get_selection() + local picker = actions.get_current_picker(prompt_bufnr) + local entry = actions.get_selected_entry(prompt_bufnr) if not entry then print("[telescope] Nothing currently selected") diff --git a/lua/telescope/pickers.lua b/lua/telescope/pickers.lua index 95476f1..05eb7fe 100644 --- a/lua/telescope/pickers.lua +++ b/lua/telescope/pickers.lua @@ -395,7 +395,11 @@ function Picker:set_selection(row) -- Handle adding '> ' to beginning of selections if self._selection_row then - a.nvim_buf_set_lines(results_bufnr, self._selection_row, self._selection_row + 1, false, {' ' .. a.nvim_buf_get_lines(results_bufnr, self._selection_row, self._selection_row + 1, false)[1]:sub(3)}) + local old_selection = a.nvim_buf_get_lines(results_bufnr, self._selection_row, self._selection_row + 1, false)[1] + + if old_selection then + a.nvim_buf_set_lines(results_bufnr, self._selection_row, self._selection_row + 1, false, {' ' .. old_selection:sub(3)}) + end end a.nvim_buf_set_lines(results_bufnr, row, row + 1, false, {'> ' .. a.nvim_buf_get_lines(results_bufnr, row, row + 1, false)[1]:sub(3)})