From bdfc9d5ecd6a6b5423837eceae5b35fbd8ec7059 Mon Sep 17 00:00:00 2001
From: olliy <55839810+olliy1or@users.noreply.github.com>
Date: Fri, 30 Apr 2021 01:17:13 +0500
Subject: [PATCH] Update ranged.lua

---
 mods/ctf/ctf_classes/ranged.lua | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/mods/ctf/ctf_classes/ranged.lua b/mods/ctf/ctf_classes/ranged.lua
index 81702d8..f83f1d6 100644
--- a/mods/ctf/ctf_classes/ranged.lua
+++ b/mods/ctf/ctf_classes/ranged.lua
@@ -66,10 +66,30 @@ local function check_grapple(itemname)
 	})
 end
 
+local function check_rocket(itemname)
+	local def = minetest.registered_items[itemname]
+	local old_func = def.on_use
+	minetest.override_item(itemname, {
+		on_use = function(itemstack, user, ...)
+			if not ctf_classes.get(user).properties.allow_rockets then
+				minetest.chat_send_player(user:get_player_name(),
+					"You can't use that weapon! Change your class at base")
+				return itemstack
+			end
+
+			return old_func(itemstack, user, ...)
+		end,
+
+	})
+end
+
 check_grapple("shooter_hook:grapple_gun_loaded")
 check_grapple("shooter_hook:grapple_gun")
 check_grapple("shooter_hook:grapple_hook")
 
+check_rocket("shooter_rocket:rocket_gun_loaded")
+check_rocket("shooter_rocket:rocket_gun")
+
 -- Override grappling hook entity to check if player has flag before teleporting
 local old_grapple_step = minetest.registered_entities["shooter_hook:hook"].on_step
 minetest.registered_entities["shooter_hook:hook"].on_step = function(self, dtime, ...)