pkg/analyzer/lib/src/summary2/constructor_initializer_resolver.dart

// Copyright (c) 2019, 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:analyzer/src/dart/ast/ast.dart';
import 'package:analyzer/src/dart/element/element.dart';
import 'package:analyzer/src/dart/resolver/ast_resolver.dart';
import 'package:analyzer/src/summary2/library_builder.dart';
import 'package:analyzer/src/summary2/link.dart';

class ConstructorInitializerResolver {
  final Linker _linker;
  final LibraryBuilder _libraryBuilder;

  ConstructorInitializerResolver(this._linker, this._libraryBuilder);

  void resolve() {
    var libraryElement = _libraryBuilder.element;
    var interfaceElements = <InterfaceElementImpl>[
      ...libraryElement.classes,
      ...libraryElement.enums,
      ...libraryElement.extensionTypes,
      ...libraryElement.mixins,
    ];

    for (var interfaceElement in interfaceElements) {
      for (var constructorElement in interfaceElement.constructors) {
        _constructorElement(interfaceElement, constructorElement);
      }
    }
  }

  void _constructorElement(
    InterfaceElementImpl interfaceElement,
    ConstructorElementImpl element,
  ) {
    if (!element.isOriginDeclaration) return;

    for (var fragment in element.fragments) {
      var node = _linker.getLinkingNode(fragment);
      switch (node) {
        case ConstructorDeclarationImpl():
          var initializerScope = node.formalParameterInitializerScope!;
          var analysisOptions = _libraryBuilder.kind.file.analysisOptions;
          var astResolver = AstResolver.forLinking(
            inheritance: _linker.inheritance,
            libraryFragment: fragment.libraryFragment,
            nameScope: initializerScope,
            analysisOptions: analysisOptions,
            enclosingClassElement: interfaceElement,
            enclosingExecutableElement: element,
          );
          astResolver.resolveConstructorDeclaration(node);

          if (node.factoryKeyword != null) {
            element.redirectedConstructor =
                node.factoryRedirectionTarget?.element;
          } else {
            for (var initializer in node.initializers) {
              if (initializer is RedirectingConstructorInvocationImpl) {
                element.redirectedConstructor = initializer.element;
              }
            }
          }
        case PrimaryConstructorDeclarationImpl():
          if (node.body case var body?) {
            var initializerScope = body.formalParameterInitializerScope!;
            var analysisOptions = _libraryBuilder.kind.file.analysisOptions;
            var astResolver = AstResolver.forLinking(
              inheritance: _linker.inheritance,
              libraryFragment: fragment.libraryFragment,
              nameScope: initializerScope,
              analysisOptions: analysisOptions,
              enclosingClassElement: interfaceElement,
              enclosingExecutableElement: element,
            );

            astResolver.resolvePrimaryConstructor(node, body);
          }
      }
    }
  }
}