From 810ff86999a84a7c0154551d612121018ed82767 Mon Sep 17 00:00:00 2001
From: Hagnimik <hgn1m@outlook.com>
Date: Thu, 12 Dec 2024 11:52:32 +0800
Subject: [PATCH] =?UTF-8?q?1.1.3=20->=201.1.4=20=E7=AC=AC24=E5=8F=B7?=
 =?UTF-8?q?=E6=8B=89=E5=8F=96=20(#24)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

添加了开机自启功能的代码
---
 Wins/SettingsWin.xaml    |  6 ++++
 Wins/SettingsWin.xaml.cs | 69 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 75 insertions(+)

diff --git a/Wins/SettingsWin.xaml b/Wins/SettingsWin.xaml
index c7d75f7..35f5681 100644
--- a/Wins/SettingsWin.xaml
+++ b/Wins/SettingsWin.xaml
@@ -14,6 +14,7 @@
         <Grid.RowDefinitions>
             <RowDefinition Height="*" />
             <RowDefinition Height="*" />
+            <RowDefinition Height="*" />
         </Grid.RowDefinitions>
         <Grid.ColumnDefinitions>
             <ColumnDefinition Width="*" />
@@ -55,5 +56,10 @@
                 </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" />
     </Grid>
 </Window>
\ No newline at end of file
diff --git a/Wins/SettingsWin.xaml.cs b/Wins/SettingsWin.xaml.cs
index 982420f..65f8665 100644
--- a/Wins/SettingsWin.xaml.cs
+++ b/Wins/SettingsWin.xaml.cs
@@ -1,4 +1,6 @@
 using System;
+using System.ComponentModel;
+using System.Diagnostics;
 using System.Windows;
 using System.Windows.Controls;
 using System.Windows.Input;
@@ -62,4 +64,71 @@ public partial class SettingsWin : Window
         if (e.Key == Key.Escape)
             Close();
     }
+
+    private void Enabled_SelfStarting_Click(object sender, RoutedEventArgs e) {
+        SelfStarting ss = new();
+        if (ss.RunCommand(SelfStarting.Action.add))
+            MessageBox.Show("已启用开机自启");
+    }
+    private void Disabled_SelfStarting_Click(object sender, RoutedEventArgs e) {
+        SelfStarting ss = new();
+        if (ss.RunCommand(SelfStarting.Action.remove))
+            MessageBox.Show("已移除开机自启");
+    }
+    private class SelfStarting {
+        /// <summary>
+        /// 自启动注册表路径
+        /// </summary>
+        readonly string regPath = @"HKEY_CURRENT_USER\Software\Microsoft\Windows\CurrentVersion\Run";
+        /// <summary>
+        /// 应用名
+        /// </summary>
+        readonly string appName = "SheasCealer";
+        /// <summary>
+        /// 当前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;
+        }
+    }
 }
\ No newline at end of file