[DotnetTrace][CollectLinux] Build record-trace args

Author: mdh1418

src/Tools/dotnet-trace/CommandLine/Commands/CollectLinuxCommand.cs

@@ -5,7 +5,9 @@
 using System.CommandLine;
 using System.Diagnostics;
 using System.IO;
+using System.Text;
 using System.Threading.Tasks;
+using Microsoft.Diagnostics.NETCore.Client;
 using Microsoft.Internal.Common.Utils;
 
 namespace Microsoft.Diagnostics.Tools.Trace
@@ -35,6 +37,12 @@ private static int CollectLinux(CollectLinuxArgs args)
                 return (int)ReturnCode.ArgumentError;
             }
 
+            if (args.ProcessId != 0 && !string.IsNullOrEmpty(args.Name))
+            {
+                Console.Error.WriteLine("Only one of --process-id or --name can be specified.");
+                return (int)ReturnCode.ArgumentError;
+            }
+
             return RunRecordTrace(args);
         }
 
@@ -145,14 +153,98 @@ private static bool TryResolveRecordTracePath(out string path)
 
         private static List<string> BuildRecordTraceArgs(CollectLinuxArgs args)
         {
-            Console.WriteLine($"{args.ProcessId}");
             List<string> recordTraceArgs = new();
 
-            recordTraceArgs.Add("--on-cpu");
+            foreach (string profile in args.Profiles)
+            {
+                if (profile.Equals("kernel-cpu", StringComparison.OrdinalIgnoreCase))
+                {
+                    recordTraceArgs.Add("--on-cpu");
+                }
+                if (profile.Equals("kernel-cswitch", StringComparison.OrdinalIgnoreCase))
+                {
+                    recordTraceArgs.Add("--off-cpu");
+                }
+            }
+
+            recordTraceArgs.Add("--script");
+            StringBuilder scriptArg = new();
+
+            string[] profiles = args.Profiles;
+            if (args.Profiles.Length == 0 && args.Providers.Length == 0 && string.IsNullOrEmpty(args.ClrEvents))
+            {
+                Console.WriteLine("No profile or providers specified, defaulting to trace profile 'dotnet-common'");
+                profiles = new[] { "cpu-sampling" };
+            }
+            List<EventPipeProvider> providerCollection = ProviderUtils.ToProviders(args.Providers, args.ClrEvents, args.ClrEventLevel, profiles);
+            foreach (EventPipeProvider provider in providerCollection)
+            {
+                Console.WriteLine($"{provider}");
+            }
+            // Dummy event for now, until FFI is released
+            scriptArg.Append($"let gc_start = event_from_dotnet(\"Microsoft-Windows-DotNETRuntime\", 1, 4, 1, \"GCStart_V2\"); record_event(gc_start);");
+
+            foreach (string perfEvent in args.PerfEvents)
+            {
+                if (string.IsNullOrWhiteSpace(perfEvent) || !perfEvent.Contains(':', StringComparison.Ordinal))
+                {
+                    throw new ArgumentException($"Invalid perf event specification '{perfEvent}'. Expected format 'provider:event'.");
+                }
+
+                string[] split = perfEvent.Split(':', 2, StringSplitOptions.TrimEntries);
+                if (split.Length != 2 || string.IsNullOrEmpty(split[0]) || string.IsNullOrEmpty(split[1]))
+                {
+                    throw new ArgumentException($"Invalid perf event specification '{perfEvent}'. Expected format 'provider:event'.");
+                }
+
+                string perfProvider = split[0];
+                string perfEventName = split[1];
+                scriptArg.Append($"let {perfEventName} = event_from_tracefs(\"{perfProvider}\", \"{perfEventName}\"); record_event({perfEventName});");
+            }
+            recordTraceArgs.Add(scriptArg.ToString());
+
+            int pid = args.ProcessId;
+            if (!string.IsNullOrEmpty(args.Name))
+            {
+                pid = CommandUtils.FindProcessIdWithName(args.Name);
+            }
+            if (pid > 0)
+            {
+                recordTraceArgs.Add($"--pid");
+                recordTraceArgs.Add($"{pid}");
+            }
+
+            string resolvedOutput = ResolveOutputPath(args.Output, pid);
+            recordTraceArgs.Add($"--out");
+            recordTraceArgs.Add(resolvedOutput);
+
+            if (args.Duration != default(TimeSpan))
+            {
+                recordTraceArgs.Add($"--duration");
+                recordTraceArgs.Add(args.Duration.ToString());
+            }
 
             return recordTraceArgs;
         }
 
+        private static string ResolveOutputPath(FileInfo output, int processId)
+        {
+            if (!string.Equals(output.Name, CommonOptions.DefaultTraceName, StringComparison.OrdinalIgnoreCase))
+            {
+                return output.Name;
+            }
+
+            DateTime now = DateTime.Now;
+            if (processId > 0)
+            {
+                Process process = Process.GetProcessById(processId);
+                FileInfo processMainModuleFileInfo = new(process.MainModule.FileName);
+                return $"{processMainModuleFileInfo.Name}_{now:yyyyMMdd}_{now:HHmmss}.nettrace";
+            }
+
+            return $"collect_linux_{now:yyyyMMdd}_{now:HHmmss}.nettrace";
+        }
+
         private static readonly Option<string> PerfEventsOption =
             new("--perf-events")
             {