Elements. Add PartInclude.partKeywordOffset

Bug: dart-lang/source_gen#769
Change-Id: I2316e3185326c2b9662a4cfa28dfc7b23a36e5c5
Reviewed-on: https://dart-review.googlesource.com/c/sdk/+/443903
Reviewed-by: Johnni Winther <[email protected]>
Reviewed-by: Brian Wilkerson <[email protected]>
Commit-Queue: Konstantin Shcheglov <[email protected]>

Author: scheglov

pkg/analyzer/CHANGELOG.md

@@ -4,6 +4,7 @@
 * Add `documentationComment` to `Element` and `Fragment`.
 * Add `metadata` to `Element` and `Fragment`.
 * Add `metadata` to `ElementDirective`.
+* Add `partKeywordOffset` to `PartInclude`.
 * Fix draining analysis events when used by `package:build`.
 * Deprecate `LibraryElementResult.element2`, use `element` instead.
 * Deprecate `ResolvedLibraryResult.element2`, use `element` instead.

pkg/analyzer/api.txt

@@ -4034,6 +4034,7 @@ package:analyzer/dart/element/element.dart:
   PartInclude (class extends Object implements ElementDirective):
     new (constructor: PartInclude Function())
     includedFragment (getter: LibraryFragment?)
+    partKeywordOffset (getter: int)
   PatternVariableElement (class extends Object implements LocalVariableElement):
     new (constructor: PatternVariableElement Function())
     firstFragment (getter: PatternVariableFragment)

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

@@ -3422,6 +3422,9 @@ sealed class NamespaceCombinator {
 abstract class PartInclude implements ElementDirective {
   /// The [LibraryFragment], if [uri] is a [DirectiveUriWithUnit].
   LibraryFragment? get includedFragment;
+
+  /// The offset of the `part` keyword.
+  int get partKeywordOffset;
 }
 
 /// A pattern variable.

pkg/analyzer/lib/src/dart/analysis/driver.dart

@@ -109,7 +109,7 @@ testFineAfterLibraryAnalyzerHook;
 // TODO(scheglov): Clean up the list of implicitly analyzed files.
 class AnalysisDriver {
   /// The version of data format, should be incremented on every format change.
-  static const int DATA_VERSION = 510;
+  static const int DATA_VERSION = 511;
 
   /// The number of exception contexts allowed to write. Once this field is
   /// zero, we stop writing any new exception contexts in this process.

pkg/analyzer/lib/src/dart/analysis/file_state.dart

@@ -1161,6 +1161,7 @@ class FileState {
   static UnlinkedPartDirective _serializePart(PartDirective node) {
     return UnlinkedPartDirective(
       configurations: _serializeConfigurations(node.configurations),
+      partKeywordOffset: node.partKeyword.offset,
       uri: node.uri.stringValue,
     );
   }

pkg/analyzer/lib/src/dart/analysis/unlinked_data.dart

@@ -342,13 +342,20 @@ class UnlinkedNamespaceDirectiveConfiguration {
 }
 
 class UnlinkedPartDirective extends UnlinkedConfigurableUriDirective {
-  UnlinkedPartDirective({required super.configurations, required super.uri});
+  final int partKeywordOffset;
+
+  UnlinkedPartDirective({
+    required super.configurations,
+    required this.partKeywordOffset,
+    required super.uri,
+  });
 
   factory UnlinkedPartDirective.read(SummaryDataReader reader) {
     return UnlinkedPartDirective(
       configurations: reader.readTypedList(
         () => UnlinkedNamespaceDirectiveConfiguration.read(reader),
       ),
+      partKeywordOffset: reader.readUInt30(),
       uri: reader.readOptionalStringUtf8(),
     );
   }
@@ -359,6 +366,7 @@ class UnlinkedPartDirective extends UnlinkedConfigurableUriDirective {
     ) {
       x.write(sink);
     });
+    sink.writeUInt30(partKeywordOffset);
     sink.writeOptionalStringUtf8(uri);
   }
 }

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

@@ -8391,7 +8391,10 @@ abstract class NonParameterVariableFragmentImpl extends VariableFragmentImpl {
 }
 
 class PartIncludeImpl extends ElementDirectiveImpl implements PartInclude {
-  PartIncludeImpl({required super.uri});
+  @override
+  int partKeywordOffset;
+
+  PartIncludeImpl({required super.uri, required this.partKeywordOffset});
 
   @override
   LibraryFragmentImpl? get includedFragment {

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

@@ -1095,7 +1095,7 @@ class LibraryReader {
   }) {
     var uri = _readDirectiveUri(containerUnit: containerUnit);
 
-    return PartIncludeImpl(uri: uri);
+    return PartIncludeImpl(partKeywordOffset: -1, uri: uri);
   }
 
   /// Read the reference of a non-local element.

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

@@ -72,6 +72,7 @@ class InformativeDataApplier {
     unitElement.withoutLoadingResolution(() {
       _applyToImports(unitElement.libraryImports, unitInfo);
       _applyToExports(unitElement.libraryExports, unitInfo);
+      _applyToPartIncludes(unitElement.parts, unitInfo);
     });
 
     unitElement.deferConstantOffsets(unitInfo.libraryConstantOffsets, (
@@ -563,6 +564,12 @@ class InformativeDataApplier {
     });
   }
 
+  void _applyToPartIncludes(List<PartIncludeImpl> imports, _InfoUnit info) {
+    forCorrespondingPairs(imports, info.parts, (element, info) {
+      element.partKeywordOffset = info.nameOffset;
+    });
+  }
+
   void _applyToTopLevelVariable(
     TopLevelVariableFragmentImpl element,
     _InfoTopLevelVariable info,

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

@@ -712,7 +712,10 @@ class LibraryBuilder {
         }
     }
 
-    return PartIncludeImpl(uri: directiveUri);
+    return PartIncludeImpl(
+      partKeywordOffset: state.unlinked.partKeywordOffset,
+      uri: directiveUri,
+    );
   }
 
   /// We want to have stable references for `loadLibrary` function. But we