chore: backward/forward navigating keymapping for column

Author: nolleh

README.md

@@ -131,10 +131,13 @@ The following mappings are available for SQL-related file types (e.g., `.sql`, `
 | `SQLFlickExecute`    | v    | `<leader>sq` | Execute selected query     |
 
 The following mappings are available in specific pages. (not user command)
-| name | mode |Key binding | runnable view | description |
-| ------------------ | ----|- | ------------- | ----------------------------------------------------- |
-| Toggle column wrap | n | `W`| result view | Place your cursor to long data column, to toggle wrap |
-| Navigate column | n | `c{i}` | result view | Navigate cursor to specified column number |
+
+| name               | mode | Key binding | runnable view | description                                           |
+| ------------------ | ---- | ----------- | ------------- | ----------------------------------------------------- |
+| Toggle column wrap | n    | `W`         | result view   | Place your cursor to long data column, to toggle wrap |
+| Navigate column    | n    | `c{i}`      | result view   | Navigate cursor to specified column number            |
+| Previous column    | n    | `[c`        | result view   | Navigate cursor to previous column                    |
+| Next column        | n    | `]c`        | result view   | Navigate cursor to next column                        |
 
 ## Supported Database
 

lua/sqlflick/display.lua

@@ -56,6 +56,38 @@ function M.create_display_window()
 
   -- vim.keymap.set("n", "c", M.map_column_navigator)
 
+  vim.keymap.set("n", "[c", function()
+    local query = require("sqlflick.query")
+    local cursor_column = query.get_column_under_cursor()
+    local cursor_pos = vim.api.nvim_win_get_cursor(0)
+    local line_num = cursor_pos[1]
+    if nil == cursor_column then
+      if cursor_pos[2] == 0 then
+        cursor_column = 1
+      else
+        cursor_column = query.get_column_number()
+      end
+    end
+    local column_start_pos = query.get_column_start_pos(cursor_column - 1)
+    vim.fn.cursor(line_num, column_start_pos)
+  end, { desc = "Move cursor to previous column" })
+
+  vim.keymap.set("n", "]c", function()
+    local query = require("sqlflick.query")
+    local cursor_pos = vim.api.nvim_win_get_cursor(0)
+    local line_num = cursor_pos[1]
+    local cursor_column = query.get_column_under_cursor()
+    if nil == cursor_column then
+      if cursor_pos[2] == 0 then
+        cursor_column = 1
+      else
+        cursor_column = query.get_column_number()
+      end
+    end
+    local column_start_pos = query.get_column_start_pos(cursor_column + 1)
+    vim.fn.cursor(line_num, column_start_pos)
+  end, { desc = "Move cursor to next column" })
+
   vim.keymap.set("n", "W", function()
     local query = require("sqlflick.query")
     local column_index = query.get_column_under_cursor()
@@ -119,7 +151,7 @@ function M.map_column_navigator()
       local cursor_pos = vim.api.nvim_win_get_cursor(0)
       local line_num = cursor_pos[1]
       vim.fn.cursor(line_num, column_start_pos)
-    end)
+    end, { desc = "Navigate cursor to column" })
   end
 end
 

lua/sqlflick/query.lua

@@ -177,7 +177,7 @@ end
 
 function M.get_column_number()
   if not M.table_data then
-    return nil
+    return 0
   end
 
   return #M.table_data.columns
@@ -186,7 +186,11 @@ end
 ---@param i integer
 function M.get_column_start_pos(i)
   if not M.table_data then
-    return nil
+    return 0
+  end
+
+  if #M.table_data.column_start_pos < i or i < 1 then
+    return 0
   end
 
   return M.table_data.column_start_pos[i]