Graves are removed if the last item is taken out

1 files changed, 10 insertions, 7 deletions

diff --git a/entities/entities/ws_grave/init.lua b/entities/entities/ws_grave/init.lua
index 17995ba..21da15b 100644
--- a/entities/entities/ws_grave/init.lua
+++ b/entities/entities/ws_grave/init.lua
@@ -9,13 +9,13 @@ function ENT:Initialize()
 	self:SetMoveType(MOVETYPE_NONE)

 	self:SetSolid(SOLID_VPHYSICS)

 	self:SetUseType(SIMPLE_USE)

-	

+

 	local phys = self:GetPhysicsObject()

 	phys:EnableMotion(false)

 	phys:Sleep()

-	

+

 	self:SetHealth(30)

-	

+

 	self.StoredItems = {}

 end

 

@@ -26,7 +26,7 @@ function ENT:AddItem(item,quantity)
 			return

 		end

 	end

-	

+

 	table.insert(self.StoredItems,{Name = item, Quantity = quantity})

 end

 

@@ -35,13 +35,16 @@ function ENT:TakeItem(pl,item,quantity)
 		if (v.Name == item) then

 			quantity = math.min(quantity,v.Quantity)

 			v.Quantity = v.Quantity - quantity

-			

+

 			pl:AddItem(item,quantity)

-			

+

 			if (v.Quantity <= 0) then table.remove(self.StoredItems,k) end

 			break

 		end

 	end

+	if(self.StoreDItems == {}) then

+		self:Remove()

+	end

 end

 

 function ENT:GetItems()

@@ -55,6 +58,6 @@ end
 

 function ENT:OnTakeDamage(dmg)

 	self:SetHealth(self:Health()-dmg)

-	

+

 	if (self:Health() <= 0) then self:Remove() end

 end