fix: selected_refactor and telescope extension after expr refactor

Fixes #501

Author: TheLeoP

lua/refactoring/init.lua

@@ -78,12 +78,6 @@ end
 function M.select_refactor(opts)
     local prefer_ex_cmd = opts and opts.prefer_ex_cmd or false
 
-    -- vim.ui.select exits visual mode without setting the `<` and `>` marks
-    local utils = require("refactoring.utils")
-    if utils.is_visual_mode() then
-        utils.exit_to_normal_mode()
-    end
-
     require("plenary.async").void(function()
         local ui = require("refactoring.ui")
         local selected_refactor = ui.select(
@@ -103,7 +97,11 @@ function M.select_refactor(opts)
                 require("refactoring.refactor").refactor_names[selected_refactor]
             api.nvim_input((":Refactor %s "):format(refactor_name))
         else
-            M.refactor(selected_refactor, opts)
+            local keys = M.refactor(selected_refactor, opts)
+            if keys == "g@" then
+                keys = "gvg@"
+            end
+            vim.cmd.normal(keys)
         end
     end)()
 end

lua/telescope/_extensions/refactoring.lua

@@ -4,7 +4,12 @@ local function refactor(prompt_bufnr)
     local content =
         require("telescope.actions.state").get_selected_entry(prompt_bufnr)
     require("telescope.actions").close(prompt_bufnr)
-    refactoring.refactor(content.value)
+    local keys = refactoring.refactor(content.value)
+
+    if keys == "g@" then
+        keys = "gvg@"
+    end
+    vim.cmd.normal(keys)
 end
 
 local function telescope_refactoring(opts)