From a38bbf28e1fb76d57e74e37218ba45852b82c6fb Mon Sep 17 00:00:00 2001
From: "Juster.zhu" <zhuzhen723723@163.com>
Date: Fri, 23 Aug 2024 00:58:09 +0800
Subject: [PATCH] =?UTF-8?q?feature:=20=E4=BF=AE=E6=94=B9=E5=90=AF=E5=8A=A8?=
 =?UTF-8?q?=E6=96=B9=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/StartManager/Application.cs | 29 ++++++++++---------------
 src/StartManager/Program.cs     | 38 +++++++++++----------------------
 src/resource.bat                |  1 +
 3 files changed, 24 insertions(+), 44 deletions(-)

diff --git a/src/StartManager/Application.cs b/src/StartManager/Application.cs
index 0f95847..8dd6d24 100644
--- a/src/StartManager/Application.cs
+++ b/src/StartManager/Application.cs
@@ -4,29 +4,22 @@ namespace StartManager;
 
 public class Application
 {
+    public static void StartFileServer() => Start("hfs.exe", GetAppDirectory());
     
-    public static void StartFileServer()
-    {
-        var name = "hfs.exe";
-    }
-    
-    public static void StartClient()
-    {
-        var name = "ClientSample.exe";
-    }
+    public static void StartClient() => Start("ClientSample.exe", GetAppDirectory());
     
-    public static void StartServer()
-    {
-        var name = "ServerSample.exe";
-    }
+    public static void StartServer() => Start("ServerSample.exe", GetAppDirectory());
     
-    public static void StartUpgrade()
-    {
-        var name = "UpgradeSample.exe";
-    }
+    public static void Reset() => Start("resource.bat", GetParent());
+
+    private static string GetAppDirectory() => Path.Combine(GetParent(), "run", "app");
 
-    public static void Reset()
+    private static string GetParent()
     {
+        var currentDirectory = Directory.GetCurrentDirectory();
+        int indexOfSrc = currentDirectory.IndexOf("src", StringComparison.OrdinalIgnoreCase);
+        string truncatedPath = currentDirectory.Substring(0, indexOfSrc + "src".Length);
+        return truncatedPath;
     }
 
     private static void Start(string appName, string path)
diff --git a/src/StartManager/Program.cs b/src/StartManager/Program.cs
index 79745f4..90899a2 100644
--- a/src/StartManager/Program.cs
+++ b/src/StartManager/Program.cs
@@ -5,22 +5,25 @@ class Program
     static void Main(string[] args)
     {
         bool exit = false;
-        ShowMenu();
         while (!exit)
         {
+            ShowMenu();
             string input = Console.ReadLine();
             switch (input)
             {
                 case "1":
-                    Option1();
+                    Application.Reset();
                     break;
                 case "2":
-                    Option2();
+                    Application.StartFileServer();
                     break;
                 case "3":
-                    Option3();
+                    Application.StartServer();
                     break;
                 case "4":
+                    Application.StartClient();
+                    break;
+                case "E":
                     exit = true;
                     break;
                 default:
@@ -33,28 +36,11 @@ static void Main(string[] args)
     static void ShowMenu()
     {
         Console.WriteLine("请选择一个选项：");
-        Console.WriteLine("1. 启动服务端");
-        Console.WriteLine("2. 启动客户端");
-        Console.WriteLine("3. 还原");
-        Console.WriteLine("4. 退出");
+        Console.WriteLine("1. 还原run目录");
+        Console.WriteLine("2. 启动文件服务端");
+        Console.WriteLine("3. 启动服务端");
+        Console.WriteLine("4. 启动客户端");
+        Console.WriteLine("E. 退出");
         Console.Write("输入你的选择: ");
     }
-
-    static void Option1()
-    {
-        Console.WriteLine("你选择了选项1");
-        // 在这里添加选项1的逻辑
-    }
-
-    static void Option2()
-    {
-        Console.WriteLine("你选择了选项2");
-        // 在这里添加选项2的逻辑
-    }
-
-    static void Option3()
-    {
-        Console.WriteLine("你选择了选项3");
-        // 在这里添加选项3的逻辑
-    }
 }
\ No newline at end of file
diff --git a/src/resource.bat b/src/resource.bat
index e20b2ec..e9c4e3f 100644
--- a/src/resource.bat
+++ b/src/resource.bat
@@ -22,6 +22,7 @@ REM Copy files from Client, Server, Upgrade to app directory
 xcopy "%BASE_DIR%Client\bin\Release\net8.0\*" "%BASE_DIR%run\app\" /s /e /y
 xcopy "%BASE_DIR%Server\bin\Release\net8.0\*" "%BASE_DIR%run\app\" /s /e /y
 xcopy "%BASE_DIR%Upgrade\bin\Release\net8.0\*" "%BASE_DIR%run\app\" /s /e /y
+xcopy "%BASE_DIR%FileService\*" "%BASE_DIR%run\app\" /s /e /y
 
 REM Create test.txt files with specified content
 echo 123456 > "%BASE_DIR%run\files\old\test.txt"