Merge pull request #54 from razroo/ZETA-6939-optional-classname-add-p…

…roperty-html Zeta 6939 optional classname for add property html
razroo · Oct 20, 2023 · e1ae714 · e1ae714
2 parents 7e3ef5f + edb5654
commit e1ae714
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 6 deletions.
diff --git a/src/rz/angular/add-property-to-html-tag/add-property-to-html-tag.spec.ts b/src/rz/angular/add-property-to-html-tag/add-property-to-html-tag.spec.ts
@@ -29,6 +29,34 @@ describe('addPropertyToHtmlTag', () => {
   </devgen-eureka-seven-global-header>
   <devgen-eureka-seven-global-header> </devgen-eureka-seven-global-header>
 </div>
+`;
+    expect(result).toEqual(expected);
+  });
+
+  it('should add a property to an html tag and use the class name to determine which element', () => {
+      const fileToBeAddedTo = `<div></div>
+<div class="Toolbar__Icons"></div>
+`;
+      const codeBlock = '{"*ngIf": "authenticated"}';
+
+      const insertIntoHtmlTag: EditHtmlFile = {
+         nodeType: 'addPropertyToHtmlTag',
+         codeBlock: codeBlock,
+         tagNameToInsertInto: 'div',
+         className: 'Toolbar__Icons'
+      };
+
+      const editInput = {
+        fileType: 'html',
+        fileToBeAddedTo: fileToBeAddedTo,
+        edits: [
+          insertIntoHtmlTag
+        ]
+      };
+
+      const result = morphCode(editInput);
+      const expected = `<div></div>
+<div class="Toolbar__Icons" *ngIf="authenticated"></div>
 `;
     expect(result).toEqual(expected);
   });

diff --git a/src/rz/angular/add-property-to-html-tag/add-property-to-html-tag.ts b/src/rz/angular/add-property-to-html-tag/add-property-to-html-tag.ts
@@ -9,12 +9,14 @@ export function addPropertyToHtmlTag(editFile: EditHtmlFile, fileToBeModified: a
   let firstMatchFound = false;
   visit(fileToBeModified, { type: 'element', tagName: editFile.tagNameToInsertInto }, (node: any, index) => {
     // Check if the tag is the one you want to modify
-    if (!firstMatchFound && node.tagName === editFile.tagNameToInsertInto) {
-      node.properties = {
-        ...(node.properties || {}),
-        ...codeBlock
-      };
-      firstMatchFound = true;
+    if(!editFile.className || editFile.className === '' || node.properties.className?.includes(editFile.className)) {
+      if (!firstMatchFound && node.tagName === editFile.tagNameToInsertInto) {
+        node.properties = {
+          ...(node.properties || {}),
+          ...codeBlock
+        };
+        firstMatchFound = true;
+      }
     }
   });