v2rayN/v2rayN/ServiceLib/ViewModels/CheckUpdateViewModel.cs

using DynamicData;
using DynamicData.Binding;
using ReactiveUI;
using ReactiveUI.Fody.Helpers;
using Splat;
using System.Reactive;

namespace ServiceLib.ViewModels
{
    public class CheckUpdateViewModel : MyReactiveObject
    {
        private const string _geo = "GeoFiles";
        private string _v2rayN = ECoreType.v2rayN.ToString();
        private List<CheckUpdateItem> _lstUpdated = [];

        private IObservableCollection<CheckUpdateItem> _checkUpdateItem = new ObservableCollectionExtended<CheckUpdateItem>();
        public IObservableCollection<CheckUpdateItem> CheckUpdateItems => _checkUpdateItem;
        public ReactiveCommand<Unit, Unit> CheckUpdateCmd { get; }
        [Reactive] public bool EnableCheckPreReleaseUpdate { get; set; }

        public CheckUpdateViewModel(Func<EViewAction, object?, Task<bool>>? updateView)
        {
            _config = AppHandler.Instance.Config;
            _updateView = updateView;

            CheckUpdateCmd = ReactiveCommand.CreateFromTask(async () =>
            {
                await CheckUpdate();
            });

            EnableCheckPreReleaseUpdate = _config.GuiItem.CheckPreReleaseUpdate;

            this.WhenAnyValue(
            x => x.EnableCheckPreReleaseUpdate,
            y => y == true)
                .Subscribe(c => { _config.GuiItem.CheckPreReleaseUpdate = EnableCheckPreReleaseUpdate; });

            RefreshSubItems();
        }

        private void RefreshSubItems()
        {
            _checkUpdateItem.Clear();

            _checkUpdateItem.Add(new CheckUpdateItem()
            {
                IsSelected = false,
                CoreType = _v2rayN,
                Remarks = ResUI.menuCheckUpdate,
            });
            _checkUpdateItem.Add(new CheckUpdateItem()
            {
                IsSelected = true,
                CoreType = ECoreType.Xray.ToString(),
                Remarks = ResUI.menuCheckUpdate,
            });
            _checkUpdateItem.Add(new CheckUpdateItem()
            {
                IsSelected = true,
                CoreType = ECoreType.mihomo.ToString(),
                Remarks = ResUI.menuCheckUpdate,
            });
            _checkUpdateItem.Add(new CheckUpdateItem()
            {
                IsSelected = true,
                CoreType = ECoreType.sing_box.ToString(),
                Remarks = ResUI.menuCheckUpdate,
            });
            _checkUpdateItem.Add(new CheckUpdateItem()
            {
                IsSelected = true,
                CoreType = _geo,
                Remarks = ResUI.menuCheckUpdate,
            });
        }

        private async Task CheckUpdate()
        {
            _lstUpdated.Clear();
            _lstUpdated = _checkUpdateItem.Where(x => x.IsSelected == true)
                    .Select(x => new CheckUpdateItem() { CoreType = x.CoreType }).ToList();

            for (var k = _checkUpdateItem.Count - 1; k >= 0; k--)
            {
                var item = _checkUpdateItem[k];
                if (item.IsSelected != true) continue;

                UpdateView(item.CoreType, "...");
                if (item.CoreType == _geo)
                {
                    await CheckUpdateGeo();
                }
                else if (item.CoreType == _v2rayN)
                {
                    await CheckUpdateN(EnableCheckPreReleaseUpdate);
                }
                else if (item.CoreType == ECoreType.mihomo.ToString())
                {
                    await CheckUpdateCore(item, false);
                }
                else
                {
                    await CheckUpdateCore(item, EnableCheckPreReleaseUpdate);
                }
            }

            await UpdateFinished();
        }

        private void UpdatedPlusPlus(string coreType, string fileName)
        {
            var item = _lstUpdated.FirstOrDefault(x => x.CoreType == coreType);
            if (item == null)
            {
                return;
            }
            item.IsFinished = true;
            if (!fileName.IsNullOrEmpty())
            {
                item.FileName = fileName;
            }
        }

        private async Task CheckUpdateGeo()
        {
            void _updateUI(bool success, string msg)
            {
                UpdateView(_geo, msg);
                if (success)
                {
                    UpdatedPlusPlus(_geo, "");
                }
            }
            await (new UpdateService()).UpdateGeoFileAll(_config, _updateUI)
                .ContinueWith(t =>
                {
                    UpdatedPlusPlus(_geo, "");
                });
        }

        private async Task CheckUpdateN(bool preRelease)
        {
            void _updateUI(bool success, string msg)
            {
                UpdateView(_v2rayN, msg);
                if (success)
                {
                    UpdateView(_v2rayN, ResUI.OperationSuccess);
                    UpdatedPlusPlus(_v2rayN, msg);
                }
            }
            await (new UpdateService()).CheckUpdateGuiN(_config, _updateUI, preRelease)
                .ContinueWith(t =>
                {
                    UpdatedPlusPlus(_v2rayN, "");
                });
        }

        private async Task CheckUpdateCore(CheckUpdateItem item, bool preRelease)
        {
            void _updateUI(bool success, string msg)
            {
                UpdateView(item.CoreType, msg);
                if (success)
                {
                    UpdateView(item.CoreType, ResUI.MsgUpdateV2rayCoreSuccessfullyMore);

                    UpdatedPlusPlus(item.CoreType, msg);
                }
            }
            var type = (ECoreType)Enum.Parse(typeof(ECoreType), item.CoreType);
            await (new UpdateService()).CheckUpdateCore(type, _config, _updateUI, preRelease)
                .ContinueWith(t =>
                {
                    UpdatedPlusPlus(item.CoreType, "");
                });
        }

        private async Task UpdateFinished()
        {
            if (_lstUpdated.Count > 0 && _lstUpdated.Count(x => x.IsFinished == true) == _lstUpdated.Count)
            {
                _updateView?.Invoke(EViewAction.DispatcherCheckUpdateFinished, false);
                await Task.Delay(2000);
                await UpgradeCore();

                if (_lstUpdated.Any(x => x.CoreType == _v2rayN && x.IsFinished == true))
                {
                    await Task.Delay(1000);
                    UpgradeN();
                }
                await Task.Delay(1000);
                _updateView?.Invoke(EViewAction.DispatcherCheckUpdateFinished, true);
            }
        }

        public void UpdateFinishedResult(bool blReload)
        {
            if (blReload)
            {
                Locator.Current.GetService<MainWindowViewModel>()?.Reload();
            }
            else
            {
                Locator.Current.GetService<MainWindowViewModel>()?.CloseCore();
            }
        }

        private void UpgradeN()
        {
            try
            {
                var fileName = _lstUpdated.FirstOrDefault(x => x.CoreType == _v2rayN)?.FileName;
                if (fileName.IsNullOrEmpty())
                {
                    return;
                }
                if (!Utils.UpgradeAppExists(out _))
                {
                    UpdateView(_v2rayN, ResUI.UpgradeAppNotExistTip);
                    return;
                }
                Locator.Current.GetService<MainWindowViewModel>()?.UpgradeApp(fileName);
            }
            catch (Exception ex)
            {
                UpdateView(_v2rayN, ex.Message);
            }
        }

        private async Task UpgradeCore()
        {
            foreach (var item in _lstUpdated)
            {
                if (item.FileName.IsNullOrEmpty())
                {
                    continue;
                }

                var fileName = item.FileName;
                if (!File.Exists(fileName))
                {
                    continue;
                }
                var toPath = Utils.GetBinPath("", item.CoreType);

                if (fileName.Contains(".tar.gz"))
                {
                    FileManager.DecompressTarFile(fileName, toPath);
                    var dir = new DirectoryInfo(toPath);
                    if (dir.Exists)
                    {
                        foreach (var subDir in dir.GetDirectories())
                        {
                            FileManager.CopyDirectory(subDir.FullName, toPath, false, null);
                            subDir.Delete(true);
                        }
                    }
                }
                else if (fileName.Contains(".gz"))
                {
                    FileManager.DecompressFile(fileName, toPath, item.CoreType);
                }
                else
                {
                    FileManager.ZipExtractToFile(fileName, toPath, _config.GuiItem.IgnoreGeoUpdateCore ? "geo" : "");
                }

                if (Utils.IsLinux())
                {
                    var filesList = (new DirectoryInfo(toPath)).GetFiles().Select(u => u.FullName).ToList();
                    foreach (var file in filesList)
                    {
                        await Utils.SetLinuxChmod(Path.Combine(toPath, item.CoreType));
                    }
                }

                UpdateView(item.CoreType, ResUI.MsgUpdateV2rayCoreSuccessfully);

                if (File.Exists(fileName))
                {
                    File.Delete(fileName);
                }
            }
        }

        private void UpdateView(string coreType, string msg)
        {
            var item = new CheckUpdateItem()
            {
                CoreType = coreType,
                Remarks = msg,
            };
            _updateView?.Invoke(EViewAction.DispatcherCheckUpdate, item);
        }

        public void UpdateViewResult(CheckUpdateItem item)
        {
            var found = _checkUpdateItem.FirstOrDefault(t => t.CoreType == item.CoreType);
            if (found == null) return;
            var itemCopy = JsonUtils.DeepCopy(found);
            itemCopy.Remarks = item.Remarks;
            _checkUpdateItem.Replace(found, itemCopy);
        }
    }
}