test

C3032

2025-12-20 38c333f8091e5062be0bbdd81960315ebce69008

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows;
 
namespace SmartScanner
{
    public static class AppRestartHelper
    {
        private const string TaskName = "MyApp_Restart_Task";
 
        public static async Task RestartAsync()
        {
            try
            {
                // 1. 创建临时批处理文件
                string batPath = CreateRestartBatchFile();
 
                // 2. 创建任务计划
                CreateScheduledTask(batPath);
 
                // 3. 延迟确保任务已注册
                await Task.Delay(500);
 
                // 4. 执行任务并退出当前实例
                ExecuteRestartTask();
 
                Application.Current.Shutdown();
            }
            catch (Exception ex)
            {
                MessageBox.Show($"重启失败: {ex.Message}", "错误", MessageBoxButton.OK, MessageBoxImage.Error);
            }
        }
 
        private static string CreateRestartBatchFile()
        {
            string tempPath = Path.GetTempPath();
            string batPath = Path.Combine(tempPath, $"{Guid.NewGuid()}_restart.bat");
 
            string exePath = Process.GetCurrentProcess().MainModule.FileName;
            string workingDir = Path.GetDirectoryName(exePath);
 
            string batContent = $@"
@echo off
timeout /t 1 /nobreak > nul
cd /d ""{workingDir}""
start """" ""{exePath}""
del ""%~f0""
";
 
            File.WriteAllText(batPath, batContent);
            return batPath;
        }
 
        private static void CreateScheduledTask(string batPath)
        {
            // 删除可能存在的旧任务
            DeleteExistingTask();
 
            // 创建新任务
            ProcessStartInfo psi = new ProcessStartInfo
            {
                FileName = "schtasks.exe",
                Arguments = $"/Create /TN \"{TaskName}\" /TR \"{batPath}\" /SC ONCE /ST {DateTime.Now.AddSeconds(1):HH:mm:ss} /F",
                WindowStyle = ProcessWindowStyle.Hidden,
                CreateNoWindow = true
            };
 
            Process.Start(psi)?.WaitForExit(2000);
        }
 
        private static void ExecuteRestartTask()
        {
            Process.Start(new ProcessStartInfo
            {
                FileName = "schtasks.exe",
                Arguments = $"/Run /TN \"{TaskName}\"",
                WindowStyle = ProcessWindowStyle.Hidden,
                CreateNoWindow = true
            })?.WaitForExit(1000);
 
            // 删除任务
            DeleteExistingTask();
        }
 
        private static void DeleteExistingTask()
        {
            Process.Start(new ProcessStartInfo
            {
                FileName = "schtasks.exe",
                Arguments = $"/Delete /TN \"{TaskName}\" /F",
                WindowStyle = ProcessWindowStyle.Hidden,
                CreateNoWindow = true
            })?.WaitForExit(1000);
        }
    }
}