Add sensitive files option

Author: CodingWonders

Program.cs

@@ -152,6 +152,8 @@ static async Task Main(string[] args)
             List<Schneegans.Unattend.ProcessorArchitecture> defaultArchitectures = new List<Schneegans.Unattend.ProcessorArchitecture>();
             defaultArchitectures.Add(Schneegans.Unattend.ProcessorArchitecture.amd64);
 
+            bool noSensitiveFiles = false;
+
             Console.WriteLine($"UnattendGen{(File.Exists(Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "DT")) ? " for DISMTools" : "")}, version {Assembly.GetEntryAssembly().GetName().Version.ToString()}");
             Console.WriteLine("-------------------------------------------------");
             Console.WriteLine($"Program: (c) {GetCopyrightTimespan(2024, DateTime.Today.Year)}. CodingWonders Software\nLibrary: (c) {GetCopyrightTimespan(2024, DateTime.Today.Year)}. Christoph Schneegans");
@@ -765,6 +767,11 @@ static async Task Main(string[] args)
                             generator.SystemComponents = defaultComponents;
                         }
                     }
+                    else if (cmdLine.StartsWith("--nosensitivefiles", StringComparison.OrdinalIgnoreCase))
+                    {
+                        Console.WriteLine("INFO: removing sensitive files after installation");
+                        noSensitiveFiles = true;
+                    }
                     if (cmdLine != Assembly.GetExecutingAssembly().Location)
                         DebugWrite($"Successfully parsed command-line switch {cmdLine}", (debugMode | Debugger.IsAttached));
                 }
@@ -799,6 +806,8 @@ static async Task Main(string[] args)
                 generator.processorArchitectures = defaultArchitectures;
             }
 
+            generator.noSensitiveFiles = noSensitiveFiles;
+
             await generator.GenerateAnswerFile(targetPath != "" ? targetPath : "unattend.xml");
         }
     }
@@ -893,6 +902,8 @@ public enum SystemTelemetry
 
         public List<SystemComponent>? SystemComponents = new List<SystemComponent>();
 
+        public bool noSensitiveFiles;
+
         public async Task GenerateAnswerFile(string targetPath)
         {
             await Task.Run(() =>
@@ -1092,7 +1103,8 @@ Configuration.Default with
                             ParallelsTools = (virtualMachine == VirtualMachineSolution.Parallels),
                             UseConfigurationSet = UseConfigSet,
                             HidePowerShellWindows = HideScriptWindows,
-                            DeleteWindowsOld = true
+                            DeleteWindowsOld = true,
+                            KeepSensitiveFiles = !noSensitiveFiles
                         }
                         );
                     /*
@@ -1134,6 +1146,7 @@ Configuration.Default with
                     Console.WriteLine("Pausing execution state and exiting after 5 seconds...");
 
                     Thread.Sleep(5000);
+                    Environment.Exit(ex.HResult);
                 }
             });
         }