try to update the visitor to get it working but failed

Author: ArcturusZhang

eng/packages/http-client-csharp-mgmt/generator/Azure.Generator.Management/src/Visitors/PaginationVisitor.cs

@@ -17,10 +17,24 @@ internal class PaginationVisitor : ScmLibraryVisitor
     {
         if (method.EnclosingType.Name.AsSpan().Contains("CollectionResult".AsSpan(), StringComparison.Ordinal) && method.Signature.Name.Equals("AsPages"))
         {
-            var doWhileStatement = statements.OfType<DoWhileStatement>().FirstOrDefault();
+            var doWhileStatement = statements.OfType<WhileStatement>().FirstOrDefault();
             if (doWhileStatement is not null)
             {
                 var body = doWhileStatement.Body;
+                var bodyStatements = body.ToList();
+                // find the yield return statement
+                var yieldReturnStatementIndex = bodyStatements.FindIndex(s => s is YieldReturnStatement);
+                if (yieldReturnStatementIndex >= 0)
+                {
+                    // this statement looks like: yield return Page<T>.FromValues((IReadOnlyList<T>)((TListResult)response).Value, nextPage?.AbsoluteUri, response);
+                    // we need to replace the cast part `(TListResult)response` to a call to `FromResponse` method
+                    var yieldReturnStatement = (YieldReturnStatement)bodyStatements[yieldReturnStatementIndex];
+                    var value = yieldReturnStatement.Value as InvokeMethodExpression;
+                    if (value != null)
+                    {
+                        var first = value.Arguments[0] as CastExpression;
+                    }
+                }
 
                 // get the response to model casting expression
                 var responseToModelStatement = body.OfType<ExpressionStatement>().Skip(1).FirstOrDefault();