mirror of
https://github.com/ppy/osu
synced 2024-12-13 18:37:04 +00:00
dd06c617b1
The structure here has changed a bit - rather than deriving from the donwload tracker I moved the additional logic into the callbacks. I think this feels better?
114 lines
4.5 KiB
C#
114 lines
4.5 KiB
C#
// Copyright (c) ppy Pty Ltd <contact@ppy.sh>. Licensed under the MIT Licence.
|
|
// See the LICENCE file in the repository root for full licence text.
|
|
|
|
using System;
|
|
using osu.Framework.Allocation;
|
|
using osu.Framework.Bindables;
|
|
using osu.Framework.Graphics.Containers;
|
|
using osu.Framework.Logging;
|
|
using osu.Framework.Threading;
|
|
using osu.Game.Beatmaps;
|
|
|
|
namespace osu.Game.Online.Rooms
|
|
{
|
|
/// <summary>
|
|
/// Represent a checksum-verifying beatmap availability tracker usable for online play screens.
|
|
///
|
|
/// This differs from a regular download tracking composite as this accounts for the
|
|
/// databased beatmap set's checksum, to disallow from playing with an altered version of the beatmap.
|
|
/// </summary>
|
|
public class OnlinePlayBeatmapAvailabilityTracker : CompositeDrawable
|
|
{
|
|
public readonly IBindable<PlaylistItem> SelectedItem = new Bindable<PlaylistItem>();
|
|
|
|
[Resolved]
|
|
private BeatmapManager beatmapManager { get; set; }
|
|
|
|
/// <summary>
|
|
/// The availability state of the currently selected playlist item.
|
|
/// </summary>
|
|
public IBindable<BeatmapAvailability> Availability => availability;
|
|
|
|
private readonly Bindable<BeatmapAvailability> availability = new Bindable<BeatmapAvailability>(BeatmapAvailability.LocallyAvailable());
|
|
|
|
private ScheduledDelegate progressUpdate;
|
|
|
|
private BeatmapDownloadTracker downloadTracker;
|
|
|
|
protected override void LoadComplete()
|
|
{
|
|
base.LoadComplete();
|
|
|
|
SelectedItem.BindValueChanged(item =>
|
|
{
|
|
// the underlying playlist is regularly cleared for maintenance purposes (things which probably need to be fixed eventually).
|
|
// to avoid exposing a state change when there may actually be none, ignore all nulls for now.
|
|
if (item.NewValue == null)
|
|
return;
|
|
|
|
downloadTracker?.Expire();
|
|
downloadTracker = new BeatmapDownloadTracker(item.NewValue.Beatmap.Value.BeatmapSet);
|
|
|
|
downloadTracker.Progress.BindValueChanged(_ =>
|
|
{
|
|
if (downloadTracker.State.Value != DownloadState.Downloading)
|
|
return;
|
|
|
|
// incoming progress changes are going to be at a very high rate.
|
|
// we don't want to flood the network with this, so rate limit how often we send progress updates.
|
|
if (progressUpdate?.Completed != false)
|
|
progressUpdate = Scheduler.AddDelayed(updateAvailability, progressUpdate == null ? 0 : 500);
|
|
});
|
|
|
|
downloadTracker.State.BindValueChanged(_ => updateAvailability(), true);
|
|
|
|
AddInternal(downloadTracker);
|
|
}, true);
|
|
}
|
|
|
|
private void updateAvailability()
|
|
{
|
|
if (downloadTracker == null)
|
|
return;
|
|
|
|
switch (downloadTracker.State.Value)
|
|
{
|
|
case DownloadState.NotDownloaded:
|
|
availability.Value = BeatmapAvailability.NotDownloaded();
|
|
break;
|
|
|
|
case DownloadState.Downloading:
|
|
availability.Value = BeatmapAvailability.Downloading((float)downloadTracker.Progress.Value);
|
|
break;
|
|
|
|
case DownloadState.Importing:
|
|
availability.Value = BeatmapAvailability.Importing();
|
|
break;
|
|
|
|
case DownloadState.LocallyAvailable:
|
|
bool hashMatches = checkHashValidity();
|
|
|
|
availability.Value = hashMatches ? BeatmapAvailability.LocallyAvailable() : BeatmapAvailability.NotDownloaded();
|
|
|
|
// only display a message to the user if a download seems to have just completed.
|
|
if (!hashMatches && downloadTracker.Progress.Value == 1)
|
|
Logger.Log("The imported beatmap set does not match the online version.", LoggingTarget.Runtime, LogLevel.Important);
|
|
|
|
break;
|
|
|
|
default:
|
|
throw new ArgumentOutOfRangeException();
|
|
}
|
|
}
|
|
|
|
private bool checkHashValidity()
|
|
{
|
|
int onlineId = SelectedItem.Value.Beatmap.Value.OnlineID;
|
|
string checksum = SelectedItem.Value.Beatmap.Value.MD5Hash;
|
|
|
|
var beatmap = beatmapManager.QueryBeatmap(b => b.OnlineBeatmapID == onlineId && b.MD5Hash == checksum);
|
|
return beatmap?.BeatmapSet.DeletePending == false;
|
|
}
|
|
}
|
|
}
|