添加编辑目标功能

整理优化代码

Author: SamHou0

TimeControl/Tools/SystemControl.cs

@@ -1,4 +1,5 @@
-using System.Diagnostics;
+using System;
+using System.Diagnostics;
 
 namespace TimeControl.Tools
 {
@@ -15,5 +16,10 @@ public static void Shutdown()
             Process process = Process.Start(processStartInfo);
             process.StandardInput.WriteLine("shutdown -s -f -t 0");
         }
+        public static void ProgramRestart()
+        {
+            Process.Start(AppDomain.CurrentDomain.BaseDirectory + "TimeControl.exe");
+            Environment.Exit(0);
+        }
     }
 }

TimeControl/Tools/TCFile.cs

@@ -35,21 +35,35 @@ public static class TCFile
 
         //Title management
         public static readonly string TitleFile = BaseLocation + "Title.txt";
-        //自动关机
+        //Auto shutdown
         public static readonly string ShutdownSpan = BaseLocation + "\\Shutdown.txt";
 
-        //数据显示
+        //TimeData dir
         public static readonly string SavedData = BaseLocation + "\\SavedData.xml";
 
         public static readonly string SavedDataDir = BaseLocation + "\\SavedData";
-
-        public static string[] SavedDataFiles
+        //Title management
+        public static void SaveTitle(ListBox titleListBox)
         {
-            get
+            List<string> titleList = new();
+            foreach (string s in titleListBox.Items)
             {
-                return Directory.GetFiles(SavedDataDir);
+                titleList.Add(s);
             }
+            File.WriteAllLines(TCFile.TitleFile, titleList);
         }
+        public static void ReadTitle(ListBox titleListBox)
+        {
+            if (File.Exists(TitleFile))
+            {
+                string[] strings = File.ReadAllLines(TitleFile);
+                foreach (string s in strings)
+                {
+                    titleListBox.Items.Add(s);
+                }
+            }
+        }
+        //Apps
 
         public static void SaveApps(List<App> apps)
         {
@@ -122,6 +136,15 @@ public static FileInfo GetLatestAppsFile()
             return latestFile;
         }
 
+        //TimeData and Goals
+
+        public static string[] SavedDataFiles
+        {
+            get
+            {
+                return Directory.GetFiles(SavedDataDir);
+            }
+        }
         public static void SaveTimeData(TimeData time)
         {
             using (StreamWriter sw = new(SavedData))

TimeControl/Windows/ControlPanel.cs

@@ -229,23 +229,11 @@ private void RemoveTitleButton_Click(object sender, EventArgs e)
         }
         private void SaveTitles()
         {
-            List<string> titleList = new();
-            foreach (string s in titleListBox.Items)
-            {
-                titleList.Add(s);
-            }
-            File.WriteAllLines(TCFile.TitleFile, titleList);
+            TCFile.SaveTitle(titleListBox);
         }
         private void InitTitles()
         {
-            if (File.Exists(TCFile.TitleFile))
-            {
-                string[] strings = File.ReadAllLines(TCFile.TitleFile);
-                foreach (string s in strings)
-                {
-                    titleListBox.Items.Add(s);
-                }
-            }
+            TCFile.ReadTitle(titleListBox);
         }
         #endregion
 
@@ -319,6 +307,7 @@ private void ProcessMonitorTimer_Tick(object sender, EventArgs e)
                 Process.Start(process);
             }
             Process[] processes=Process.GetProcesses();
+            //Title Management
             foreach (Process process in processes)
             {
                 if(!string.IsNullOrWhiteSpace(process.MainWindowTitle))

TimeControl/Windows/GoalChangeWindow.Designer.cs

@@ -28,8 +28,7 @@ private void RefreshForm()
 
         private void GoalChangeWindow_FormClosed(object sender, FormClosedEventArgs e)
         {
-            Process.Start(AppDomain.CurrentDomain.BaseDirectory + "TimeControl.exe");
-            Environment.Exit(0);
+            SystemControl.ProgramRestart();
         }
 
         private void GoalChangeWindow_FormClosing(object sender, FormClosingEventArgs e)
@@ -88,5 +87,13 @@ private void deleteButton_Click(object sender, EventArgs e)
                 RefreshForm();
             }
         }
+
+        private void EditButton_Click(object sender, EventArgs e)
+        {
+            TimeData timeData = TCFile.ReadTimeData();
+            timeData.GoalName = goalTextBox.Text;
+            TCFile.SaveTimeData(timeData);
+            RefreshForm();
+        }
     }
 }