feat(replacer): end of word replacer implementation

Author: Sasha Gurevich

lua/toggle/init.lua

@@ -37,14 +37,16 @@ end
 
 function M.toggle()
     local replacers = {
-        replacer.__cWORD_replacer(),
-        replacer.__cword_replacer(),
-        replacer.__character_replacer(),
+        replacer.__get_cWORD_replacer,
+        replacer.__get_cword_replacer,
+        replacer.__get_end_of_word_replacer,
+        replacer.__character_replacer,
     }
 
     local current_cursor_position = vim.fn.getcurpos()
 
-    for _, r in ipairs(replacers) do
+    for _, get_replacer in ipairs(replacers) do
+        local r = get_replacer()
         if r.can_handle() then
             r.replace()
             if config.keep_cursor_position then

lua/toggle/replacer.lua

@@ -1,54 +1,69 @@
 local mapping = require('toggle.mapping')
 
+-- TODO provide luadocs for replacers
 local M = {}
 
-M.__cword_replacer = function()
+M.__get_cword_replacer = function()
+    local word = vim.fn.expand('<cword>')
     return {
         can_handle = function()
-            local word = vim.fn.expand('<cword>')
             return mapping.__has_mapping(word)
         end,
 
         replace = function()
-            local word = vim.fn.expand('<cword>')
             vim.api.nvim_command('normal! ciw' .. mapping.__get_mapping(word))
         end,
     }
 end
 
-M.__cWORD_replacer = function()
+M.__get_cWORD_replacer = function()
+    local word = vim.fn.expand('<cWORD>')
     return {
         can_handle = function()
-            local word = vim.fn.expand('<cWORD>')
             return mapping.__has_mapping(word)
         end,
 
         replace = function()
-            local word = vim.fn.expand('<cWORD>')
             vim.api.nvim_command('normal! ciW' .. mapping.__get_mapping(word))
         end,
     }
 end
 
-M.__character_replacer = function()
+M.__get_character_replacer = function()
+    local coords = vim.api.nvim_win_get_cursor(0)
+    local line = vim.api.nvim_get_current_line()
+    local character = line:sub(coords[2] + 1, coords[2] + 1)
+
     return {
         can_handle = function()
-            local character = M.__get_character_under_cursor()
             return mapping.__has_mapping(character)
         end,
 
         replace = function()
-            local character = M.__get_character_under_cursor()
             vim.api.nvim_command('normal! r' .. mapping.__get_mapping(character))
         end,
     }
 end
 
--- TODO write tests
-M.__get_character_under_cursor = function()
-    local coords = vim.api.nvim_win_get_cursor(0)
+M.__get_end_of_word_replacer = function()
+    local current_cursor_position = vim.fn.getcurpos()
+    vim.api.nvim_command('normal! e') -- jump to the end of word
+
+    local end_of_word_column = vim.api.nvim_win_get_cursor(0)[2]
+    vim.fn.setpos('.', current_cursor_position) -- restore cursor position
+
     local line = vim.api.nvim_get_current_line()
-    return line:sub(coords[2] + 1, coords[2] + 1)
+    local end_of_word_under_cursor = line:sub(current_cursor_position[3], end_of_word_column + 1)
+
+    return {
+        can_handle = function()
+            return mapping.__has_mapping(end_of_word_under_cursor)
+        end,
+
+        replace = function()
+            vim.api.nvim_command('normal! ce' .. mapping.__get_mapping(end_of_word_under_cursor))
+        end,
+    }
 end
 
 return M

test/test_files/test_1.txt

@@ -1,3 +1,4 @@
 false
 false,
 <
+goto_prev

test/toggle_spec.lua

@@ -15,24 +15,39 @@ local function get_character_under_cursor()
 end
 
 describe('toggle', function()
-    it('toggle cword', function()
+    it('toggle - cword', function()
         open_file_at('test/test_files/test_1.txt', 1, 1, function()
             toggle.toggle()
             assert(vim.fn.expand('<cword>') == 'true')
+            toggle.toggle()
+            assert(vim.fn.expand('<cword>') == 'false')
         end)
     end)
 
-    it('toggle cWORD', function()
+    it('toggle - cWORD', function()
         open_file_at('test/test_files/test_1.txt', 2, 1, function()
             toggle.toggle()
             assert(vim.fn.expand('<cword>') == 'true')
+            toggle.toggle()
+            assert(vim.fn.expand('<cword>') == 'false')
         end)
     end)
 
-    it('toggle char', function()
+    it('toggle - char', function()
         open_file_at('test/test_files/test_1.txt', 3, 1, function()
             toggle.toggle()
             assert(get_character_under_cursor() == '>')
+            toggle.toggle()
+            assert(get_character_under_cursor() == '<')
+        end)
+    end)
+
+    it('toggle - end of word', function()
+        open_file_at('test/test_files/test_1.txt', 4, 5, function()
+            toggle.toggle()
+            assert(vim.fn.expand('<cword>') == 'goto_next')
+            toggle.toggle()
+            assert(vim.fn.expand('<cword>') == 'goto_prev')
         end)
     end)
 end)