@coreui/react-pro 4.11.0-next.0 → 4.11.0

package/src/components/smart-table/utils.ts

@@ -2,6 +2,7 @@ import type {
   Column,
   ColumnFilter,
   ColumnFilterValue,
+  Group,
   Item,
   Sorter,
   SorterValue,
@@ -77,12 +78,130 @@ export const getColumnLabel = (column: Column | string) =>
       : pretifyName(column.key)
     : pretifyName(column)
 
-export const getColumnNames = (columns: (string | Column)[] | undefined, items: Item[]) =>
-  columns
-    ? columns.map((column: Column | string) => {
-        return typeof column === 'object' ? column.key : column
+// export const getColumnNames = (columns: (string | Column)[] | undefined, items: Item[]) =>
+//   columns
+//     ? columns.map((column: Column | string) => {
+//         return typeof column === 'object' ? column.key : column
+//       })
+//     : getColumnNamesFromItems(items)
+
+export const getColumnNames = (
+  columns: (string | Column)[] | undefined,
+  items: Item[],
+): string[] => {
+  if (columns) {
+    const _columns = []
+
+    for (const column of columns) {
+      if (typeof column === 'object' && column.children) {
+        _columns.push(...getColumnNames(column.children, []))
+        continue
+      }
+
+      typeof column === 'object' ? _columns.push(column.key) : _columns.push(column)
+    }
+
+    return _columns
+  }
+
+  return getColumnNamesFromItems(items)
+}
+
+export const getColumns = (_columns: (Column | Group | string)[]): (Column | string)[] => {
+  const columns = []
+
+  for (const column of _columns) {
+    if (typeof column === 'object' && column.group && column.children) {
+      columns.push(...getColumns(column.children))
+      continue
+    }
+
+    if (typeof column === 'object' && column.children) {
+      columns.push(...getColumns(column.children))
+    }
+
+    columns.push(column)
+  }
+
+  return columns
+}
+
+export const countColumns = (columns: Column[], counter = 0) => {
+  let _counter = counter
+  for (const column of columns) {
+    if (!column.children) {
+      _counter++
+    }
+
+    if (column.children) {
+      _counter = countColumns(column.children, _counter)
+    }
+  }
+
+  return _counter
+}
+
+export const getColumnGroups = (columns: (string | Column)[] | undefined) => {
+  const groups = []
+
+  const traverseColumns = (column: Column, deep = 0, colSpan = 0): Group[] => {
+    const groups = []
+
+    if (column.children) {
+      for (const _column of column.children) {
+        if (!_column.group) {
+          colSpan++
+        }
+        groups.push(...traverseColumns(_column, deep + 1, colSpan))
+      }
+    }
+
+    if (typeof column === 'object' && column.group) {
+      const { children, group, ...rest } = column
+      groups.push({
+        deep: deep,
+        label: group,
+        ...(children && { colspan: countColumns(children) }),
+        ...rest,
       })
-    : getColumnNamesFromItems(items)
+    }
+
+    return groups
+  }
+
+  if (columns) {
+    for (const column of columns) {
+      if (typeof column === 'object' && column.group) {
+        const objects = traverseColumns(column)
+
+        if (objects) {
+          for (const object of objects) {
+            const { deep, ...rest } = object
+
+            if (deep === undefined) {
+              continue
+            }
+
+            for (let i = 0; i < deep; i++) {
+              if (groups[i]) {
+                continue
+              }
+
+              groups.push([])
+            }
+
+            if (groups[deep]) {
+              groups[deep].push(rest)
+            } else {
+              groups.push([rest])
+            }
+          }
+        }
+      }
+    }
+  }
+  return groups
+}
 
 export const getColumnNamesFromItems = (items: Item[]) =>
   Object.keys(items[0] || {}).filter((el) => el.charAt(0) !== '_')