Add backup and restore

2026-01-30 22:49:47 +05:00 · 2024-09-07 14:52:33 +08:00
parent d5f1cc99ac
commit beddc71ed8
17 changed files with 889 additions and 14 deletions
--- a/v2rayN/ServiceLib/ViewModels/BackupAndRestoreViewModel.cs
+++ b/v2rayN/ServiceLib/ViewModels/BackupAndRestoreViewModel.cs
@@ -0,0 +1,151 @@
+using ReactiveUI;
+using ReactiveUI.Fody.Helpers;
+using Splat;
+using System.Reactive;
+
+namespace ServiceLib.ViewModels
+{
+    public class BackupAndRestoreViewModel : MyReactiveObject
+    {
+        public ReactiveCommand<Unit, Unit> RemoteBackupCmd { get; }
+        public ReactiveCommand<Unit, Unit> RemoteRestoreCmd { get; }
+        public ReactiveCommand<Unit, Unit> WebDavCheckCmd { get; }
+
+        [Reactive]
+        public WebDavItem SelectedSource { get; set; }
+
+        [Reactive]
+        public string OperationMsg { get; set; }
+
+        public BackupAndRestoreViewModel(Func<EViewAction, object?, Task<bool>>? updateView)
+        {
+            _config = LazyConfig.Instance.Config;
+            _updateView = updateView;
+            _noticeHandler = Locator.Current.GetService<NoticeHandler>();
+
+            WebDavCheckCmd = ReactiveCommand.CreateFromTask(async () =>
+            {
+                await WebDavCheck();
+            });
+
+            RemoteBackupCmd = ReactiveCommand.CreateFromTask(async () =>
+            {
+                await RemoteBackup();
+            });
+            RemoteRestoreCmd = ReactiveCommand.CreateFromTask(async () =>
+            {
+                await RemoteRestore();
+            });
+
+            SelectedSource = JsonUtils.DeepCopy(_config.webDavItem);
+        }
+
+        private void DisplayOperationMsg(string msg = "")
+        {
+            OperationMsg = msg;
+        }
+
+        private async Task WebDavCheck()
+        {
+            DisplayOperationMsg();
+            _config.webDavItem = SelectedSource;
+            ConfigHandler.SaveConfig(_config);
+
+            var result = await WebDavHandler.Instance.CheckConnection();
+            if (result)
+            {
+                DisplayOperationMsg(ResUI.OperationSuccess);
+            }
+            else
+            {
+                DisplayOperationMsg(WebDavHandler.Instance.GetLastError());
+            }
+        }
+
+        private async Task RemoteBackup()
+        {
+            DisplayOperationMsg();
+            var fileName = Utils.GetBackupPath($"backup_{DateTime.Now:yyyyMMddHHmmss}.zip");
+            var result = await CreateZipFileFromDirectory(fileName);
+            if (result)
+            {
+                var result2 = await WebDavHandler.Instance.PutFile(fileName);
+                if (result2)
+                {
+                    DisplayOperationMsg(ResUI.OperationSuccess);
+                    return;
+                }
+            }
+
+            DisplayOperationMsg(WebDavHandler.Instance.GetLastError());
+        }
+
+        private async Task RemoteRestore()
+        {
+            DisplayOperationMsg();
+            var fileName = Utils.GetTempPath(Utils.GetGUID());
+            var result = await WebDavHandler.Instance.GetRawFile(fileName);
+            if (result)
+            {
+                await LocalRestore(fileName);
+                return;
+            }
+
+            DisplayOperationMsg(WebDavHandler.Instance.GetLastError());
+        }
+
+        public async Task<bool> LocalBackup(string fileName)
+        {
+            DisplayOperationMsg();
+            var result = await CreateZipFileFromDirectory(fileName);
+            if (result)
+            {
+                DisplayOperationMsg(ResUI.OperationSuccess);
+            }
+            else
+            {
+                DisplayOperationMsg(WebDavHandler.Instance.GetLastError());
+            }
+
+            return result;
+        }
+
+        public async Task LocalRestore(string fileName)
+        {
+            DisplayOperationMsg();
+            if (Utils.IsNullOrEmpty(fileName))
+            {
+                return;
+            }
+
+            //backup first
+            var fileBackup = Utils.GetBackupPath($"backup_{DateTime.Now:yyyyMMddHHmmss}.zip");
+            var result = await CreateZipFileFromDirectory(fileBackup);
+            if (result)
+            {
+                Locator.Current.GetService<MainWindowViewModel>()?.V2rayUpgrade(fileName);
+            }
+            else
+            {
+                DisplayOperationMsg(WebDavHandler.Instance.GetLastError());
+            }
+        }
+
+        private async Task<bool> CreateZipFileFromDirectory(string fileName)
+        {
+            if (Utils.IsNullOrEmpty(fileName))
+            {
+                return false;
+            }
+
+            var configDir = Utils.GetConfigPath();
+            var configDirZipTemp = Utils.GetTempPath($"v2rayN_{DateTime.Now:yyyyMMddHHmmss}");
+            var configDirTemp = Path.Combine(configDirZipTemp, "guiConfigs");
+
+            await Task.Run(() => FileManager.CopyDirectory(configDir, configDirTemp, true, "cache.db"));
+            var ret = await Task.Run(() => FileManager.CreateFromDirectory(configDirZipTemp, fileName));
+            await Task.Run(() => Directory.Delete(configDirZipTemp, true));
+            return ret;
+        }
+    }
+}