Bubberstation/code/datums/components/defaceable.dm

/// Lets you graffiti on an object
/datum/component/defaceable
	/// Icon file from which to draw our overlay
	var/icon
	/// Icon states to draw from our file, optionally use as a list key for TRUE to skip recolouring that icon state
	var/list/icon_states
	/// String description of what you have drawn
	var/drawing_of
	/// What colour have we been drawn with?
	var/ink_colour
	/// Optional callback called when we are drawn on, we pass in the ink colour
	var/datum/callback/on_defaced

/datum/component/defaceable/Initialize(icon, list/icon_states, drawing_of, datum/callback/on_defaced)
	if (!isatom(parent))
		return COMPONENT_INCOMPATIBLE

	src.icon = icon
	src.icon_states = icon_states
	src.drawing_of = drawing_of
	src.on_defaced = on_defaced

/datum/component/defaceable/RegisterWithParent()
	RegisterSignal(parent, COMSIG_ATOM_ITEM_INTERACTION, PROC_REF(on_drawn))
	RegisterSignal(parent, COMSIG_ATOM_REQUESTING_CONTEXT_FROM_ITEM, PROC_REF(on_hovered))
	var/atom/atom_parent = parent
	atom_parent.flags_1 |= HAS_CONTEXTUAL_SCREENTIPS_1

/datum/component/defaceable/UnregisterFromParent()
	UnregisterSignal(parent, list(COMSIG_ATOM_ITEM_INTERACTION, COMSIG_ATOM_REQUESTING_CONTEXT_FROM_ITEM, COMSIG_ATOM_EXAMINE, COMSIG_COMPONENT_CLEAN_ACT, COMSIG_ATOM_UPDATE_OVERLAYS))
	var/atom/atom_parent = parent
	atom_parent.update_appearance(UPDATE_OVERLAYS)

/datum/component/defaceable/Destroy(force = FALSE)
	on_defaced = null
	return ..()

/// Inform people that they can mess us up
/datum/component/defaceable/proc/on_hovered(atom/source, list/context, obj/item/held_item, mob/user)
	if (HAS_TRAIT(source, TRAIT_DEFACED))
		if (is_type_in_list(held_item, list(/obj/item/reagent_containers/spray, /obj/item/soap, /obj/item/rag)))
			context[SCREENTIP_CONTEXT_LMB] = "Clean"
	else
		if (istype(held_item, /obj/item/pen) || istype(held_item, /obj/item/toy/crayon))
			context[SCREENTIP_CONTEXT_LMB] = "Draw [drawing_of ? drawing_of : "on"]"
	return CONTEXTUAL_SCREENTIP_SET

/// See if someone can draw on us
/datum/component/defaceable/proc/on_drawn(atom/source, mob/living/user, obj/item/tool)
	SIGNAL_HANDLER
	if (user.combat_mode)
		return
	if(istype(tool, /obj/item/toy/crayon))
		var/obj/item/toy/crayon/crayon = tool
		ink_colour = crayon.paint_color
	else if(istype(tool, /obj/item/pen))
		var/obj/item/pen/pen = tool
		ink_colour = pen.colour

	if (!ink_colour)
		return

	ADD_TRAIT(parent, TRAIT_DEFACED, ref(src))

	on_defaced?.Invoke(ink_colour)

	RegisterSignal(parent, COMSIG_COMPONENT_CLEAN_ACT, PROC_REF(on_cleaned))

	if (drawing_of)
		RegisterSignal(parent, COMSIG_ATOM_EXAMINE, PROC_REF(on_examined))

	if (icon && icon_states)
		RegisterSignal(parent, COMSIG_ATOM_UPDATE_OVERLAYS, PROC_REF(on_update_overlays))
	source.update_appearance(UPDATE_OVERLAYS)

	UnregisterSignal(parent, COMSIG_ATOM_ITEM_INTERACTION)

	return ITEM_INTERACT_SUCCESS

/// Render our beautiful drawing
/datum/component/defaceable/proc/on_update_overlays(atom/source, list/overlays)
	SIGNAL_HANDLER
	for (var/state in icon_states)
		var/mutable_appearance/appearance = mutable_appearance(icon, state)
		if (!icon_states[state])
			appearance.color = ink_colour
		overlays += appearance

/// Wash it off
/datum/component/defaceable/proc/on_cleaned(atom/source, clean_types)
	SIGNAL_HANDLER
	if (!(clean_types & (CLEAN_WASH|CLEAN_SCRUB)))
		return

	REMOVE_TRAIT(parent, TRAIT_DEFACED, ref(src))
	RegisterSignal(parent, COMSIG_ATOM_ITEM_INTERACTION, PROC_REF(on_drawn))
	UnregisterSignal(parent, list(COMSIG_ATOM_EXAMINE, COMSIG_COMPONENT_CLEAN_ACT, COMSIG_ATOM_UPDATE_OVERLAYS))
	source.update_appearance(UPDATE_OVERLAYS)

/// See it there
/datum/component/defaceable/proc/on_examined(atom/source, mob/user, list/examine_list)
	SIGNAL_HANDLER
	examine_list += span_notice("Someone has crudely drawn [drawing_of] on [source.p_them()].")