/**
 * @license
 * Copyright Google LLC 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 {bold, green, red} from 'chalk';
import * as ts from 'typescript';
import {MigrationRule} from '../../update-tool/migration-rule';
import {PropertyNameUpgradeData} from '../data/property-names';
import {determineBaseTypes} from '../typescript/base-types';
import {getVersionUpgradeData, RuleUpgradeData} from '../upgrade-data';

/**
 * Rule that identifies class declarations that extend CDK or Material classes
 * which had a public property change.
 */
export class ClassInheritanceRule extends MigrationRule<RuleUpgradeData> {
  /**
   * Map of classes that have been updated. Each class name maps to the according property
   * change data.
   */
  propertyNames = new Map<string, PropertyNameUpgradeData>();

  // Only enable the migration rule if there is upgrade data.
  ruleEnabled = this.propertyNames.size !== 0;

  init(): void {
    getVersionUpgradeData(this, 'propertyNames')
        .filter(data => data.whitelist && data.whitelist.classes)
        .forEach(
            data => data.whitelist.classes.forEach(name => this.propertyNames.set(name, data)));
  }

  visitNode(node: ts.Node): void {
    if (ts.isClassDeclaration(node)) {
      this._visitClassDeclaration(node);
    }
  }

  private _visitClassDeclaration(node: ts.ClassDeclaration) {
    const baseTypes = determineBaseTypes(node);
    const className = node.name ? node.name.text : '{unknown-name}';

    if (!baseTypes) {
      return;
    }

    baseTypes.forEach(typeName => {
      const data = this.propertyNames.get(typeName);

      if (data) {
        this.createFailureAtNode(
            node,
            `Found class "${bold(className)}" which extends class ` +
                `"${bold(typeName)}". Please note that the base class property ` +
                `"${red(data.replace)}" has changed to "${green(data.replaceWith)}". ` +
                `You may need to update your class as well.`);
      }
    });
  }
}