fix: display/column option not applicable issue

Author: nolleh

lua/sqlflick/init.lua

@@ -174,7 +174,6 @@ end
 
 -- Setup function that will be called by users
 function M.setup(opts)
-  -- Set up configuration
   config.setup(opts)
 
   -- Check and install dependencies
@@ -185,12 +184,12 @@ function M.setup(opts)
     end
   end
 
-  -- Ensure backend is installed
   install.ensure_installed()
 
-  -- Set up custom highlights
   highlights.setup()
 
+  query.setup(config.opts)
+
   -- Create display commands
   vim.api.nvim_create_user_command("SQLFlickDebug", function()
     print("SQLFlick Debug Info:")

lua/sqlflick/query.lua

@@ -3,11 +3,13 @@ local M = {}
 -- Store table data for manipulation (word wrapping, etc.)
 M.table_data = nil
 
-local config = require("sqlflick.config")
+M.MAX_COLUMN_WIDTH = 200 -- Maximum width for any column
+M.MIN_COLUMN_WIDTH = 0 -- Minimum width for readability
 
--- Configuration
-local MAX_COLUMN_WIDTH = math.max(config.opts.display.column.max_width, 200) -- Maximum width for any column
-local MIN_COLUMN_WIDTH = math.min(config.opts.display.column.min_width, 0) -- Minimum width for readability
+function M.setup(opts)
+  M.MAX_COLUMN_WIDTH = math.min(opts.display.column.max_width, M.MAX_COLUMN_WIDTH)
+  M.MIN_COLUMN_WIDTH = math.max(opts.display.column.min_width, M.MIN_COLUMN_WIDTH)
+end
 
 -- Helper function to truncate text and add ellipsis if needed
 local function truncate_text(text, max_width)
@@ -108,7 +110,7 @@ function M.format_query_results(result)
     end
 
     local ideal_width = math.max(header_width, max_data_width)
-    col_widths[i] = math.max(MIN_COLUMN_WIDTH, math.min(MAX_COLUMN_WIDTH, ideal_width))
+    col_widths[i] = math.max(M.MIN_COLUMN_WIDTH, math.min(M.MAX_COLUMN_WIDTH, ideal_width))
   end
 
   for i, width in ipairs(col_widths) do
@@ -315,7 +317,7 @@ function M.format_query_results_with_wrapping(result)
     end
 
     local ideal_width = math.max(header_width, max_data_width)
-    col_widths[i] = math.max(MIN_COLUMN_WIDTH, math.min(MAX_COLUMN_WIDTH, ideal_width)) + 2
+    col_widths[i] = math.max(M.MIN_COLUMN_WIDTH, math.min(M.MAX_COLUMN_WIDTH, ideal_width)) + 2
   end
 
   -- Format header and borders

test/sql/mysql.sql

@@ -73,6 +73,14 @@ WHERE USERNAME LIKE '%long%'
    OR PASSWORD LIKE '%long%'
 ORDER BY CREATED_ON DESC;
 
+SELECT USERNAME as "Very Long Column Header That Should Test Word Wrap",
+       PASSWORD as "Another Extremely Long Column Header For Testing Display",
+       EMAIL as "Yet Another Long Column Header To Verify Word Wrap Functionality",
+       CREATED_ON as "Creation Timestamp With Long Header",
+       LAST_LOGIN as "Last Login Time With Extended Header Name"
+FROM ACCOUNT_TEST
+ORDER BY CREATED_ON DESC;
+
 -- Create table with varied column widths for word wrap testing
 DROP TABLE IF EXISTS WORDWRAP_TEST;
 CREATE TABLE WORDWRAP_TEST (

test/test.lua

@@ -94,7 +94,7 @@ sqlflick.setup({
 
   display = {
     column = {
-      max_width = 200,
+      -- max_width = 10,
     },
   },