address feedback

Author: mmalerba

src/lib/stepper/step-header.html

@@ -1,15 +1,11 @@
 <div class="mat-step-header-ripple" mat-ripple [matRippleTrigger]="_getHostElement()"></div>
-<div class="mat-step-icon-state-{{state}} mat-step-icon" [class.mat-step-icon-selected]="selected"
-     [ngSwitch]="state">
-
-  <div class="mat-step-icon-content">
-    <ng-container *ngSwitchDefault [ngSwitch]="!!(iconOverrides && iconOverrides[state])">
-      <ng-container
-        *ngSwitchCase="true"
-        [ngTemplateOutlet]="iconOverrides[state]"
-        [ngTemplateOutletContext]="_getIconContext()"></ng-container>
-      <mat-icon *ngSwitchDefault>{{_getDefaultIconForState(state)}}</mat-icon>
-    </ng-container>
+<div class="mat-step-icon-state-{{state}} mat-step-icon" [class.mat-step-icon-selected]="selected">
+  <div class="mat-step-icon-content" [ngSwitch]="!!(iconOverrides && iconOverrides[state])">
+    <ng-container
+      *ngSwitchCase="true"
+      [ngTemplateOutlet]="iconOverrides[state]"
+      [ngTemplateOutletContext]="_getIconContext()"></ng-container>
+    <mat-icon *ngSwitchDefault>{{_getDefaultIconForState(state)}}</mat-icon>
   </div>
 </div>
 <div class="mat-step-label"

src/lib/stepper/step-header.ts

@@ -6,7 +6,7 @@
  * found in the LICENSE file at https://angular.io/license
  */
 
-import {FocusMonitor} from '@angular/cdk/a11y';
+import { FocusMonitor } from '@angular/cdk/a11y';
 import {
   ChangeDetectionStrategy,
   ChangeDetectorRef,
@@ -15,26 +15,25 @@ import {
   Input,
   OnDestroy,
   ViewEncapsulation,
-  TemplateRef,
+  TemplateRef
 } from '@angular/core';
-import {Subscription} from 'rxjs';
-import {MatStepLabel} from './step-label';
-import {MatStepperIntl} from './stepper-intl';
-import {MatStepperIconContext} from './stepper-icon';
-import {CdkStepHeader, StepState} from '@angular/cdk/stepper';
-
+import { Subscription } from 'rxjs';
+import { MatStepLabel } from './step-label';
+import { MatStepperIntl } from './stepper-intl';
+import { MatStepperIconContext } from './stepper-icon';
+import { CdkStepHeader, StepState } from '@angular/cdk/stepper';
 
 @Component({
   moduleId: module.id,
   selector: 'mat-step-header',
   templateUrl: 'step-header.html',
   styleUrls: ['step-header.css'],
   host: {
-    'class': 'mat-step-header',
-    'role': 'tab',
+    class: 'mat-step-header',
+    role: 'tab'
   },
   encapsulation: ViewEncapsulation.None,
-  changeDetection: ChangeDetectionStrategy.OnPush,
+  changeDetection: ChangeDetectionStrategy.OnPush
 })
 export class MatStepHeader extends CdkStepHeader implements OnDestroy {
   private _intlSubscription: Subscription;
@@ -49,7 +48,7 @@ export class MatStepHeader extends CdkStepHeader implements OnDestroy {
   @Input() errorMessage: string;
 
   /** Overrides for the header icons, passed in via the stepper. */
-  @Input() iconOverrides: {[key: string]: TemplateRef<MatStepperIconContext>};
+  @Input() iconOverrides: { [key: string]: TemplateRef<MatStepperIconContext> };
 
   /** Index of the given step. */
   @Input() index: number;
@@ -67,10 +66,13 @@ export class MatStepHeader extends CdkStepHeader implements OnDestroy {
     public _intl: MatStepperIntl,
     private _focusMonitor: FocusMonitor,
     _elementRef: ElementRef<HTMLElement>,
-    changeDetectorRef: ChangeDetectorRef) {
+    changeDetectorRef: ChangeDetectorRef
+  ) {
     super(_elementRef);
     _focusMonitor.monitor(_elementRef, true);
-    this._intlSubscription = _intl.changes.subscribe(() => changeDetectorRef.markForCheck());
+    this._intlSubscription = _intl.changes.subscribe(() =>
+      changeDetectorRef.markForCheck()
+    );
   }
 
   ngOnDestroy() {
@@ -102,13 +104,16 @@ export class MatStepHeader extends CdkStepHeader implements OnDestroy {
     };
   }
 
-  _getDefaultIconForState(state: string): string {
+  _getDefaultIconForState(state: StepState): string {
     if (state == 'number') {
       return `${this.index + 1}`;
     }
-    return ({
-      'edit': 'create',
-      'error': 'warning',
-    } as {[key: string]: string})[state] || state;
+    if (state == 'edit') {
+      return 'create';
+    }
+    if (state == 'error') {
+      return 'warning';
+    }
+    return state;
   }
 }