Battery_Down/Program.cs at main · LiteZero/Battery_Down · GitHub

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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
namespace BatteryDown;

internal static class Program
{
    [STAThread]
    private static async Task Main(string[] args)
    {
        ApplicationConfiguration.Initialize();

        try
        {
            if (await TryHandleCommandAsync(args))
            {
                return;
            }

            Application.Run(new SettingsForm());
        }
        catch (Exception exception)
        {
            Environment.ExitCode = 1;
            MessageBox.Show(
                exception.Message,
                "Battery Down",
                MessageBoxButtons.OK,
                MessageBoxIcon.Error);
        }
    }

    private static async Task<bool> TryHandleCommandAsync(string[] args)
    {
        if (args.Length == 0)
        {
            return false;
        }

        var command = NormalizeCommand(args[0]);
        switch (command)
        {
            case "prompt":
            case "unlock":
                if (PowerStatus.IsOnBattery)
                {
                    var config = AppConfig.Load();
                    if (!LockScreenPolicy.HandleIfLocked(config))
                    {
                        await ToastService.ShowPromptAsync(config);
                    }
                }
                else
                {
                    ShutdownPlanService.Cancel();
                    ToastService.ClearPrompt();
                }

                return true;

            case "continue":
            case "cancel":
                ShutdownPlanService.Cancel();
                ToastService.ClearPrompt();
                return true;

            case "preview-continue":
            case "preview-shutdown":
                ToastService.ClearPrompt();
                return true;

            case "preview":
                var previewConfig = AppConfig.Load();
                if (args.Length >= 2 && int.TryParse(args[1], out var previewSeconds))
                {
                    previewConfig.ShutdownDelaySeconds = Math.Clamp(previewSeconds, 1, 3600);
                }

                await ToastService.ShowPromptAsync(previewConfig, isPreview: true);
                return true;

            case "shutdown":
                ShutdownPlanService.Cancel();
                ToastService.ClearPrompt();
                if (PowerStatus.IsOnBattery)
                {
                    ShutdownService.ShutdownNow("用户选择了关机。");
                }

                return true;

            case "timeout" when args.Length >= 2:
                ShutdownPlanService.ExecuteTimeout(args[1]);
                return true;

            case "locked-power-check":
                if (args.Length >= 2 &&
                    AppConfig.LoadFrom(args[1]).ShutdownImmediatelyWhenLocked &&
                    SessionStateService.IsLocked &&
                    PowerStatus.IsOnBattery)
                {
                    ShutdownService.ShutdownNow("锁屏状态下检测到拔出电源。");
                }

                return true;

            case "configure-lock-task" when args.Length >= 2:
                EnsureAdministrator();
                if (args[1].Equals("enable", StringComparison.OrdinalIgnoreCase))
                {
                    TaskSchedulerService.InstallLockedSystemTask();
                }
                else if (args[1].Equals("disable", StringComparison.OrdinalIgnoreCase))
                {
                    TaskSchedulerService.RemoveLockedSystemTask();
                }
                else
                {
                    throw new ArgumentException("无效的锁屏系统任务配置参数。");
                }

                return true;

            default:
                return false;
        }
    }

    private static string NormalizeCommand(string argument)
    {
        if (Uri.TryCreate(argument, UriKind.Absolute, out var uri) &&
            uri.Scheme.Equals(RegistrationService.ProtocolName, StringComparison.OrdinalIgnoreCase))
        {
            return uri.Host.ToLowerInvariant();
        }

        return argument.TrimStart('-', '/').ToLowerInvariant();
    }

    private static void EnsureAdministrator()
    {
        using var identity = System.Security.Principal.WindowsIdentity.GetCurrent();
        var principal = new System.Security.Principal.WindowsPrincipal(identity);
        if (!principal.IsInRole(System.Security.Principal.WindowsBuiltInRole.Administrator))
        {
            throw new UnauthorizedAccessException("此操作需要管理员权限。");
        }
    }
}