From 31b736ad62ad3228d210226510ee5442a99f779a Mon Sep 17 00:00:00 2001
From: Soichi Takamura <thepiglovesyou@gmail.com>
Date: Thu, 4 Jan 2024 10:37:47 +0900
Subject: [PATCH] respectColumnLevels

---
 README.md               | 38 ++++++++--------------------
 package.json            |  4 +++
 scripts/index.pug       |  4 +--
 src/merge-cells.test.ts | 26 ++++++++++++++++---
 src/merge-cells.ts      | 55 ++++++++++++++++++++++++++++++-----------
 5 files changed, 79 insertions(+), 48 deletions(-)

diff --git a/README.md b/README.md
index 897a957..d2df51d 100644
--- a/README.md
+++ b/README.md
@@ -1,6 +1,6 @@
 # merge-cells [![Node.js CI](https://github.com/piglovesyou/merge-cells/actions/workflows/node.js.yml/badge.svg)](https://github.com/piglovesyou/merge-cells/actions/workflows/node.js.yml)
 
-Provide JavaScript/TypeScript functions to help you merge cells of an HTML `GROUP BY`-like table.
+Provide JavaScript/TypeScript functions to help you merge cells with vertically the same values.
 
 👉 Example: https://piglovesyou.github.io/merge-cells/
 
@@ -10,7 +10,7 @@ Example usage in a browser:
 
 ```html
 <script type="module" defer>
-    import { mergeCells } from 'https://cdn.jsdelivr.net/npm/merge-cells@latest/dist/merge-cells.min.js'
+    import { mergeCells } from 'https://cdn.jsdelivr.net/npm/merge-cells@latest'
 
     const tableEl = document.getElementById('table')
     mergeCells(tableEl)
@@ -41,20 +41,14 @@ calcRowspanFromObjectArray(
 // ]
 ```
 
+## Available functions
+
 ### `mergeCells`
 
 Apply `rowspan` to an HTML table, calculated by `calcRowspanFromObjectArray`. Signature:
 
 ```ts
-function mergeCells(tableEl: HTMLTableElement, columnIndexes?: number[]): void
-```
-
-Example:
-
-```ts
-const tableEl = document.getElementById('table') as HTMLTableElement
-
-mergeCells(tableEl)
+function mergeCells(tableEl: HTMLTableElement, columnIndexes?: number[], options?: Options): void
 ```
 
 ### `calcRowspanFromObjectArray`
@@ -66,27 +60,15 @@ rendering. Signature:
 function calcRowspanFromObjectArray<E extends Record<string, any>>(
     rows: E[],
     columnKeys: string[],
+    options?: Options,
 ): Record<string, number>
 ```
 
-Example:
+### `Options`
 
-```ts
-expect(
-    calcRowspanFromObjectArray(
-        [
-            { c_1: 'a', c_2: 'a' },
-            { c_1: 'a', c_2: 'a' },
-            { c_1: 'b', c_2: 'a' },
-        ],
-        ['c_1', 'c_2'],
-    ),
-).toStrictEqual([
-    { c_1: 2, c_2: 2 },
-    { c_1: 0, c_2: 0 },
-    { c_1: 1, c_2: 1 },
-])
-```
+| property              | default value | description                                                                                                                 |
+| --------------------- | ------------- | --------------------------------------------------------------------------------------------------------------------------- |
+| `respectColumnLevels` | `true`        | Whether to merge/separate cells based on column levels. Leave it to `true` if you can aggregate your table with `GROUP BY`. |
 
 ## Publishing
 
diff --git a/package.json b/package.json
index 5256f6e..1e8d1ab 100644
--- a/package.json
+++ b/package.json
@@ -7,6 +7,10 @@
     ],
     "author": "Soichi Takamura <thepiglovesyou@gmail.com>",
     "license": "MIT",
+    "bugs": {
+        "url": "https://github.com/piglovesyou/merge-cells/issues"
+    },
+    "homepage": "https://github.com/piglovesyou/merge-cells#readme",
     "devDependencies": {
         "@babel/preset-env": "^7.23.6",
         "@babel/preset-typescript": "^7.23.3",
diff --git a/scripts/index.pug b/scripts/index.pug
index 4d58331..be119f6 100644
--- a/scripts/index.pug
+++ b/scripts/index.pug
@@ -10,8 +10,8 @@ html(lang="en")
 
         .container
             h1= pageTitle
-            p Provide JavaScript/TypeScript functions to help you merge cells of an HTML #[code GROUP BY]-like table.
-            p Go to #[a(href="https://github.com/piglovesyou/merge-cells") github.com/piglovesyou/merge-cells] for usage and more.
+            p Provide JavaScript/TypeScript functions to help you merge cells with vertically the same values.
+            p Go to #[a(href="https://github.com/piglovesyou/merge-cells") github.com/piglovesyou/merge-cells] for the usage and more.
 
             table#table.table.table-bordered
                 thead
diff --git a/src/merge-cells.test.ts b/src/merge-cells.test.ts
index 8af68b1..5bb2bbc 100644
--- a/src/merge-cells.test.ts
+++ b/src/merge-cells.test.ts
@@ -86,9 +86,29 @@ describe('lib.test.ts', () => {
                 ['c_1', 'c_2'],
             ),
         ).toStrictEqual([
-            { c_2: 1, c_1: 1 },
-            { c_2: 1, c_1: 1 },
-            { c_2: 1, c_1: 1 },
+            { c_1: 1, c_2: 1 },
+            { c_1: 1, c_2: 1 },
+            { c_1: 1, c_2: 1 },
+        ])
+    })
+
+    test('merge vertically regardless of levels', () => {
+        expect(
+            calcRowspanFromObjectArray(
+                [
+                    { c_1: 'a', c_2: 'a', c_3: 'c' },
+                    { c_1: 'a', c_2: 'b', c_3: 'c' },
+                    { c_1: 'b', c_2: 'b', c_3: 'a' },
+                ],
+                ['c_1', 'c_2', 'c_3'],
+                {
+                    respectColumnLevels: false,
+                },
+            ),
+        ).toStrictEqual([
+            { c_1: 2, c_2: 1, c_3: 2 },
+            { c_1: 0, c_2: 2, c_3: 0 },
+            { c_1: 1, c_2: 0, c_3: 1 },
         ])
     })
 })
diff --git a/src/merge-cells.ts b/src/merge-cells.ts
index 8ce3db1..bba26d5 100644
--- a/src/merge-cells.ts
+++ b/src/merge-cells.ts
@@ -7,12 +7,21 @@ interface BaseRow<E, K extends ColumnKey> {
     rowspan: Rowspan<K>
 }
 
+interface Options {
+    respectColumnLevels?: boolean
+}
+
 type ValueProvider<E, K extends ColumnKey> = (row: E, columnKey: K) => any
 
+const defaultOptions: Options = {
+    respectColumnLevels: true,
+}
+
 export function calcRowspanRecur<E, K extends ColumnKey>(
     rows: E[],
     fields: K[], // Object keys or array indexes
     valueProvider: ValueProvider<E, K>,
+    options: Options,
     baseRows: (BaseRow<E, K> | undefined)[] = [],
     acc: Rowspan<K>[] = [],
 ): Rowspan<K>[] {
@@ -30,28 +39,37 @@ export function calcRowspanRecur<E, K extends ColumnKey>(
     } as BaseRow<E, K>
     acc.push(rowspan)
 
-    const nextBaseRows = [] as BaseRow<E, K>[]
+    const nextBaseRows = options.respectColumnLevels ? ([] as BaseRow<E, K>[]) : baseRows
+
     for (let i = 0; i < fields.length; i++) {
         const field = fields[i]
         const baseRow = baseRows[i]
 
         const isNewValue = !baseRow || valueProvider(top, field) !== valueProvider(baseRow.row, field)
+
         if (isNewValue) {
-            return calcRowspanRecur(
-                rest,
-                fields,
-                valueProvider,
-                [...nextBaseRows, ...Array(fields.length - nextBaseRows.length).fill(topAcc)],
-                acc,
-            )
+            if (options.respectColumnLevels) {
+                return calcRowspanRecur(
+                    rest,
+                    fields,
+                    valueProvider,
+                    options,
+                    [...nextBaseRows, ...Array(fields.length - nextBaseRows.length).fill(topAcc)],
+                    acc,
+                )
+            } else {
+                baseRows[i] = topAcc
+            }
+        } else {
+            baseRows[i]!!.rowspan[field]++
+            topAcc.rowspan[field] = 0
+            if (options.respectColumnLevels) {
+                nextBaseRows.push(baseRow)
+            }
         }
-
-        baseRow.rowspan[field]++
-        topAcc.rowspan[field] = 0
-        nextBaseRows.push(baseRow)
     }
 
-    return calcRowspanRecur(rest, fields, valueProvider, nextBaseRows, acc)
+    return calcRowspanRecur(rest, fields, valueProvider, options, nextBaseRows, acc)
 }
 
 export function applyRowspanToTable(tableEl: HTMLTableElement, columnIndexes?: number[]) {
@@ -78,15 +96,22 @@ export function applyRowspanToTable(tableEl: HTMLTableElement, columnIndexes?: n
 export function calcRowspanWithTableRows(
     rows: HTMLCollectionOf<HTMLTableRowElement>,
     columnIndexes: number[] = rows[0] ? Array.from(Array(rows[0].cells.length).keys()) : [],
+    options: Options = defaultOptions,
 ) {
-    return calcRowspanRecur(Array.from(rows), columnIndexes, (row, fieldIndex) => row.cells[fieldIndex]!!.textContent)
+    return calcRowspanRecur(
+        Array.from(rows),
+        columnIndexes,
+        (row, fieldIndex) => row.cells[fieldIndex]!!.textContent,
+        options,
+    )
 }
 
 export function calcRowspanFromObjectArray<E extends Record<string, any>>(
     rows: E[],
     columnKeys: string[],
+    options: Options = defaultOptions,
 ): Rowspan<string>[] {
-    return calcRowspanRecur(rows, columnKeys, (row, field) => row[field]!!)
+    return calcRowspanRecur(rows, columnKeys, (row, field) => row[field]!!, options)
 }
 
 export const mergeCells = applyRowspanToTable