chore: fix dnd/menu alignment issue

Signed-off-by: Scott Dover <[email protected]>

Author: Scott Dover

client/src/webview/DataViewer.tsx

@@ -74,6 +74,7 @@ const DataViewer = () => {
           defaultColDef={{
             sortable: true,
           }}
+          maintainColumnOrder
           infiniteInitialRowCount={100}
           maxBlocksInCache={10}
           onGridReady={onGridReady}

client/src/webview/useDataViewer.ts

@@ -146,7 +146,6 @@ const useDataViewer = (theme: string) => {
           });
         },
       };
-
       event.api.setGridOption("datasource", dataSource);
     },
     [columns],
@@ -203,14 +202,17 @@ const useDataViewer = (theme: string) => {
             params.event.key === "Enter" ||
             (params.event.key === "F10" && params.event.shiftKey)
           ) {
-            const dropdownButton =
-              // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
-              (params.event.target as HTMLElement).querySelector(
-                ".dropdown > button",
-              );
-            if (!dropdownButton) {
+            // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
+            const dropdown = (params.event.target as HTMLElement).querySelector(
+              ".dropdown",
+            );
+            if (!dropdown) {
               return true;
             }
+            if (!dropdown.classList.contains("active")) {
+              dropdown.classList.add("active");
+            }
+            const dropdownButton = dropdown.querySelector("button");
             displayMenuForColumn(
               params.api,
               // eslint-disable-next-line @typescript-eslint/consistent-type-assertions