2019-06-28 19:37:53 +08:00
|
|
|
// Copyright (c) ppy Pty Ltd <contact@ppy.sh>. Licensed under the MIT Licence.
|
2019-01-24 16:43:03 +08:00
|
|
|
// See the LICENCE file in the repository root for full licence text.
|
2018-04-13 17:19:50 +08:00
|
|
|
|
|
|
|
using System;
|
2018-07-06 15:39:27 +08:00
|
|
|
using System.Threading.Tasks;
|
2018-04-13 17:19:50 +08:00
|
|
|
using osu.Framework.Allocation;
|
|
|
|
using osu.Framework.Logging;
|
|
|
|
using osu.Game;
|
|
|
|
using osu.Game.Overlays;
|
|
|
|
using osu.Game.Overlays.Notifications;
|
2023-05-23 17:06:04 +08:00
|
|
|
using osu.Game.Screens.Play;
|
2024-07-05 06:00:45 +08:00
|
|
|
using Velopack;
|
2024-07-05 05:28:49 +08:00
|
|
|
using Velopack.Sources;
|
2018-04-13 17:19:50 +08:00
|
|
|
|
2018-07-06 15:39:27 +08:00
|
|
|
namespace osu.Desktop.Updater
|
2018-04-13 17:19:50 +08:00
|
|
|
{
|
2024-08-31 22:03:10 +08:00
|
|
|
public partial class VelopackUpdateManager : Game.Updater.UpdateManager
|
2018-04-13 17:19:50 +08:00
|
|
|
{
|
2024-07-05 06:00:45 +08:00
|
|
|
private readonly UpdateManager updateManager;
|
2022-08-02 22:23:54 +08:00
|
|
|
private INotificationOverlay notificationOverlay = null!;
|
2018-04-13 17:19:50 +08:00
|
|
|
|
2022-09-12 13:52:11 +08:00
|
|
|
[Resolved]
|
|
|
|
private OsuGameBase game { get; set; } = null!;
|
|
|
|
|
2023-05-23 17:06:04 +08:00
|
|
|
[Resolved]
|
|
|
|
private ILocalUserPlayInfo? localUserInfo { get; set; }
|
|
|
|
|
2024-09-03 18:11:34 +08:00
|
|
|
private UpdateInfo? pendingUpdate;
|
|
|
|
|
2024-08-31 22:03:10 +08:00
|
|
|
public VelopackUpdateManager()
|
2024-07-05 05:28:49 +08:00
|
|
|
{
|
2024-09-02 13:23:05 +08:00
|
|
|
updateManager = new UpdateManager(new GithubSource(@"https://github.com/ppy/osu", null, false), new UpdateOptions
|
2024-07-05 06:00:45 +08:00
|
|
|
{
|
2024-09-03 18:11:34 +08:00
|
|
|
AllowVersionDowngrade = true,
|
2024-07-05 06:00:45 +08:00
|
|
|
});
|
2024-07-05 05:28:49 +08:00
|
|
|
}
|
|
|
|
|
2018-04-13 17:19:50 +08:00
|
|
|
[BackgroundDependencyLoader]
|
2022-04-19 05:18:10 +08:00
|
|
|
private void load(INotificationOverlay notifications)
|
2018-04-13 17:19:50 +08:00
|
|
|
{
|
2022-04-19 05:18:10 +08:00
|
|
|
notificationOverlay = notifications;
|
2018-04-13 17:19:50 +08:00
|
|
|
}
|
|
|
|
|
2021-03-08 11:57:16 +08:00
|
|
|
protected override async Task<bool> PerformUpdateCheck() => await checkForUpdateAsync().ConfigureAwait(false);
|
2020-05-07 14:07:22 +08:00
|
|
|
|
2024-06-27 03:25:41 +08:00
|
|
|
private async Task<bool> checkForUpdateAsync(UpdateProgressNotification? notification = null)
|
2018-04-13 17:19:50 +08:00
|
|
|
{
|
2024-09-08 22:07:17 +08:00
|
|
|
// whether to check again in 30 minutes. generally only if there's an error or no update was found (yet).
|
|
|
|
bool scheduleRecheck = false;
|
2018-04-13 17:19:50 +08:00
|
|
|
|
|
|
|
try
|
|
|
|
{
|
2023-05-23 17:06:04 +08:00
|
|
|
// Avoid any kind of update checking while gameplay is running.
|
|
|
|
if (localUserInfo?.IsPlaying.Value == true)
|
2024-09-08 22:07:17 +08:00
|
|
|
{
|
|
|
|
scheduleRecheck = true;
|
2023-05-23 17:06:04 +08:00
|
|
|
return false;
|
2024-09-08 22:07:17 +08:00
|
|
|
}
|
2023-05-23 17:06:04 +08:00
|
|
|
|
2024-09-04 18:41:40 +08:00
|
|
|
// TODO: we should probably be checking if there's a more recent update, rather than shortcutting here.
|
|
|
|
// Velopack does support this scenario (see https://github.com/ppy/osu/pull/28743#discussion_r1743495975).
|
2024-09-03 18:11:34 +08:00
|
|
|
if (pendingUpdate != null)
|
2020-10-06 12:18:42 +08:00
|
|
|
{
|
2024-07-05 05:28:49 +08:00
|
|
|
// If there is an update pending restart, show the notification to restart again.
|
|
|
|
notificationOverlay.Post(new UpdateApplicationCompleteNotification
|
2020-10-06 12:18:42 +08:00
|
|
|
{
|
2024-07-05 05:28:49 +08:00
|
|
|
Activated = () =>
|
2022-09-12 13:52:11 +08:00
|
|
|
{
|
2024-09-26 00:11:23 +08:00
|
|
|
Task.Run(restartToApplyUpdate);
|
2024-07-05 05:28:49 +08:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
});
|
2024-09-08 22:07:17 +08:00
|
|
|
|
2024-07-05 05:28:49 +08:00
|
|
|
return true;
|
2020-10-06 12:18:42 +08:00
|
|
|
}
|
2018-04-13 17:19:50 +08:00
|
|
|
|
2024-09-03 18:11:34 +08:00
|
|
|
pendingUpdate = await updateManager.CheckForUpdatesAsync().ConfigureAwait(false);
|
|
|
|
|
2024-09-08 22:07:17 +08:00
|
|
|
// No update is available. We'll check again later.
|
2024-09-03 18:11:34 +08:00
|
|
|
if (pendingUpdate == null)
|
2024-09-08 22:07:17 +08:00
|
|
|
{
|
|
|
|
scheduleRecheck = true;
|
2024-09-03 18:11:34 +08:00
|
|
|
return false;
|
2024-09-08 22:07:17 +08:00
|
|
|
}
|
2024-09-03 18:11:34 +08:00
|
|
|
|
2024-09-08 22:07:17 +08:00
|
|
|
// An update is found, let's notify the user and start downloading it.
|
2018-04-13 17:19:50 +08:00
|
|
|
if (notification == null)
|
|
|
|
{
|
2022-09-12 13:52:11 +08:00
|
|
|
notification = new UpdateProgressNotification
|
|
|
|
{
|
2024-09-26 00:11:23 +08:00
|
|
|
CompletionClickAction = () =>
|
|
|
|
{
|
|
|
|
Task.Run(restartToApplyUpdate);
|
|
|
|
return true;
|
|
|
|
},
|
2022-09-12 13:52:11 +08:00
|
|
|
};
|
|
|
|
|
2018-04-13 17:19:50 +08:00
|
|
|
Schedule(() => notificationOverlay.Post(notification));
|
|
|
|
}
|
|
|
|
|
2022-09-12 13:52:11 +08:00
|
|
|
notification.StartDownload();
|
2018-04-13 17:19:50 +08:00
|
|
|
|
|
|
|
try
|
|
|
|
{
|
2024-09-03 18:11:34 +08:00
|
|
|
await updateManager.DownloadUpdatesAsync(pendingUpdate, p => notification.Progress = p / 100f).ConfigureAwait(false);
|
2018-04-13 17:19:50 +08:00
|
|
|
|
|
|
|
notification.State = ProgressNotificationState.Completed;
|
|
|
|
}
|
|
|
|
catch (Exception e)
|
|
|
|
{
|
2024-07-05 05:45:34 +08:00
|
|
|
// In the case of an error, a separate notification will be displayed.
|
2024-09-09 00:58:09 +08:00
|
|
|
scheduleRecheck = true;
|
2024-07-05 05:45:34 +08:00
|
|
|
notification.FailDownload();
|
|
|
|
Logger.Error(e, @"update failed!");
|
2018-04-13 17:19:50 +08:00
|
|
|
}
|
|
|
|
}
|
2024-07-05 05:28:49 +08:00
|
|
|
catch (Exception e)
|
2018-04-13 17:19:50 +08:00
|
|
|
{
|
|
|
|
// we'll ignore this and retry later. can be triggered by no internet connection or thread abortion.
|
2021-07-12 13:55:09 +08:00
|
|
|
scheduleRecheck = true;
|
2024-09-03 18:16:10 +08:00
|
|
|
Logger.Log($@"update check failed ({e.Message})");
|
2018-04-13 17:19:50 +08:00
|
|
|
}
|
|
|
|
finally
|
|
|
|
{
|
2019-06-28 19:37:53 +08:00
|
|
|
if (scheduleRecheck)
|
2018-04-13 17:19:50 +08:00
|
|
|
{
|
2021-07-02 13:43:48 +08:00
|
|
|
Scheduler.AddDelayed(() => Task.Run(async () => await checkForUpdateAsync().ConfigureAwait(false)), 60000 * 30);
|
2018-04-13 17:19:50 +08:00
|
|
|
}
|
|
|
|
}
|
2020-10-06 12:00:02 +08:00
|
|
|
|
|
|
|
return true;
|
2018-04-13 17:19:50 +08:00
|
|
|
}
|
|
|
|
|
2024-09-26 00:11:23 +08:00
|
|
|
private async Task restartToApplyUpdate()
|
2020-10-06 12:18:42 +08:00
|
|
|
{
|
2024-09-26 00:11:23 +08:00
|
|
|
await updateManager.WaitExitThenApplyUpdatesAsync(pendingUpdate?.TargetFullRelease).ConfigureAwait(false);
|
2024-06-27 03:25:41 +08:00
|
|
|
Schedule(() => game.AttemptExit());
|
2018-11-19 20:29:29 +08:00
|
|
|
}
|
2018-04-13 17:19:50 +08:00
|
|
|
}
|
|
|
|
}
|