050e157005
* clean up a bunch of R&D code * don't store components * brug * speedrun some sloth review * technology disks * expand functionality, begin work on printer * disk printer ui * file * fix the rebase * disk console is finito * Update DiskConsoleSystem.cs
95 lines
3.6 KiB
C#
95 lines
3.6 KiB
C#
using System.Linq;
|
|
using Content.Server.Popups;
|
|
using Content.Server.Research.Components;
|
|
using Content.Server.Research.Systems;
|
|
using Content.Server.Research.TechnologyDisk.Components;
|
|
using Content.Shared.Examine;
|
|
using Content.Shared.Interaction;
|
|
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(database, recipe, false);
|
|
}
|
|
Dirty(database);
|
|
}
|
|
_popup.PopupEntity(Loc.GetString("tech-disk-inserted"), target, args.User);
|
|
EntityManager.DeleteEntity(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;
|
|
|
|
//get a list of every distinct recipe in all the technologies.
|
|
var allTechs = new List<string>();
|
|
foreach (var tech in _prototype.EnumeratePrototypes<TechnologyPrototype>())
|
|
{
|
|
allTechs.AddRange(tech.UnlockedRecipes);
|
|
}
|
|
allTechs = allTechs.Distinct().ToList();
|
|
|
|
//get a list of every distinct unlocked tech across all databases
|
|
var allUnlocked = new List<string>();
|
|
foreach (var database in EntityQuery<TechnologyDatabaseComponent>())
|
|
{
|
|
allUnlocked.AddRange(database.RecipeIds);
|
|
}
|
|
allUnlocked = allUnlocked.Distinct().ToList();
|
|
|
|
//make a list of every single non-unlocked tech
|
|
var validTechs = allTechs.Where(tech => !allUnlocked.Contains(tech)).ToList();
|
|
|
|
//pick one
|
|
component.Recipes = new();
|
|
component.Recipes.Add(_random.Pick(validTechs));
|
|
}
|
|
}
|