From 5d3d3173d92a61a5e66188a15878cdec3c3e886d Mon Sep 17 00:00:00 2001
From: mdnapo <mdd.napo@gmail.com>
Date: Fri, 3 May 2024 11:03:14 +0200
Subject: [PATCH] Supplemented script command

---
 .../Commands/MycroForge.Script.Create.cs      | 65 +++++++++++++++++++
 .../Commands/MycroForge.Script.Edit.cs        | 55 ++++++++++++++++
 .../Commands/MycroForge.Script.Run.cs         |  2 +-
 .../Extensions/ServiceCollectionExtensions.cs |  2 +
 4 files changed, 123 insertions(+), 1 deletion(-)
 create mode 100644 MycroForge.CLI/Commands/MycroForge.Script.Create.cs
 create mode 100644 MycroForge.CLI/Commands/MycroForge.Script.Edit.cs

diff --git a/MycroForge.CLI/Commands/MycroForge.Script.Create.cs b/MycroForge.CLI/Commands/MycroForge.Script.Create.cs
new file mode 100644
index 0000000..10635d4
--- /dev/null
+++ b/MycroForge.CLI/Commands/MycroForge.Script.Create.cs
@@ -0,0 +1,65 @@
+using System.CommandLine;
+using System.Diagnostics;
+using MycroForge.CLI.Commands.Interfaces;
+
+namespace MycroForge.CLI.Commands;
+
+public partial class MycroForge
+{
+    public partial class Script
+    {
+        public class Create : Command, ISubCommandOf<Script>
+        {
+            private static readonly Argument<string> NameArgument =
+                new(name: "name", description: "The name of the script");
+
+            public Create() : base(name: "create", description: "Create a script")
+            {
+                AddArgument(NameArgument);
+                this.SetHandler(ExecuteAsync, NameArgument);
+            }
+
+            private async Task ExecuteAsync(string name)
+            {
+                var path = await CreateFile(name);
+                await OpenFile(path);
+            }
+
+            private static async Task<string> CreateFile(string? name = null, string fileExtension = "py")
+            {
+                var folder = Path.Join(
+                    Environment.GetFolderPath(Environment.SpecialFolder.UserProfile), ".m4g"
+                );
+
+                Directory.CreateDirectory(folder);
+
+                var filePath = Path.Join(folder, $"{name}.{fileExtension}");
+
+                if (File.Exists(filePath))
+                    throw new Exception($"File {filePath} already exists.");
+
+                await File.WriteAllTextAsync(filePath, string.Empty);
+
+                return filePath;
+            }
+
+            private static async Task OpenFile(string file)
+            {
+                var process = new Process
+                {
+                    StartInfo = new ProcessStartInfo
+                    {
+                        FileName = "code",
+                        Arguments = $"--wait {file}",
+                        WindowStyle = ProcessWindowStyle.Hidden,
+                        UseShellExecute = true,
+                    }
+                };
+
+                process.Start();
+
+                await process.WaitForExitAsync();
+            }
+        }
+    }
+}
\ No newline at end of file
diff --git a/MycroForge.CLI/Commands/MycroForge.Script.Edit.cs b/MycroForge.CLI/Commands/MycroForge.Script.Edit.cs
new file mode 100644
index 0000000..116e0ea
--- /dev/null
+++ b/MycroForge.CLI/Commands/MycroForge.Script.Edit.cs
@@ -0,0 +1,55 @@
+using System.CommandLine;
+using System.Diagnostics;
+using MycroForge.CLI.Commands.Interfaces;
+
+namespace MycroForge.CLI.Commands;
+
+public partial class MycroForge
+{
+    public partial class Script
+    {
+        public class Edit : Command, ISubCommandOf<Script>
+        {
+            private static readonly Argument<string> NameArgument =
+                new(name: "name", description: "The name of the script");
+
+            public Edit() : base(name: "edit", description: "Edit a script")
+            {
+                AddArgument(NameArgument);
+                this.SetHandler(ExecuteAsync, NameArgument);
+            }
+
+            private async Task ExecuteAsync(string name)
+            {
+                await EditFile(name);
+            }
+
+            private static async Task EditFile(string name)
+            {
+                var folder = Path.Join(
+                    Environment.GetFolderPath(Environment.SpecialFolder.UserProfile), ".m4g"
+                );
+
+                var file = Path.Join(folder, $"{name}.py");
+
+                if (!File.Exists(file))
+                    throw new Exception($"File {file} does not exists.");
+
+                var process = new Process
+                {
+                    StartInfo = new ProcessStartInfo
+                    {
+                        FileName = "code",
+                        Arguments = $"--wait {file}",
+                        WindowStyle = ProcessWindowStyle.Hidden,
+                        UseShellExecute = true,
+                    }
+                };
+
+                process.Start();
+
+                await process.WaitForExitAsync();
+            }
+        }
+    }
+}
\ No newline at end of file
diff --git a/MycroForge.CLI/Commands/MycroForge.Script.Run.cs b/MycroForge.CLI/Commands/MycroForge.Script.Run.cs
index 0a3a71e..1f4b492 100644
--- a/MycroForge.CLI/Commands/MycroForge.Script.Run.cs
+++ b/MycroForge.CLI/Commands/MycroForge.Script.Run.cs
@@ -43,7 +43,7 @@ public partial class MycroForge
                 try
                 {
                     var scriptPath = Path.Combine(
-                        Environment.GetFolderPath(Environment.SpecialFolder.UserProfile), ".m4g", name
+                        Environment.GetFolderPath(Environment.SpecialFolder.UserProfile), ".m4g", $"{name}.py"
                     );
                     engine.ExecuteFile(scriptPath, scope);
 
diff --git a/MycroForge.CLI/Extensions/ServiceCollectionExtensions.cs b/MycroForge.CLI/Extensions/ServiceCollectionExtensions.cs
index ef7a466..2361641 100644
--- a/MycroForge.CLI/Extensions/ServiceCollectionExtensions.cs
+++ b/MycroForge.CLI/Extensions/ServiceCollectionExtensions.cs
@@ -43,6 +43,8 @@ public static class ServiceCollectionExtensions
 
         // Register "m4g script"
         services.AddScoped<ISubCommandOf<Commands.MycroForge>, Commands.MycroForge.Script>();
+        services.AddScoped<ISubCommandOf<Commands.MycroForge.Script>, Commands.MycroForge.Script.Create>();
+        services.AddScoped<ISubCommandOf<Commands.MycroForge.Script>, Commands.MycroForge.Script.Edit>();
         services.AddScoped<ISubCommandOf<Commands.MycroForge.Script>, Commands.MycroForge.Script.Run>();
 
         return services;