baseline: OpenMU upstream b5a0961 (fresh source)

src/Persistence/AccountDataSource.cs

@@ -0,0 +1,58 @@
+// <copyright file="AccountDataSource.cs" company="MUnique">
+// Licensed under the MIT License. See LICENSE file in the project root for full license information.
+// </copyright>
+
+namespace MUnique.OpenMU.Persistence;
+
+using System.Collections;
+using Microsoft.Extensions.Logging;
+using MUnique.OpenMU.AttributeSystem;
+using MUnique.OpenMU.DataModel.Configuration;
+using MUnique.OpenMU.DataModel.Entities;
+using MUnique.OpenMU.Interfaces;
+
+/// <summary>
+/// Implementation of a <see cref="IDataSource{Account}"/>.
+/// </summary>
+public sealed class AccountDataSource : DataSourceBase<Account>
+{
+    private static readonly IReadOnlyDictionary<Type, Func<Account, IEnumerable>> Enumerables = new Dictionary<Type, Func<Account, IEnumerable>>
+    {
+        { typeof(Account), a => Enumerable.Repeat(a, 1) },
+        { typeof(Character), a => a.Characters },
+        { typeof(ItemStorage), a => a.Characters.Select(c => c.Inventory).Concat(Enumerable.Repeat(a.Vault, 1)).Where(i => i is not null) },
+        { typeof(Item), a => a.Characters.SelectMany(c => c.Inventory?.Items ?? Enumerable.Empty<Item>()).Concat(a.Vault?.Items ?? Enumerable.Empty<Item>()) },
+        { typeof(StatAttribute), a => a.Characters.SelectMany(c => c.Attributes) },
+        { typeof(LetterHeader), a => a.Characters.SelectMany(c => c.Letters) },
+        { typeof(SkillEntry), a => a.Characters.SelectMany(c => c.LearnedSkills) },
+        { typeof(CharacterQuestState), a => a.Characters.SelectMany(c => c.QuestStates) },
+        { typeof(QuestMonsterKillRequirementState), a => a.Characters.SelectMany(c => c.QuestStates).SelectMany(q => q.RequirementStates) },
+    }.AsReadOnly();
+
+    private readonly IDataSource<GameConfiguration> _gameConfigurationSource;
+
+    /// <summary>
+    /// Initializes a new instance of the <see cref="AccountDataSource"/> class.
+    /// </summary>
+    /// <param name="logger">The logger.</param>
+    /// <param name="persistenceContextProvider">The persistence context provider.</param>
+    /// <param name="gameConfigurationSource">The game configuration source.</param>
+    public AccountDataSource(
+        ILogger<DataSourceBase<Account>> logger,
+        IPersistenceContextProvider persistenceContextProvider,
+        IDataSource<GameConfiguration> gameConfigurationSource)
+        : base(logger, persistenceContextProvider)
+    {
+        this._gameConfigurationSource = gameConfigurationSource;
+    }
+
+    /// <inheritdoc />
+    protected override IReadOnlyDictionary<Type, Func<Account, IEnumerable>> TypeToEnumerables => Enumerables;
+
+    /// <inheritdoc />
+    protected override async ValueTask<IContext> CreateNewContextAsync()
+    {
+        var gameConfiguration = await this._gameConfigurationSource.GetOwnerAsync(Guid.Empty).ConfigureAwait(false);
+        return this.ContextProvider.CreateNewPlayerContext(gameConfiguration);
+    }
+}