// robot_upgrades.dm
// Contains various borg upgrades.

/obj/item/borg/upgrade
	name = "borg upgrade module."
	desc = "Protected by FRM."
	icon = 'icons/obj/module.dmi'
	icon_state = "cyborg_upgrade"
	origin_tech = "programming=2"
	materials = list(MAT_METAL = 10000)
	/// Whether or not the cyborg needs to have a chosen module before they can receive this upgrade.
	var/require_module = FALSE
	/// The type of module this upgrade is compatible with: Engineering, Medical, etc.
	var/module_type = null
	/// A list of items, and their replacements that this upgrade should replace on installation, in the format of `item_type_to_replace = replacement_item_type`.
	var/list/items_to_replace = list()
	/// A list of items to add, rather than replace
	var/list/items_to_add = list()
	/// A list of replacement items will need to be placed into a cyborg module's `special_rechargable` list after this upgrade is installed.
	var/list/special_rechargables = list()
	/// Allow the same upgrade to be installed multiple times, FALSE by default
	var/allow_duplicate = FALSE
	/// Delete the module after installing it. For deleting upgrades that might be installed multiple times, like the rename/reset upgrades.
	var/delete_after_install = FALSE

/**
 * Called when someone clicks on a borg with an upgrade in their hand.
 *
 * Arguments:
 * * R - the cyborg that was clicked on with an upgrade.
 */
/obj/item/borg/upgrade/proc/action(mob/user, mob/living/silicon/robot/R)
	if(!pre_install_checks(user, R))
		return
	if(!user.drop_item())
		to_chat(user, SPAN_NOTICE("\The [src] is stuck to your hand, you cannot install it in [R]"))
		return FALSE
	if(!do_install(R))
		return
	after_install(R)
	if(delete_after_install)
		qdel(src)
	else
		forceMove(R)
	return TRUE

/**
 * Checks if the upgrade is able to be applied to the cyborg, before actually applying it.
 *
 * Arguments:
 * * R - the cyborg that was clicked on with an upgrade.
 */
/obj/item/borg/upgrade/proc/pre_install_checks(mob/user, mob/living/silicon/robot/R)
	if(R.stat == DEAD)
		to_chat(user, SPAN_WARNING("[src] will not function on a deceased cyborg."))
		return
	if(module_type && !istype(R.module, module_type))
		to_chat(R, SPAN_WARNING("Upgrade mounting error! No suitable hardpoint detected!"))
		to_chat(user, SPAN_WARNING("There's no mounting point for the module!"))
		return
	var/obj/item/borg/upgrade/u = locate(type) in R
	if(u && !allow_duplicate)
		to_chat(user, SPAN_NOTICE("This unit already has [src] installed!"))
		return
	return TRUE

/**
 * Executes code that will modify the cyborg or its module.
 *
 * Arguments:
 * * R - the cyborg we're applying the upgrade to.
 */
/obj/item/borg/upgrade/proc/do_install(mob/living/silicon/robot/R)
	return TRUE

/**
 * Executes code after the module has been installed and the cyborg has been modified in some way.
 *
 * Arguments:
 * * R - the cyborg that we've applied the upgrade to.
 */
/obj/item/borg/upgrade/proc/after_install(mob/living/silicon/robot/R)
	for(var/item in items_to_replace)
		var/replacement_type = items_to_replace[item]
		var/obj/item/replacement = new replacement_type(R.module)
		R.module.remove_item_from_lists(item)
		R.module.basic_modules += replacement

		if(replacement_type in special_rechargables)
			R.module.special_rechargables += replacement

	for(var/item in items_to_add)
		var/obj/item/replacement = new item(R.module)
		R.module.basic_modules += replacement

	R.module?.rebuild_modules()
	return TRUE

/*******************/
// MARK: COMMON
/*******************/

/obj/item/borg/upgrade/reset
	name = "cyborg module reset board"
	desc = "Used to reset a cyborg's module. Destroys any other upgrades applied to the cyborg."
	icon_state = "cyborg_upgrade1"
	require_module = TRUE
	delete_after_install = TRUE

/obj/item/borg/upgrade/reset/do_install(mob/living/silicon/robot/R)
	R.reset_module()
	return TRUE

/obj/item/borg/upgrade/reset/after_install(mob/living/silicon/robot/R)
	return // We don't need to give them replacement items, or rebuild their module list. It's going to be a blank borg.

/obj/item/borg/upgrade/rename
	name = "cyborg reclassification board"
	desc = "Used to rename a cyborg."
	icon_state = "cyborg_upgrade1"
	delete_after_install = TRUE
	materials = list(MAT_METAL = 35000)
	var/heldname = "default name"

/obj/item/borg/upgrade/rename/attack_self__legacy__attackchain(mob/user)
	var/new_heldname = tgui_input_text(user, "Enter new robot name", "Cyborg Reclassification", heldname, MAX_NAME_LEN)
	if(!new_heldname)
		return
	heldname = new_heldname

/obj/item/borg/upgrade/rename/do_install(mob/living/silicon/robot/R)
	if(!R.allow_rename)
		to_chat(R, SPAN_WARNING("Internal diagnostic error: incompatible upgrade module detected."))
		return
	R.notify_ai(3, R.name, heldname)
	R.name = heldname
	R.custom_name = heldname
	R.real_name = heldname
	if(R.mmi && R.mmi.brainmob)
		R.mmi.brainmob.name = R.name
	return TRUE

/obj/item/borg/upgrade/restart
	name = "cyborg emergency reboot module"
	desc = "Used to force a reboot of a disabled-but-repaired cyborg, bringing it back online."
	icon_state = "cyborg_upgrade1"
	delete_after_install = TRUE
	materials = list(MAT_METAL = 60000 , MAT_GLASS = 5000)

/obj/item/borg/upgrade/restart/do_install(mob/living/silicon/robot/R)
	if(R.health < 0)
		to_chat(usr, SPAN_WARNING("You have to repair the cyborg before using this module!"))
		return

	if(!R.key)
		for(var/mob/dead/observer/ghost in GLOB.player_list)
			if(ghost.mind && ghost.mind.current == R)
				R.key = ghost.key

	R.set_stat(CONSCIOUS)
	GLOB.dead_mob_list -= R //please never forget this ever kthx
	GLOB.alive_mob_list += R
	R.notify_ai(1)

	return TRUE

/obj/item/borg/upgrade/thrusters
	name = "ion thruster upgrade"
	desc = "A energy-operated thruster system for cyborgs."
	icon_state = "cyborg_upgrade3"
	origin_tech = "engineering=4;powerstorage=4"
	materials = list(MAT_METAL = 10000, MAT_PLASMA = 5000, MAT_URANIUM = 6000)

/obj/item/borg/upgrade/thrusters/do_install(mob/living/silicon/robot/R)
	R.ionpulse = TRUE
	return TRUE

/obj/item/borg/upgrade/selfrepair
	name = "self-repair module"
	desc = "This module will repair the cyborg over time."
	icon_state = "cyborg_upgrade5"
	require_module = TRUE
	materials = list(MAT_METAL = 15000, MAT_GLASS = 15000)
	var/repair_amount = -1
	var/repair_tick = 1
	var/msg_cooldown = 0
	var/on = FALSE
	var/powercost = 10
	var/mob/living/silicon/robot/cyborg

/obj/item/borg/upgrade/selfrepair/do_install(mob/living/silicon/robot/R)
	cyborg = R
	icon_state = "selfrepair_off"
	var/datum/action/A = new /datum/action/item_action/toggle(src)
	A.Grant(R)
	return TRUE

/obj/item/borg/upgrade/selfrepair/Destroy()
	cyborg = null
	STOP_PROCESSING(SSobj, src)
	on = FALSE
	return ..()

/obj/item/borg/upgrade/selfrepair/ui_action_click()
	on = !on
	if(on)
		to_chat(cyborg, SPAN_NOTICE("You activate the self-repair module."))
		START_PROCESSING(SSobj, src)
	else
		to_chat(cyborg, SPAN_NOTICE("You deactivate the self-repair module."))
		STOP_PROCESSING(SSobj, src)
	update_icon(UPDATE_ICON_STATE)

/obj/item/borg/upgrade/selfrepair/update_icon_state()
	if(cyborg)
		icon_state = "selfrepair_[on ? "on" : "off"]"
		update_action_buttons()
	else
		icon_state = "cyborg_upgrade5"

/obj/item/borg/upgrade/selfrepair/proc/deactivate()
	STOP_PROCESSING(SSobj, src)
	on = FALSE
	update_icon(UPDATE_ICON_STATE)

/obj/item/borg/upgrade/selfrepair/process()
	if(!repair_tick)
		repair_tick = 1
		return

	if(cyborg && (cyborg.stat != DEAD) && on)
		if(!cyborg.cell)
			to_chat(cyborg, SPAN_WARNING("Self-repair module deactivated. Please, insert the power cell."))
			deactivate()
			return

		if(cyborg.cell.charge < powercost * 2)
			to_chat(cyborg, SPAN_WARNING("Self-repair module deactivated. Please recharge."))
			deactivate()
			return

		if(cyborg.health < cyborg.maxHealth)
			if(cyborg.health < 0)
				repair_amount = 2.5
				powercost = 30
			else
				repair_amount = 1
				powercost = 10
			cyborg.heal_overall_damage(repair_amount, repair_amount)
			cyborg.cell.use(powercost)
		else
			cyborg.cell.use(5)
		repair_tick = 0

		if((world.time - 2000) > msg_cooldown)
			var/msgmode = "standby"
			if(cyborg.health < 0)
				msgmode = "critical"
			else if(cyborg.health < cyborg.maxHealth)
				msgmode = "normal"
			to_chat(cyborg, SPAN_NOTICE("Self-repair is active in [SPAN_BOLDNOTICE("[msgmode]")] mode."))
			msg_cooldown = world.time
	else
		deactivate()

/obj/item/borg/upgrade/vtec
	name = "robotic VTEC Module"
	desc = "Used to activate a cyborg's VTEC systems, allowing them to retain more speed when damaged. Alternatively speeds up slow vehicles."
	icon_state = "cyborg_upgrade2"
	require_module = TRUE
	origin_tech = "engineering=4;materials=5;programming=4"
	materials = list(MAT_METAL = 80000 , MAT_GLASS = 6000 , MAT_URANIUM = 5000)

/obj/item/borg/upgrade/vtec/do_install(mob/living/silicon/robot/R)
	R.slowdown_cap = 3.5
	return TRUE

/***********************/
// MARK: Security
/***********************/

/obj/item/borg/upgrade/disablercooler
	name = "cyborg rapid disabler cooling module"
	desc = "Used to cool a mounted disabler, increasing the potential current in it and thus its recharge rate."
	icon_state = "cyborg_upgrade3"
	origin_tech = "engineering=4;powerstorage=4;combat=4"
	require_module = TRUE
	module_type = /obj/item/robot_module/security

/obj/item/borg/upgrade/disablercooler/do_install(mob/living/silicon/robot/R)
	var/obj/item/gun/energy/disabler/cyborg/T = locate() in R.module.modules
	if(!T)
		to_chat(usr, SPAN_NOTICE("There's no disabler in this unit!"))
		return
	if(T.charge_delay <= 2)
		to_chat(R, SPAN_NOTICE("A cooling unit is already installed!"))
		to_chat(usr, SPAN_NOTICE("There's no room for another cooling unit!"))
		return

	T.charge_delay = max(2 , T.charge_delay - 4)

	return TRUE

/*******************/
// MARK: Mining
/*******************/

/obj/item/borg/upgrade/ddrill
	name = "mining cyborg diamond drill"
	desc = "A diamond drill replacement for the mining cyborg's standard drill."
	icon_state = "cyborg_upgrade3"
	origin_tech = "engineering=4;materials=5"
	materials = list(MAT_METAL = 10000, MAT_GOLD = 250, MAT_URANIUM = 500)
	require_module = TRUE
	module_type = /obj/item/robot_module/miner
	items_to_replace = list(
		/obj/item/pickaxe/drill/cyborg = /obj/item/pickaxe/drill/cyborg/diamond
	)

/obj/item/borg/upgrade/soh
	name = "mining cyborg satchel of holding"
	desc = "A satchel of holding replacement for mining cyborg's ore satchel module."
	icon_state = "cyborg_upgrade3"
	origin_tech = "engineering=4;materials=4;bluespace=4"
	require_module = TRUE
	module_type = /obj/item/robot_module/miner
	items_to_replace = list(
		/obj/item/storage/bag/ore/cyborg = /obj/item/storage/bag/ore/cyborg/holding
	)

/obj/item/borg/upgrade/lavaproof
	name = "mining cyborg lavaproof chassis"
	desc = "An upgrade kit to apply specialized coolant systems and insulation layers to a mining cyborg's chassis, enabling them to withstand exposure to molten rock."
	icon_state = "ash_plating"
	materials = list(MAT_METAL = 10000, MAT_PLASMA = 4000, MAT_TITANIUM = 5000)
	resistance_flags = LAVA_PROOF | FIRE_PROOF
	require_module = TRUE
	module_type = /obj/item/robot_module/miner

/obj/item/borg/upgrade/lavaproof/do_install(mob/living/silicon/robot/R)
	if(istype(R))
		R.weather_immunities += "lava"
	return TRUE

/***********************/
// MARK: Engineer
/***********************/

/obj/item/borg/upgrade/rcd
	name = "R.C.D. upgrade"
	desc = "A modified Rapid Construction Device, able to pull energy directly from a cyborg's internal power cell."
	icon_state = "cyborg_upgrade5"
	origin_tech = "engineering=4;materials=5;powerstorage=4"
	materials = list(MAT_METAL = 30000, MAT_GLASS = 15000)
	require_module = TRUE
	module_type = /obj/item/robot_module/engineering
	items_to_add = list(/obj/item/rcd/borg)

/obj/item/borg/upgrade/rcd/after_install(mob/living/silicon/robot/R)
	if(R.emagged) // Emagged engi-borgs have already have the RCD added.
		return
	R.module.remove_item_from_lists(/obj/item/rcd) // So emagging them in the future won't grant another RCD.
	..()

/obj/item/borg/upgrade/rped
	name = "Rapid Part Exchange Device upgrade"
	desc = "A modified Rapid Part Exchange Device designed to be used by engineering robots."
	icon_state = "cyborg_upgrade5"
	materials = list(MAT_METAL = 40000, MAT_GLASS = 15000)
	require_module = TRUE
	module_type = /obj/item/robot_module/engineering
	items_to_add = list(/obj/item/storage/part_replacer)

/***********************/
// MARK: Janitor
/***********************/

/obj/item/borg/upgrade/floorbuffer
	name = "janitorial floor buffer upgrade"
	desc = "A floor buffer upgrade kit that can be attached to janitor cyborgs and mobile janicarts."
	icon = 'icons/obj/vehicles.dmi'
	icon_state = "upgrade"
	origin_tech = "materials=3;engineering=4"
	materials = list(MAT_METAL = 9000, MAT_GLASS = 7600)
	require_module = TRUE
	module_type = /obj/item/robot_module/janitor
	/// How much speed the cyborg loses while the buffer is active
	var/buffer_speed = 1
	var/mob/living/silicon/robot/cyborg

/obj/item/borg/upgrade/floorbuffer/do_install(mob/living/silicon/robot/R)
	cyborg = R
	var/datum/action/A = new /datum/action/item_action/floor_buffer(src)
	A.Grant(R)
	return TRUE

/obj/item/borg/upgrade/floorbuffer/ui_action_click()
	if(!cyborg.floorbuffer)
		cyborg.floorbuffer = TRUE
		cyborg.speed += buffer_speed
	else
		cyborg.floorbuffer = FALSE
		cyborg.speed -= buffer_speed
	to_chat(cyborg, SPAN_NOTICE("The floor buffer is now [cyborg.floorbuffer ? "active" : "deactivated"]."))

/obj/item/borg/upgrade/floorbuffer/Destroy()
	if(cyborg)
		cyborg.floorbuffer = FALSE
		cyborg = null
	return ..()

/obj/item/borg/upgrade/bluespace_trash_bag
	name = "janitor cyborg trash bag of holding upgrade"
	desc = "An advanced trash bag upgrade board with bluespace properties that can be attached to janitorial cyborgs."
	icon_state = "cyborg_upgrade4"
	require_module = TRUE
	module_type = /obj/item/robot_module/janitor
	materials = list(MAT_GOLD = 1500, MAT_URANIUM = 250, MAT_PLASMA = 1500)
	items_to_replace = list(
		/obj/item/storage/bag/trash/cyborg = /obj/item/storage/bag/trash/bluespace/cyborg
	)

/***********************/
// MARK: Service
/***********************/

/obj/item/borg/upgrade/rsf_executive
	name = "executive service upgrade"
	desc = "An upgrade that replaces a service cyborg's Rapid Service Fabricator with a classy Executive version."
	icon_state = "cyborg_upgrade5"
	origin_tech = "biotech=2;materials=1"
	materials = list(MAT_METAL = 10000, MAT_GLASS = 6000, MAT_GOLD = 2000)
	require_module = TRUE
	module_type = /obj/item/robot_module/butler
	items_to_add = list(/obj/item/kitchen/knife/cheese)
	items_to_replace = list(/obj/item/rsf = /obj/item/rsf/executive)

/***********************/
// MARK: Medical
/***********************/

/obj/item/borg/upgrade/holo_stretcher
	name = "holo stretcher rack upgrade"
	desc = "An upgrade that allows medical cyborgs to carry holo stretchers in addition to basic roller beds."
	icon = 'icons/obj/rollerbed.dmi'
	icon_state = "holo_retracted"
	origin_tech = "magnets=3;biotech=4;powerstorage=3"
	materials = list(MAT_METAL = 1000, MAT_SILVER = 500, MAT_GLASS = 500, MAT_DIAMOND = 200)
	require_module = TRUE
	module_type = /obj/item/robot_module/medical
	items_to_replace = list(/obj/item/roller_holder = /obj/item/roller_holder/holo)

/***********************/
// MARK: Syndicate
/***********************/

/obj/item/borg/upgrade/syndicate
	name = "safety override module"
	desc = "Unlocks the hidden, deadlier functions of a cyborg."
	icon_state = "cyborg_upgrade3"
	origin_tech = "combat=6;materials=6"
	materials = list(MAT_METAL = 10000, MAT_GLASS = 15000, MAT_DIAMOND = 10000)
	require_module = TRUE

/obj/item/borg/upgrade/syndicate/do_install(mob/living/silicon/robot/R)
	if(R.weapons_unlock)
		return // They already had the safety override upgrade, or they're a cyborg type which has this by default.
	R.weapons_unlock = TRUE
	to_chat(R, SPAN_WARNING("Warning: safety protocols have been disabled!"))
	return TRUE

/obj/item/borg/upgrade/syndie_soap
	name = "janitor cyborg syndicate soap"
	desc = "Using forbidden technology and some red dye, upgrade a janitorial cyborg's soap performance by 90 percent!"
	icon_state = "cyborg_upgrade4"
	require_module = TRUE
	module_type = /obj/item/robot_module/janitor
	materials = list(MAT_GOLD = 1250, MAT_PLASMA = 2500, MAT_SILVER = 1250)
	items_to_replace = list(
		/obj/item/soap/nanotrasen = /obj/item/soap/syndie
	)

/***********************/
// MARK: Abductor
/***********************/

/obj/item/borg/upgrade/abductor_engi
	name = "engineering cyborg abductor upgrade"
	desc = "An experimental upgrade that replaces an engineering cyborg's tools with the abductor versions."
	icon_state = "abductor_mod"
	origin_tech = "engineering=6;materials=6;abductor=3"
	materials = list(MAT_METAL = 25000, MAT_SILVER = 12500, MAT_PLASMA = 5000, MAT_TITANIUM = 10000, MAT_DIAMOND = 10000) //Base abductor engineering tools * 4
	require_module = TRUE
	module_type = /obj/item/robot_module/engineering
	items_to_replace = list(
		/obj/item/weldingtool = /obj/item/weldingtool/abductor,
		/obj/item/wrench = /obj/item/wrench/abductor,
		/obj/item/screwdriver = /obj/item/screwdriver/abductor,
		/obj/item/crowbar = /obj/item/crowbar/abductor,
		/obj/item/wirecutters = /obj/item/wirecutters/abductor,
		/obj/item/multitool = /obj/item/multitool/abductor
	)
	special_rechargables = list(
		/obj/item/weldingtool/abductor
	)

/obj/item/borg/upgrade/abductor_medi
	name = "medical cyborg abductor upgrade"
	desc = "An experimental upgrade that replaces a medical cyborg's tools with the abductor versions."
	icon_state = "abductor_mod"
	origin_tech = "biotech=6;materials=6;abductor=2"
	materials = list(MAT_METAL = 18000, MAT_GLASS = 1500, MAT_SILVER = 13000, MAT_GOLD = 1000, MAT_PLASMA = 4000, MAT_TITANIUM = 12000, MAT_DIAMOND = 1000) //Base abductor engineering tools *8 + IMS cost
	require_module = TRUE
	module_type = /obj/item/robot_module/medical
	items_to_replace = list(
		/obj/item/scalpel/laser = /obj/item/scalpel/laser/alien,
		/obj/item/hemostat = /obj/item/hemostat/alien,
		/obj/item/retractor = /obj/item/retractor/alien,
		/obj/item/bonegel = /obj/item/bonegel/alien,
		/obj/item/fix_o_vein = /obj/item/fix_o_vein/alien,
		/obj/item/bonesetter = /obj/item/bonesetter/alien,
		/obj/item/circular_saw = /obj/item/circular_saw/alien,
		/obj/item/surgicaldrill = /obj/item/surgicaldrill/alien,
		/obj/item/reagent_containers/borghypo = /obj/item/reagent_containers/borghypo/abductor
	)

/obj/item/borg/upgrade/abductor_medi/after_install(mob/living/silicon/robot/R)
	. = ..()
	if(!R.emagged) // Emagged Mediborgs that are upgraded need the evil chems.
		return
	for(var/obj/item/reagent_containers/borghypo/F in R.module.modules)
		F.emag_act()

/obj/item/borg/upgrade/abductor_jani
	name = "janitorial cyborg abductor upgrade"
	desc = "An experimental upgrade that replaces a janitorial cyborg's tools with the abductor versions."
	icon_state = "abductor_mod"
	origin_tech = "biotech=6;materials=6;abductor=2"
	materials = list(MAT_METAL = 10000, MAT_SILVER = 7500, MAT_PLASMA = 2500, MAT_TITANIUM = 7500, MAT_DIAMOND = 5000) //Base abductor jani tools *5
	require_module = TRUE
	module_type = /obj/item/robot_module/janitor
	items_to_replace = list(
		/obj/item/mop/advanced/cyborg = /obj/item/mop/advanced/abductor,
		/obj/item/soap = /obj/item/soap/syndie/abductor,
		/obj/item/lightreplacer/cyborg = /obj/item/lightreplacer/bluespace/abductor,
		/obj/item/melee/flyswatter = /obj/item/melee/flyswatter/abductor
	)
	items_to_add = list(
		/obj/item/reagent_containers/spray/cleaner/safety/abductor
	)
	special_rechargables = list(
		/obj/item/reagent_containers/spray/cleaner/safety/abductor,
		/obj/item/lightreplacer/bluespace/abductor
	)