93 lines
3.3 KiB
C#
93 lines
3.3 KiB
C#
using System.Linq;
|
|
using Content.Server.Popups;
|
|
using Content.Server.Research.Systems;
|
|
using Content.Server.Research.TechnologyDisk.Components;
|
|
using Content.Shared.Examine;
|
|
using Content.Shared.Interaction;
|
|
using Content.Shared.Random;
|
|
using Content.Shared.Random.Helpers;
|
|
using Content.Shared.Research.Components;
|
|
using Content.Shared.Research.Prototypes;
|
|
using Robust.Shared.Prototypes;
|
|
using Robust.Shared.Random;
|
|
|
|
namespace Content.Server.Research.TechnologyDisk.Systems;
|
|
|
|
public sealed class TechnologyDiskSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly PopupSystem _popup = default!;
|
|
[Dependency] private readonly ResearchSystem _research = default!;
|
|
[Dependency] private readonly IPrototypeManager _prototype = default!;
|
|
[Dependency] private readonly IRobustRandom _random = default!;
|
|
|
|
/// <inheritdoc/>
|
|
public override void Initialize()
|
|
{
|
|
SubscribeLocalEvent<TechnologyDiskComponent, AfterInteractEvent>(OnAfterInteract);
|
|
SubscribeLocalEvent<TechnologyDiskComponent, ExaminedEvent>(OnExamine);
|
|
SubscribeLocalEvent<TechnologyDiskComponent, MapInitEvent>(OnMapInit);
|
|
}
|
|
|
|
private void OnAfterInteract(EntityUid uid, TechnologyDiskComponent component, AfterInteractEvent args)
|
|
{
|
|
if (args.Handled || !args.CanReach || args.Target is not { } target)
|
|
return;
|
|
|
|
if (!HasComp<ResearchServerComponent>(target) || !TryComp<TechnologyDatabaseComponent>(target, out var database))
|
|
return;
|
|
|
|
if (component.Recipes != null)
|
|
{
|
|
foreach (var recipe in component.Recipes)
|
|
{
|
|
_research.AddLatheRecipe(target, recipe, database);
|
|
}
|
|
}
|
|
_popup.PopupEntity(Loc.GetString("tech-disk-inserted"), target, args.User);
|
|
QueueDel(uid);
|
|
args.Handled = true;
|
|
}
|
|
|
|
private void OnExamine(EntityUid uid, TechnologyDiskComponent component, ExaminedEvent args)
|
|
{
|
|
var message = Loc.GetString("tech-disk-examine-none");
|
|
if (component.Recipes != null && component.Recipes.Any())
|
|
{
|
|
var prototype = _prototype.Index<LatheRecipePrototype>(component.Recipes[0]);
|
|
var resultPrototype = _prototype.Index<EntityPrototype>(prototype.Result);
|
|
message = Loc.GetString("tech-disk-examine", ("result", resultPrototype.Name));
|
|
|
|
if (component.Recipes.Count > 1) //idk how to do this well. sue me.
|
|
message += " " + Loc.GetString("tech-disk-examine-more");
|
|
}
|
|
args.PushMarkup(message);
|
|
}
|
|
|
|
private void OnMapInit(EntityUid uid, TechnologyDiskComponent component, MapInitEvent args)
|
|
{
|
|
if (component.Recipes != null)
|
|
return;
|
|
|
|
var weightedRandom = _prototype.Index<WeightedRandomPrototype>(component.TierWeightPrototype);
|
|
var tier = int.Parse(weightedRandom.Pick(_random));
|
|
|
|
//get a list of every distinct recipe in all the technologies.
|
|
var techs = new List<string>();
|
|
foreach (var tech in _prototype.EnumeratePrototypes<TechnologyPrototype>())
|
|
{
|
|
if (tech.Tier != tier)
|
|
continue;
|
|
|
|
techs.AddRange(tech.RecipeUnlocks);
|
|
}
|
|
techs = techs.Distinct().ToList();
|
|
|
|
if (!techs.Any())
|
|
return;
|
|
|
|
//pick one
|
|
component.Recipes = new();
|
|
component.Recipes.Add(_random.Pick(techs));
|
|
}
|
|
}
|