Merge pull request #283 from microsoft/zt/280-cleanup-failed-exports

#280: Clean up files on failed export

Author: nmetulev

AIDevGallery/Pages/Scenarios/ScenarioPage.xaml.cs

@@ -256,6 +256,7 @@ private async void ExportSampleToggle_Click(object sender, RoutedEventArgs e)
         }
 
         ContentDialog? dialog = null;
+        var generator = new Generator();
         try
         {
             var totalSize = cachedModels.Sum(cm => cm.Value.ModelSize);
@@ -280,8 +281,6 @@ private async void ExportSampleToggle_Click(object sender, RoutedEventArgs e)
                 var folder = await picker.PickSingleFolderAsync();
                 if (folder != null)
                 {
-                    var generator = new Generator();
-
                     dialog = new ContentDialog
                     {
                         XamlRoot = this.XamlRoot,
@@ -327,6 +326,7 @@ private async void ExportSampleToggle_Click(object sender, RoutedEventArgs e)
         catch (Exception ex)
         {
             Debug.WriteLine(ex);
+            generator.CleanUp();
             dialog?.Hide();
 
             var message = "Please try again, or report this issue.";

AIDevGallery/ProjectGenerator/Generator.cs

@@ -21,6 +21,7 @@ namespace AIDevGallery.ProjectGenerator;
 internal partial class Generator
 {
     private readonly string templatePath = Path.Join(Package.Current.InstalledLocation.Path, "ProjectGenerator", "Template");
+    private string generatedProjectPath = string.Empty;
 
     [GeneratedRegex(@"[^a-zA-Z0-9_]")]
     private static partial Regex SafeNameRegex();
@@ -64,6 +65,8 @@ internal async Task<string> GenerateAsync(Sample sample, Dictionary<ModelType, E
             dirIndexCount++;
         }
 
+        generatedProjectPath = outputPath;
+
         var modelTypes = sample.Model1Types.Concat(sample.Model2Types ?? Enumerable.Empty<ModelType>())
                 .Where(models.ContainsKey);
 
@@ -426,4 +429,13 @@ private async Task AddFilesFromSampleAsync(
             await File.WriteAllTextAsync(Path.Join(outputPath, $"Sample.xaml.cs"), sample.GetCleanCSCode(modelInfos), cancellationToken);
         }
     }
+
+    internal void CleanUp()
+    {
+        if(!string.IsNullOrEmpty(generatedProjectPath) && Directory.Exists(generatedProjectPath))
+        {
+            Directory.Delete(generatedProjectPath, true);
+            generatedProjectPath = string.Empty;
+        }
+    }
 }