From ea354cac72ad6936e6039513bbd4fdd8202b3220 Mon Sep 17 00:00:00 2001
From: Hgnim <hgn1m@outlook.com>
Date: Mon, 6 Jan 2025 19:10:56 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E4=BA=86=E5=BC=80=E6=9C=BA?=
 =?UTF-8?q?=E8=87=AA=E5=90=AF=E5=8A=9F=E8=83=BD=EF=BC=9A=20=E6=94=B9?=
 =?UTF-8?q?=E4=B8=BA=E7=A8=8B=E5=BA=8F=E5=86=85=E9=83=A8=E6=93=8D=E4=BD=9C?=
 =?UTF-8?q?=E6=B3=A8=E5=86=8C=E8=A1=A8=EF=BC=8C=E4=B8=8D=E9=9C=80=E8=A6=81?=
 =?UTF-8?q?=E9=A2=9D=E5=A4=96=E5=90=AF=E5=8A=A8=E5=91=BD=E4=BB=A4=E8=A1=8C?=
 =?UTF-8?q?=E7=AA=97=E5=8F=A3=EF=BC=8C=E4=BD=BF=E5=85=B6=E5=88=87=E6=8D=A2?=
 =?UTF-8?q?=E8=AE=BE=E7=BD=AE=E6=97=B6=E6=9B=B4=E5=8A=A0=E5=BF=AB=E6=8D=B7?=
 =?UTF-8?q?=20=E5=8F=AF=E4=BB=A5=E6=A3=80=E6=B5=8B=E5=BD=93=E5=89=8D?=
 =?UTF-8?q?=E6=98=AF=E5=90=A6=E5=90=AF=E7=94=A8=E4=BA=86=E5=BC=80=E6=9C=BA?=
 =?UTF-8?q?=E8=87=AA=E5=90=AF=20=E5=B0=BD=E5=8F=AF=E8=83=BD=E7=9A=84?=
 =?UTF-8?q?=E7=AE=80=E5=8C=96=E4=BA=86=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Wins/SettingsWin.xaml    |  10 ++--
 Wins/SettingsWin.xaml.cs | 100 ++++++++++++++++++---------------------
 2 files changed, 51 insertions(+), 59 deletions(-)

diff --git a/Wins/SettingsWin.xaml b/Wins/SettingsWin.xaml
index 35f5681..74c7f8e 100644
--- a/Wins/SettingsWin.xaml
+++ b/Wins/SettingsWin.xaml
@@ -9,7 +9,7 @@
         mc:Ignorable="d"
         d:DataContext="{d:DesignInstance preses:SettingsPres}"
         Style="{DynamicResource CommonWindow}" WindowStartupLocation="CenterScreen" ResizeMode="NoResize" SizeToContent="Height" Width="500"
-        KeyDown="SettingsWin_KeyDown">
+        KeyDown="SettingsWin_KeyDown" Loaded="Window_Loaded">
     <Grid Margin="5">
         <Grid.RowDefinitions>
             <RowDefinition Height="*" />
@@ -56,10 +56,8 @@
                 </Binding>
             </Button.Content>
         </Button>
-        
-        <Button Grid.Row="2" Grid.Column="0" Margin="5" Content="启用/修复 程序开机自启动" ToolTip="{Binding Source={x:Static consts:SettingsConst.ColorsButtonToolTip}}"
-                Click="Enabled_SelfStarting_Click"/>
-        <Button Grid.Row="2" Grid.Column="1" Margin="5" Content="禁用 程序开机自启动" ToolTip="{Binding Source={x:Static consts:SettingsConst.ColorsButtonToolTip}}"
-                Click="Disabled_SelfStarting_Click" />
+
+        <Button x:Name="SelfStartingToggle" Grid.Row="2" Grid.Column="0" Margin="5" Content="/ / / /" ToolTip="{Binding Source={x:Static consts:SettingsConst.ColorsButtonToolTip}}"
+                Click="SelfStartingToggle_Click"/>
     </Grid>
 </Window>
\ No newline at end of file
diff --git a/Wins/SettingsWin.xaml.cs b/Wins/SettingsWin.xaml.cs
index 65f8665..d3dbedf 100644
--- a/Wins/SettingsWin.xaml.cs
+++ b/Wins/SettingsWin.xaml.cs
@@ -10,6 +10,7 @@ using Sheas_Cealer.Consts;
 using Sheas_Cealer.Preses;
 using Sheas_Cealer.Props;
 using Sheas_Cealer.Utils;
+using Microsoft.Win32;
 
 namespace Sheas_Cealer.Wins;
 
@@ -65,21 +66,24 @@ public partial class SettingsWin : Window
             Close();
     }
 
-    private void Enabled_SelfStarting_Click(object sender, RoutedEventArgs e) {
-        SelfStarting ss = new();
-        if (ss.RunCommand(SelfStarting.Action.add))
-            MessageBox.Show("已启用开机自启");
+	readonly SelfStarting selfStarting = new();
+    private void SelfStartingToggle_Click(object sender, RoutedEventArgs e) {
+        //在此处为了方便演示，就直接使用了按钮名字进行判断，不建议这么做
+        if( SelfStartingToggle.Content as string == "启用>>禁用 程序开机自启动") {
+            selfStarting.Setting = false;
+            SelfStartingToggle.Content = "禁用>>启用 程序开机自启动";
+		}
+        else {
+			selfStarting.Setting = true;
+			SelfStartingToggle.Content = "启用>>禁用 程序开机自启动";
+		}
     }
-    private void Disabled_SelfStarting_Click(object sender, RoutedEventArgs e) {
-        SelfStarting ss = new();
-        if (ss.RunCommand(SelfStarting.Action.remove))
-            MessageBox.Show("已移除开机自启");
-    }
-    private class SelfStarting {
+
+	private class SelfStarting {
         /// <summary>
         /// 自启动注册表路径
         /// </summary>
-        readonly string regPath = @"HKEY_CURRENT_USER\Software\Microsoft\Windows\CurrentVersion\Run";
+        readonly string regPath = @"Software\Microsoft\Windows\CurrentVersion\Run";
         /// <summary>
         /// 应用名
         /// </summary>
@@ -88,47 +92,37 @@ public partial class SettingsWin : Window
         /// 当前exe文件所在目录
         /// </summary>
         internal string ThisExeFilePath => System.Windows.Forms.Application.ExecutablePath;
-        //自启动程序时会添加启动参数，可以在程序启动时加个判断，判断有-selfStarting参数时进行自动最小化程序等操作
-        string AddCommands => $"{RemoveCommands} & reg add {regPath} /f /v \"{appName}\" /t REG_SZ /d \"{ThisExeFilePath} -selfStarting\"";// & pause";        
-        string RemoveCommands => $"reg delete {regPath} /f /v \"{appName}\"";
-        internal enum Action {
-            add,remove
-        }
-        /// <summary>
-        /// 运行命令
-        /// </summary>
-        /// <param name="action">操作方式</param>
-        /// <returns>返回布尔值代表执行是否成功</returns>
-        internal bool RunCommand(Action action) {
-            // 检查当前进程是否以管理员身份运行
-            //if (!new WindowsPrincipal(WindowsIdentity.GetCurrent()).IsInRole(WindowsBuiltInRole.Administrator))
-            string commands = "";
-            switch (action) {
-                case Action.add:
-                    commands = AddCommands; break;
-                case Action.remove:
-                    commands = RemoveCommands; break;
-            }
 
-            Process process = new() {
-                StartInfo = new ProcessStartInfo {
-                    UseShellExecute = true,
-                    Verb = "RunAs", // 请求管理员权限
-                    CreateNoWindow = true,
-                    FileName = "cmd.exe",
-                    Arguments = $" /c \"{ commands }\""
-                }
-            };
-            bool isTrue = false;
-            try {
-                process.Start();
-                process.WaitForExit();
-                isTrue = true;
-            } 
-            catch (Win32Exception) { System.Windows.Forms.MessageBox.Show("用户取消了授权", "", System.Windows.Forms.MessageBoxButtons.OK, System.Windows.Forms.MessageBoxIcon.Error); }
-            catch { System.Windows.Forms.MessageBox.Show("发生未知错误！", "", System.Windows.Forms.MessageBoxButtons.OK, System.Windows.Forms.MessageBoxIcon.Error); }
-            process.Close();
-            return isTrue;
-        }
-    }
+        /// <summary>
+        /// 设置开机自启
+        /// </summary>
+		internal bool Setting {
+			get {
+				using RegistryKey? key = Registry.CurrentUser.OpenSubKey(regPath, false);
+				return key?.GetValue(appName) != null;
+			}
+			set {
+				void DeleteReg() {
+					using RegistryKey? key = Registry.CurrentUser.OpenSubKey(regPath, true);
+					key?.DeleteValue(appName, false);
+				}
+				switch (value) {
+					case true: {
+							DeleteReg();
+							using RegistryKey? key = Registry.CurrentUser.OpenSubKey(regPath, true);
+							//自启动程序时会添加启动参数，可以在程序启动时加个判断，判断有-selfStarting参数时进行自动最小化程序等操作
+							key?.SetValue(appName, $"{ThisExeFilePath} -selfStarting", RegistryValueKind.String);
+							break;
+						}
+					case false:
+						DeleteReg();
+						break;
+				}
+			}
+		}
+	}
+
+	private void Window_Loaded(object sender, RoutedEventArgs e) => 
+        //检查当前选项状态
+        SelfStartingToggle.Content = selfStarting.Setting ? "启用>>禁用 程序开机自启动" : "禁用>>启用 程序开机自启动";
 }
\ No newline at end of file