1
0
mirror of https://github.com/ppy/osu.git synced 2024-12-16 17:02:55 +08:00
osu-lazer/osu.Game/Rulesets/RulesetStore.cs

125 lines
4.1 KiB
C#
Raw Normal View History

2017-04-17 16:43:48 +08:00
// Copyright (c) 2007-2017 ppy Pty Ltd <contact@ppy.sh>.
// Licensed under the MIT Licence - https://raw.githubusercontent.com/ppy/osu/master/LICENCE
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Reflection;
using osu.Game.Database;
2017-04-17 16:43:48 +08:00
namespace osu.Game.Rulesets
2017-04-17 16:43:48 +08:00
{
/// <summary>
/// Todo: All of this needs to be moved to a RulesetStore.
2017-04-17 16:43:48 +08:00
/// </summary>
public class RulesetStore : DatabaseBackedStore
2017-04-17 16:43:48 +08:00
{
2017-09-19 16:19:37 +08:00
private static readonly Dictionary<Assembly, Type> loaded_assemblies = new Dictionary<Assembly, Type>();
2017-09-19 16:19:37 +08:00
static RulesetStore()
2017-04-17 16:43:48 +08:00
{
2017-09-19 16:19:37 +08:00
AppDomain.CurrentDomain.AssemblyResolve += currentDomain_AssemblyResolve;
2017-04-17 18:44:03 +08:00
foreach (string file in Directory.GetFiles(Environment.CurrentDirectory, $"{ruleset_library_prefix}.*.dll"))
loadRulesetFromFile(file);
2017-09-19 16:19:37 +08:00
}
public RulesetStore(Func<OsuDbContext> factory)
: base(factory)
2017-10-16 16:02:31 +08:00
{
AddMissingRulesets();
2017-10-16 16:02:31 +08:00
}
/// <summary>
/// Retrieve a ruleset using a known ID.
/// </summary>
/// <param name="id">The ruleset's internal ID.</param>
/// <returns>A ruleset, if available, else null.</returns>
public RulesetInfo GetRuleset(int id) => AvailableRulesets.FirstOrDefault(r => r.ID == id);
/// <summary>
/// All available rulesets.
/// </summary>
public IEnumerable<RulesetInfo> AvailableRulesets;
2017-10-16 16:02:31 +08:00
2017-09-19 16:19:37 +08:00
private static Assembly currentDomain_AssemblyResolve(object sender, ResolveEventArgs args) => loaded_assemblies.Keys.FirstOrDefault(a => a.FullName == args.Name);
2017-04-17 16:43:48 +08:00
2017-09-19 16:19:37 +08:00
private const string ruleset_library_prefix = "osu.Game.Rulesets";
2017-04-17 16:43:48 +08:00
protected void AddMissingRulesets()
2017-09-19 16:19:37 +08:00
{
2017-10-17 14:50:42 +08:00
var context = GetContext();
2017-10-16 16:02:31 +08:00
var instances = loaded_assemblies.Values.Select(r => (Ruleset)Activator.CreateInstance(r, new RulesetInfo())).ToList();
2017-09-19 16:19:37 +08:00
2017-10-05 03:52:12 +08:00
//add all legacy modes in correct order
foreach (var r in instances.Where(r => r.LegacyID >= 0).OrderBy(r => r.LegacyID))
2017-04-17 16:43:48 +08:00
{
2017-10-05 03:52:12 +08:00
var rulesetInfo = createRulesetInfo(r);
2017-10-17 14:50:42 +08:00
if (context.RulesetInfo.SingleOrDefault(rsi => rsi.ID == rulesetInfo.ID) == null)
2017-04-17 16:43:48 +08:00
{
2017-10-17 14:50:42 +08:00
context.RulesetInfo.Add(rulesetInfo);
2017-04-17 16:43:48 +08:00
}
2017-10-05 03:52:12 +08:00
}
2017-10-16 16:02:31 +08:00
2017-10-17 14:50:42 +08:00
context.SaveChanges();
2017-09-19 16:19:37 +08:00
2017-10-05 03:52:12 +08:00
//add any other modes
foreach (var r in instances.Where(r => r.LegacyID < 0))
{
var us = createRulesetInfo(r);
2017-04-17 16:43:48 +08:00
2017-10-17 14:50:42 +08:00
var existing = context.RulesetInfo.FirstOrDefault(ri => ri.InstantiationInfo == us.InstantiationInfo);
2017-09-19 16:19:37 +08:00
2017-10-05 03:52:12 +08:00
if (existing == null)
2017-10-17 14:50:42 +08:00
context.RulesetInfo.Add(us);
2017-10-05 03:52:12 +08:00
}
2017-10-16 16:02:31 +08:00
2017-10-17 14:50:42 +08:00
context.SaveChanges();
2017-04-17 16:43:48 +08:00
2017-10-05 03:52:12 +08:00
//perform a consistency check
2017-10-17 14:50:42 +08:00
foreach (var r in context.RulesetInfo)
2017-09-19 16:19:37 +08:00
{
2017-10-05 03:52:12 +08:00
try
{
r.CreateInstance();
r.Available = true;
}
catch
2017-09-19 16:19:37 +08:00
{
2017-10-05 03:52:12 +08:00
r.Available = false;
2017-09-19 16:19:37 +08:00
}
2017-10-05 03:52:12 +08:00
}
2017-10-16 16:02:31 +08:00
2017-10-17 14:50:42 +08:00
context.SaveChanges();
AvailableRulesets = context.RulesetInfo.Where(r => r.Available).ToList();
2017-04-17 16:43:48 +08:00
}
2017-09-19 16:19:37 +08:00
private static void loadRulesetFromFile(string file)
{
var filename = Path.GetFileNameWithoutExtension(file);
2017-09-19 16:19:37 +08:00
if (loaded_assemblies.Values.Any(t => t.Namespace == filename))
return;
try
{
2017-09-19 16:19:37 +08:00
var assembly = Assembly.LoadFrom(file);
loaded_assemblies[assembly] = assembly.GetTypes().First(t => t.IsSubclassOf(typeof(Ruleset)));
}
2017-10-16 12:11:35 +08:00
catch (Exception)
{
}
}
2017-04-17 16:43:48 +08:00
private RulesetInfo createRulesetInfo(Ruleset ruleset) => new RulesetInfo
{
Name = ruleset.Description,
InstantiationInfo = ruleset.GetType().AssemblyQualifiedName,
2017-10-14 13:28:25 +08:00
ID = ruleset.LegacyID
2017-04-17 16:43:48 +08:00
};
}
}