Compute source spans for IR nodes and J/K-elements

pkg/compiler/lib/src/kernel/element_map_impl.dart

 // Copyright (c) 2017, 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 dart2js.kernel.element_map;
 
 import 'package:kernel/ast.dart' as ir;
 
 import '../common.dart';
 import '../common/names.dart' show Identifiers;
@@ skipping 107 matching lines @@
   }
 
   LibraryEntity get _mainLibrary {
     return _env.mainMethod != null
         ? _getLibrary(_env.mainMethod.enclosingLibrary)
         : null;
   }
 
   Iterable<LibraryEntity> get _libraries;
 
+  SourceSpan _getSourceSpanFromTreeNode(ir.TreeNode node) {
+    // TODO(johnniwinther): Use [ir.Location] directly as a [SourceSpan].
+    Uri uri;
+    int offset;
+    while (node != null) {
+      if (node.fileOffset != ir.TreeNode.noOffset) {
+        offset = node.fileOffset;
+        uri = Uri.parse(node.location.file);
+        break;
+      }
+      node = node.parent;
+    }
+    if (uri != null) {
+      return new SourceSpan(uri, offset, offset + 1);
+    }
+    return null;
+  }
+
+  SourceSpan getSourceSpan(Spannable spannable, Entity currentElement) {
+    SourceSpan fromSpannable(Spannable spannable) {
+      if (spannable is IndexedLibrary &&
+          spannable.libraryIndex < _libraryEnvs.length) {
+        LibraryEnv env = _libraryEnvs[spannable.libraryIndex];
+        return _getSourceSpanFromTreeNode(env.library);
+      } else if (spannable is IndexedClass &&
+          spannable.classIndex < _classEnvs.length) {
+        ClassEnv env = _classEnvs[spannable.classIndex];
+        return _getSourceSpanFromTreeNode(env.cls);
+      } else if (spannable is IndexedMember &&
+          spannable.memberIndex < _memberData.length) {
+        MemberData data = _memberData[spannable.memberIndex];
+        return _getSourceSpanFromTreeNode(data.node);
+      }
+      return null;
+    }
+
+    SourceSpan sourceSpan = fromSpannable(spannable);
+    sourceSpan ??= fromSpannable(currentElement);
+    return sourceSpan;
+  }
+
   LibraryEntity lookupLibrary(Uri uri) {
     LibraryEnv libraryEnv = _env.lookupLibrary(uri);
     if (libraryEnv == null) return null;
     return _getLibrary(libraryEnv.library, libraryEnv);
   }
 
   String _getLibraryName(IndexedLibrary library) {
     assert(checkFamily(library));
     LibraryEnv libraryEnv = _libraryEnvs[library.libraryIndex];
     return libraryEnv.library.name ?? '';
@@ skipping 391 matching lines @@
   }
 
   Iterable<InterfaceType> _getInterfaces(IndexedClass cls) {
     assert(checkFamily(cls));
     ClassData data = _classData[cls.classIndex];
     _ensureSupertypes(cls, data);
     return data.interfaces;
   }
 
   Spannable _getSpannable(MemberEntity member, ir.Node node) {
-    return member;
+    SourceSpan sourceSpan;
+    if (node is ir.TreeNode) {
+      sourceSpan = _getSourceSpanFromTreeNode(node);
+    }
+    sourceSpan ??= getSourceSpan(member, null);
+    return sourceSpan;
   }
 
   ir.Member _getMemberNode(covariant IndexedMember member) {
     assert(checkFamily(member));
     return _memberData[member.memberIndex].node;
   }
 
   ir.Class _getClassNode(covariant IndexedClass cls) {
     assert(checkFamily(cls));
     return _classEnvs[cls.classIndex].cls;
@@ skipping 1375 matching lines @@
   @override
   ir.Member getMemberNode(MemberEntity member) {
     return _getMemberNode(member);
   }
 
   @override
   ir.Class getClassNode(ClassEntity cls) {
     return _getClassNode(cls);
   }
 }