[element model] migrate no_logic_in_create_state

Bug: https://github.com/dart-lang/linter/issues/5099
Change-Id: Idddc688b6a66f1dd03e994af5648d33ca4953441
Reviewed-on: https://dart-review.googlesource.com/c/sdk/+/390083
Reviewed-by: Brian Wilkerson <[email protected]>
Auto-Submit: Phil Quitslund <[email protected]>
Commit-Queue: Brian Wilkerson <[email protected]>

Author: pq

pkg/linter/lib/src/rules/no_logic_in_create_state.dart

@@ -41,7 +41,7 @@ class _Visitor extends SimpleAstVisitor<void> {
 
     var parent = node.parent;
     if (parent is! ClassDeclaration ||
-        !isStatefulWidget(parent.declaredElement)) {
+        !isStatefulWidget2(parent.declaredFragment?.element)) {
       return;
     }
     var body = node.body;

pkg/linter/lib/src/util/flutter_utils.dart

@@ -45,6 +45,9 @@ bool isState(InterfaceElement element) => _flutter.isState(element);
 bool isStatefulWidget(ClassElement? element) =>
     element != null && _flutter.isStatefulWidget(element);
 
+bool isStatefulWidget2(ClassElement2? element) =>
+    element != null && _flutter.isStatefulWidget2(element);
+
 bool isWidgetProperty(DartType? type) {
   if (isWidgetType(type)) {
     return true;
@@ -114,6 +117,11 @@ class _Flutter {
   bool isExactly(InterfaceElement element, String type, Uri uri) =>
       element.name == type && element.source.uri == uri;
 
+  /// Whether [element] is exactly the element named [type], from Flutter.
+  bool isExactly2(InterfaceElement2 element, String type, Uri uri) =>
+      element.name == type &&
+      element.firstFragment.libraryFragment.source.uri == uri;
+
   bool isExactWidget(ClassElement element) =>
       isExactly(element, _nameWidget, _uriFramework);
 
@@ -145,6 +153,11 @@ class _Flutter {
       element.allSupertypes.any((type) =>
           isExactly(type.element, _nameStatefulWidget, _uriFramework));
 
+  bool isStatefulWidget2(ClassElement2 element) =>
+      isExactly2(element, _nameStatefulWidget, _uriFramework) ||
+      element.allSupertypes.any((type) =>
+          isExactly(type.element, _nameStatefulWidget, _uriFramework));
+
   bool isWidget(InterfaceElement element) {
     if (isExactly(element, _nameWidget, _uriFramework)) {
       return true;