fix: Fixed init command issue with binary files.

tryAGI · Sep 7, 2024 · afdfbd1 · afdfbd1
1 parent 4574e57
commit afdfbd1
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/src/libs/OpenApiGenerator.Cli/Commands/InitializeCommand.cs b/src/libs/OpenApiGenerator.Cli/Commands/InitializeCommand.cs
@@ -117,9 +117,19 @@ private static async Task HandleAsync(
                 Directory.CreateDirectory(directory);
             }
 
-            await File.WriteAllTextAsync(
-                path,
-                Replace(resource.AsString())).ConfigureAwait(false);
+            var extension = Path.GetExtension(path);
+            if (extension.ToUpperInvariant() is ".PNG" or ".JPG" or ".JPEG" or ".GIF" or ".SVG" or ".SNK")
+            {
+                await File.WriteAllBytesAsync(
+                    path,
+                    resource.AsBytes()).ConfigureAwait(false);
+            }
+            else
+            {
+                await File.WriteAllTextAsync(
+                    path,
+                    Replace(resource.AsString())).ConfigureAwait(false);
+            }
         }
 
         Console.WriteLine("Done.");