feat(snack-bar): inject data and MdSnackBarRef into custom snack-bar component

* Injects the `MdSnackBarRef` into custom snack bar instances.
* Adds the option to inject arbitrary data into a snack bar.
* Turns the `DialogInjector` into a `PortalInjector` to allow it to be used elsewhere (e.g. the snack-bar).
* Minor tweaks to the `SimpleSnackBar` to get it to use the new DI tokens instead of assigning data directly to the component instance.

Fixes #5371.

Author: crisbeto

src/lib/dialog/dialog-injector.ts renamed to src/lib/core/portal/portal-injector.ts

@@ -1,28 +1,30 @@
-/**
- * @license
- * Copyright Google Inc. All Rights Reserved.
- *
- * Use of this source code is governed by an MIT-style license that can be
- * found in the LICENSE file at https://angular.io/license
- */
-
-import {Injector} from '@angular/core';
-import {MdDialogRef} from './dialog-ref';
-import {MdDialogContainer} from './dialog-container';
-
-/** Custom injector type specifically for instantiating components with a dialog. */
-export class DialogInjector implements Injector {
-  constructor(
-    private _parentInjector: Injector,
-    private _customTokens: WeakMap<any, any>) { }
-
-  get(token: any, notFoundValue?: any): any {
-    const value = this._customTokens.get(token);
-
-    if (typeof value !== 'undefined') {
-      return value;
-    }
-
-    return this._parentInjector.get<any>(token, notFoundValue);
-  }
-}
+/**
+ * @license
+ * Copyright Google Inc. All Rights Reserved.
+ *
+ * Use of this source code is governed by an MIT-style license that can be
+ * found in the LICENSE file at https://angular.io/license
+ */
+
+import {Injector} from '@angular/core';
+
+/**
+ * Custom injector to be used when providing custom
+ * injection tokens to components inside a portal.
+ * @docs-private
+ */
+export class PortalInjector implements Injector {
+  constructor(
+    private _parentInjector: Injector,
+    private _customTokens: WeakMap<any, any>) { }
+
+  get(token: any, notFoundValue?: any): any {
+    const value = this._customTokens.get(token);
+
+    if (typeof value !== 'undefined') {
+      return value;
+    }
+
+    return this._parentInjector.get<any>(token, notFoundValue);
+  }
+}

src/lib/dialog/dialog.ts

@@ -25,9 +25,9 @@ import {
   OverlayState,
   ComponentPortal,
 } from '../core';
+import {PortalInjector} from '../core/portal/portal-injector';
 import {extendObject} from '../core/util/object-extend';
 import {ESCAPE} from '../core/keyboard/keycodes';
-import {DialogInjector} from './dialog-injector';
 import {MdDialogConfig} from './dialog-config';
 import {MdDialogRef} from './dialog-ref';
 import {MdDialogContainer} from './dialog-container';
@@ -222,7 +222,7 @@ export class MdDialog {
   private _createInjector<T>(
       config: MdDialogConfig,
       dialogRef: MdDialogRef<T>,
-      dialogContainer: MdDialogContainer): DialogInjector {
+      dialogContainer: MdDialogContainer): PortalInjector {
 
     let userInjector = config && config.viewContainerRef && config.viewContainerRef.injector;
     let injectionTokens = new WeakMap();
@@ -231,7 +231,7 @@ export class MdDialog {
     injectionTokens.set(MdDialogContainer, dialogContainer);
     injectionTokens.set(MD_DIALOG_DATA, config.data);
 
-    return new DialogInjector(userInjector || this._injector, injectionTokens);
+    return new PortalInjector(userInjector || this._injector, injectionTokens);
   }
 
   /**

src/lib/snack-bar/simple-snack-bar.html

@@ -1,2 +1,6 @@
-{{message}}
-<button class="mat-simple-snackbar-action" *ngIf="hasAction" (click)="dismiss()">{{action}}</button>
+{{data.message}}
+
+<button
+  class="mat-simple-snackbar-action"
+  *ngIf="hasAction"
+  (click)="dismiss()">{{data.action}}</button>

src/lib/snack-bar/simple-snack-bar.ts

@@ -6,8 +6,8 @@
  * found in the LICENSE file at https://angular.io/license
  */
 
-import {Component, ViewEncapsulation} from '@angular/core';
-import {MdSnackBarRef} from './snack-bar-ref';
+import {Component, ViewEncapsulation, Inject} from '@angular/core';
+import {MdSnackBarRef, MD_SNACK_BAR_DATA} from './index';
 
 
 /**
@@ -25,14 +25,9 @@ import {MdSnackBarRef} from './snack-bar-ref';
   }
 })
 export class SimpleSnackBar {
-  /** The message to be shown in the snack bar. */
-  message: string;
-
-  /** The label for the button in the snack bar. */
-  action: string;
-
-  /** The instance of the component making up the content of the snack bar. */
-  snackBarRef: MdSnackBarRef<SimpleSnackBar>;
+  constructor(
+    public snackBarRef: MdSnackBarRef<SimpleSnackBar>,
+    @Inject(MD_SNACK_BAR_DATA) public data: { message: string, action: string }) { }
 
   /** Dismisses the snack bar. */
   dismiss(): void {
@@ -41,6 +36,6 @@ export class SimpleSnackBar {
 
   /** If the action button should be shown. */
   get hasAction(): boolean {
-    return !!this.action;
+    return !!this.data.action;
   }
 }

src/lib/snack-bar/snack-bar-config.ts

@@ -30,4 +30,7 @@ export class MdSnackBarConfig {
 
   /** Text layout direction for the snack bar. */
   direction?: Direction = 'ltr';
+
+  /** Data being injected into the child component. */
+  data?: any = null;
 }

src/lib/snack-bar/snack-bar-ref.ts

@@ -17,12 +17,8 @@ import {MdSnackBarContainer} from './snack-bar-container';
  * Reference to a snack bar dispatched from the snack bar service.
  */
 export class MdSnackBarRef<T> {
-  private _instance: T;
-
   /** The instance of the component making up the content of the snack bar. */
-  get instance(): T {
-    return this._instance;
-  }
+  instance: T;
 
   /**
    * The instance of the component making up the content of the snack bar.
@@ -45,11 +41,8 @@ export class MdSnackBarRef<T> {
    */
   private _durationTimeoutId: number;
 
-  constructor(instance: T,
-              containerInstance: MdSnackBarContainer,
+  constructor(containerInstance: MdSnackBarContainer,
               private _overlayRef: OverlayRef) {
-    // Sets the readonly instance of the snack bar content component.
-    this._instance = instance;
     this.containerInstance = containerInstance;
     // Dismiss snackbar on action.
     this.onAction().subscribe(() => this.dismiss());

src/lib/snack-bar/snack-bar.md

@@ -15,9 +15,11 @@ let snackBarRef = snackBar.open('Message archived', 'Undo');
 let snackBarRef = snackbar.openFromComponent(MessageArchivedComponent);
 ```
 
-In either case, an `MdSnackBarRef` is returned. This can be used to dismiss the snack-bar or to 
-receive notification of when the snack-bar is dismissed. For simple messages with an action, the 
+In either case, a `MdSnackBarRef` is returned. This can be used to dismiss the snack-bar or to
+receive notification of when the snack-bar is dismissed. For simple messages with an action, the
 `MdSnackBarRef` exposes an observable for when the action is triggered.
+If you want to close a custom snack-bar that was opened via `openFromComponent`, from within the
+component itself, you can inject the `MdSnackBarRef`.
 
 ```ts
 snackBarRef.afterDismissed().subscribe(() => {
@@ -33,7 +35,7 @@ snackBarRef.dismiss();
 ```
 
 ### Dismissal
-A snack-bar can be dismissed manually by calling the `dismiss` method on the `MdSnackBarRef` 
+A snack-bar can be dismissed manually by calling the `dismiss` method on the `MdSnackBarRef`
 returned from the call to `open`.
 
 Only one snack-bar can ever be opened at one time. If a new snackbar is opened while a previous
@@ -45,3 +47,28 @@ snackbar.open('Message archived', 'Undo', {
   duration: 3000
 });
 ```
+
+### Sharing data with a custom snack-bar.
+You can share data with the custom snack-bar, that you opened via the `openFromComponent` method,
+by passing it through the `data` property.
+
+```ts
+snackbar.openFromComponent(MessageArchivedComponent, {
+  data: 'some data'
+});
+```
+
+To access the data in your component, you have to use the `MD_SNACK_BAR_DATA` injection token:
+
+```ts
+import {Component, Inject} from '@angular/core';
+import {MD_SNACK_BAR_DATA} from '@angular/material';
+
+@Component({
+  selector: 'your-snack-bar',
+  template: 'passed in {{ data }}',
+})
+export class MessageArchivedComponent {
+  constructor(@Inject(MD_SNACK_BAR_DATA) public data: any) { }
+}
+```

src/lib/snack-bar/snack-bar.spec.ts

@@ -7,11 +7,18 @@ import {
   flushMicrotasks,
   tick
 } from '@angular/core/testing';
-import {NgModule, Component, Directive, ViewChild, ViewContainerRef} from '@angular/core';
+import {NgModule, Component, Directive, ViewChild, ViewContainerRef, Inject} from '@angular/core';
 import {CommonModule} from '@angular/common';
 import {NoopAnimationsModule} from '@angular/platform-browser/animations';
-import {MdSnackBarModule, MdSnackBar, MdSnackBarConfig, SimpleSnackBar} from './index';
 import {OverlayContainer, LiveAnnouncer} from '../core';
+import {
+  MdSnackBarModule,
+  MdSnackBar,
+  MdSnackBarConfig,
+  MdSnackBarRef,
+  SimpleSnackBar,
+  MD_SNACK_BAR_DATA,
+} from './index';
 
 
 // TODO(josephperrott): Update tests to mock waiting for time to complete for animations.
@@ -174,17 +181,6 @@ describe('MdSnackBar', () => {
     });
   }));
 
-  it('should open a custom component', () => {
-    let config = {viewContainerRef: testViewContainerRef};
-    let snackBarRef = snackBar.openFromComponent(BurritosNotification, config);
-
-    expect(snackBarRef.instance instanceof BurritosNotification)
-      .toBe(true, 'Expected the snack bar content component to be BurritosNotification');
-    expect(overlayContainerElement.textContent!.trim())
-        .toBe('Burritos are on the way.',
-              `Expected the overlay text content to be 'Burritos are on the way'`);
-  });
-
   it('should set the animation state to visible on entry', () => {
     let config = {viewContainerRef: testViewContainerRef};
     let snackBarRef = snackBar.open(simpleMessage, undefined, config);
@@ -361,6 +357,37 @@ describe('MdSnackBar', () => {
     expect(pane.getAttribute('dir')).toBe('rtl', 'Expected the pane to be in RTL mode.');
   });
 
+  describe('with custom component', () => {
+    it('should open a custom component', () => {
+      const snackBarRef = snackBar.openFromComponent(BurritosNotification);
+
+      expect(snackBarRef.instance instanceof BurritosNotification)
+        .toBe(true, 'Expected the snack bar content component to be BurritosNotification');
+      expect(overlayContainerElement.textContent!.trim())
+          .toBe('Burritos are on the way.', 'Expected component to have the proper text.');
+    });
+
+    it('should inject the snack bar reference into the component', () => {
+      const snackBarRef = snackBar.openFromComponent(BurritosNotification);
+
+      expect(snackBarRef.instance.snackBarRef)
+        .toBe(snackBarRef, 'Expected component to have an injected snack bar reference.');
+    });
+
+    it('should be able to inject arbitrary user data', () => {
+      const snackBarRef = snackBar.openFromComponent(BurritosNotification, {
+        data: {
+          burritoType: 'Chimichanga'
+        }
+      });
+
+      expect(snackBarRef.instance.data).toBeTruthy('Expected component to have a data object.');
+      expect(snackBarRef.instance.data.burritoType)
+        .toBe('Chimichanga', 'Expected the injected data object to be the one the user provided.');
+    });
+
+  });
+
 });
 
 describe('MdSnackBar with parent MdSnackBar', () => {
@@ -453,7 +480,11 @@ class ComponentWithChildViewContainer {
 
 /** Simple component for testing ComponentPortal. */
 @Component({template: '<p>Burritos are on the way.</p>'})
-class BurritosNotification {}
+class BurritosNotification {
+  constructor(
+    public snackBarRef: MdSnackBarRef<BurritosNotification>,
+    @Inject(MD_SNACK_BAR_DATA) public data: any) { }
+}
 
 
 @Component({