mirror of https://github.com/ppy/osu
89 lines
3.7 KiB
C#
89 lines
3.7 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 System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Linq.Expressions;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using osu.Framework.Logging;
|
|
using osu.Framework.Platform;
|
|
using osu.Game.Beatmaps;
|
|
using osu.Game.Database;
|
|
using osu.Game.IO.Archives;
|
|
using osu.Game.Rulesets;
|
|
using osu.Game.Scoring.Legacy;
|
|
|
|
namespace osu.Game.Scoring
|
|
{
|
|
public class ScoreModelManager : ArchiveModelManager<ScoreInfo, ScoreFileInfo>
|
|
{
|
|
public override IEnumerable<string> HandledExtensions => new[] { ".osr" };
|
|
|
|
protected override string[] HashableFileTypes => new[] { ".osr" };
|
|
|
|
protected override string ImportFromStablePath => Path.Combine("Data", "r");
|
|
|
|
private readonly RulesetStore rulesets;
|
|
private readonly Func<BeatmapManager> beatmaps;
|
|
|
|
public ScoreModelManager(RulesetStore rulesets, Func<BeatmapManager> beatmaps, Storage storage, IDatabaseContextFactory contextFactory, IIpcHost importHost = null)
|
|
: base(storage, contextFactory, new ScoreStore(contextFactory, storage), importHost)
|
|
{
|
|
this.rulesets = rulesets;
|
|
this.beatmaps = beatmaps;
|
|
}
|
|
|
|
protected override ScoreInfo CreateModel(ArchiveReader archive)
|
|
{
|
|
if (archive == null)
|
|
return null;
|
|
|
|
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);
|
|
|
|
public List<ScoreInfo> GetAllUsableScores() => ModelStore.ConsumableItems.Where(s => !s.DeletePending).ToList();
|
|
|
|
public IEnumerable<ScoreInfo> QueryScores(Expression<Func<ScoreInfo, bool>> query) => ModelStore.ConsumableItems.AsNoTracking().Where(query);
|
|
|
|
public ScoreInfo Query(Expression<Func<ScoreInfo, bool>> query) => ModelStore.ConsumableItems.AsNoTracking().FirstOrDefault(query);
|
|
|
|
protected override Task Populate(ScoreInfo model, ArchiveReader archive, CancellationToken cancellationToken = default)
|
|
=> Task.CompletedTask;
|
|
|
|
protected override bool CheckLocalAvailability(ScoreInfo model, IQueryable<ScoreInfo> items)
|
|
=> base.CheckLocalAvailability(model, items)
|
|
|| (model.OnlineScoreID != null && items.Any(i => i.OnlineScoreID == model.OnlineScoreID));
|
|
|
|
public override void ExportModelTo(ScoreInfo model, Stream outputStream)
|
|
{
|
|
var file = model.Files.SingleOrDefault();
|
|
if (file == null)
|
|
return;
|
|
|
|
using (var inputStream = Files.Storage.GetStream(file.FileInfo.StoragePath))
|
|
inputStream.CopyTo(outputStream);
|
|
}
|
|
|
|
protected override IEnumerable<string> GetStableImportPaths(Storage storage)
|
|
=> storage.GetFiles(ImportFromStablePath).Where(p => HandledExtensions.Any(ext => Path.GetExtension(p)?.Equals(ext, StringComparison.OrdinalIgnoreCase) ?? false))
|
|
.Select(path => storage.GetFullPath(path));
|
|
}
|
|
}
|