Visit ScopedApi (#7368)

Author: JoshLove-msft

packages/http-client-csharp/generator/Microsoft.TypeSpec.Generator/src/LibraryVisitor.cs

@@ -5,6 +5,7 @@
 using Microsoft.TypeSpec.Generator.Expressions;
 using Microsoft.TypeSpec.Generator.Input;
 using Microsoft.TypeSpec.Generator.Providers;
+using Microsoft.TypeSpec.Generator.Snippets;
 using Microsoft.TypeSpec.Generator.Statements;
 
 namespace Microsoft.TypeSpec.Generator
@@ -202,6 +203,11 @@ internal virtual void Visit(OutputLibrary library)
             return statement;
         }
 
+        protected internal virtual ValueExpression? VisitScopedApiExpression(ScopedApi expression, MethodProvider method)
+        {
+            return expression;
+        }
+
         protected internal virtual MethodBodyStatement? VisitIfElseStatement(IfElseStatement statement, MethodProvider method)
         {
             return statement;

packages/http-client-csharp/generator/Microsoft.TypeSpec.Generator/src/Snippets/ScopedApi.cs

@@ -3,14 +3,15 @@
 
 using Microsoft.TypeSpec.Generator.Expressions;
 using Microsoft.TypeSpec.Generator.Primitives;
+using Microsoft.TypeSpec.Generator.Providers;
 using Microsoft.TypeSpec.Generator.Statements;
 
 namespace Microsoft.TypeSpec.Generator.Snippets
 {
     public record ScopedApi : ValueExpression
     {
-        public ValueExpression Original { get; }
-        public CSharpType Type { get; }
+        public ValueExpression Original { get; private set; }
+        public CSharpType Type { get; private set; }
 
         public ScopedApi(CSharpType type, ValueExpression original)
             : base(original)
@@ -28,5 +29,32 @@ internal override void Write(CodeWriter writer)
 
         protected internal override bool IsEmptyExpression() => Original.IsEmptyExpression();
         public MethodBodyStatement Terminate() => _terminated ??= new ExpressionStatement(this);
+
+        internal override ValueExpression? Accept(LibraryVisitor visitor, MethodProvider method)
+        {
+            var updatedExpression = visitor.VisitScopedApiExpression(this, method);
+
+            if (updatedExpression is not ScopedApi scopedApi)
+            {
+                return updatedExpression?.Accept(visitor, method);
+            }
+
+            scopedApi.Original = scopedApi.Original.Accept(visitor, method)!;
+
+            return scopedApi;
+        }
+
+        public void Update(ValueExpression? original = null, CSharpType? type = null)
+        {
+            if (original != null)
+            {
+                Original = original;
+            }
+
+            if (type != null)
+            {
+                Type = type;
+            }
+        }
     }
 }