/**
 * @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 * as t from '@babel/types';

import {FatalLinkerError} from '../../fatal_linker_error';
import {AstHost, Range} from '../ast_host';
import {assert} from '../utils';

/**
 * This implementation of `AstHost` is able to get information from Babel AST nodes.
 */
export class BabelAstHost implements AstHost<t.Expression> {
  getSymbolName(node: t.Expression): string|null {
    if (t.isIdentifier(node)) {
      return node.name;
    } else if (t.isMemberExpression(node) && t.isIdentifier(node.property)) {
      return node.property.name;
    } else {
      return null;
    }
  }

  isStringLiteral = t.isStringLiteral;

  parseStringLiteral(str: t.Expression): string {
    assert(str, t.isStringLiteral, 'a string literal');
    return str.value;
  }

  isNumericLiteral = t.isNumericLiteral;

  parseNumericLiteral(num: t.Expression): number {
    assert(num, t.isNumericLiteral, 'a numeric literal');
    return num.value;
  }

  isBooleanLiteral = t.isBooleanLiteral;

  parseBooleanLiteral(bool: t.Expression): boolean {
    assert(bool, t.isBooleanLiteral, 'a boolean literal');
    return bool.value;
  }

  isArrayLiteral = t.isArrayExpression;

  parseArrayLiteral(array: t.Expression): t.Expression[] {
    assert(array, t.isArrayExpression, 'an array literal');
    return array.elements.map(element => {
      assert(element, isNotEmptyElement, 'element in array not to be empty');
      assert(element, isNotSpreadElement, 'element in array not to use spread syntax');
      return element;
    });
  }

  isObjectLiteral = t.isObjectExpression;

  parseObjectLiteral(obj: t.Expression): Map<string, t.Expression> {
    assert(obj, t.isObjectExpression, 'an object literal');

    const result = new Map<string, t.Expression>();
    for (const property of obj.properties) {
      assert(property, t.isObjectProperty, 'a property assignment');
      assert(property.value, t.isExpression, 'an expression');
      assert(property.key, isPropertyName, 'a property name');
      const key = t.isIdentifier(property.key) ? property.key.name : property.key.value;
      result.set(key, property.value);
    }
    return result;
  }

  isFunctionExpression(node: t.Expression): node is Extract<t.Function, t.Expression> {
    return t.isFunction(node);
  }

  parseReturnValue(fn: t.Expression): t.Expression {
    assert(fn, this.isFunctionExpression, 'a function');
    if (!t.isBlockStatement(fn.body)) {
      // it is a simple array function expression: `(...) => expr`
      return fn.body;
    }

    // it is a function (arrow or normal) with a body. E.g.:
    // * `(...) => { stmt; ... }`
    // * `function(...) { stmt; ... }`

    if (fn.body.body.length !== 1) {
      throw new FatalLinkerError(
          fn.body, 'Unsupported syntax, expected a function body with a single return statement.');
    }
    const stmt = fn.body.body[0];
    assert(stmt, t.isReturnStatement, 'a function body with a single return statement');
    if (stmt.argument === null) {
      throw new FatalLinkerError(stmt, 'Unsupported syntax, expected function to return a value.');
    }

    return stmt.argument;
  }

  getRange(node: t.Expression): Range {
    if (node.loc == null || node.start === null || node.end === null) {
      throw new FatalLinkerError(
          node, 'Unable to read range for node - it is missing location information.');
    }
    return {
      startLine: node.loc.start.line - 1,  // Babel lines are 1-based
      startCol: node.loc.start.column,
      startPos: node.start,
      endPos: node.end,
    };
  }
}

/**
 * Return true if the expression does not represent an empty element in an array literal.
 * For example in `[,foo]` the first element is "empty".
 */
function isNotEmptyElement(e: t.Expression|t.SpreadElement|null): e is t.Expression|
    t.SpreadElement {
  return e !== null;
}

/**
 * Return true if the expression is not a spread element of an array literal.
 * For example in `[x, ...rest]` the `...rest` expression is a spread element.
 */
function isNotSpreadElement(e: t.Expression|t.SpreadElement): e is t.Expression {
  return !t.isSpreadElement(e);
}


/**
 * Return true if the expression can be considered a text based property name.
 */
function isPropertyName(e: t.Expression): e is t.Identifier|t.StringLiteral|t.NumericLiteral {
  return t.isIdentifier(e) || t.isStringLiteral(e) || t.isNumericLiteral(e);
}
