Refactor: CreateMissingTableHead and gettingTableBody into their own …

…functions.
LeeWannacott · May 7, 2023 · 0e5ec50 · 0e5ec50
1 parent f6ca95f
commit 0e5ec50
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/public/table-sort.js b/public/table-sort.js
@@ -35,26 +35,32 @@ function tableSortJs(testingTableSortJS = false, domDocumentWindow = document) {
     }
   }
 
-  function makeTableSortable(sortableTable) {
+  function createMissingTableHead(sortableTable) {
     let createTableHead;
-    let tableBody;
-    if (sortableTable.getElementsByTagName("thead").length === 0) {
       if (testingTableSortJS === true) {
         createTableHead = domDocumentWindow.createElement("thead");
       } else {
         createTableHead = document.createElement("thead");
       }
       createTableHead.appendChild(sortableTable.rows[0]);
       sortableTable.insertBefore(createTableHead, sortableTable.firstChild);
+  }
+
+  function getTableBody(sortableTable) {
+    if (sortableTable.getElementsByTagName("thead").length === 0) {
+      createMissingTableHead(sortableTable)
       if (sortableTable.querySelectorAll("tbody").length > 1) {
-        tableBody = sortableTable.querySelectorAll("tbody")[1];
+        return sortableTable.querySelectorAll("tbody")[1];
       } else {
-        tableBody = sortableTable.querySelector("tbody");
+        return sortableTable.querySelector("tbody");
       }
     } else {
-      tableBody = sortableTable.querySelector("tbody");
+      return sortableTable.querySelector("tbody");
     }
+  }
 
+  function makeTableSortable(sortableTable) {
+    const tableBody = getTableBody(sortableTable);
     const tableHead = sortableTable.querySelector("thead");
     const tableHeadHeaders = tableHead.querySelectorAll("th");