Address CR feedback

Workiva · Dec 19, 2023 · d36a656 · d36a656
1 parent 1bf39c4
commit d36a656
Showing 1 changed file with 4 additions and 13 deletions.
diff --git a/lib/src/dart3_suggestors/null_safety_prep/required_flux_props.dart b/lib/src/dart3_suggestors/null_safety_prep/required_flux_props.dart
@@ -46,10 +46,9 @@ class RequiredFluxProps extends RecursiveAstVisitor with ClassSuggestor {
             .whereType<MethodDeclaration>()
             .firstOrNull
             ?.name
-            .value()
-            .toString()
-            .contains(RegExp(r'getDefaultProps|defaultProps')) ==
-        true;
+            .lexeme
+            .contains(RegExp(r'getDefaultProps|defaultProps')) ??
+        false;
     if (isReturnedAsDefaultProps) return;
 
     final maybeFluxUiPropsMixin = cascadeWriteEl.mixins
@@ -86,15 +85,7 @@ class RequiredFluxProps extends RecursiveAstVisitor with ClassSuggestor {
   }
 
   void yieldNewCascadeSection(CascadeExpression node, String newSection) {
-    final sf = context.sourceFile;
-    final targetLineOffset = sf.getOffset(sf.getLine(node.target.offset));
-    int offset;
-    if (targetLineOffset == sf.getOffset(sf.getLine(node.target.end))) {
-      // Cascade on a single line / same line as the target, add the new setter(s) before the semicolon
-      offset = node.target.end;
-    } else {
-      offset = sf.getOffsetOfLineAfter(node.target.offset);
-    }
+    final offset = node.target.end;
     yieldPatch(newSection, offset, offset);
   }