Apply headings to <th> in body too. Trim labels.

zachleat · Oct 4, 2023 · 9c8cf7d · 9c8cf7d
1 parent 2b531d9
commit 9c8cf7d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/table-saw.js b/table-saw.js
@@ -101,15 +101,19 @@ table-saw.${this._identifier} {
 	}
 
 	addHeaders() {
-		let labels = Array.from(this.querySelectorAll("thead th")).map(entry => entry.innerText);
+		let labels = Array.from(this.querySelectorAll("thead th")).map(entry => entry.innerText.trim());
 		if(labels.length === 0) {
 			this._needsStylesheet = false;
 			console.error("No `<th>` elements for Tablesaw were found:", this);
 			return;
 		}
 
-		let cells = this.querySelectorAll("tbody td");
+		let cells = this.querySelectorAll("tbody :is(td, th)");
 		for(let cell of cells) {
+			if(!labels[cell.cellIndex]) {
+				continue;
+			}
+
 			cell.setAttribute(this.attrs.label, labels[cell.cellIndex]);
 
 			// wrap if this cell has child nodes for correct grid alignment