Resynthesize codeOffset/codeLength properties.

pkg/analyzer/lib/src/summary/summarize_ast.dart

 // Copyright (c) 2016, 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 serialization.summarize_ast;
 
 import 'package:analyzer/dart/ast/ast.dart';
 import 'package:analyzer/dart/ast/token.dart';
 import 'package:analyzer/dart/ast/visitor.dart';
 import 'package:analyzer/src/generated/utilities_dart.dart';
@@ skipping 75 matching lines @@
           'Unexpected identifier type: ${identifier.runtimeType}');
     }
     return b;
   }
 
   @override
   EntityRefBuilder serializePropertyAccess(PropertyAccess access) {
     Expression target = access.target;
     if (target is Identifier) {
       EntityRefBuilder targetRef = serializeIdentifier(target);
-      return new EntityRefBuilder(
-          reference: visitor.serializeReference(
-              targetRef.reference, access.propertyName.name));
+      return new EntityRefBuilder(reference: visitor.serializeReference(
+          targetRef.reference, access.propertyName.name));
     } else {
       // TODO(scheglov) should we handle other targets in malformed constants?
       throw new StateError('Unexpected target type: ${target.runtimeType}');
     }
   }
 
   @override
   EntityRefBuilder serializeType(TypeName node) {
     return visitor.serializeTypeName(node);
   }
@@ skipping 233 matching lines @@
       serializer.serializeAnnotation(a);
       return serializer.toBuilder();
     }).toList();
   }
 
   /**
    * Serialize a [ClassDeclaration] or [ClassTypeAlias] into an [UnlinkedClass]
    * and store the result in [classes].
    */
   void serializeClass(
+      AstNode node,
       Token abstractKeyword,
       String name,
       int nameOffset,
       TypeParameterList typeParameters,
       TypeName superclass,
       WithClause withClause,
       ImplementsClause implementsClause,
       NodeList<ClassMember> members,
       bool isMixinApplication,
       Comment documentationComment,
@@ skipping 26 matching lines @@
       for (ClassMember member in members) {
         member.accept(this);
       }
       scopes.removeLast();
     }
     b.executables = executables;
     b.fields = variables;
     b.isAbstract = abstractKeyword != null;
     b.documentationComment = serializeDocumentation(documentationComment);
     b.annotations = serializeAnnotations(annotations);
+    serializeCodeRange(b, node);
     classes.add(b);
     scopes.removeLast();
     assert(scopes.length == oldScopesLength);
     executables = oldExecutables;
     variables = oldVariables;
   }
 
+  void serializeCodeRange(Object b, AstNode node) {
+    if (b is UnlinkedClassBuilder) {
+      b.hasCodeRange = true;
+      b.codeOffset = node.offset;
+      b.codeLength = node.length;
+    } else if (b is UnlinkedEnumBuilder) {
+      b.hasCodeRange = true;
+      b.codeOffset = node.offset;
+      b.codeLength = node.length;
+    } else if (b is UnlinkedExecutableBuilder) {
+      b.hasCodeRange = true;
+      b.codeOffset = node.offset;
+      b.codeLength = node.length;
+    } else if (b is UnlinkedParamBuilder) {
+      b.hasCodeRange = true;
+      b.codeOffset = node.offset;
+      b.codeLength = node.length;
+    } else if (b is UnlinkedTypedefBuilder) {
+      b.hasCodeRange = true;
+      b.codeOffset = node.offset;
+      b.codeLength = node.length;
+    } else if (b is UnlinkedTypeParamBuilder) {
+      b.hasCodeRange = true;
+      b.codeOffset = node.offset;
+      b.codeLength = node.length;
+    } else if (b is UnlinkedUnitBuilder) {
+      b.hasCodeRange = true;
+      b.codeOffset = node.offset;
+      b.codeLength = node.length;
+    } else if (b is UnlinkedVariableBuilder) {
+      b.hasCodeRange = true;
+      b.codeOffset = node.offset;
+      b.codeLength = node.length;
+    }
+  }
+
   /**
    * Serialize a [Combinator] into an [UnlinkedCombinator].
    */
   UnlinkedCombinatorBuilder serializeCombinator(Combinator combinator) {
     UnlinkedCombinatorBuilder b = new UnlinkedCombinatorBuilder();
     if (combinator is ShowCombinator) {
       b.shows =
           combinator.shownNames.map((SimpleIdentifier id) => id.name).toList();
       b.offset = combinator.offset;
       b.end = combinator.end;
@@ skipping 16 matching lines @@
     if (!hasCoreBeenImported) {
       unlinkedImports.add(new UnlinkedImportBuilder(isImplicit: true));
     }
     compilationUnit.declarations.accept(this);
     UnlinkedUnitBuilder b = new UnlinkedUnitBuilder();
     b.libraryName = libraryName;
     b.libraryNameOffset = libraryNameOffset;
     b.libraryNameLength = libraryNameLength;
     b.libraryDocumentationComment = libraryDocumentationComment;
     b.libraryAnnotations = libraryAnnotations;
+    serializeCodeRange(b, compilationUnit);
     b.classes = classes;
     b.enums = enums;
     b.executables = executables;
     b.exports = exports;
     b.imports = unlinkedImports;
     b.parts = parts;
     b.references = unlinkedReferences;
     b.typedefs = typedefs;
     b.variables = variables;
     b.publicNamespace = computePublicNamespace(compilationUnit);
@@ skipping 27 matching lines @@
         text: text,
         offset: documentationComment.offset,
         length: documentationComment.length);
   }
 
   /**
    * Serialize a [FunctionDeclaration] or [MethodDeclaration] into an
    * [UnlinkedExecutable].
    */
   UnlinkedExecutableBuilder serializeExecutable(
+      AstNode node,
       String name,
       int nameOffset,
       bool isGetter,
       bool isSetter,
       TypeName returnType,
       FormalParameterList formalParameters,
       FunctionBody body,
       bool isTopLevel,
       bool isDeclaredStatic,
       Comment documentationComment,
@@ skipping 32 matching lines @@
         for (int i = 0; i < formalParameters.parameters.length; i++) {
           if (!b.parameters[i].isFunctionTyped &&
               b.parameters[i].type == null) {
             b.parameters[i].inferredTypeSlot = assignSlot();
           }
         }
       }
     }
     b.documentationComment = serializeDocumentation(documentationComment);
     b.annotations = serializeAnnotations(annotations);
+    serializeCodeRange(b, node);
     if (returnType == null && !isSemanticallyStatic) {
       b.inferredReturnTypeSlot = assignSlot();
     }
     b.visibleOffset = enclosingBlock?.offset;
     b.visibleLength = enclosingBlock?.length;
     serializeFunctionBody(b, body);
     scopes.removeLast();
     assert(scopes.length == oldScopesLength);
     return b;
   }
@@ skipping 58 matching lines @@
 
   /**
    * Serialize a [FieldFormalParameter], [FunctionTypedFormalParameter], or
    * [SimpleFormalParameter] into an [UnlinkedParam].
    */
   UnlinkedParamBuilder serializeParameter(NormalFormalParameter node) {
     UnlinkedParamBuilder b = new UnlinkedParamBuilder();
     b.name = node.identifier.name;
     b.nameOffset = node.identifier.offset;
     b.annotations = serializeAnnotations(node.metadata);
+    serializeCodeRange(b, node);
     switch (node.kind) {
       case ParameterKind.REQUIRED:
         b.kind = UnlinkedParamKind.required;
         break;
       case ParameterKind.POSITIONAL:
         b.kind = UnlinkedParamKind.positional;
         break;
       case ParameterKind.NAMED:
         b.kind = UnlinkedParamKind.named;
         break;
@@ skipping 137 matching lines @@
     for (VariableDeclaration variable in variables.variables) {
       UnlinkedVariableBuilder b = new UnlinkedVariableBuilder();
       b.isFinal = variables.isFinal;
       b.isConst = variables.isConst;
       b.isStatic = isDeclaredStatic;
       b.name = variable.name.name;
       b.nameOffset = variable.name.offset;
       b.type = serializeTypeName(variables.type);
       b.documentationComment = serializeDocumentation(documentationComment);
       b.annotations = serializeAnnotations(annotations);
+      serializeCodeRange(b, variables.parent);
       if (variable.isConst ||
           variable.isFinal && isField && !isDeclaredStatic) {
         Expression initializer = variable.initializer;
         if (initializer != null) {
           b.constExpr = serializeConstExpr(initializer);
         }
       }
       if (variable.initializer != null &&
           (variables.isFinal || variables.isConst)) {
         b.propagatedTypeSlot = assignSlot();
@@ skipping 16 matching lines @@
     enclosingBlock = node;
     super.visitBlock(node);
     enclosingBlock = oldBlock;
   }
 
   @override
   void visitClassDeclaration(ClassDeclaration node) {
     TypeName superclass =
         node.extendsClause == null ? null : node.extendsClause.superclass;
     serializeClass(
+        node,
         node.abstractKeyword,
         node.name.name,
         node.name.offset,
         node.typeParameters,
         superclass,
         node.withClause,
         node.implementsClause,
         node.members,
         false,
         node.documentationComment,
         node.metadata);
   }
 
   @override
   void visitClassTypeAlias(ClassTypeAlias node) {
     serializeClass(
+        node,
         node.abstractKeyword,
         node.name.name,
         node.name.offset,
         node.typeParameters,
         node.superclass,
         node.withClause,
         node.implementsClause,
         null,
         true,
         node.documentationComment,
@@ skipping 71 matching lines @@
     b.nameOffset = node.name.offset;
     b.values = node.constants
         .map((EnumConstantDeclaration value) => new UnlinkedEnumValueBuilder(
             documentationComment:
                 serializeDocumentation(value.documentationComment),
             name: value.name.name,
             nameOffset: value.name.offset))
         .toList();
     b.documentationComment = serializeDocumentation(node.documentationComment);
     b.annotations = serializeAnnotations(node.metadata);
+    serializeCodeRange(b, node);
     enums.add(b);
   }
 
   @override
   void visitExportDirective(ExportDirective node) {
     UnlinkedExportNonPublicBuilder b = new UnlinkedExportNonPublicBuilder(
         uriOffset: node.uri.offset, uriEnd: node.uri.end, offset: node.offset);
     b.annotations = serializeAnnotations(node.metadata);
     exports.add(b);
   }
@@ skipping 15 matching lines @@
       } else {
         b.type = serializeTypeName(node.type);
       }
     }
     return b;
   }
 
   @override
   void visitFunctionDeclaration(FunctionDeclaration node) {
     executables.add(serializeExecutable(
+        node,
         node.name.name,
         node.name.offset,
         node.isGetter,
         node.isSetter,
         node.returnType,
         node.functionExpression.parameters,
         node.functionExpression.body,
         true,
         false,
         node.documentationComment,
         node.metadata,
         node.functionExpression.typeParameters,
         node.externalKeyword != null));
   }
 
   @override
   void visitFunctionExpression(FunctionExpression node) {
     if (node.parent is! FunctionDeclaration) {
       executables.add(serializeExecutable(
+          node,
           null,
           node.offset,
           false,
           false,
           null,
           node.parameters,
           node.body,
           false,
           false,
           null,
@@ skipping 15 matching lines @@
         serializeTypeParameters(node.typeParameters, typeParameterScope);
     EntityRefBuilder serializedReturnType = serializeTypeName(node.returnType);
     if (serializedReturnType != null) {
       b.returnType = serializedReturnType;
     }
     b.parameters = node.parameters.parameters
         .map((FormalParameter p) => p.accept(this))
         .toList();
     b.documentationComment = serializeDocumentation(node.documentationComment);
     b.annotations = serializeAnnotations(node.metadata);
+    serializeCodeRange(b, node);
     typedefs.add(b);
     scopes.removeLast();
     assert(scopes.length == oldScopesLength);
   }
 
   @override
   UnlinkedParamBuilder visitFunctionTypedFormalParameter(
       FunctionTypedFormalParameter node) {
     UnlinkedParamBuilder b = serializeParameter(node);
     b.isFunctionTyped = true;
@@ skipping 39 matching lines @@
     libraryNameOffset = node.name.offset;
     libraryNameLength = node.name.length;
     libraryDocumentationComment =
         serializeDocumentation(node.documentationComment);
     libraryAnnotations = serializeAnnotations(node.metadata);
   }
 
   @override
   void visitMethodDeclaration(MethodDeclaration node) {
     executables.add(serializeExecutable(
+        node,
         node.name.name,
         node.name.offset,
         node.isGetter,
         node.isSetter,
         node.returnType,
         node.parameters,
         node.body,
         false,
         node.isStatic,
         node.documentationComment,
@@ skipping 28 matching lines @@
 
   @override
   UnlinkedTypeParamBuilder visitTypeParameter(TypeParameter node) {
     UnlinkedTypeParamBuilder b = new UnlinkedTypeParamBuilder();
     b.name = node.name.name;
     b.nameOffset = node.name.offset;
     if (node.bound != null) {
       b.bound = serializeTypeName(node.bound);
     }
     b.annotations = serializeAnnotations(node.metadata);
+    serializeCodeRange(b, node);
     return b;
   }
 
   @override
   void visitVariableDeclarationStatement(VariableDeclarationStatement node) {
     serializeVariables(node.variables, false, null, null, false);
   }
 
   /**
    * Helper method to determine if a given [typeName] refers to `dynamic`.
    */
   static bool isDynamic(TypeName typeName) {
     Identifier name = typeName.name;
     return name is SimpleIdentifier && name.name == 'dynamic';
   }
 }
 
 /**
  * A [_TypeParameterScope] is a [_Scope] which defines [_ScopedTypeParameter]s.
  */
 class _TypeParameterScope extends _Scope {
   /**
    * Get the number of [_ScopedTypeParameter]s defined in this
    * [_TypeParameterScope].
    */
   int get length => _definedNames.length;
 }