using System; using System.Collections.Generic; using System.Diagnostics; using System.IO; using System.Linq; using System.Text; using System.Threading.Tasks; using Tango.Core.Commands; using Tango.Core.Helpers; using Tango.Explorer; using Tango.PPC.Common; using Tango.PPC.Common.MachineUpdate; using Tango.PPC.UI.Dialogs; using Tango.PPC.UI.ViewsContracts; namespace Tango.PPC.UI.ViewModels { public class MachineUpdateViewVM : PPCViewModel { public enum MachineUpdateView { UpdateCheckView, UpdateCheckErrorView, UpdateAvailableView, UpToDateView, UpdateProgressView, UpdateDbProgressView, UpdateCompletedView, UpdateFailedView, UpdateFromPackageView, UpdateFailedFromPackageView, } private MachineUpdateResult _update_result; private DbCompareResult _db_compare_result; #region Properties /// /// Gets or sets the machine update manager. /// public IMachineUpdateManager MachineUpdateManager { get; set; } private String _latestVersion; /// /// Gets or sets the latest version. /// public String LatestVersion { get { return _latestVersion; } set { _latestVersion = value; RaisePropertyChangedAuto(); } } private bool _isDbUpdate; /// /// Gets or sets a value indicating whether this instance is database update. /// public bool IsDbUpdate { get { return _isDbUpdate; } set { _isDbUpdate = value; RaisePropertyChangedAuto(); } } private String _failedError; /// /// Gets or sets the setup failed error. /// public String FailedError { get { return _failedError; } set { _failedError = value; RaisePropertyChangedAuto(); } } #endregion #region Commands /// /// Gets or sets the complete command. /// public RelayCommand CompleteCommand { get; set; } /// /// Gets or sets the install command. /// public RelayCommand UpdateCommand { get; set; } /// /// Gets or sets the restart command. /// public RelayCommand RestartCommand { get; set; } /// /// Gets or sets the close command. /// public RelayCommand CloseCommand { get; set; } /// /// Gets or sets to application command. /// public RelayCommand ToApplicationCommand { get; set; } #endregion #region Constructors public MachineUpdateViewVM(IMachineUpdateManager machineUpdateManager) { MachineUpdateManager = machineUpdateManager; CompleteCommand = new RelayCommand(CompleteUpdate); UpdateCommand = new RelayCommand(Update); RestartCommand = new RelayCommand(CheckForUpdates); CloseCommand = new RelayCommand(() => { NavigationManager.NavigateTo(Common.Navigation.NavigationView.HomeModule); NavigateTo(MachineUpdateView.UpdateCheckView); }); ToApplicationCommand = new RelayCommand(() => { NavigationManager.NavigateTo(Common.Navigation.NavigationView.HomeModule); NavigateTo(MachineUpdateView.UpdateCheckView); }); } #endregion #region Update public async void CheckForUpdates() { await NavigateTo(MachineUpdateView.UpdateCheckView); try { IsDbUpdate = false; var response = await MachineUpdateManager.CheckForUpdate(MachineProvider.Machine.SerialNumber); if (response.IsUpdateAvailable) { LatestVersion = response.Version; await NavigateTo(MachineUpdateView.UpdateAvailableView); } else { _db_compare_result = await MachineUpdateManager.UpdateDBCheck(MachineProvider.Machine.SerialNumber); if (_db_compare_result.RequiresUpdate) { IsDbUpdate = true; await NavigateTo(MachineUpdateView.UpdateAvailableView); } else { await NavigateTo(MachineUpdateView.UpToDateView); } } } catch (Exception ex) { FailedError = ex.FlattenMessage(); LogManager.Log(ex, "Error while trying to check for updates."); await NavigateTo(MachineUpdateView.UpdateFailedView); } } private async void Update() { if (!IsDbUpdate) { await NavigateTo(MachineUpdateView.UpdateProgressView); LogManager.Log("Starting machine update..."); try { _update_result = await MachineUpdateManager.Update(MachineProvider.Machine.SerialNumber); LogManager.Log("Machine update completed."); await NavigateTo(MachineUpdateView.UpdateCompletedView); } catch (Exception ex) { FailedError = ex.FlattenMessage(); LogManager.Log(ex, "Machine update failed."); await NavigateTo(MachineUpdateView.UpdateFailedView); } } else { await NavigateTo(MachineUpdateView.UpdateDbProgressView); LogManager.Log("Starting database update..."); try { await MachineUpdateManager.UpdateDB(_db_compare_result); LogManager.Log("Database update completed."); await NavigateTo(MachineUpdateView.UpdateCompletedView); } catch (Exception ex) { FailedError = ex.FlattenMessage(); LogManager.Log(ex, "Database update failed."); await NavigateTo(MachineUpdateView.UpdateFailedView); } } } #endregion #region Complete private void CompleteUpdate() { if (!IsDbUpdate) { String updater_exe = Path.Combine(_update_result.UpdatePackagePath, "Tango.PPC.Updater.exe"); LogManager.Log("Completing machine setup..."); LogManager.Log($"Executing '{updater_exe}' with arguments '{PathHelper.GetStartupPath()}'..."); Process.Start(updater_exe, PathHelper.GetStartupPath()); LogManager.Log("Terminating application process!"); ApplicationManager.ShutDown(); } else { LogManager.Log("Restarting Application..."); ApplicationManager.Restart(); } } #endregion #region Override Methods /// /// Called when the application has been started. /// public override void OnApplicationStarted() { } /// /// Navigates to the specified view. /// /// The view. private Task NavigateTo(MachineUpdateView view) { return View.NavigateTo(view); } /// /// Called when the application is ready and all modules views are loaded. /// public override void OnApplicationReady() { base.OnApplicationReady(); StorageProvider.RegisterFileHandler(ExplorerFileDefinition.Update.Extension, HandleSoftwareUpdatePackageLoaded); } #endregion #region Handle USB Update private async void HandleSoftwareUpdatePackageLoaded(ExplorerFileItem fileItem) { UpdatePackageFile packageFile = null; try { packageFile = await MachineUpdateManager.GetUpdatePackageFileInfo(fileItem.Path); } catch (Exception ex) { LogManager.Log(ex, $"Error loading update package file from {fileItem.Path}."); await NotificationProvider.ShowError("An error occurred while trying to load the selected software update package. Please make sure the package is valid."); return; } if (ApplicationManager.Version <= packageFile.Version) { await NotificationProvider.ShowError($"The selected update package (v{packageFile.Version.ToString()}) contains an older software version."); return; } UpdateFromFileViewVM vm = new UpdateFromFileViewVM(); vm.Version = packageFile.Version.ToString(); await NotificationProvider.ShowDialog(vm); if (vm.DialogResult) { await NavigationManager.NavigateTo(Common.Navigation.NavigationView.MachineUpdateView); await NavigateTo(MachineUpdateView.UpdateFromPackageView); LogManager.Log("Starting machine update from package..."); try { _update_result = await MachineUpdateManager.UpdateFromTUP(fileItem.Path); LogManager.Log("Machine update from package completed."); await NavigateTo(MachineUpdateView.UpdateCompletedView); } catch (Exception ex) { LogManager.Log(ex, "Machine update from package failed."); await NavigateTo(MachineUpdateView.UpdateFailedFromPackageView); } } } #endregion } }