mirror of
https://github.com/misternebula/quantum-space-buddies.git
synced 2025-12-11 20:15:10 +01:00
90 lines
3.2 KiB
C#
90 lines
3.2 KiB
C#
using Cysharp.Threading.Tasks;
|
|
using HarmonyLib;
|
|
using OWML.Common;
|
|
using QSB.ItemSync.WorldObjects;
|
|
using QSB.ItemSync.WorldObjects.Items;
|
|
using QSB.ItemSync.WorldObjects.Sockets;
|
|
using QSB.Utility;
|
|
using QSB.Utility.Deterministic;
|
|
using QSB.WorldSync;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading;
|
|
using UnityEngine;
|
|
|
|
namespace QSB.ItemSync;
|
|
|
|
public class ItemManager : WorldObjectManager
|
|
{
|
|
public override WorldObjectScene WorldObjectScene => WorldObjectScene.Both;
|
|
|
|
public override async UniTask BuildWorldObjects(OWScene scene, CancellationToken ct)
|
|
{
|
|
DebugLog.DebugWrite("Building OWItems...", MessageType.Info);
|
|
|
|
// Items
|
|
QSBWorldSync.Init<QSBSlideReelItem, SlideReelItem>();
|
|
QSBWorldSync.Init<QSBWarpCoreItem, WarpCoreItem>();
|
|
// dream lantern and vision torch are set up in their own managers
|
|
// the rest can just use the generic thing below because they dont have special things using them
|
|
|
|
// Use the basic QSBItem class for any items that do not require custom code through a derived class (mod compatibility)
|
|
// QSB addons can still define their own QSBItem derived classes and they will just get skipped here
|
|
var handledItemTypes = new HashSet<Type>(GetHandledItemTypes()); // set cuz we do Contains below
|
|
DebugLog.DebugWrite($"Handled OWItem types (the rest will get generic QSBItem support) are: {handledItemTypes.Join()}");
|
|
var otherItemlistToInitFrom = QSBWorldSync.GetUnityObjects<OWItem>()
|
|
.Where(x => !handledItemTypes.Contains(x.GetType()))
|
|
.SortDeterministic();
|
|
// could make a subclass for this but i dont care, and would have to filter out from reflection thing below
|
|
QSBWorldSync.Init<QSBItem<OWItem>, OWItem>(otherItemlistToInitFrom);
|
|
|
|
// Sockets
|
|
QSBWorldSync.Init<QSBItemSocket, OWItemSocket>();
|
|
|
|
// other drop targets that don't already have world objects
|
|
// breaks if mod adds custom subclass of IItemDropTarget and then makes WorldObject of that but can just fix that when that happens
|
|
var listToInitFrom = QSBWorldSync.GetUnityObjects<MonoBehaviour>()
|
|
.Where(x => x is IItemDropTarget and not (RaftDock or RaftController or PrisonCellElevator))
|
|
.SortDeterministic();
|
|
QSBWorldSync.Init<QSBOtherDropTarget, MonoBehaviour>(listToInitFrom);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets all types that extend QSBItem and returns the list of OWItem types that are already handled by dedicated classes
|
|
/// </summary>
|
|
private static IEnumerable<Type> GetHandledItemTypes()
|
|
{
|
|
var assemblies = QSBCore.Addons.Values
|
|
.Select(x => x.GetType().Assembly)
|
|
.Append(typeof(QSBCore).Assembly);
|
|
|
|
if (QSBCore.QSBNHAssembly != null)
|
|
{
|
|
assemblies = assemblies.Append(QSBCore.QSBNHAssembly);
|
|
}
|
|
|
|
// If the class inherits from QSBItem<T>, this will return what T is else null
|
|
static Type GetTypeFromQSBItem(Type type)
|
|
{
|
|
if (type.IsInterface || type.IsAbstract || type.BaseType == null)
|
|
{
|
|
return null;
|
|
}
|
|
if (type.BaseType.IsGenericType && type.BaseType.GetGenericTypeDefinition() == typeof(QSBItem<>))
|
|
{
|
|
return type.BaseType.GetGenericArguments()[0];
|
|
}
|
|
else
|
|
{
|
|
return GetTypeFromQSBItem(type.BaseType);
|
|
}
|
|
}
|
|
|
|
return assemblies.SelectMany(x => x.GetTypes())
|
|
.Select(GetTypeFromQSBItem)
|
|
.Where(x => x != null)
|
|
.OrderBy(x => x.FullName);
|
|
}
|
|
}
|