fix(menu): unable to open same sub-menu from different triggers and not picking up indirect descendant items

* Reworks the relationship between the menu items and the menu panel to allow for the items to be indirect descendants of the menu, while also allowing for `mat-menu` instances to be declared inside of other `mat-menu` instances without having the root `mat-menu` pick up all of the descendant items.
* Adds the ability to pass in an array to the `ListKeyManager`, in addition to a `QueryList`.
* Fixes not being able to re-use the same sub-menu between multiple sub-menu triggers.

Fixes #9969.
Fixes #9987.

Author: crisbeto

src/cdk/a11y/key-manager/list-key-manager.ts

@@ -50,17 +50,22 @@ export class ListKeyManager<T extends ListKeyManagerOption> {
   // Buffer for the letters that the user has pressed when the typeahead option is turned on.
   private _pressedLetters: string[] = [];
 
-  constructor(private _items: QueryList<T>) {
-    _items.changes.subscribe((newItems: QueryList<T>) => {
-      if (this._activeItem) {
-        const itemArray = newItems.toArray();
-        const newIndex = itemArray.indexOf(this._activeItem);
-
-        if (newIndex > -1 && newIndex !== this._activeItemIndex) {
-          this._activeItemIndex = newIndex;
+  constructor(private _items: QueryList<T> | T[]) {
+    // We allow for the items to be an array because, in some cases, the consumer may
+    // not have access to a QueryList of the items they want to manage (e.g. when the
+    // items aren't being collected via `ViewChildren` or `ContentChildren`).
+    if (_items instanceof QueryList) {
+      _items.changes.subscribe((newItems: QueryList<T>) => {
+        if (this._activeItem) {
+          const itemArray = newItems.toArray();
+          const newIndex = itemArray.indexOf(this._activeItem);
+
+          if (newIndex > -1 && newIndex !== this._activeItemIndex) {
+            this._activeItemIndex = newIndex;
+          }
         }
-      }
-    });
+      });
+    }
   }
 
   /**
@@ -120,7 +125,7 @@ export class ListKeyManager<T extends ListKeyManagerOption> {
       filter(() => this._pressedLetters.length > 0),
       map(() => this._pressedLetters.join(''))
     ).subscribe(inputString => {
-      const items = this._items.toArray();
+      const items = this._getItemsArray();
 
       // Start at 1 because we want to start searching at the item immediately
       // following the current active item.
@@ -148,7 +153,7 @@ export class ListKeyManager<T extends ListKeyManagerOption> {
     const previousIndex = this._activeItemIndex;
 
     this._activeItemIndex = index;
-    this._activeItem = this._items.toArray()[index];
+    this._activeItem = this._getItemsArray()[index];
 
     if (this._activeItemIndex !== previousIndex) {
       this.change.next(index);
@@ -267,17 +272,18 @@ export class ListKeyManager<T extends ListKeyManagerOption> {
    * currently active item and the new active item. It will calculate differently
    * depending on whether wrap mode is turned on.
    */
-  private _setActiveItemByDelta(delta: -1 | 1, items = this._items.toArray()): void {
-    this._wrap ? this._setActiveInWrapMode(delta, items)
-               : this._setActiveInDefaultMode(delta, items);
+  private _setActiveItemByDelta(delta: -1 | 1): void {
+    this._wrap ? this._setActiveInWrapMode(delta) : this._setActiveInDefaultMode(delta);
   }
 
   /**
    * Sets the active item properly given "wrap" mode. In other words, it will continue to move
    * down the list until it finds an item that is not disabled, and it will wrap if it
    * encounters either end of the list.
    */
-  private _setActiveInWrapMode(delta: -1 | 1, items: T[]): void {
+  private _setActiveInWrapMode(delta: -1 | 1): void {
+    const items = this._getItemsArray();
+
     for (let i = 1; i <= items.length; i++) {
       const index = (this._activeItemIndex + (delta * i) + items.length) % items.length;
       const item = items[index];
@@ -294,17 +300,18 @@ export class ListKeyManager<T extends ListKeyManagerOption> {
    * continue to move down the list until it finds an item that is not disabled. If
    * it encounters either end of the list, it will stop and not wrap.
    */
-  private _setActiveInDefaultMode(delta: -1 | 1, items: T[]): void {
-    this._setActiveItemByIndex(this._activeItemIndex + delta, delta, items);
+  private _setActiveInDefaultMode(delta: -1 | 1): void {
+    this._setActiveItemByIndex(this._activeItemIndex + delta, delta);
   }
 
   /**
    * Sets the active item to the first enabled item starting at the index specified. If the
    * item is disabled, it will move in the fallbackDelta direction until it either
    * finds an enabled item or encounters the end of the list.
    */
-  private _setActiveItemByIndex(index: number, fallbackDelta: -1 | 1,
-                                items = this._items.toArray()): void {
+  private _setActiveItemByIndex(index: number, fallbackDelta: -1 | 1): void {
+    const items = this._getItemsArray();
+
     if (!items[index]) {
       return;
     }
@@ -319,4 +326,9 @@ export class ListKeyManager<T extends ListKeyManagerOption> {
 
     this.setActiveItem(index);
   }
+
+  /** Returns the items as an array. */
+  private _getItemsArray(): T[] {
+    return this._items instanceof QueryList ? this._items.toArray() : this._items;
+  }
 }

src/lib/menu/menu-directive.ts

@@ -17,15 +17,13 @@ import {
   ChangeDetectionStrategy,
   Component,
   ContentChild,
-  ContentChildren,
   ElementRef,
   EventEmitter,
   Inject,
   InjectionToken,
   Input,
   OnDestroy,
   Output,
-  QueryList,
   TemplateRef,
   ViewChild,
   ViewEncapsulation,
@@ -35,14 +33,16 @@ import {
 import {Observable} from 'rxjs/Observable';
 import {merge} from 'rxjs/observable/merge';
 import {Subscription} from 'rxjs/Subscription';
+import {Subject} from 'rxjs/Subject';
 import {matMenuAnimations} from './menu-animations';
 import {throwMatMenuInvalidPositionX, throwMatMenuInvalidPositionY} from './menu-errors';
 import {MatMenuItem} from './menu-item';
-import {MatMenuPanel} from './menu-panel';
+import {MatMenuPanel, MAT_MENU_PANEL} from './menu-panel';
 import {MatMenuContent} from './menu-content';
 import {MenuPositionX, MenuPositionY} from './menu-positions';
 import {coerceBooleanProperty} from '@angular/cdk/coercion';
 import {FocusOrigin} from '@angular/cdk/a11y';
+import {AnimationEvent} from '@angular/animations';
 
 
 /** Default `mat-menu` options that can be overridden. */
@@ -79,18 +79,27 @@ const MAT_MENU_BASE_ELEVATION = 2;
   changeDetection: ChangeDetectionStrategy.OnPush,
   encapsulation: ViewEncapsulation.None,
   preserveWhitespaces: false,
+  exportAs: 'matMenu',
   animations: [
     matMenuAnimations.transformMenu,
     matMenuAnimations.fadeInItems
   ],
-  exportAs: 'matMenu'
+  providers: [
+    {provide: MAT_MENU_PANEL, useExisting: MatMenu}
+  ]
 })
-export class MatMenu implements OnInit, AfterContentInit, MatMenuPanel, OnDestroy {
+export class MatMenu implements OnInit, AfterContentInit, MatMenuPanel<MatMenuItem>, OnDestroy {
   private _keyManager: FocusKeyManager<MatMenuItem>;
   private _xPosition: MenuPositionX = this._defaultOptions.xPosition;
   private _yPosition: MenuPositionY = this._defaultOptions.yPosition;
   private _previousElevation: string;
 
+  /** Menu items inside the current menu. */
+  private _items: MatMenuItem[] = [];
+
+  /** Emits whenever the amount of menu items changes. */
+  private _itemChanges = new Subject<MatMenuItem[]>();
+
   /** Subscription to tab events on the menu panel */
   private _tabSubscription = Subscription.EMPTY;
 
@@ -100,6 +109,12 @@ export class MatMenu implements OnInit, AfterContentInit, MatMenuPanel, OnDestro
   /** Current state of the panel animation. */
   _panelAnimationState: 'void' | 'enter' = 'void';
 
+  /** Emits whenever an animation on the menu completes. */
+  _animationDone = new Subject<AnimationEvent>();
+
+  /** Whether the menu is animating. */
+  _isAnimating: boolean;
+
   /** Parent menu of the current menu panel. */
   parentMenu: MatMenuPanel | undefined;
 
@@ -134,9 +149,6 @@ export class MatMenu implements OnInit, AfterContentInit, MatMenuPanel, OnDestro
   /** @docs-private */
   @ViewChild(TemplateRef) templateRef: TemplateRef<any>;
 
-  /** List of the items inside of a menu. */
-  @ContentChildren(MatMenuItem) items: QueryList<MatMenuItem>;
-
   /**
    * Menu content that will be rendered lazily.
    * @docs-private
@@ -202,7 +214,7 @@ export class MatMenu implements OnInit, AfterContentInit, MatMenuPanel, OnDestro
   }
 
   ngAfterContentInit() {
-    this._keyManager = new FocusKeyManager<MatMenuItem>(this.items).withWrap().withTypeAhead();
+    this._keyManager = new FocusKeyManager<MatMenuItem>(this._items).withWrap().withTypeAhead();
     this._tabSubscription = this._keyManager.tabOut.subscribe(() => this.close.emit('keydown'));
   }
 
@@ -213,16 +225,10 @@ export class MatMenu implements OnInit, AfterContentInit, MatMenuPanel, OnDestro
 
   /** Stream that emits whenever the hovered menu item changes. */
   _hovered(): Observable<MatMenuItem> {
-    if (this.items) {
-      return this.items.changes.pipe(
-        startWith(this.items),
-        switchMap(items => merge(...items.map(item => item._hovered)))
-      );
-    }
-
-    return this._ngZone.onStable
-      .asObservable()
-      .pipe(take(1), switchMap(() => this._hovered()));
+    return this._itemChanges.pipe(
+      startWith(this._items),
+      switchMap(items => merge(...items.map(item => item._hovered)))
+    );
   }
 
   /** Handle a keyboard event from the menu, delegating to the appropriate action. */
@@ -300,6 +306,35 @@ export class MatMenu implements OnInit, AfterContentInit, MatMenuPanel, OnDestro
     }
   }
 
+  /**
+   * Registers a menu item with the menu.
+   * @docs-private
+   */
+  addItem(item: MatMenuItem) {
+    // We register the items through this method, rather than picking them up through
+    // `ContentChildren`, because we need the items to be picked up by their closest
+    // `mat-menu` ancestor. If we used `@ContentChildren(MatMenuItem, {descendants: true})`,
+    // all descendant items will bleed into the top-level menu in the case where the consumer
+    // has `mat-menu` instances nested inside each other.
+    if (this._items.indexOf(item) === -1) {
+      this._items.push(item);
+      this._itemChanges.next(this._items);
+    }
+  }
+
+  /**
+   * Removes an item from the menu.
+   * @docs-private
+   */
+  removeItem(item: MatMenuItem) {
+    const index = this._items.indexOf(item);
+
+    if (this._items.indexOf(item) > -1) {
+      this._items.splice(index, 1);
+      this._itemChanges.next(this._items);
+    }
+  }
+
   /** Starts the enter animation. */
   _startAnimation() {
     // @deletion-target 6.0.0 Combine with _resetAnimation.
@@ -313,7 +348,8 @@ export class MatMenu implements OnInit, AfterContentInit, MatMenuPanel, OnDestro
   }
 
   /** Callback that is invoked when the panel animation completes. */
-  _onAnimationDone(_event: AnimationEvent) {
-    // @deletion-target 6.0.0 Not being used anymore. To be removed.
+  _onAnimationDone(event: AnimationEvent) {
+    this._animationDone.next(event);
+    this._isAnimating = false;
   }
 }

src/lib/menu/menu-item.ts

@@ -14,6 +14,7 @@ import {
   OnDestroy,
   ViewEncapsulation,
   Inject,
+  Optional,
 } from '@angular/core';
 import {
   CanDisable,
@@ -23,6 +24,7 @@ import {
 } from '@angular/material/core';
 import {Subject} from 'rxjs/Subject';
 import {DOCUMENT} from '@angular/common';
+import {MAT_MENU_PANEL, MatMenuPanel} from './menu-panel';
 
 // Boilerplate for applying mixins to MatMenuItem.
 /** @docs-private */
@@ -71,7 +73,8 @@ export class MatMenuItem extends _MatMenuItemMixinBase
   constructor(
     private _elementRef: ElementRef,
     @Inject(DOCUMENT) document?: any,
-    private _focusMonitor?: FocusMonitor) {
+    private _focusMonitor?: FocusMonitor,
+    @Inject(MAT_MENU_PANEL) @Optional() private _parentMenu?: MatMenuPanel<MatMenuItem>) {
 
     // @deletion-target 6.0.0 make `_focusMonitor` and `document` required params.
     super();
@@ -83,6 +86,10 @@ export class MatMenuItem extends _MatMenuItemMixinBase
       _focusMonitor.monitor(this._getHostElement(), false);
     }
 
+    if (_parentMenu && _parentMenu.addItem) {
+      _parentMenu.addItem(this);
+    }
+
     this._document = document;
   }
 
@@ -100,6 +107,10 @@ export class MatMenuItem extends _MatMenuItemMixinBase
       this._focusMonitor.stopMonitoring(this._getHostElement());
     }
 
+    if (this._parentMenu && this._parentMenu.removeItem) {
+      this._parentMenu.removeItem(this);
+    }
+
     this._hovered.complete();
   }
 

src/lib/menu/menu-panel.ts

@@ -6,17 +6,23 @@
  * found in the LICENSE file at https://angular.io/license
  */
 
-import {EventEmitter, TemplateRef} from '@angular/core';
+import {EventEmitter, TemplateRef, InjectionToken} from '@angular/core';
 import {MenuPositionX, MenuPositionY} from './menu-positions';
 import {Direction} from '@angular/cdk/bidi';
 import {FocusOrigin} from '@angular/cdk/a11y';
 import {MatMenuContent} from './menu-content';
 
+/**
+ * Injection token used to provide the parent menu to menu-specific components.
+ * @docs-private
+ */
+export const MAT_MENU_PANEL = new InjectionToken<MatMenuPanel>('MAT_MENU_PANEL');
+
 /**
  * Interface for a custom menu panel that can be used with `matMenuTriggerFor`.
  * @docs-private
  */
-export interface MatMenuPanel {
+export interface MatMenuPanel<T = any> {
   xPosition: MenuPositionX;
   yPosition: MenuPositionY;
   overlapTrigger: boolean;
@@ -30,4 +36,6 @@ export interface MatMenuPanel {
   setElevation?(depth: number): void;
   lazyContent?: MatMenuContent;
   backdropClass?: string;
+  addItem?: (item: T) => void;
+  removeItem?: (item: T) => void;
 }