feat: added hooks to html

Bdaya-Dev · Aug 6, 2023 · 2c738d1 · 2c738d1
1 parent ce86b25
commit 2c738d1
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/packages/reporter_html/lib/src/config.dart b/packages/reporter_html/lib/src/config.dart
@@ -8,7 +8,15 @@ typedef ElementProcessor = void Function(Element element);
 class ReporterHtmlConfig {
   final Node Function(ReportCalculatedRange range)? createElementFromValue;
 
+  ///td or th
+  final void Function(Element th, ReportCalculatedRange range)? postProcessCell;
+  //tr
+  final void Function(Element th, List<ReportCalculatedRange> cells)?
+      postProcessRow;
+
   const ReporterHtmlConfig({
     this.createElementFromValue,
+    this.postProcessCell,
+    this.postProcessRow,
   });
 }
diff --git a/packages/reporter_html/lib/src/reporter_html_base.dart b/packages/reporter_html/lib/src/reporter_html_base.dart
@@ -37,6 +37,7 @@ class TabularReporterHtml {
           row.groupListsBy((element) => element.range.columnIndex);
       final minColKey = colGroups.keys.min;
       final tr = _registerElement('tr', parent: res);
+
       for (var colIndex = minColKey; colIndex <= maxColIndex; colIndex++) {
         final cells = colGroups[colIndex];
         if (cells == null) {
@@ -48,6 +49,7 @@ class TabularReporterHtml {
         }
         final cellElement = _registerElement(cellElementTag, parent: tr);
         final cell = cells.first;
+
         cellElement.attributes['rowspan'] = cell.range.rowSpan.toString();
         cellElement.attributes['colspan'] = cell.range.colSpan.toString();
         final valueNode = config.createElementFromValue?.call(cell);
@@ -56,7 +58,9 @@ class TabularReporterHtml {
         } else {
           cellElement.text = cell.value.toString();
         }
+        config.postProcessCell?.call(cellElement, cell);
       }
+      config.postProcessRow?.call(tr, cells);
     }
     return res;
   }