refactor(refactor): use higher local apis and some cleanup

1 files changed, 3 insertions, 11 deletions

diff --git a/lua/nvim-treesitter/refactor/smart_rename.lua b/lua/nvim-treesitter/refactor/smart_rename.lua
index 5b7562eb..8aab9538 100644
--- a/lua/nvim-treesitter/refactor/smart_rename.lua
+++ b/lua/nvim-treesitter/refactor/smart_rename.lua
@@ -3,6 +3,7 @@
 
 local ts_utils = require'nvim-treesitter.ts_utils'
 local configs = require'nvim-treesitter.configs'
+local utils = require'nvim-treesitter.utils'
 local api = vim.api
 
 local M = {}
@@ -12,7 +13,7 @@ function M.smart_rename(bufnr)
   local node_at_point = ts_utils.get_node_at_cursor()
 
   if not node_at_point then
-    print('No node to rename!')
+    utils.print_warning("No node to rename!")
     return
   end
 
@@ -29,23 +30,16 @@ function M.smart_rename(bufnr)
     table.insert(nodes_to_rename, node_at_point)
   end
 
-  if definition and not vim.tbl_contains(nodes_to_rename, definition) then
+  if not vim.tbl_contains(nodes_to_rename, definition) then
     table.insert(nodes_to_rename, definition)
   end
 
-  if #nodes_to_rename < 1 then
-    print('No nodes to rename!')
-    return
-  end
-
   for _, node in ipairs(nodes_to_rename) do
     local start_row, start_col, end_row, end_col = node:range() 
-
     local line = api.nvim_buf_get_lines(bufnr, start_row, start_row + 1, false)[1]
 
     if line then
       local new_line = line:sub(1, start_col) .. new_name .. line:sub(end_col + 1, -1)
-
       api.nvim_buf_set_lines(bufnr, start_row, start_row + 1, false, { new_line })
     end
   end
@@ -53,12 +47,10 @@ end
 
 function M.attach(bufnr)
   local bufnr = bufnr or api.nvim_get_current_buf()
-
   local config = configs.get_module('refactor.smart_rename')
 
   for fn_name, mapping in pairs(config.keymaps) do
     local cmd = string.format([[:lua require'nvim-treesitter.refactor.smart_rename'.%s(%d)<CR>]], fn_name, bufnr)
-
     api.nvim_buf_set_keymap(bufnr, 'n', mapping, cmd, { silent = true })
   end
 end