Fixed column visibility bug.

src/components/Dashboard/Inventory/StockEvents.jsx

@@ -14,7 +14,7 @@ const StockEvents = () => {
   const [viewMode, setViewMode] = useViewMode('stockEvents')
 
   const [columnVisibility, setColumnVisibility] =
-    useColumnVisibility('stockEvents')
+    useColumnVisibility('stockEvent')
 
   const actionItems = {
     items: [

src/components/Dashboard/Management/AuditLogs.jsx

@@ -11,7 +11,7 @@ const AuditLogs = () => {
   const tableRef = useRef()
 
   const [columnVisibility, updateColumnVisibility] =
-    useColumnVisibility('auditLogs')
+    useColumnVisibility('auditLog')
 
   const actionItems = {
     items: [

src/components/Dashboard/Management/CourierServices.jsx

@@ -54,8 +54,8 @@ const CourierServices = () => {
             <ColumnViewButton
               type='courierService'
               loading={false}
-              collapseState={columnVisibility}
-              updateCollapseState={setColumnVisibility}
+              visibleState={columnVisibility}
+              updateVisibleState={setColumnVisibility}
             />
             <ExportListButton objectType='courierService' />
           </Space>

src/components/Dashboard/Management/Couriers.jsx

@@ -53,8 +53,8 @@ const Couriers = () => {
             <ColumnViewButton
               type='courier'
               loading={false}
-              collapseState={columnVisibility}
-              updateCollapseState={setColumnVisibility}
+              visibleState={columnVisibility}
+              updateVisibleState={setColumnVisibility}
             />
             <ExportListButton objectType='courier' />
           </Space>

src/components/Dashboard/Management/DocumentJobs.jsx

@@ -54,8 +54,8 @@ const DocumentJobs = () => {
             <ColumnViewButton
               type='documentJob'
               loading={false}
-              collapseState={columnVisibility}
-              updateCollapseState={setColumnVisibility}
+              visibleState={columnVisibility}
+              updateVisibleState={setColumnVisibility}
             />
             <ExportListButton objectType='documentJob' />
           </Space>

src/components/Dashboard/Management/DocumentPrinters.jsx

@@ -53,8 +53,8 @@ const DocumentPrinters = () => {
             <ColumnViewButton
               type='documentPrinter'
               loading={false}
-              collapseState={columnVisibility}
-              updateCollapseState={setColumnVisibility}
+              visibleState={columnVisibility}
+              updateVisibleState={setColumnVisibility}
             />
             <ExportListButton objectType='documentPrinter' />
           </Space>

src/components/Dashboard/Management/DocumentTemplates.jsx

@@ -54,8 +54,8 @@ const DocumentTemplates = () => {
             <ColumnViewButton
               type='documentTemplate'
               loading={false}
-              collapseState={columnVisibility}
-              updateCollapseState={setColumnVisibility}
+              visibleState={columnVisibility}
+              updateVisibleState={setColumnVisibility}
             />
             <ExportListButton objectType='documentTemplate' />
           </Space>

src/components/Dashboard/Management/Files.jsx

@@ -42,8 +42,8 @@ const Files = () => {
             <ColumnViewButton
               type='file'
               loading={false}
-              collapseState={columnVisibility}
-              updateCollapseState={setColumnVisibility}
+              visibleState={columnVisibility}
+              updateVisibleState={setColumnVisibility}
             />
             <ExportListButton objectType='file' />
           </Space>

src/components/Dashboard/Management/NoteTypes.jsx

@@ -54,8 +54,8 @@ const NoteTypes = () => {
             <ColumnViewButton
               type='noteType'
               loading={false}
-              collapseState={columnVisibility}
-              updateCollapseState={setColumnVisibility}
+              visibleState={columnVisibility}
+              updateVisibleState={setColumnVisibility}
             />
             <ExportListButton objectType='noteType' />
           </Space>

src/components/Dashboard/Management/Parts.jsx

@@ -58,8 +58,8 @@ const Parts = (filter) => {
             <ColumnViewButton
               type='part'
               loading={false}
-              collapseState={columnVisibility}
-              updateCollapseState={setColumnVisibility}
+              visibleState={columnVisibility}
+              updateVisibleState={setColumnVisibility}
             />
             <ExportListButton objectType='part' />
           </Space>

src/components/Dashboard/Management/Products.jsx

@@ -231,7 +231,7 @@ const Products = () => {
   ]
 
   const [columnVisibility, updateColumnVisibility] = useColumnVisibility(
-    'Products',
+    'product',
     columns
   )
 

src/components/Dashboard/Management/TaxRates.jsx

@@ -53,8 +53,8 @@ const TaxRates = () => {
             <ColumnViewButton
               type='taxRate'
               loading={false}
-              collapseState={columnVisibility}
-              updateCollapseState={setColumnVisibility}
+              visibleState={columnVisibility}
+              updateVisibleState={setColumnVisibility}
             />
             <ExportListButton objectType='taxRate' />
           </Space>

src/components/Dashboard/Management/Vendors.jsx

@@ -53,8 +53,8 @@ const Vendors = () => {
             <ColumnViewButton
               type='vendor'
               loading={false}
-              collapseState={columnVisibility}
-              updateCollapseState={setColumnVisibility}
+              visibleState={columnVisibility}
+              updateVisibleState={setColumnVisibility}
             />
             <ExportListButton objectType='vendor' />
           </Space>

src/components/Dashboard/hooks/useColumnVisibility.jsx

@@ -1,13 +1,22 @@
 import { useState, useEffect } from 'react'
 import { getModelByName } from '../../../database/ObjectModels'
 
-const useColumnVisibility = (type) => {
+const useColumnVisibility = (type, customColumns) => {
   const getInitialVisibility = () => {
-    const stored = sessionStorage.getItem(`${type}_columnVisibility`)
+    const storageKey = `${type}_columnVisibility`
+    const stored = sessionStorage.getItem(storageKey)
     if (stored) {
       return JSON.parse(stored)
     }
     // Default visibility - all columns visible
+    if (customColumns && Array.isArray(customColumns)) {
+      return customColumns.reduce((acc, col) => {
+        if (col.key) {
+          acc[col.key] = true
+        }
+        return acc
+      }, {})
+    }
     const model = getModelByName(type)
     const columns = model.columns || []
     return columns.reduce((acc, columnName) => {