guh

Simple-Station · Mar 28, 2024 · 6187158 · 6187158
1 parent 85c83e4
commit 6187158
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/Content.Server/DeltaV/Lamiae/LamiaSystem.cs b/Content.Server/DeltaV/Lamiae/LamiaSystem.cs
@@ -61,7 +61,7 @@ public override void Update(float frameTime)
                 if (segment.segment.SegmentNumber == 1)
                 {
                     Transform(segmentUid).Coordinates = Transform(attachedUid).Coordinates;
-                    var revoluteJoint = _jointSystem.CreateWeldJoint(attachedUid, segmentUid, id: ("Segment" + segment.segment.SegmentNumber + segment.segment.Lamia));
+                    var revoluteJoint = _jointSystem.CreateWeldJoint(attachedUid, segmentUid, id: "Segment" + segment.segment.SegmentNumber + segment.segment.Lamia);
                     revoluteJoint.CollideConnected = false;
                 }
                 if (segment.segment.SegmentNumber < segment.segment.MaxSegments)
@@ -144,7 +144,7 @@ private void OnShutdown(EntityUid uid, LamiaComponent component, ComponentShutdo
 
         private void OnJointRemoved(EntityUid uid, LamiaComponent component, JointRemovedEvent args)
         {
-            if (!component.Segments.Contains(args.OtherBody.Owner))
+            if (!component.Segments.Contains(args.OtherEntity))
                 return;
 
             if (HasComp<PortalTimeoutComponent>(uid)) return;
@@ -199,7 +199,7 @@ public void SpawnSegments(EntityUid uid, LamiaComponent component)
 
         private EntityUid AddSegment(EntityUid uid, EntityUid lamia, LamiaComponent lamiaComponent, int segmentNumber)
         {
-            LamiaSegmentComponent segmentComponent = new();
+            EnsureComp<LamiaSegmentComponent>(uid, out var segmentComponent);
             segmentComponent.MaxSegments = lamiaComponent.NumberOfSegments;
             segmentComponent.BulletPassover = lamiaComponent.BulletPassover;
             segmentComponent.Lamia = lamia;
@@ -219,7 +219,8 @@ private EntityUid AddSegment(EntityUid uid, EntityUid lamia, LamiaComponent lami
             {
                 segmentComponent.OffsetSwitching = lamiaComponent.StaticOffset * MathF.Pow(lamiaComponent.OffsetConstant, segmentNumber - taperConstant);
                 segmentComponent.ScaleFactor = lamiaComponent.StaticScale * MathF.Pow(1f / lamiaComponent.OffsetConstant, segmentNumber - taperConstant);
-            } else
+            }
+            else
             {
                 segmentComponent.OffsetSwitching = lamiaComponent.StaticOffset;
                 segmentComponent.ScaleFactor = lamiaComponent.StaticScale;
@@ -229,12 +230,10 @@ private EntityUid AddSegment(EntityUid uid, EntityUid lamia, LamiaComponent lami
                 segmentComponent.OffsetSwitching *= -1;
             }
 
-            segmentComponent.Owner = segment;
             segmentComponent.SegmentNumber = segmentNumber;
-            EntityManager.AddComponent(segment, segmentComponent, true);
 
             _segments.Enqueue((segmentComponent, lamia));
-            lamiaComponent.Segments.Add(segmentComponent.Owner);
+            lamiaComponent.Segments.Add(uid);
             return segment;
         }