Abstract Assets replication methods

Neakita · Sep 6, 2024 · 8fdcb2f · 8fdcb2f
1 parent 7f842d2
commit 8fdcb2f
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/SightKeeper.Data/Binary/Model/DataSets/PackableDataSet.cs b/SightKeeper.Data/Binary/Model/DataSets/PackableDataSet.cs
@@ -39,6 +39,7 @@ public PackableDataSet(
 	}
 
 	public abstract ImmutableArray<PackableTag> GetTags();
+	public abstract ImmutableArray<PackableAsset> GetAssets();
 }
 
 /// <summary>
@@ -73,4 +74,9 @@ public sealed override ImmutableArray<PackableTag> GetTags()
 	{
 		return ImmutableArray<PackableTag>.CastUp(Tags);
 	}
+
+	public sealed override ImmutableArray<PackableAsset> GetAssets()
+	{
+		return ImmutableArray<PackableAsset>.CastUp(Assets);
+	}
 }
diff --git a/SightKeeper.Data/Binary/Replication/DataSetReplicator.cs b/SightKeeper.Data/Binary/Replication/DataSetReplicator.cs
@@ -1,18 +1,22 @@
 using System.Collections.Immutable;
 using CommunityToolkit.Diagnostics;
 using SightKeeper.Data.Binary.Model.DataSets;
+using SightKeeper.Data.Binary.Model.DataSets.Assets;
 using SightKeeper.Data.Binary.Model.DataSets.Compositions;
 using SightKeeper.Data.Binary.Model.DataSets.Tags;
 using SightKeeper.Data.Binary.Services;
 using SightKeeper.Domain.Model;
 using SightKeeper.Domain.Model.DataSets;
+using SightKeeper.Domain.Model.DataSets.Assets;
 using SightKeeper.Domain.Model.DataSets.Screenshots;
 using SightKeeper.Domain.Model.DataSets.Tags;
 
 namespace SightKeeper.Data.Binary.Replication;
 
 internal abstract class DataSetReplicator
 {
+	protected delegate Tag TagGetter(byte TagId, byte? KeyPointTagId = null);
+
 	public DataSetReplicator(FileSystemScreenshotsDataAccess screenshotsDataAccess)
 	{
 		_screenshotsDataAccess = screenshotsDataAccess;
@@ -26,6 +30,7 @@ public DataSet Replicate(PackableDataSet packed, ReplicationSession session)
 		var dataSet = CreateDataSet(packed.Name, packed.Description, game, composition);
 		ReplicateScreenshots(dataSet.Screenshots, packed.Screenshots);
 		var tagsLookup = ReplicateTags(dataSet.Tags, packed.GetTags());
+		ReplicateAssets(dataSet.Assets, packed.GetAssets(), (tagId, keyPointTagId) => tagsLookup[(tagId, keyPointTagId)]);
 		return dataSet;
 	}
 
@@ -43,6 +48,8 @@ protected virtual Tag ReplicateTag(
 		return tag;
 	}
 
+	protected abstract void ReplicateAsset(AssetsLibrary library, PackableAsset asset, TagGetter getTag);
+
 	private readonly FileSystemScreenshotsDataAccess _screenshotsDataAccess;
 
 	private static Composition ReplicateComposition(PackableComposition? composition)
@@ -74,4 +81,10 @@ private void ReplicateScreenshots(ScreenshotsLibrary library, ImmutableArray<Pac
 			ReplicateTag(library, tag, lookupBuilder);
 		return lookupBuilder.ToImmutable();
 	}
+
+	private void ReplicateAssets(AssetsLibrary library, ImmutableArray<PackableAsset> assets, TagGetter getTag)
+	{
+		foreach (var asset in assets)
+			ReplicateAsset(library, asset, getTag);
+	}
 }