Merge pull request Simple-Station#164 from Mnemotechnician/feat/bette…

…r-vampires

Vampireal

Content.Server/Body/Components/MetabolizerComponent.cs

@@ -1,4 +1,5 @@
 using Content.Server.Body.Systems;
+using Content.Server.Floofstation.Traits;
 using Content.Server.Traits.Assorted;
 using Content.Shared.Body.Prototypes;
 using Content.Shared.FixedPoint;
@@ -45,7 +46,7 @@ public sealed partial class MetabolizerComponent : Component
         ///     List of metabolizer types that this organ is. ex. Human, Slime, Felinid, w/e.
         /// </summary>
         [DataField]
-        [Access(typeof(MetabolizerSystem), typeof(LiquorLifelineSystem), Other = AccessPermissions.ReadExecute)] // FIXME Friends
+        [Access(typeof(MetabolizerSystem), typeof(LiquorLifelineSystem), typeof(VampirismSystem), Other = AccessPermissions.ReadExecute)] // Floofstation
         public HashSet<ProtoId<MetabolizerTypePrototype>>? MetabolizerTypes = null;
 
         /// <summary>

Content.Server/Body/Components/StomachComponent.cs

@@ -1,4 +1,5 @@
 using Content.Server.Body.Systems;
+using Content.Server.Floofstation.Traits;
 using Content.Server.Nutrition.EntitySystems;
 using Content.Shared.Chemistry.Components;
 using Content.Shared.Chemistry.Reagent;
@@ -45,6 +46,7 @@ public sealed partial class StomachComponent : Component
         ///     A whitelist for what special-digestible-required foods this stomach is capable of eating.
         /// </summary>
         [DataField]
+        [Access(Other = AccessPermissions.ReadWriteExecute)] // Floofstation
         public EntityWhitelist? SpecialDigestible = null;
 
         /// <summary>

Content.Server/FloofStation/Traits/Components/VampirismComponent.cs

@@ -0,0 +1,29 @@
+using Content.Server.Body.Components;
+using Content.Shared.Body.Prototypes;
+using Content.Shared.Whitelist;
+using Robust.Shared.Prototypes;
+
+namespace Content.Server.Floofstation.Traits.Components;
+
+/// <summary>
+///     Enables the mob to suck blood from other mobs to replenish its own saturation.
+///     Must be fully initialized before being added to a mob.
+/// </summary>
+[RegisterComponent]
+public sealed partial class VampirismComponent : Component
+{
+    [DataField]
+    public HashSet<ProtoId<MetabolizerTypePrototype>> MetabolizerPrototypes = new() { "Vampiric", "Animal" };
+
+    /// <summary>
+    ///     A whitelist for what special-digestible-required foods the vampire's stomach is capable of eating.
+    /// </summary>
+    [DataField]
+    public EntityWhitelist? SpecialDigestible = null;
+
+    [DataField]
+    public TimeSpan SuccDelay = TimeSpan.FromSeconds(1);
+
+    [DataField]
+    public float UnitsToSucc = 10;
+}

Content.Server/FloofStation/Traits/VampirismSystem.cs

@@ -0,0 +1,51 @@
+using Content.Server.Body.Components;
+using Content.Server.Floofstation.Traits.Components;
+using Content.Server.Vampiric;
+using Content.Shared.Body.Components;
+using Content.Shared.Body.Systems;
+
+namespace Content.Server.Floofstation.Traits;
+
+public sealed class VampirismSystem : EntitySystem
+{
+    [Dependency] private readonly SharedBodySystem _body = default!;
+
+    public override void Initialize()
+    {
+        SubscribeLocalEvent<VampirismComponent, MapInitEvent>(OnInitVampire);
+    }
+
+    private void OnInitVampire(Entity<VampirismComponent> ent, ref MapInitEvent args)
+    {
+        EnsureBloodSucker(ent);
+
+        if (!TryComp<BodyComponent>(ent, out var body)
+		    || !_body.TryGetBodyOrganComponents<MetabolizerComponent>(ent, out var comps, body))
+            return;
+
+        foreach (var (metabolizer, organ) in comps)
+        {
+            if (!TryComp<StomachComponent>(organ.Owner, out var stomach))
+                continue;
+
+            metabolizer.MetabolizerTypes = ent.Comp.MetabolizerPrototypes;
+
+            if (ent.Comp.SpecialDigestible is {} whitelist)
+                stomach.SpecialDigestible = whitelist;
+        }
+    }
+
+    private void EnsureBloodSucker(Entity<VampirismComponent> uid)
+    {
+        if (HasComp<BloodSuckerComponent>(uid))
+            return;
+
+        AddComp(uid, new BloodSuckerComponent
+        {
+            Delay = uid.Comp.SuccDelay,
+            InjectWhenSucc = false, // The code for it is deprecated, might wanna make it inject something when (if?) it gets reworked
+            UnitsToSucc = uid.Comp.UnitsToSucc,
+            WebRequired = false
+        });
+    }
+}

Resources/Locale/en-US/Floof/traits/traits.ftl

@@ -1,5 +1,11 @@
 trait-name-Vampirism = Vampirism
-trait-description-Vampirism = Whether through implantation, genetic modification, or evolution, you have a pair of hollow, sharp fangs used to drink iron-based blood from the beings that contain it.
+trait-description-Vampirism =
+    Your body has evolved to be able to suck blood from beings that contain it and metabolize it into useful compounds.
+    You cannot eat normal food, but drinking blood satiates your hunger and thirst, and also improves your health.
+
+trait-name-HollowFangs = Hollow fangs
+trait-description-HollowFangs =
+    Whether through implantation, genetic modification, or evolution, you have a pair of hollow, sharp fangs used to drink iron-based blood from the beings that contain it.
 
 trait-name-CumProducer = Cock
 trait-description-CumProducer = You have a schlong between your legs.

Resources/Prototypes/Floof/Traits/physical.yml

@@ -1,27 +1,54 @@
 - type: trait
   id: Vampirism # You may port this to EE, you have my permission!
   category: Physical
-  points: -1
+  points: 3
   requirements:
     - !type:CharacterJobRequirement
       inverted: true
       jobs:
         - Borg
         - MedicalBorg
-    - !type:CharacterSpeciesRequirement # This will be removed once you can edit organs with traits
+    - !type:CharacterSpeciesRequirement
       inverted: true
       species:
-        - Moth
-        - Oni
-        - Diona
-        - SlimePerson
-        - Human
         - IPC
+    - !type:CharacterTraitRequirement
+      inverted: true
+      traits:
+      - HollowFangs
+  components:
+    - type: Vampirism
+      succDelay: 3
+      specialDigestible: # Vampires cannot eat food chat is that real
+        tags:
+        - IceCream
+        - Pill
+        - Crayon
+        - Paper
+
+- type: trait
+  id: HollowFangs
+  category: Physical
+  points: -2
+  requirements:
+  - !type:CharacterJobRequirement
+    inverted: true
+    jobs:
+    - Borg
+    - MedicalBorg
+  - !type:CharacterSpeciesRequirement
+    inverted: true
+    species:
+    - IPC
+  - !type:CharacterTraitRequirement
+    inverted: true
+    traits:
+    - Vampirism
   components:
-  - type: BloodSucker
-    unitsToSucc: 10
-    injectWhenSucc: false
-    webRequired: false
+  - type: Vampirism
+    succDelay: 5
+    metabolizerPrototypes:
+    - Animal
 
 - type: trait
   id: Weakness

Resources/Prototypes/Reagents/biological.yml

@@ -51,6 +51,11 @@
         - !type:AdjustReagent
           reagent: UncookedAnimalProteins
           amount: 0.5
+          # Floofstation - prevent vampires from getting sick by drinking blood
+          conditions:
+          - !type:OrganType
+            type: Vampiric
+            shouldHave: false
     Medicine:
       effects:
       - !type:HealthChange