Technology Disks (#13077)

* 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
2022-12-20 17:39:57 -05:00
parent 88ed188b42
commit 050e157005
34 changed files with 659 additions and 151 deletions
--- a/Content.Server/Research/TechnologyDisk/Systems/TechnologyDiskSystem.cs
+++ b/Content.Server/Research/TechnologyDisk/Systems/TechnologyDiskSystem.cs
@@ -0,0 +1,94 @@
+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));
+    }
+}