Track families on contraptions

Author: LengthenedGradient

lua/cfw/classes/contraption_sv.lua

@@ -8,7 +8,8 @@ function CFW.createContraption()
         ents        = {},
         entsbyclass = {},
         count       = 0,
-        color       = ColorRand(50, 255)
+        color       = ColorRand(50, 255),
+        families    = {}
     }
 
     setmetatable(con, CFW.Classes.Contraption)

lua/cfw/classes/family_sv.lua

@@ -29,6 +29,9 @@ do -- Class def
     function CLASS:Init()
         CFW.Families[self] = true
 
+        local con = self.ancestor:GetContraption()
+        if con then con.families[self] = true end
+
         hook.Run("cfw.family.created", self)
     end
 
@@ -39,6 +42,9 @@ do -- Class def
     function CLASS:Delete()
         self:Sub(self.ancestor, true)
 
+        local con = self.ancestor:GetContraption()
+        if con then con.families[self] = nil end
+
         hook.Run("cfw.family.deleted", self)
 
         CFW.Families[self] = nil