Add support for distributed tracing in generated clients (Azure#49691)

* feat: add support for distributed tracing in clients

* refactor to use more visitor APIs

* address initial feedback

* address feedback

* more feedback & bump versions

* nits & regen

Author: jorgerangel-msft

eng/Packages.Data.props

@@ -206,8 +206,8 @@
   </ItemGroup>
 
   <ItemGroup Condition="'$(IsGeneratorLibrary)' == 'true'">
-    <PackageReference Update="Microsoft.TypeSpec.Generator.ClientModel" Version="1.0.0-alpha.20250429.2" />
-    <PackageReference Update="Microsoft.TypeSpec.Generator.Input" Version="1.0.0-alpha.20250429.2" />
+    <PackageReference Update="Microsoft.TypeSpec.Generator.ClientModel" Version="1.0.0-alpha.20250501.3" />
+    <PackageReference Update="Microsoft.TypeSpec.Generator.Input" Version="1.0.0-alpha.20250501.3" />
     <PackageReference Update="Azure.Generator" Version="1.0.0-alpha.20250429.2" />
     <PackageReference Update="System.ClientModel" Version="1.3.0" />
   </ItemGroup>

eng/packages/http-client-csharp/generator/Azure.Generator/src/Azure.Generator.csproj

@@ -15,7 +15,7 @@
 		<!-- Check if the folder does not exist -->
 		<Error Condition="!Exists('..\..\..\node_modules\@typespec\http-client-csharp')" Text="Emitter has not been built please run `npm ci` from repo root folder, and followed by `npm run build` from ./eng/packages/http-client-csharp folder." />
 	</Target>
-  
+
 	<!-- Copy output to package dist path for local execution -->
 	<Target Name="CopyForNpmPackage" AfterTargets="Build">
 		<Message Text="Copying output to dist path" Importance="high" />
@@ -40,6 +40,21 @@
     <Compile Include="$(MSBuildThisFileDirectory)..\..\..\..\..\..\sdk\core\Azure.Core\src\Shared\RequestHeaderExtensions.cs" LinkBase="Shared/Core">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </Compile>
+		<Compile Include="$(MSBuildThisFileDirectory)..\..\..\..\..\..\sdk\core\Azure.Core\src\Shared\AppContextSwitchHelper.cs" LinkBase="Shared/Core">
+		  <CopyToOutputDirectory>Always</CopyToOutputDirectory>
+		</Compile>
+		<Compile Include="$(MSBuildThisFileDirectory)..\..\..\..\..\..\sdk\core\Azure.Core\src\Shared\ClientDiagnostics.cs" LinkBase="Shared/Core">
+		  <CopyToOutputDirectory>Always</CopyToOutputDirectory>
+		</Compile>
+		<Compile Include="$(MSBuildThisFileDirectory)..\..\..\..\..\..\sdk\core\Azure.Core\src\Shared\DiagnosticScopeFactory.cs" LinkBase="Shared/Core">
+		  <CopyToOutputDirectory>Always</CopyToOutputDirectory>
+		</Compile>
+		<Compile Include="$(MSBuildThisFileDirectory)..\..\..\..\..\..\sdk\core\Azure.Core\src\Shared\DiagnosticScope.cs" LinkBase="Shared/Core">
+		  <CopyToOutputDirectory>Always</CopyToOutputDirectory>
+		</Compile>
+		<Compile Include="$(MSBuildThisFileDirectory)..\..\..\..\..\..\sdk\core\Azure.Core\src\Shared\HttpMessageSanitizer.cs" LinkBase="Shared/Core">
+		  <CopyToOutputDirectory>Always</CopyToOutputDirectory>
+		</Compile>
 	</ItemGroup>
 
 </Project>

eng/packages/http-client-csharp/generator/Azure.Generator/src/AzureClientGenerator.cs

@@ -1,6 +1,7 @@
 // Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT License.
 
+using Azure.Generator.Visitors;
 using Microsoft.CodeAnalysis;
 using Microsoft.TypeSpec.Generator;
 using Microsoft.TypeSpec.Generator.ClientModel;
@@ -49,5 +50,6 @@ protected override void Configure()
         var sharedSourceDirectory = Path.Combine(Path.GetDirectoryName(typeof(AzureClientGenerator).Assembly.Location)!, "Shared", "Core");
         AddSharedSourceDirectory(sharedSourceDirectory);
         AddVisitor(new NamespaceVisitor());
+        AddVisitor(new DistributedTracingVisitor());
     }
 }

eng/packages/http-client-csharp/generator/Azure.Generator/src/Primitives/NewAzureProjectScaffolding.cs

@@ -7,6 +7,7 @@
 using System;
 using System.Collections.Generic;
 using System.Linq;
+using System.Numerics;
 
 namespace Azure.Generator.Primitives
 {
@@ -52,31 +53,40 @@ protected override string GetSourceProjectFileContent()
 
             if (hasOperation)
             {
-                builder.CompileIncludes.Add(new CSharpProjectWriter.CSProjCompileInclude(GetCompileInclude("RawRequestUriBuilder.cs", pathSegmentCount), SharedSourceLinkBase));
-                builder.CompileIncludes.Add(new CSharpProjectWriter.CSProjCompileInclude(GetCompileInclude("TypeFormatters.cs", pathSegmentCount), SharedSourceLinkBase));
-                builder.CompileIncludes.Add(new CSharpProjectWriter.CSProjCompileInclude(GetCompileInclude("RequestHeaderExtensions.cs", pathSegmentCount), SharedSourceLinkBase));
+                foreach (var file in _operationSharedFiles)
+                {
+                    builder.CompileIncludes.Add(new CSharpProjectWriter.CSProjCompileInclude(GetCompileInclude(file, pathSegmentCount), SharedSourceLinkBase));
+                }
             }
 
             if (hasLongRunningOperation)
             {
                 foreach (var file in _lroSharedFiles)
                 {
-                    builder.CompileIncludes.Add(new CSharpProjectWriter.CSProjCompileInclude(GetCompileInclude(file, pathSegmentCount), "Shared/Core"));
+                    builder.CompileIncludes.Add(new CSharpProjectWriter.CSProjCompileInclude(GetCompileInclude(file, pathSegmentCount), SharedSourceLinkBase));
                 }
             }
 
             return builder.Write();
         }
 
-        private static readonly IReadOnlyList<string> _lroSharedFiles =
+        private static readonly IReadOnlyList<string> _operationSharedFiles =
         [
+            "RawRequestUriBuilder.cs",
+            "TypeFormatters.cs",
+            "RequestHeaderExtensions.cs",
             "AppContextSwitchHelper.cs",
-            "AsyncLockWithValue.cs",
-            "FixedDelayWithNoJitterStrategy.cs",
             "ClientDiagnostics.cs",
             "DiagnosticScopeFactory.cs",
             "DiagnosticScope.cs",
             "HttpMessageSanitizer.cs",
+            "TrimmingAttribute.cs",
+        ];
+
+        private static readonly IReadOnlyList<string> _lroSharedFiles =
+        [
+            "AsyncLockWithValue.cs",
+            "FixedDelayWithNoJitterStrategy.cs",
             "IOperationSource.cs",
             "NextLinkOperationImplementation.cs",
             "OperationFinalStateVia.cs",
@@ -86,7 +96,6 @@ protected override string GetSourceProjectFileContent()
             "OperationPoller.cs",
             "SequentialDelayStrategy.cs",
             "TaskExtensions.cs",
-            "TrimmingAttribute.cs",
             "VoidValue.cs"
         ];
 
@@ -131,4 +140,4 @@ private string GetCompileInclude(string fileName, int pathSegmentCount)
             return $"{MSBuildThisFileDirectory}{string.Concat(Enumerable.Repeat(ParentDirectory, pathSegmentCount))}{RelativeCoreSegment}{fileName}";
         }
     }
-}
+}