pkg/analysis_server/lib/src/services/correction/fix/data_driven/transform_set_parser.dart

// Copyright (c) 2020, the Dart project authors. Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

import 'package:analysis_server/src/services/correction/fix/data_driven/change.dart';
import 'package:analysis_server/src/services/correction/fix/data_driven/element_descriptor.dart';
import 'package:analysis_server/src/services/correction/fix/data_driven/rename.dart';
import 'package:analysis_server/src/services/correction/fix/data_driven/transform.dart';
import 'package:analysis_server/src/services/correction/fix/data_driven/transform_set.dart';
import 'package:analysis_server/src/services/correction/fix/data_driven/transform_set_error_code.dart';
import 'package:analysis_server/src/utilities/extensions/yaml.dart';
import 'package:analyzer/error/listener.dart';
import 'package:yaml/yaml.dart';

/// A parser used to read a transform set from a file.
class TransformSetParser {
  static const String _changesKey = 'changes';

  static const String _componentsKey = 'components';

  static const String _elementKey = 'element';

  static const String _kindKey = 'kind';

  static const String _newNameKey = 'newName';

  static const String _titleKey = 'title';

  static const String _transformsKey = 'transforms';

  static const String _urisKey = 'uris';

  static const String _versionKey = 'version';

  static const String _renameKind = 'rename';

  static const int currentVersion = 1;

  /// The error reporter to which diagnostics will be reported.
  final ErrorReporter errorReporter;

  /// Initialize a newly created parser to report diagnostics to the
  /// [errorReporter].
  TransformSetParser(this.errorReporter);

  /// Return the result of parsing the file [content] into a transform set.
  TransformSet parse(String content) {
    assert(content != null);
    var map = _parseYaml(content);
    if (map == null) {
      return TransformSet();
    }
    return _translateTransformSet(map);
  }

  /// Return the result of parsing the file [content] into a YAML node.
  YamlNode _parseYaml(String content) {
    try {
      return loadYamlNode(content);
    } on YamlException catch (e) {
      var span = e.span;
      errorReporter.reportErrorForOffset(TransformSetErrorCode.yamlSyntaxError,
          span.start.offset, span.length, [e.message]);
    }
    return null;
  }

  /// Report a diagnostic with the given [code] associated with the given
  /// [node]. A list of [arguments] should be provided if the diagnostic message
  /// has parameters.
  void _reportError(TransformSetErrorCode code, YamlNode node,
      [List<String> arguments]) {
    var span = node.span;
    errorReporter.reportErrorForOffset(
        code, span.start.offset, span.length, arguments);
  }

  /// Report any keys in the [map] whose values are not in [validKeys].
  void _reportUnsupportedKeys(YamlMap map, Set<String> validKeys) {
    for (var keyNode in map.nodes.keys) {
      if (keyNode is YamlScalar) {
        var key = _translateString(keyNode);
        if (key != null && !validKeys.contains(key)) {
          _reportError(TransformSetErrorCode.unsupportedKey, keyNode, [key]);
        }
      } else {
        // TODO(brianwilkerson) Report the unsupported key.
      }
    }
  }

  /// Translate the [node] into a change.
  Change _translateChange(YamlNode node) {
    if (node is YamlMap) {
      var kind = _translateString(node.valueAt(_kindKey));
      // TODO(brianwilkerson) Implement additional change kinds.
      if (kind == _renameKind) {
        return _translateRenameChange(node);
      }
      // TODO(brianwilkerson) Report the invalid change kind.
      return null;
    } else if (node == null) {
      // TODO(brianwilkerson) Report the missing YAML.
      return null;
    } else {
      // TODO(brianwilkerson) Report the invalid YAML.
      return null;
    }
  }

  /// Translate the [node] into an element descriptor.
  ElementDescriptor _translateElement(YamlNode node) {
    if (node is YamlMap) {
      var uris = _translateList(node.valueAt(_urisKey), _translateString);
      var components =
          _translateList(node.valueAt(_componentsKey), _translateString);
      return ElementDescriptor(libraryUris: uris, components: components);
    } else if (node == null) {
      // TODO(brianwilkerson) Report the missing YAML.
      return null;
    } else {
      // TODO(brianwilkerson) Report the invalid YAML.
      return null;
    }
  }

  /// Translate the [node] into an integer.
  int _translateInteger(YamlNode node) {
    if (node is YamlScalar) {
      var value = node.value;
      if (value is int) {
        return value;
      }
      // TODO(brianwilkerson) Report the invalid YAML. For the best UX we
      //  probably need to pass in the code to report.
      return null;
    } else if (node == null) {
      // TODO(brianwilkerson) Report the missing YAML. For the best UX we
      //  probably need to pass in the code to report.
      return null;
    } else {
      // TODO(brianwilkerson) Report the invalid YAML. For the best UX we
      //  probably need to pass in the code to report.
      return null;
    }
  }

  /// Translate the [node] into a list of objects using the [elementTranslator].
  List<R> _translateList<R>(
      YamlNode node, R Function(YamlNode) elementTranslator) {
    if (node is YamlList) {
      var translatedList = <R>[];
      for (var element in node.nodes) {
        var result = elementTranslator(element);
        if (result != null) {
          translatedList.add(result);
        }
      }
      return translatedList;
    } else if (node == null) {
      // TODO(brianwilkerson) Report the missing YAML.
      return null;
    } else {
      // TODO(brianwilkerson) Report the invalid YAML.
      return null;
    }
  }

  /// Translate the [node] into a rename change.
  Change _translateRenameChange(YamlMap node) {
    _reportUnsupportedKeys(node, const {_kindKey, _newNameKey});
    var newName = _translateString(node.valueAt(_newNameKey));
    return Rename(newName: newName);
  }

  /// Translate the [node] into a string.
  String _translateString(YamlNode node) {
    if (node is YamlScalar) {
      var value = node.value;
      if (value is String) {
        return value;
      }
      // TODO(brianwilkerson) Report the invalid YAML. For the best UX we
      //  probably need to pass in the code to report.
      return null;
    } else if (node == null) {
      // TODO(brianwilkerson) Report the missing YAML. For the best UX we
      //  probably need to pass in the code to report.
      return null;
    } else {
      // TODO(brianwilkerson) Report the invalid YAML. For the best UX we
      //  probably need to pass in the code to report.
      return null;
    }
  }

  /// Translate the [node] into a transform.
  Transform _translateTransform(YamlNode node) {
    if (node is YamlMap) {
      _reportUnsupportedKeys(node, const {_changesKey, _elementKey, _titleKey});
      var title = _translateString(node.valueAt(_titleKey));
      var element = _translateElement(node.valueAt(_elementKey));
      var changes =
          _translateList<Change>(node.valueAt(_changesKey), _translateChange);
      return Transform(title: title, element: element, changes: changes);
    } else if (node == null) {
      // TODO(brianwilkerson) Report the missing YAML.
      return null;
    } else {
      // TODO(brianwilkerson) Report the invalid YAML.
      return null;
    }
  }

  /// Translate the [node] into a transform set.
  TransformSet _translateTransformSet(YamlNode node) {
    if (node is YamlMap) {
      _reportUnsupportedKeys(node, const {_transformsKey, _versionKey});
      var set = TransformSet();
      // TODO(brianwilkerson) Version information is currently being ignored,
      //  but needs to be used to select a translator.
      var version = _translateInteger(node.valueAt(_versionKey));
      if (version != currentVersion) {
        // TODO(brianwilkerson) Report the unsupported version.
      }
      var transformations =
          _translateList(node.valueAt(_transformsKey), _translateTransform);
      for (var transform in transformations) {
        set.addTransform(transform);
      }
      return set;
    } else if (node == null) {
      // TODO(brianwilkerson) Report the missing YAML.
      return null;
    } else {
      // TODO(brianwilkerson) Report the invalid YAML.
      return null;
    }
  }
}