fix(migrations): fixes CF migration i18n ng-template offsets (#53212)

thePunderWoman · pkozlowski-opensource · commit ba6d7fe0184d · 2023-11-28T16:44:11.000+01:00
This addresses an issue where multiple ng-templates are present with i18n attributes. The offsets would be incorrectly accounted for when being replaced with an ng-container. fixes: #53149 PR Close #53212
diff --git a/packages/core/schematics/ng-generate/control-flow-migration/util.ts b/packages/core/schematics/ng-generate/control-flow-migration/util.ts
@@ -302,8 +302,14 @@ export function getTemplates(template: string): Map<string, Template> {
 }
 
 function wrapIntoI18nContainer(i18nAttr: Attribute, content: string) {
+  const {start, middle, end} = generatei18nContainer(i18nAttr, content);
+  return `${start}${middle}${end}`;
+}
+
+function generatei18nContainer(
+    i18nAttr: Attribute, middle: string): {start: string, middle: string, end: string} {
   const i18n = i18nAttr.value === '' ? 'i18n' : `i18n="${i18nAttr.value}"`;
-  return `<ng-container ${i18n}>${content}</ng-container>`;
+  return {start: `<ng-container ${i18n}>`, middle, end: `</ng-container>`};
 }
 
 /**
@@ -431,8 +437,7 @@ export function getMainBlock(etm: ElementToMigrate, tmpl: string, offset: number
   } else if (isI18nTemplate(etm, i18nAttr)) {
     const childStart = etm.el.children[0].sourceSpan.start.offset - offset;
     const childEnd = etm.el.children[etm.el.children.length - 1].sourceSpan.end.offset - offset;
-    const middle = wrapIntoI18nContainer(i18nAttr!, tmpl.slice(childStart, childEnd));
-    return {start: '', middle, end: ''};
+    return generatei18nContainer(i18nAttr!, tmpl.slice(childStart, childEnd));
   }
 
   const attrStart = etm.attr.keySpan!.start.offset - 1 - offset;
diff --git a/packages/core/schematics/test/control_flow_migration_spec.ts b/packages/core/schematics/test/control_flow_migration_spec.ts
@@ -620,6 +620,45 @@ describe('control flow migration', () => {
       ].join('\n'));
     });
 
+    it('should migrate an NgIfElse case with ng-templates with multiple i18n attributes',
+       async () => {
+         writeFile('/comp.ts', `
+        import {Component} from '@angular/core';
+        import {NgIf} from '@angular/common';
+
+        @Component({
+          templateUrl: './comp.html'
+        })
+        class Comp {
+          show = false;
+        }
+      `);
+
+         writeFile('/comp.html', [
+           `<ng-template *ngIf="false; else barTempl" i18n="@@foo">`,
+           `  Foo`,
+           `</ng-template>`,
+           `<ng-template i18n="@@bar" #barTempl> Bar </ng-template>`,
+           `<a *ngIf="true" i18n="@@bam">Bam</a>`,
+         ].join('\n'));
+
+         await runMigration();
+         const content = tree.readContent('/comp.html');
+
+         expect(content).toBe([
+           `@if (false) {`,
+           `  <ng-container i18n="@@foo">`,
+           `    Foo`,
+           `  </ng-container>`,
+           `} @else {`,
+           `  <ng-container i18n="@@bar"> Bar </ng-container>`,
+           `}`,
+           `@if (true) {`,
+           `  <a i18n="@@bam">Bam</a>`,
+           `}`,
+         ].join('\n'));
+       });
+
     it('should migrate an if else case', async () => {
       writeFile('/comp.ts', `
         import {Component} from '@angular/core';
