Helmet light respects sprites

Author: Oblisk234

code/modules/clothing/chameleon.dm

@@ -275,9 +275,16 @@
 	if(istype(atom_target, /obj/item/clothing/suit/space/hardsuit/infiltration)) //YOGS START
 		var/obj/item/clothing/suit/space/hardsuit/infiltration/I = target
 		var/obj/item/clothing/suit/space/hardsuit/HS = picked_item
-		var/obj/item/clothing/head/helmet/helmet = initial(HS.helmettype)
+		var/obj/item/clothing/head/helmet/space/hardsuit/helmet = initial(HS.helmettype)
 		I.head_piece.initial_state = initial(helmet.icon_state)
 		I.head_piece.update_appearance(UPDATE_ICON)
+		I.head_piece.current_disguise = picked_item
+		I.head_piece.new_type = helmet.hardsuit_type
+		var/datum/action/A
+		for(var/X in I.actions)
+			A = X
+			A.build_all_button_icons()
+		
 		qdel(helmet)
 		//YOGS END
 

yogstation/code/modules/antagonists/infiltrator/items/hardsuit.dm

@@ -4,13 +4,48 @@
 	item_state = "eng_helm"
 	armor = list(MELEE = 35, BULLET = 15, LASER = 30,ENERGY = 10, BOMB = 10, BIO = 100, RAD = 50, FIRE = 75, ACID = 75)
 	syndicate = TRUE
+	var/current_disguise = /obj/item/clothing/suit/space/hardsuit/infiltration
+	var/new_type = "engineering"
+	var/list/bad_hardsuits = list(
+		/obj/item/clothing/suit/space/hardsuit/darktemplar, 
+		/obj/item/clothing/suit/space/hardsuit/darktemplar/chap, 
+		/obj/item/clothing/suit/space/hardsuit/cult, 
+		/obj/item/clothing/suit/space/hardsuit/syndi, 
+		/obj/item/clothing/suit/space/hardsuit/syndi/elite, 
+		/obj/item/clothing/suit/space/hardsuit/syndi/owl, 
+		/obj/item/clothing/suit/space/hardsuit/syndi/debug, 
+		/obj/item/clothing/suit/space/hardsuit/carp, 
+		/obj/item/clothing/suit/space/hardsuit/carp/dragon, 
+		/obj/item/clothing/suit/space/hardsuit/swat, 
+		/obj/item/clothing/suit/space/hardsuit/swat/captain, 
+		/obj/item/clothing/suit/space/hardsuit/ert/paranormal, 
+		/obj/item/clothing/suit/space/hardsuit/ert/paranormal/inquisitor, 
+		/obj/item/clothing/suit/space/hardsuit/ert/paranormal/beserker, 
+		/obj/item/clothing/suit/space/hardsuit/shielded/swat, 
+		/obj/item/clothing/suit/space/hardsuit/shielded/swat/honk, 
+		/obj/item/clothing/suit/space/hardsuit/deathsquad
+	)
 
 /obj/item/clothing/head/helmet/space/hardsuit/infiltration/Initialize(mapload)
 	. = ..()
 	if(istype(loc, /obj/item/clothing/suit/space/hardsuit/infiltration))
 		var/obj/item/clothing/suit/space/hardsuit/infiltration/I = loc
 		I.head_piece = src
 
+/obj/item/clothing/head/helmet/space/hardsuit/infiltration/attack_self(mob/user)
+	if(bad_hardsuits.Find(current_disguise))
+		to_chat(user, span_warning("You can't use the hardsuit's helmet light with this current disguise, change to another one!"))
+	else //Copied from original hardsuit attack_self and modified slightly
+		on = !on
+		icon_state = "[basestate][on]-[new_type]"
+		user.update_inv_head()	//so our mob-overlays update
+
+		set_light_on(on)
+
+		for(var/X in actions)
+			var/datum/action/A = X
+			A.build_all_button_icons()
+
 /obj/item/clothing/suit/space/hardsuit/infiltration
 	name = "engineering hardsuit"
 	icon_state = "hardsuit-engineering"