mirror of
https://github.com/ppy/osu
synced 2024-12-09 08:30:36 +00:00
142 lines
4.5 KiB
C#
142 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.Game.Extensions;
|
|
using osu.Game.Online.API;
|
|
using osu.Game.Scoring;
|
|
|
|
#nullable enable
|
|
|
|
namespace osu.Game.Online
|
|
{
|
|
public class ScoreDownloadTracker : DownloadTracker<ScoreInfo>
|
|
{
|
|
[Resolved(CanBeNull = true)]
|
|
protected ScoreManager? Manager { get; private set; }
|
|
|
|
[Resolved(CanBeNull = true)]
|
|
protected ScoreModelDownloader? Downloader { get; private set; }
|
|
|
|
private ArchiveDownloadRequest<IScoreInfo>? attachedRequest;
|
|
|
|
public ScoreDownloadTracker(ScoreInfo trackedItem)
|
|
: base(trackedItem)
|
|
{
|
|
}
|
|
|
|
[BackgroundDependencyLoader(true)]
|
|
private void load()
|
|
{
|
|
if (Manager == null || Downloader == null)
|
|
return;
|
|
|
|
// Used to interact with manager classes that don't support interface types. Will eventually be replaced.
|
|
var scoreInfo = new ScoreInfo
|
|
{
|
|
ID = TrackedItem.ID,
|
|
OnlineID = TrackedItem.OnlineID
|
|
};
|
|
|
|
if (Manager.IsAvailableLocally(scoreInfo))
|
|
UpdateState(DownloadState.LocallyAvailable);
|
|
else
|
|
attachDownload(Downloader.GetExistingDownload(scoreInfo));
|
|
|
|
Downloader.DownloadBegan += downloadBegan;
|
|
Downloader.DownloadFailed += downloadFailed;
|
|
Manager.ItemUpdated += itemUpdated;
|
|
Manager.ItemRemoved += itemRemoved;
|
|
}
|
|
|
|
private void downloadBegan(ArchiveDownloadRequest<IScoreInfo> request) => Schedule(() =>
|
|
{
|
|
if (checkEquality(request.Model, TrackedItem))
|
|
attachDownload(request);
|
|
});
|
|
|
|
private void downloadFailed(ArchiveDownloadRequest<IScoreInfo> request) => Schedule(() =>
|
|
{
|
|
if (checkEquality(request.Model, TrackedItem))
|
|
attachDownload(null);
|
|
});
|
|
|
|
private void attachDownload(ArchiveDownloadRequest<IScoreInfo>? request)
|
|
{
|
|
if (attachedRequest != null)
|
|
{
|
|
attachedRequest.Failure -= onRequestFailure;
|
|
attachedRequest.DownloadProgressed -= onRequestProgress;
|
|
attachedRequest.Success -= onRequestSuccess;
|
|
}
|
|
|
|
attachedRequest = request;
|
|
|
|
if (attachedRequest != null)
|
|
{
|
|
if (attachedRequest.Progress == 1)
|
|
{
|
|
UpdateProgress(1);
|
|
UpdateState(DownloadState.Importing);
|
|
}
|
|
else
|
|
{
|
|
UpdateProgress(attachedRequest.Progress);
|
|
UpdateState(DownloadState.Downloading);
|
|
|
|
attachedRequest.Failure += onRequestFailure;
|
|
attachedRequest.DownloadProgressed += onRequestProgress;
|
|
attachedRequest.Success += onRequestSuccess;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
UpdateState(DownloadState.NotDownloaded);
|
|
}
|
|
}
|
|
|
|
private void onRequestSuccess(string _) => Schedule(() => UpdateState(DownloadState.Importing));
|
|
|
|
private void onRequestProgress(float progress) => Schedule(() => UpdateProgress(progress));
|
|
|
|
private void onRequestFailure(Exception e) => Schedule(() => attachDownload(null));
|
|
|
|
private void itemUpdated(ScoreInfo item) => Schedule(() =>
|
|
{
|
|
if (checkEquality(item, TrackedItem))
|
|
UpdateState(DownloadState.LocallyAvailable);
|
|
});
|
|
|
|
private void itemRemoved(ScoreInfo item) => Schedule(() =>
|
|
{
|
|
if (checkEquality(item, TrackedItem))
|
|
UpdateState(DownloadState.NotDownloaded);
|
|
});
|
|
|
|
private bool checkEquality(IScoreInfo x, IScoreInfo y) => x.MatchesOnlineID(y);
|
|
|
|
#region Disposal
|
|
|
|
protected override void Dispose(bool isDisposing)
|
|
{
|
|
base.Dispose(isDisposing);
|
|
attachDownload(null);
|
|
|
|
if (Downloader != null)
|
|
{
|
|
Downloader.DownloadBegan -= downloadBegan;
|
|
Downloader.DownloadFailed -= downloadFailed;
|
|
}
|
|
|
|
if (Manager != null)
|
|
{
|
|
Manager.ItemUpdated -= itemUpdated;
|
|
Manager.ItemRemoved -= itemRemoved;
|
|
}
|
|
}
|
|
|
|
#endregion
|
|
}
|
|
}
|