Store parts in Kernel Library, resynthesize parts in Analyzer.

pkg/analyzer/lib/src/dart/element/element.dart

 // Copyright (c) 2014, 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.
 
 library analyzer.src.dart.element.element;
 
 import 'dart:collection';
 import 'dart:math' show min;
 
 import 'package:analyzer/dart/ast/ast.dart';
@@ skipping 832 matching lines @@
     }
     return true;
   }
 
   @override
   List<kernel.TypeParameter> get kernelTypeParams => _kernel?.typeParameters;
 
   @override
   List<ElementAnnotation> get metadata {
     if (_kernel != null) {
-      _metadata ??= enclosingUnit._kernelContext
-          .buildAnnotations(enclosingUnit, _kernel.annotations);
+      _metadata ??=
+          enclosingUnit._kernelContext.buildAnnotations(_kernel.annotations);
     }
     if (_unlinkedClass != null) {
       return _metadata ??=
           _buildAnnotations(enclosingUnit, _unlinkedClass.annotations);
     }
     return super.metadata;
   }
 
   @override
   List<MethodElement> get methods {
@@ skipping 582 matching lines @@
    * The unlinked representation of the unit in the summary.
    */
   final UnlinkedUnit _unlinkedUnit;
 
   /**
    * The unlinked representation of the part in the summary.
    */
   final UnlinkedPart _unlinkedPart;
 
   /**
-   * The kernel context in which the library is resynthesized.
+   * The kernel context in which the unit is resynthesized.
    */
-  final KernelLibraryResynthesizerContext _kernelContext;
+  final KernelUnitResynthesizerContext _kernelContext;
 
   /**
    * The source that corresponds to this compilation unit.
    */
   @override
   Source source;
 
   @override
   LineInfo lineInfo;
 
@@ skipping 102 matching lines @@
         super.forSerialized(null) {
     _enclosingElement = enclosingLibrary;
     _name = name;
     _nameOffset = -1;
   }
 
   @override
   List<PropertyAccessorElement> get accessors {
     if (_accessors == null) {
       if (_kernelContext != null) {
-        _explicitTopLevelAccessors ??=
-            _kernelContext.buildTopLevelAccessors(this);
-        _explicitTopLevelVariables ??=
-            _kernelContext.buildTopLevelVariables(this);
+        _explicitTopLevelAccessors ??= _kernelContext.buildTopLevelAccessors();
+        _explicitTopLevelVariables ??= _kernelContext.buildTopLevelVariables();
       }
       if (_unlinkedUnit != null) {
         _explicitTopLevelAccessors ??=
             resynthesizerContext.buildTopLevelAccessors();
         _explicitTopLevelVariables ??=
             resynthesizerContext.buildTopLevelVariables();
       }
       if (_explicitTopLevelAccessors != null) {
         _accessors = <PropertyAccessorElementImpl>[]
           ..addAll(_explicitTopLevelAccessors.accessors)
@@ skipping 35 matching lines @@
       super.enclosingElement as LibraryElement;
 
   @override
   CompilationUnitElementImpl get enclosingUnit {
     return this;
   }
 
   @override
   List<ClassElement> get enums {
     if (_kernelContext != null) {
-      _enums ??= _kernelContext.library.classes
+      _enums ??= _kernelContext.kernelUnit.classes
           .where((k) => k.isEnum)
           .map((k) => new EnumElementImpl.forKernel(this, k))
           .toList(growable: false);
     }
     if (_unlinkedUnit != null) {
       _enums ??= _unlinkedUnit.enums
           .map((e) => new EnumElementImpl.forSerialized(e, this))
           .toList(growable: false);
     }
     return _enums ?? const <ClassElement>[];
   }
 
   /**
    * Set the enums contained in this compilation unit to the given [enums].
    */
   void set enums(List<ClassElement> enums) {
     _assertNotResynthesized(_unlinkedUnit);
     for (ClassElement enumDeclaration in enums) {
       (enumDeclaration as EnumElementImpl).enclosingElement = this;
     }
     this._enums = enums;
   }
 
   @override
   List<FunctionElement> get functions {
     if (_kernelContext != null) {
-      _functions ??= _kernelContext.library.procedures
+      _functions ??= _kernelContext.kernelUnit.procedures
           .where((k) => k.kind == kernel.ProcedureKind.Method)
           .map((k) => new FunctionElementImpl.forKernel(this, k))
           .toList(growable: false);
     }
     if (_unlinkedUnit != null) {
       _functions ??= _unlinkedUnit.executables
           .where((e) => e.kind == UnlinkedExecutableKind.functionOrMethod)
           .map((e) => new FunctionElementImpl.forSerialized(e, this))
           .toList(growable: false);
     }
     return _functions ?? const <FunctionElement>[];
   }
 
   /**
    * Set the top-level functions contained in this compilation unit to the given
    * [functions].
    */
   void set functions(List<FunctionElement> functions) {
     for (FunctionElement function in functions) {
       (function as FunctionElementImpl).enclosingElement = this;
     }
     this._functions = functions;
   }
 
   @override
   List<FunctionTypeAliasElement> get functionTypeAliases {
     if (_kernelContext != null) {
-      _typeAliases ??= _kernelContext.library.typedefs
+      _typeAliases ??= _kernelContext.kernelUnit.typedefs
           .map((k) => new FunctionTypeAliasElementImpl.forKernel(this, k))
           .toList(growable: false);
     }
     if (_unlinkedUnit != null) {
       _typeAliases ??= _unlinkedUnit.typedefs.map((t) {
         if (t.style == TypedefStyle.functionType) {
           return new FunctionTypeAliasElementImpl.forSerialized(t, this);
         } else if (t.style == TypedefStyle.genericFunctionType) {
           return new GenericTypeAliasElementImpl.forSerialized(t, this);
         }
@@ skipping 29 matching lines @@
           library.definingCompilationUnit as CompilationUnitElementImpl,
           _unlinkedPart.annotations);
     }
     return super.metadata;
   }
 
   @override
   List<TopLevelVariableElement> get topLevelVariables {
     if (_variables == null) {
       if (_kernelContext != null) {
-        _explicitTopLevelAccessors ??=
-            _kernelContext.buildTopLevelAccessors(this);
-        _explicitTopLevelVariables ??=
-            _kernelContext.buildTopLevelVariables(this);
+        _explicitTopLevelAccessors ??= _kernelContext.buildTopLevelAccessors();
+        _explicitTopLevelVariables ??= _kernelContext.buildTopLevelVariables();
       }
       if (_unlinkedUnit != null) {
         _explicitTopLevelAccessors ??=
             resynthesizerContext.buildTopLevelAccessors();
         _explicitTopLevelVariables ??=
             resynthesizerContext.buildTopLevelVariables();
       }
       if (_explicitTopLevelVariables != null) {
         var variables = <TopLevelVariableElement>[]
           ..addAll(_explicitTopLevelVariables.variables)
@@ skipping 39 matching lines @@
     }
     this._typeAliases = typeAliases;
   }
 
   @override
   TypeParameterizedElementMixin get typeParameterContext => null;
 
   @override
   List<ClassElement> get types {
     if (_kernelContext != null) {
-      _types ??= _kernelContext.library.classes
+      _types ??= _kernelContext.kernelUnit.classes
           .where((k) => !k.isEnum && !k.isSyntheticMixinImplementation)
           .map((k) => new ClassElementImpl.forKernel(this, k))
           .toList(growable: false);
     }
     if (_unlinkedUnit != null) {
       _types ??= _unlinkedUnit.classes
           .map((c) => new ClassElementImpl.forSerialized(c, this))
           .toList(growable: false);
     }
     return _types ?? const <ClassElement>[];
@@ skipping 112 matching lines @@
       }
     }
     return null;
   }
 
   /**
    * Replace the given [from] top-level variable with [to] in this compilation unit.
    */
   void replaceTopLevelVariable(
       TopLevelVariableElement from, TopLevelVariableElement to) {
-    if (_unlinkedUnit != null) {
+    if (_kernelContext != null || _unlinkedUnit != null) {
       // Getters and setter in different units should be patched to use the
       // same variables before these variables were asked and returned.
       assert(_variables == null);
       _topLevelVariableReplaceMap ??=
           <TopLevelVariableElement, TopLevelVariableElement>{};
       _topLevelVariableReplaceMap[from] = to;
     } else {
       int index = _variables.indexOf(from);
       _variables[index] = to;
     }
@@ skipping 2195 matching lines @@
   bool get isSynchronous => !isAsynchronous;
 
   @override
   List<kernel.TypeParameter> get kernelTypeParams {
     return _kernel?.function?.typeParameters;
   }
 
   @override
   List<ElementAnnotation> get metadata {
     if (_kernel != null) {
-      _metadata ??= enclosingUnit._kernelContext
-          .buildAnnotations(enclosingUnit, _kernel.annotations);
+      _metadata ??=
+          enclosingUnit._kernelContext.buildAnnotations(_kernel.annotations);
     }
     if (serializedExecutable != null) {
       return _metadata ??=
           _buildAnnotations(enclosingUnit, serializedExecutable.annotations);
     }
     return super.metadata;
   }
 
   @override
   String get name {
@@ skipping 917 matching lines @@
 
   @override
   List<kernel.TypeParameter> get kernelTypeParams => _kernel?.typeParameters;
 
   @override
   ElementKind get kind => ElementKind.FUNCTION_TYPE_ALIAS;
 
   @override
   List<ElementAnnotation> get metadata {
     if (_kernel != null) {
-      _metadata ??= enclosingUnit._kernelContext
-          .buildAnnotations(enclosingUnit, _kernel.annotations);
+      _metadata ??=
+          enclosingUnit._kernelContext.buildAnnotations(_kernel.annotations);
     }
     if (_unlinkedTypedef != null) {
       return _metadata ??=
           _buildAnnotations(enclosingUnit, _unlinkedTypedef.annotations);
     }
     return super.metadata;
   }
 
   @override
   String get name {
@@ skipping 974 matching lines @@
   }
 }
 
 /**
  * The kernel context in which a library is resynthesized.
  */
 abstract class KernelLibraryResynthesizerContext {
   kernel.Library get library;
 
   /**
+   * Return the [LibraryElement] for the given absolute [uriStr].
+   */
+  LibraryElement getLibrary(String uriStr);
+}
+
+/**
+ * Top-level declarations of a Kernel library filtered by the unit.
+ */
+abstract class KernelUnit {
+  List<kernel.Class> get classes;
+
+  List<kernel.Field> get fields;
+
+  List<kernel.Procedure> get procedures;
+
+  List<kernel.Typedef> get typedefs;
+}
+
+/**
+ * The kernel context in which a unit is resynthesized.
+ */
+abstract class KernelUnitResynthesizerContext {
+  /**
+   * Subset of top-level declarations in the unit.
+   */
+  KernelUnit get kernelUnit;
+
+  /**
    * Build [ElementAnnotation]s for the given Kernel [annotations].
    */
-  List<ElementAnnotation> buildAnnotations(
-      CompilationUnitElementImpl unit, List<kernel.Expression> annotations);
+  List<ElementAnnotation> buildAnnotations(List<kernel.Expression> annotations);
 
   /**
    * Build explicit top-level property accessors.
    */
-  UnitExplicitTopLevelAccessors buildTopLevelAccessors(
-      CompilationUnitElementImpl unit);
+  UnitExplicitTopLevelAccessors buildTopLevelAccessors();
 
   /**
    * Build explicit top-level variables.
    */
-  UnitExplicitTopLevelVariables buildTopLevelVariables(
-      CompilationUnitElementImpl unit);
+  UnitExplicitTopLevelVariables buildTopLevelVariables();
 
   /**
    * Return the resynthesized [ConstructorInitializer] for the given Kernel
    * [initializer], or `null` if synthetic.
    */
   ConstructorInitializer getConstructorInitializer(
       ConstructorElementImpl constructor, kernel.Initializer initializer);
 
   /**
    * Return the [Expression] for the given kernel.
    */
   Expression getExpression(ElementImpl context, kernel.Expression expression);
 
   /**
    * Return the list with exactly two elements - positional and named parameter
    * lists.
    */
   List<List<kernel.VariableDeclaration>> getFunctionTypeParameters(
       kernel.FunctionType functionType);
 
   /**
    * Return the [InterfaceType] for the given Kernel [type], or `null` if the
    * [type] does not correspond to an [InterfaceType].
    */
   InterfaceType getInterfaceType(ElementImpl context, kernel.Supertype type);
 
   /**
-   * Return the [LibraryElement] for the given absolute [uriStr].
-   */
-  LibraryElement getLibrary(String uriStr);
-
-  /**
    * Return the [ConstructorElementImpl] to which the given [kernelConstructor]
    * or [kernelFactory] redirects.
    */
   ConstructorElementImpl getRedirectedConstructor(
       kernel.Constructor kernelConstructor, kernel.Procedure kernelFactory);
 
   /**
    * Return the [DartType] for the given Kernel [type], or `null` if the [type]
    * does not correspond to a [DartType].
    */
@@ skipping 167 matching lines @@
       : resynthesizerContext = null,
         _unlinkedDefiningUnit = null,
         nameLength = _kernelContext.library.name?.length ?? 0,
         super.forSerialized(null) {
     _name = _kernelContext.library.name ?? '';
     _nameOffset = _kernelContext.library.fileOffset;
     setResolutionCapability(
         LibraryResolutionCapability.resolvedTypeNames, true);
     setResolutionCapability(
         LibraryResolutionCapability.constantExpressions, true);
-
-    definingCompilationUnit = new CompilationUnitElementImpl.forKernel(
-        this, _kernelContext, '<no-name>');
-
-    // TODO(scheglov) how to support parts?
   }
 
   /**
    * Initialize a newly created library element in the given [context] to have
    * the given [name].
    */
   LibraryElementImpl.forNode(this.context, LibraryIdentifier name)
       : nameLength = name != null ? name.length : 0,
         resynthesizerContext = null,
         _kernelContext = null,
@@ skipping 1480 matching lines @@
 
   @override
   void set isFinal(bool isFinal) {
     _assertNotResynthesized(_unlinkedVariable);
     super.isFinal = isFinal;
   }
 
   @override
   List<ElementAnnotation> get metadata {
     if (_kernel != null) {
-      _metadata ??= enclosingUnit._kernelContext
-          .buildAnnotations(enclosingUnit, _kernel.annotations);
+      _metadata ??=
+          enclosingUnit._kernelContext.buildAnnotations(_kernel.annotations);
     }
     if (_unlinkedVariable != null) {
       return _metadata ??=
           _buildAnnotations(enclosingUnit, _unlinkedVariable.annotations);
     }
     return super.metadata;
   }
 
   @override
   String get name {
@@ skipping 2079 matching lines @@
 
   @override
   DartObject computeConstantValue() => null;
 
   @override
   void visitChildren(ElementVisitor visitor) {
     super.visitChildren(visitor);
     _initializer?.accept(visitor);
   }
 }