fix(material/dialog): content directives picking up wrong ref for stacked mixed type dialogs

The dialog content directives try to resolve their closest dialog using DI, and if it fails (e.g.
inside a template dialog), they fall back to resolving it through the DOM. This becomes
problematic if the `ng-template` was declared inside another dialog component, because
it'll pick up the declaration dialog, not the one that the button exists in.

These changes remove the resolution using DI and switch to only going through the DOM.

Fixes #21554.

Author: crisbeto

src/material-experimental/mdc-dialog/dialog-content-directives.ts

@@ -9,10 +9,10 @@
 import {
   Directive,
   ElementRef,
+  Inject,
   Input,
   OnChanges,
   OnInit,
-  Optional,
   SimpleChanges,
 } from '@angular/core';
 import {_closeDialogVia} from '@angular/material/dialog';
@@ -47,22 +47,22 @@ export class MatDialogClose implements OnInit, OnChanges {
 
   @Input('matDialogClose') _matDialogClose: any;
 
+  /** Reference to the dialog that the close button is placed inside of. */
+  dialogRef: MatDialogRef<any>;
+
   constructor(
-      // The dialog title directive is always used in combination with a `MatDialogRef`.
-      // tslint:disable-next-line: lightweight-tokens
-      @Optional() public dialogRef: MatDialogRef<any>,
-      private _elementRef: ElementRef<HTMLElement>,
-      private _dialog: MatDialog) {}
+    /**
+     * @deprecated `_dialogRef` parameter to be removed.
+     * @breaking-change 12.0.0
+     */
+    @Inject(ElementRef) _dialogRef: any,
+    private _elementRef: ElementRef<HTMLElement>,
+    private _dialog: MatDialog) {}
 
   ngOnInit() {
-    if (!this.dialogRef) {
-      // When this directive is included in a dialog via TemplateRef (rather than being
-      // in a Component), the DialogRef isn't available via injection because embedded
-      // views cannot be given a custom injector. Instead, we look up the DialogRef by
-      // ID. This must occur in `onInit`, as the ID binding for the dialog container won't
-      // be resolved at constructor time.
-      this.dialogRef = getClosestDialog(this._elementRef, this._dialog.openDialogs)!;
-    }
+    // Always resolve the closest dialog using the DOM, rather than DI, because DI won't work for
+    // `TemplateRef`-based dialogs and it may give us wrong results for stacked ones (see #21554).
+    this.dialogRef = getClosestDialog(this._elementRef, this._dialog.openDialogs)!;
   }
 
   ngOnChanges(changes: SimpleChanges) {
@@ -97,17 +97,19 @@ export class MatDialogClose implements OnInit, OnChanges {
 export class MatDialogTitle implements OnInit {
   @Input() id: string = `mat-mdc-dialog-title-${dialogElementUid++}`;
 
+  private _dialogRef: MatDialogRef<unknown>;
+
   constructor(
-      // The dialog title directive is always used in combination with a `MatDialogRef`.
-      // tslint:disable-next-line: lightweight-tokens
-      @Optional() private _dialogRef: MatDialogRef<any>,
+      /**
+       * @deprecated `_dialogRef` parameter to be removed.
+       * @breaking-change 12.0.0
+       */
+      @Inject(ElementRef) _dialogRef: any,
       private _elementRef: ElementRef<HTMLElement>,
       private _dialog: MatDialog) {}
 
   ngOnInit() {
-    if (!this._dialogRef) {
-      this._dialogRef = getClosestDialog(this._elementRef, this._dialog.openDialogs)!;
-    }
+    this._dialogRef = getClosestDialog(this._elementRef, this._dialog.openDialogs)!;
 
     if (this._dialogRef) {
       Promise.resolve().then(() => {

src/material-experimental/mdc-dialog/dialog.spec.ts

@@ -742,6 +742,26 @@ describe('MDC-based MatDialog', () => {
        expect(resolver.resolveComponentFactory).toHaveBeenCalled();
      }));
 
+  it('should close the correct dialog when stacked and using a template from another dialog',
+     fakeAsync(() => {
+       const dialogRef = dialog.open(MixedTypeStackedDialog);
+       viewContainerFixture.detectChanges();
+
+       dialogRef.componentInstance.open();
+       viewContainerFixture.detectChanges();
+
+       expect(overlayContainerElement.textContent).toContain('Bottom');
+       expect(overlayContainerElement.textContent).toContain('Top');
+
+       (overlayContainerElement.querySelector('.close') as HTMLButtonElement).click();
+       flushMicrotasks();
+       viewContainerFixture.detectChanges();
+       tick(500);
+
+       expect(overlayContainerElement.textContent).toContain('Bottom');
+       expect(overlayContainerElement.textContent).not.toContain('Top');
+     }));
+
   describe('passing in data', () => {
     it('should be able to pass in data', () => {
       let config = {data: {stringParam: 'hello', dateParam: new Date()}};
@@ -1905,6 +1925,26 @@ class DialogWithoutFocusableElements {
 })
 class ShadowDomComponent {}
 
+@Component({
+  template: `
+    Bottom
+    <ng-template>
+      Top
+      <button class="close" mat-dialog-close>Close</button>
+    </ng-template>
+  `,
+})
+class MixedTypeStackedDialog {
+  @ViewChild(TemplateRef) template: TemplateRef<any>;
+
+  constructor(private _dialog: MatDialog) {}
+
+  open() {
+    this._dialog.open(this.template);
+  }
+}
+
+
 // Create a real (non-test) NgModule as a workaround for
 // https://github.com/angular/angular/issues/10760
 const TEST_DIRECTIVES = [
@@ -1918,6 +1958,7 @@ const TEST_DIRECTIVES = [
   DialogWithoutFocusableElements,
   ComponentWithContentElementTemplateRef,
   ShadowDomComponent,
+  MixedTypeStackedDialog,
 ];
 
 @NgModule({
@@ -1931,6 +1972,7 @@ const TEST_DIRECTIVES = [
     ContentElementDialog,
     DialogWithInjectedData,
     DialogWithoutFocusableElements,
+    MixedTypeStackedDialog,
   ],
 })
 class DialogTestModule {

src/material/dialog/dialog-content-directives.ts

@@ -11,9 +11,9 @@ import {
   Input,
   OnChanges,
   OnInit,
-  Optional,
   SimpleChanges,
   ElementRef,
+  Inject,
 } from '@angular/core';
 import {MatDialog} from './dialog';
 import {_closeDialogVia, MatDialogRef} from './dialog-ref';
@@ -45,28 +45,26 @@ export class MatDialogClose implements OnInit, OnChanges {
 
   @Input('matDialogClose') _matDialogClose: any;
 
-  constructor(
+  /**
+   * Reference to the containing dialog.
+   * @deprecated `dialogRef` property to become private.
+   * @breaking-change 13.0.0
+   */
+  dialogRef: MatDialogRef<any>;
 
+  constructor(
     /**
-     * Reference to the containing dialog.
-     * @deprecated `dialogRef` property to become private.
-     * @breaking-change 13.0.0
+     * @deprecated `_dialogRef` parameter to be removed.
+     * @breaking-change 12.0.0
      */
-    // The dialog title directive is always used in combination with a `MatDialogRef`.
-    // tslint:disable-next-line: lightweight-tokens
-    @Optional() public dialogRef: MatDialogRef<any>,
+    @Inject(ElementRef) _dialogRef: any,
     private _elementRef: ElementRef<HTMLElement>,
     private _dialog: MatDialog) {}
 
   ngOnInit() {
-    if (!this.dialogRef) {
-      // When this directive is included in a dialog via TemplateRef (rather than being
-      // in a Component), the DialogRef isn't available via injection because embedded
-      // views cannot be given a custom injector. Instead, we look up the DialogRef by
-      // ID. This must occur in `onInit`, as the ID binding for the dialog container won't
-      // be resolved at constructor time.
-      this.dialogRef = getClosestDialog(this._elementRef, this._dialog.openDialogs)!;
-    }
+    // Always resolve the closest dialog using the DOM, rather than DI, because DI won't work for
+    // `TemplateRef`-based dialogs and it may give us wrong results for stacked ones (see #21554).
+    this.dialogRef = getClosestDialog(this._elementRef, this._dialog.openDialogs)!;
   }
 
   ngOnChanges(changes: SimpleChanges) {
@@ -102,17 +100,19 @@ export class MatDialogTitle implements OnInit {
   /** Unique id for the dialog title. If none is supplied, it will be auto-generated. */
   @Input() id: string = `mat-dialog-title-${dialogElementUid++}`;
 
+  private _dialogRef: MatDialogRef<unknown>;
+
   constructor(
-      // The dialog title directive is always used in combination with a `MatDialogRef`.
-      // tslint:disable-next-line: lightweight-tokens
-      @Optional() private _dialogRef: MatDialogRef<any>,
+      /**
+       * @deprecated `_dialogRef` parameter to be removed.
+       * @breaking-change 12.0.0
+       */
+      @Inject(ElementRef) _dialogRef: any,
       private _elementRef: ElementRef<HTMLElement>,
       private _dialog: MatDialog) {}
 
   ngOnInit() {
-    if (!this._dialogRef) {
-      this._dialogRef = getClosestDialog(this._elementRef, this._dialog.openDialogs)!;
-    }
+    this._dialogRef = getClosestDialog(this._elementRef, this._dialog.openDialogs)!;
 
     if (this._dialogRef) {
       Promise.resolve().then(() => {

src/material/dialog/dialog.spec.ts

@@ -800,6 +800,26 @@ describe('MatDialog', () => {
       expect(resolver.resolveComponentFactory).toHaveBeenCalled();
     }));
 
+  it('should close the correct dialog when stacked and using a template from another dialog',
+    fakeAsync(() => {
+      const dialogRef = dialog.open(MixedTypeStackedDialog);
+      viewContainerFixture.detectChanges();
+
+      dialogRef.componentInstance.open();
+      viewContainerFixture.detectChanges();
+
+      expect(overlayContainerElement.textContent).toContain('Bottom');
+      expect(overlayContainerElement.textContent).toContain('Top');
+
+      (overlayContainerElement.querySelector('.close') as HTMLButtonElement).click();
+      flushMicrotasks();
+      viewContainerFixture.detectChanges();
+      tick(500);
+
+      expect(overlayContainerElement.textContent).toContain('Bottom');
+      expect(overlayContainerElement.textContent).not.toContain('Top');
+    }));
+
   describe('passing in data', () => {
     it('should be able to pass in data', () => {
       let config = {
@@ -1981,6 +2001,25 @@ class DialogWithoutFocusableElements {}
 })
 class ShadowDomComponent {}
 
+@Component({
+  template: `
+    Bottom
+    <ng-template>
+      Top
+      <button class="close" mat-dialog-close>Close</button>
+    </ng-template>
+  `,
+})
+class MixedTypeStackedDialog {
+  @ViewChild(TemplateRef) template: TemplateRef<any>;
+
+  constructor(private _dialog: MatDialog) {}
+
+  open() {
+    this._dialog.open(this.template);
+  }
+}
+
 // Create a real (non-test) NgModule as a workaround for
 // https://github.com/angular/angular/issues/10760
 const TEST_DIRECTIVES = [
@@ -1994,6 +2033,7 @@ const TEST_DIRECTIVES = [
   DialogWithoutFocusableElements,
   ComponentWithContentElementTemplateRef,
   ShadowDomComponent,
+  MixedTypeStackedDialog,
 ];
 
 @NgModule({
@@ -2007,6 +2047,7 @@ const TEST_DIRECTIVES = [
     ContentElementDialog,
     DialogWithInjectedData,
     DialogWithoutFocusableElements,
+    MixedTypeStackedDialog,
   ],
 })
 class DialogTestModule { }

tools/public_api_guard/material/dialog.d.ts

@@ -89,12 +89,12 @@ export declare class MatDialogClose implements OnInit, OnChanges {
     dialogResult: any;
     type: 'submit' | 'button' | 'reset';
     constructor(
-    dialogRef: MatDialogRef<any>, _elementRef: ElementRef<HTMLElement>, _dialog: MatDialog);
+    _dialogRef: any, _elementRef: ElementRef<HTMLElement>, _dialog: MatDialog);
     _onButtonClick(event: MouseEvent): void;
     ngOnChanges(changes: SimpleChanges): void;
     ngOnInit(): void;
     static ɵdir: i0.ɵɵDirectiveDeclaration<MatDialogClose, "[mat-dialog-close], [matDialogClose]", ["matDialogClose"], { "ariaLabel": "aria-label"; "type": "type"; "dialogResult": "mat-dialog-close"; "_matDialogClose": "matDialogClose"; }, {}, never>;
-    static ɵfac: i0.ɵɵFactoryDeclaration<MatDialogClose, [{ optional: true; }, null, null]>;
+    static ɵfac: i0.ɵɵFactoryDeclaration<MatDialogClose, never>;
 }
 
 export declare class MatDialogConfig<D = any> {
@@ -172,10 +172,11 @@ export declare const enum MatDialogState {
 
 export declare class MatDialogTitle implements OnInit {
     id: string;
-    constructor(_dialogRef: MatDialogRef<any>, _elementRef: ElementRef<HTMLElement>, _dialog: MatDialog);
+    constructor(
+    _dialogRef: any, _elementRef: ElementRef<HTMLElement>, _dialog: MatDialog);
     ngOnInit(): void;
     static ɵdir: i0.ɵɵDirectiveDeclaration<MatDialogTitle, "[mat-dialog-title], [matDialogTitle]", ["matDialogTitle"], { "id": "id"; }, {}, never>;
-    static ɵfac: i0.ɵɵFactoryDeclaration<MatDialogTitle, [{ optional: true; }, null, null]>;
+    static ɵfac: i0.ɵɵFactoryDeclaration<MatDialogTitle, never>;
 }
 
 export declare function throwMatDialogContentAlreadyAttachedError(): void;