From 3362b3d405349d27010864ef62dcb35979b944a3 Mon Sep 17 00:00:00 2001
From: Alex Andreev <alex.andreev.email@gmail.com>
Date: Tue, 23 May 2023 13:48:59 +0300
Subject: [PATCH] Passing columns in the context

Signed-off-by: Alex Andreev <alex.andreev.email@gmail.com>
---
 .../kube-object-list-layout.tsx                      | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/packages/core/src/renderer/components/kube-object-list-layout/kube-object-list-layout.tsx b/packages/core/src/renderer/components/kube-object-list-layout/kube-object-list-layout.tsx
index fe8b87dc68..13140dacb7 100644
--- a/packages/core/src/renderer/components/kube-object-list-layout/kube-object-list-layout.tsx
+++ b/packages/core/src/renderer/components/kube-object-list-layout/kube-object-list-layout.tsx
@@ -31,6 +31,7 @@ import type { ClusterContext } from "../../cluster-frame-context/cluster-frame-c
 import type { GeneralKubeObjectListLayoutColumn, SpecificKubeListLayoutColumn } from "@k8slens/list-layout";
 import { kubeObjectListLayoutColumnInjectionToken } from "@k8slens/list-layout";
 import { sortBy } from "lodash";
+import { TableDataContext } from "@k8slens/table-tokens";
 
 export type KubeItemListStore<K extends KubeObject> = ItemListStore<K, false> & SubscribableStore & {
   getByPath: (path: string) => K | undefined;
@@ -186,7 +187,7 @@ class NonInjectedKubeObjectListLayout<
       ...targetColumns,
     ], (v) => -v.priority).map((col) => col.header);
 
-    return (
+    const itemsListLayout = (
       <ItemListLayout<K, false>
         className={cssNames("KubeObjectListLayout", className)}
         store={store}
@@ -233,6 +234,15 @@ class NonInjectedKubeObjectListLayout<
         {...layoutProps}
       />
     );
+
+    return (
+      <TableDataContext.Provider
+        value={{
+          columns: targetColumns as GeneralKubeObjectListLayoutColumn[],
+        }}>
+        {itemsListLayout}
+      </TableDataContext.Provider>
+    );
   }
 }