Allow adding more pump entity classes

SpaceAgeMP · Mar 6, 2024 · 1c79dd5 · 1c79dd5
1 parent 55f79c0
commit 1c79dd5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/lua/entities/rd_pump/init.lua b/lua/entities/rd_pump/init.lua
@@ -94,7 +94,7 @@ function ENT:LinkToPump(ply, ent)
 		return false, "You are not allowed to control this pump!"
 	end
 
-	if ent:GetClass() ~= self:GetClass() then
+	if not (IsValid(ent) and ent.IsRDPump and ent:IsRDPump()) then
 		return false, "The other entity is not a pump"
 	end
 

diff --git a/lua/entities/rd_pump/shared.lua b/lua/entities/rd_pump/shared.lua
@@ -12,4 +12,8 @@ ENT.IsPump = true
 
 list.Set("LSEntOverlayText", "rd_pump", {
 	HasOOO = true
-})
+})
+
+function ENT:IsRDPump()
+	return true
+end