AutoTotal/SettingsWindow.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
using Microsoft.Win32;
using System.Windows;
using System.Windows.Controls;
using MessageBox = System.Windows.MessageBox;
namespace AutoTotal {
    public partial class SettingsWindow : Window {
        public SettingsWindow() {
            InitializeComponent();
            foreach (string path in Properties.Settings.Default.Folders?.OfType<string>() ?? Enumerable.Empty<string>()) Folders.Items.Add(new TextBlock { Text = path });
            BlockCheckbox.IsChecked = Properties.Settings.Default.BlockFiles;
            AutoRunCheckbox.IsChecked = Autorun.Exists();
            ExplorerItemCheckBox.IsChecked = Registry.CurrentUser.OpenSubKey(@"Software\Classes\*\shell\AutoTotal") != null;
        }

        private void AddFolder(object sender, RoutedEventArgs e) {
            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 == System.Windows.Forms.DialogResult.OK && !string.IsNullOrWhiteSpace(folderBrowserDialog.SelectedPath)) {
                if (!Properties.Settings.Default.Folders.Contains(folderBrowserDialog.SelectedPath)) {
                    Properties.Settings.Default.Folders.Add(folderBrowserDialog.SelectedPath);
                    Properties.Settings.Default.Save();
                    FolderSpy.Add(folderBrowserDialog.SelectedPath);
                    Folders.Items.Add(new TextBlock() { Text = folderBrowserDialog.SelectedPath });
                }
            }
        }

        private void DelFolder(object sender, RoutedEventArgs e) {
            FolderSpy.Remove((Folders.SelectedItem as TextBlock)?.Text);
            Properties.Settings.Default.Folders.Remove((Folders.SelectedItem as TextBlock)?.Text);
            Properties.Settings.Default.Save();
            Folders.Items.Remove(Folders.SelectedItem);
        }

        private void Save(object sender, RoutedEventArgs e) {
            Properties.Settings.Default.BlockFiles = BlockCheckbox?.IsChecked ?? false;
            if (AutoRunCheckbox.IsChecked ?? false) {
                if (!Autorun.Exists()) Autorun.Add();
            }
            else if (Autorun.Exists()) Autorun.Remove();
            if (ExplorerItemCheckBox.IsChecked ?? false) {
                try {
                    string exePath = System.Reflection.Assembly.GetExecutingAssembly().Location;
                    if (exePath.EndsWith(".dll", System.StringComparison.OrdinalIgnoreCase)) {
                        exePath = exePath.Substring(0, exePath.Length - 4) + ".exe";
                    }
                    using (RegistryKey key = Registry.CurrentUser.CreateSubKey(@"Software\Classes\*\shell\AutoTotal")) {
                        if (key != null) {
                            key.SetValue("", AutoTotal.Properties.Resources.ScanFileOnVT);
                            key.SetValue("Icon", exePath);
                        }
                    }
                    using (RegistryKey commandKey = Registry.CurrentUser.CreateSubKey(@"Software\Classes\*\shell\AutoTotal\command")) {
                        if (commandKey != null) {
                            commandKey.SetValue("", $"\"{exePath}\" /scan \"%1\"");
                        }
                    }
                }
                catch (System.Exception ex) {
                    MessageBox.Show(ex.Message, ex.GetType().Name, MessageBoxButton.OK, MessageBoxImage.Error);
                }
            }
            else {
                try {
                    Registry.CurrentUser.DeleteSubKeyTree(@"Software\Classes\*\shell\AutoTotal", false);
                }
                catch (System.Exception ex) {
                    MessageBox.Show(ex.Message, ex.GetType().Name, MessageBoxButton.OK, MessageBoxImage.Error);
                }
            }
            Properties.Settings.Default.Save();
            Close();
        }

        private void ChangeVTKey(object sender, RoutedEventArgs e) {
            new SetKeyWindow().Show();
        }
    }
}