fix focus

Author: RafaelGB

src/components/DefaultCell.tsx

@@ -193,6 +193,7 @@ export default function DefaultCell(
         return dirtyCell ? (
           <input
             value={(contextValue.value && contextValue.value.toString()) || ""}
+            ref={editableMdRef}
             onChange={handleOnChange}
             onKeyDown={handleKeyDown}
             onBlur={handleOnBlur}

src/components/HeaderMenu.tsx

@@ -110,13 +110,12 @@ const HeaderMenu = (headerMenuProps: HeaderMenuProps) => {
       setLabelStateInvalid(true);
       return;
     }
-
     // Update state of ordered
-    const updateOrderWithNewKey = table
-      .getAllColumns()
-      .map((o: any) => (o.id === column.id ? newKey : o.id));
+    const updateOrderWithNewKey = table.options.state.columnOrder.map((o) =>
+      o === column.id ? newKey : o
+    );
     table.setColumnOrder(updateOrderWithNewKey);
-
+    // Update state of altered column
     setkeyState(newKey);
     alterColumnLabel(column.columnDef as TableColumn, labelState).then(() => {
       updateDataAfterLabelChange(

src/components/Table.tsx

@@ -94,6 +94,16 @@ export function Table(tableData: TableDataType) {
   const [columnOrder, setColumnOrder] = React.useState<ColumnOrderState>(
     columns.map((c) => c.id)
   );
+  const findColumn = React.useCallback(
+    (id: string) => {
+      const findedColumn = columns.find((c) => c.id === id);
+      return {
+        findedColumn,
+        index: columns.indexOf(findedColumn),
+      };
+    },
+    [columns]
+  );
   // Niveling number of columns
   if (columnOrder.length !== columns.length) {
     setColumnOrder(columns.map((c) => c.id));
@@ -247,17 +257,6 @@ export function Table(tableData: TableDataType) {
       current_row_template: settingsValue,
     });
   }
-
-  const findColumn = React.useCallback(
-    (id: string) => {
-      const findedColumn = columns.find((c) => c.id === id);
-      return {
-        findedColumn,
-        index: columns.indexOf(findedColumn),
-      };
-    },
-    [table.options.state.columnOrder]
-  );
   LOGGER.debug(`<= Table`);
   return (
     <>