feat: column navigate feature in result view

Author: nolleh

lua/sqlflick/display.lua

@@ -54,6 +54,8 @@ function M.create_display_window()
     vim.api.nvim_win_close(win, true)
   end, opts)
 
+  -- vim.keymap.set("n", "c", M.map_column_navigator)
+
   vim.keymap.set("n", "W", function()
     local query = require("sqlflick.query")
     local column_index = query.get_column_under_cursor()
@@ -107,6 +109,20 @@ function M.create_display_window()
   return buf, win
 end
 
+function M.map_column_navigator()
+  local query = require("sqlflick.query")
+  local col_num = query.get_column_number()
+
+  for i = 1, col_num do
+    vim.keymap.set("n", "c" .. i, function()
+      local column_start_pos = query.get_column_start_pos(i)
+      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
+end
+
 -- Display query results in display window
 function M.display_results(buf, _, error, query, results)
   -- Set buffer content

lua/sqlflick/init.lua

@@ -300,6 +300,7 @@ function M.setup(opts)
       -- Format and display results
       local lines = query.format_query_results(result)
       local error = result.error ~= nil and true or false
+      display.map_column_navigator()
       display.display_results(M.display_buf, M.display_win, error, query_text, lines)
     end
   end, { nargs = 1 })
@@ -376,6 +377,7 @@ function M.setup(opts)
       -- Format and display results
       local lines = query.format_query_results(result)
       local error = result.error ~= nil and true or false
+      display.map_column_navigator()
       display.display_results(M.display_buf, M.display_win, error, query_text, lines)
     end
   end, {})

lua/sqlflick/query.lua

@@ -1,6 +1,11 @@
 local M = {}
 
 -- Store table data for manipulation (word wrapping, etc.)
+---@class TableData
+---@field rows table
+---@field columns table
+---@field column_start_pos table
+---@field wrapped_columns table
 M.table_data = nil
 
 M.MAX_COLUMN_WIDTH = 200 -- Maximum width for any column
@@ -97,6 +102,7 @@ function M.format_query_results(result)
     columns = result.columns,
     rows = result.rows,
     wrapped_columns = {}, -- Track which columns are wrapped
+    column_start_pos = {},
   }
 
   local col_widths = {}
@@ -137,6 +143,14 @@ function M.format_query_results(result)
       .. truncated_col
       .. string.rep(" ", col_widths[i] - vim.fn.strdisplaywidth(truncated_col) - 1)
     header = header .. padded_col .. "│"
+
+    local prev_pos = 1
+    local prev_col_width = 0
+    if i > 1 then
+      prev_pos = M.table_data.column_start_pos[i - 1]
+      prev_col_width = col_widths[i - 1]
+    end
+    M.table_data.column_start_pos[i] = prev_pos + prev_col_width + 2 + 1
   end
 
   table.insert(lines, top)
@@ -161,6 +175,23 @@ function M.format_query_results(result)
   return lines
 end
 
+function M.get_column_number()
+  if not M.table_data then
+    return nil
+  end
+
+  return #M.table_data.columns
+end
+
+---@param i integer
+function M.get_column_start_pos(i)
+  if not M.table_data then
+    return nil
+  end
+
+  return M.table_data.column_start_pos[i]
+end
+
 function M.get_column_under_cursor()
   if not M.table_data then
     return nil
@@ -384,6 +415,14 @@ function M.format_query_results_with_wrapping(result)
         local text = row_lines[i][line_idx] or ""
         local padded_text = " " .. text .. string.rep(" ", col_widths[i] - vim.fn.strdisplaywidth(text) - 1)
         line = line .. padded_text .. "│"
+
+        local prev_pos = 1
+        local prev_col_width = 0
+        if i > 1 then
+          prev_pos = M.table_data.column_start_pos[i - 1]
+          prev_col_width = col_widths[i - 1]
+        end
+        M.table_data.column_start_pos[i] = prev_pos + prev_col_width + 2 + 1
       end
       table.insert(lines, line)
     end