ECS and bandaid research (#9251)
This commit is contained in:
@@ -1,73 +1,7 @@
|
||||
using Content.Shared.Research.Components;
|
||||
using Content.Shared.Research.Prototypes;
|
||||
|
||||
namespace Content.Server.Research.Components
|
||||
{
|
||||
[RegisterComponent]
|
||||
public sealed class TechnologyDatabaseComponent : SharedTechnologyDatabaseComponent
|
||||
{
|
||||
public override ComponentState GetComponentState()
|
||||
{
|
||||
return new TechnologyDatabaseState(_technologies);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Synchronizes this database against other,
|
||||
/// adding all technologies from the other that
|
||||
/// this one doesn't have.
|
||||
/// </summary>
|
||||
/// <param name="otherDatabase">The other database</param>
|
||||
/// <param name="twoway">Whether the other database should be synced against this one too or not.</param>
|
||||
public void Sync(TechnologyDatabaseComponent otherDatabase, bool twoway = true)
|
||||
{
|
||||
foreach (var tech in otherDatabase.Technologies)
|
||||
{
|
||||
if (!IsTechnologyUnlocked(tech)) AddTechnology(tech);
|
||||
}
|
||||
|
||||
if (twoway)
|
||||
otherDatabase.Sync(this, false);
|
||||
|
||||
Dirty();
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// If there's a research client component attached to the owner entity,
|
||||
/// and the research client is connected to a research server, this method
|
||||
/// syncs against the research server, and the server against the local database.
|
||||
/// </summary>
|
||||
/// <returns>Whether it could sync or not</returns>
|
||||
public bool SyncWithServer()
|
||||
{
|
||||
if (!IoCManager.Resolve<IEntityManager>().TryGetComponent(Owner, out ResearchClientComponent? client)) return false;
|
||||
if (client.Server?.Database == null) return false;
|
||||
|
||||
Sync(client.Server.Database);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// If possible, unlocks a technology on this database.
|
||||
/// </summary>
|
||||
/// <param name="technology"></param>
|
||||
/// <returns></returns>
|
||||
public bool UnlockTechnology(TechnologyPrototype technology)
|
||||
{
|
||||
if (!CanUnlockTechnology(technology)) return false;
|
||||
|
||||
AddTechnology(technology);
|
||||
Dirty();
|
||||
return true;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Adds a technology to the database without checking if it could be unlocked.
|
||||
/// </summary>
|
||||
/// <param name="technology"></param>
|
||||
public void AddTechnology(TechnologyPrototype technology)
|
||||
{
|
||||
_technologies.Add(technology);
|
||||
}
|
||||
}
|
||||
public sealed class TechnologyDatabaseComponent : SharedTechnologyDatabaseComponent {}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user