fix crash if local variable is used as provider, instead try to get a…

…ctual provider
Alexqwesa · Sep 1, 2023 · 6f1e224 · 6f1e224
1 parent 4158086
commit 6f1e224
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/packages/riverpod_graph/lib/src/analyze.dart b/packages/riverpod_graph/lib/src/analyze.dart
@@ -812,6 +812,22 @@ VariableElement parseProviderFromExpression(
     if (target != null) return parseProviderFromExpression(target, context);
   }
 
+  if (providerExpression is SimpleIdentifier) {
+    // if local variable is used as provider, try to get actual provider
+    final block = providerExpression.thisOrAncestorOfType<Block>();
+    if (block != null) {
+      final vars = block.statements.whereType<VariableDeclarationStatement>();
+      for (final vv in vars) {
+        final element = vv.variables.variables.firstWhereOrNull(
+          (element) => element.name.toString() == providerExpression.toString(),
+        );
+        if (element != null) {
+          return element.declaredElement!;
+        }
+      }
+    }
+  }
+
   throw UnsupportedError(
     'unknown expression $providerExpression ${providerExpression.runtimeType}',
   );