fix: proceed applying part to next level

Regardless of current element already having a part attribute.
bcgov · Oct 9, 2023 · 6af4f2c · 6af4f2c
1 parent ae3a94d
commit 6af4f2c
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/frontend/src/helpers/CustomDirectives.ts b/frontend/src/helpers/CustomDirectives.ts
@@ -16,13 +16,14 @@ export const masking = (shadowSelector: string) => (el: any, binding: any) => {
 const setPart = (nodes: HTMLCollection, currentLevel: number, maxLevel: number) => {
   for (const node of nodes) {
     if (node.nodeType === Node.ELEMENT_NODE) {
-      if (node.getAttribute('part')) continue
-
-      const partValues = [node.tagName.toLowerCase()]
-      if (node.className) {
-        partValues.push(node.className)
+      if (!node.getAttribute('part')) {
+        const partValues = [node.tagName.toLowerCase()]
+        if (node.className) {
+          partValues.push(node.className)
+        }
+        node.setAttribute('part', partValues.join(' '))
       }
-      node.setAttribute('part', partValues.join(' '))
+
       if (currentLevel < maxLevel) {
         setPart(node.children, currentLevel + 1, maxLevel)
       }