AutoTotal/App.xaml.cs at master · ImMALWARE/AutoTotal · 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
using Microsoft.Win32;
using Notification.Wpf;
using Notification.Wpf.Constants;
using Notification.Wpf.Classes;
using System.Collections.Specialized;
using System.IO;
using System.Windows;
using System.Windows.Media.Imaging;
using Application = System.Windows.Application;

namespace AutoTotal {

    public static class Data {
        internal static readonly NotificationManager notificationManager = new();
        internal static readonly Dictionary<string, NotifierProgress<(double?, string, string, bool?)>> progressbars = new();
        internal static Mutex? mutex;
        internal static SettingsWindow? settings;
    }

    public partial class App : Application {
        protected override void OnStartup(StartupEventArgs e) {
            base.OnStartup(e);
            NotificationConstants.DefaultProgressButtonContent = AutoTotal.Properties.Resources.Cancel;
            Data.mutex = new Mutex(true, "AutoTotalMutex", out bool isOnlyInstance);
            if (string.IsNullOrEmpty(AutoTotal.Properties.Settings.Default.VTKey)) {
                // Первый запуск
                if (!isOnlyInstance) {
                    System.Windows.MessageBox.Show(AutoTotal.Properties.Resources.FinishFirstSetup, "AutoTotal", MessageBoxButton.OK, MessageBoxImage.Error);
                    Environment.Exit(0);
                }
                new SetKeyWindow().ShowDialog();
                AutoTotal.Properties.Settings.Default.Folders = new() { Registry.GetValue(@"HKEY_CURRENT_USER\Software\Microsoft\Windows\CurrentVersion\Explorer\Shell Folders", "{374DE290-123F-4565-9164-39C4925E467B}", string.Empty)?.ToString() };
                AutoTotal.Properties.Settings.Default.Save();
                if (!Autorun.Exists()) Autorun.Add();
                Data.notificationManager.Show(new NotificationContent {
                    Title = "AutoTotal",
                    Message = AutoTotal.Properties.Resources.ProgramWillAutostart,
                    Type = NotificationType.Information,
                    TrimType = NotificationTextTrimType.NoTrim,
                    LeftButtonAction = () => ShowSettings(),
                    LeftButtonContent = AutoTotal.Properties.Resources.Settings
                }, expirationTime: TimeSpan.FromSeconds(10));
            }

            new Window {
                Width = 1,
                Height = 1,
                WindowStyle = WindowStyle.None,
                ShowInTaskbar = false,
                Visibility = Visibility.Hidden
            }.Show();

            // Сканирование из командной строки
            if (e.Args.Length == 2 && e.Args[0] == "/scan") Task.Run(async () => await Utils.ScanFile(e.Args[1], isOnlyInstance));
            if (isOnlyInstance) {
                bool dark;
                try {
                    dark = (int)(Registry.GetValue(@"HKEY_CURRENT_USER\Software\Microsoft\Windows\CurrentVersion\Themes\Personalize", "AppsUseLightTheme", 1) ?? 1) == 0;
                }
                catch {
                    dark = false;
                }
                NotifyIcon notifyIcon = new() {
                    Visible = true,
                    Icon = new Icon(AppDomain.CurrentDomain.BaseDirectory + "at.ico"),
                    Text = AutoTotal.Properties.Resources.WorkingInBackground,
                    ContextMenuStrip = new ContextMenuStrip {
                        Renderer = dark ? new DarkModeRenderer() : new ToolStripProfessionalRenderer(),
                        Items = {
                            new ToolStripMenuItem(AutoTotal.Properties.Resources.Settings, null, (s, e) => ShowSettings()),
                            new ToolStripMenuItem(AutoTotal.Properties.Resources.AddFolder, null, (s, e) => {
                                if (Data.settings?.IsVisible ?? false) {
                                    System.Windows.MessageBox.Show(AutoTotal.Properties.Resources.AddFolderForce, AutoTotal.Properties.Resources.AddingFolder, MessageBoxButton.OK, MessageBoxImage.Warning);
                                    return;
                                }
                                using var folderBrowserDialog = new FolderBrowserDialog();
                                folderBrowserDialog.SelectedPath = Registry.GetValue(@"HKEY_CURRENT_USER\Software\Microsoft\Windows\CurrentVersion\Explorer\Shell Folders", "{374DE290-123F-4565-9164-39C4925E467B}", string.Empty)!.ToString()!;
                                DialogResult result = folderBrowserDialog.ShowDialog();
                                if (result == DialogResult.OK && !string.IsNullOrWhiteSpace(folderBrowserDialog.SelectedPath)) {
                                    if (!AutoTotal.Properties.Settings.Default.Folders.Contains(folderBrowserDialog.SelectedPath)) {
                                        AutoTotal.Properties.Settings.Default.Folders.Add(folderBrowserDialog.SelectedPath);
                                        AutoTotal.Properties.Settings.Default.Save();
                                        FolderSpy.Add(folderBrowserDialog.SelectedPath);
                                    }
                                }
                            }),
                            new ToolStripMenuItem(AutoTotal.Properties.Resources.ScanFile, null, (s, e) => {
                                using var openFileDialog = new System.Windows.Forms.OpenFileDialog();
                                DialogResult result = openFileDialog.ShowDialog();
                                if (result == DialogResult.OK && !string.IsNullOrWhiteSpace(openFileDialog.FileName)) Task.Run(async () => await Utils.ScanFile(openFileDialog.FileName));
                            }),
                            new ToolStripMenuItem(AutoTotal.Properties.Resources.ChangeVTKey, null, (s, e) => new SetKeyWindow().Show()),
                            new ToolStripMenuItem(AutoTotal.Properties.Resources.Exit, null, (s, e) => Environment.Exit(1))
                        }
                    }
                };
                if (!e.Args.Contains("/autorun")) {
                    Data.notificationManager.Show(new NotificationContent {
                        Title = "AutoTotal",
                        Message = AutoTotal.Properties.Resources.Started,
                        Type = NotificationType.Notification,
                        TrimType = NotificationTextTrimType.NoTrim,
                        LeftButtonAction = () => ShowSettings(),
                        LeftButtonContent = AutoTotal.Properties.Resources.Settings,
                        Icon = new BitmapImage(new Uri("pack://application:,,,/res/virustotal.png")),
                    }, expirationTime: TimeSpan.FromSeconds(5));
                }
                StringCollection ToRemove = new();
                foreach (string? path in AutoTotal.Properties.Settings.Default.Folders) {
                    if (!Directory.Exists(path)) {
                        ToRemove.Add(path);
                        Data.notificationManager.Show(new NotificationContent {
                            Title = AutoTotal.Properties.Resources.DisappearedFolder,
                            Message = path,
                            Type = NotificationType.Information,
                            TrimType = NotificationTextTrimType.NoTrim,
                            LeftButtonAction = () => ShowSettings(),
                            LeftButtonContent = AutoTotal.Properties.Resources.Settings
                        }, expirationTime: TimeSpan.FromSeconds(10));
                    }
                    else FolderSpy.Add(path);
                };
                if (ToRemove.Count > 0) {
                    foreach (string? path in ToRemove) AutoTotal.Properties.Settings.Default.Folders.Remove(path);
                    AutoTotal.Properties.Settings.Default.Save();
                }
            }
        }

        private static void ShowSettings() {
            if (Data.settings == null) {
                Data.settings = new SettingsWindow();
                Data.settings.Closed += (sender, e) => Data.settings = null;
                Data.settings.Show();
            }
            else Data.settings.Focus();
        }
    }
}