2021-09-30 17:21:16 +08:00
|
|
|
// 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 System.Collections.Generic;
|
|
|
|
using System.Linq;
|
|
|
|
using System.Threading;
|
2022-01-07 23:40:14 +08:00
|
|
|
using Newtonsoft.Json;
|
2021-09-30 17:21:16 +08:00
|
|
|
using osu.Framework.Logging;
|
|
|
|
using osu.Framework.Platform;
|
|
|
|
using osu.Game.Beatmaps;
|
|
|
|
using osu.Game.Database;
|
|
|
|
using osu.Game.IO.Archives;
|
2021-12-14 18:47:11 +08:00
|
|
|
using osu.Game.Rulesets;
|
2021-09-30 17:21:16 +08:00
|
|
|
using osu.Game.Scoring.Legacy;
|
2021-12-06 21:47:00 +08:00
|
|
|
using Realms;
|
|
|
|
|
|
|
|
#nullable enable
|
2021-09-30 17:21:16 +08:00
|
|
|
|
|
|
|
namespace osu.Game.Scoring
|
|
|
|
{
|
2022-06-16 17:53:13 +08:00
|
|
|
public class ScoreImporter : RealmArchiveModelImporter<ScoreInfo>
|
2021-09-30 17:21:16 +08:00
|
|
|
{
|
|
|
|
public override IEnumerable<string> HandledExtensions => new[] { ".osr" };
|
|
|
|
|
|
|
|
protected override string[] HashableFileTypes => new[] { ".osr" };
|
|
|
|
|
2021-12-14 18:47:11 +08:00
|
|
|
private readonly RulesetStore rulesets;
|
2021-09-30 17:21:16 +08:00
|
|
|
private readonly Func<BeatmapManager> beatmaps;
|
|
|
|
|
2022-06-16 17:11:50 +08:00
|
|
|
public ScoreImporter(RulesetStore rulesets, Func<BeatmapManager> beatmaps, Storage storage, RealmAccess realm)
|
2022-01-24 18:59:58 +08:00
|
|
|
: base(storage, realm)
|
2021-09-30 17:21:16 +08:00
|
|
|
{
|
|
|
|
this.rulesets = rulesets;
|
|
|
|
this.beatmaps = beatmaps;
|
|
|
|
}
|
|
|
|
|
2021-12-06 21:47:00 +08:00
|
|
|
protected override ScoreInfo? CreateModel(ArchiveReader archive)
|
2021-09-30 17:21:16 +08:00
|
|
|
{
|
|
|
|
using (var stream = archive.GetStream(archive.Filenames.First(f => f.EndsWith(".osr", StringComparison.OrdinalIgnoreCase))))
|
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
|
|
|
return new DatabasedLegacyScoreDecoder(rulesets, beatmaps()).Parse(stream).ScoreInfo;
|
|
|
|
}
|
|
|
|
catch (LegacyScoreDecoder.BeatmapNotFoundException e)
|
|
|
|
{
|
|
|
|
Logger.Log(e.Message, LoggingTarget.Information, LogLevel.Error);
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
public Score GetScore(ScoreInfo score) => new LegacyDatabasedScore(score, rulesets, beatmaps(), Files.Store);
|
|
|
|
|
2022-01-13 15:27:07 +08:00
|
|
|
protected override void Populate(ScoreInfo model, ArchiveReader? archive, Realm realm, CancellationToken cancellationToken = default)
|
2022-01-07 16:27:48 +08:00
|
|
|
{
|
|
|
|
// Ensure the beatmap is not detached.
|
2022-01-12 17:05:25 +08:00
|
|
|
if (!model.BeatmapInfo.IsManaged)
|
|
|
|
model.BeatmapInfo = realm.Find<BeatmapInfo>(model.BeatmapInfo.ID);
|
2022-01-07 16:27:48 +08:00
|
|
|
|
|
|
|
if (!model.Ruleset.IsManaged)
|
|
|
|
model.Ruleset = realm.Find<RulesetInfo>(model.Ruleset.ShortName);
|
|
|
|
|
2022-01-17 13:02:15 +08:00
|
|
|
// These properties are known to be non-null, but these final checks ensure a null hasn't come from somewhere (or the refetch has failed).
|
|
|
|
// Under no circumstance do we want these to be written to realm as null.
|
|
|
|
if (model.BeatmapInfo == null) throw new ArgumentNullException(nameof(model.BeatmapInfo));
|
|
|
|
if (model.Ruleset == null) throw new ArgumentNullException(nameof(model.Ruleset));
|
|
|
|
|
2022-01-07 23:40:14 +08:00
|
|
|
if (string.IsNullOrEmpty(model.StatisticsJson))
|
|
|
|
model.StatisticsJson = JsonConvert.SerializeObject(model.Statistics);
|
2022-01-07 16:27:48 +08:00
|
|
|
}
|
2021-09-30 17:21:16 +08:00
|
|
|
}
|
|
|
|
}
|