#define NIF_CALIBRATION_STAGE_1 0
#define NIF_CALIBRATION_STAGE_1_END  0.1
#define NIF_CALIBRATION_STAGE_2 0.2
#define NIF_CALIBRATION_STAGE_2_END 0.9
#define NIF_CALIBRATION_STAGE_FINISHED 1

#define NIF_DURABILITY_LOSS_HALVED 2
#define NIF_MINIMUM_DURABILITY 0
#define NIF_MINIMUM_POWER_LEVEL 0

#define NIF_SETUP_BLINDNESS "nif_setup"
#define MAX_NIF_REWARDS_POINTS 2000

// This is the original NIF that other NIFs are based on.
/obj/item/organ/cyberimp/brain/nif
	name = "Nanite Implant Framework"
	desc = "A brain implant that infuses the user with nanites."
	icon = 'modular_skyrat/modules/modular_implants/icons/obj/nifs.dmi'
	icon_state = "base_nif"
	w_class = WEIGHT_CLASS_NORMAL
	slot = ORGAN_SLOT_BRAIN_NIF
	actions_types = list(/datum/action/item_action/nif/open_menu)

	//User Variables
	///What user is currently linked with the NIF?
	var/mob/living/carbon/human/linked_mob = null
	///What CKEY does the original user have? Used to prevent theft
	var/stored_ckey

	//Calibration variables
	///Is the NIF properly calibrated yet?
	var/is_calibrated = FALSE
	///Is the NIF currently being calibrated?
	var/calibrating = FALSE
	///How long does each step in the calibration process take in total?
	var/calibration_time = 3 MINUTES
	///How far through the calibration process is the NIF? Do not touch this outside of perform_calibration(), if you can at all help it.
	var/calibration_duration
	///Determines the likelyhood of a side effect occuring each process cycle: 1 / side_effect_risk
	var/side_effect_risk = 50

	//Power Variables
	///What is the maximum power level of the NIF?
	var/max_power_level = 1000
	///How much power is currently inside of the NIF?
	var/power_level = 0
	///How much power is the NIF currently using? Negative usage will result in power being gained.
	var/power_usage = 0

	//Nutrition variables
	///Is power being drawn from nutrition?
	var/nutrition_drain = FALSE
	///How fast is nutrition drained from the host?
	var/nutrition_drain_rate = 1.5
	///What is the rate of nutrition to power?
	var/nutrition_conversion_rate = 5
	///What is the minimum nutrition someone has to be at for the NIF to convert power?
	var/minimum_nutrition = 25

	//Blood variables
	///Is power being drawn through blood
	var/blood_drain = FALSE
	///The rate of blood to energy
	var/blood_conversion_rate = 5 //From full blood, this would get someone to 500 charge
	///How fast is blood being drained?
	var/blood_drain_rate = 1
	///When is blood draining disabled?
	var/minimum_blood_level = BLOOD_VOLUME_SAFE

	//Durability and persistence variables
	///What is the maximum durability of the NIF?
	var/max_durability = 100
	///What level of durability is the NIF at?
	var/durability = 100
	//How much durability is lost upon dying, if any.
	var/death_durability_loss = 10
	///Does the NIF stay between rounds? By default, they do.
	var/nif_persistence = TRUE
	///Is the NIF completely broken? If this is true, the user won't be able to pull up the TGUI menu at all.
	var/broken = FALSE
	///Does the NIF have theft protection? This should only be disabled if admins need to fix something.
	var/theft_protection = TRUE
	///Is the NIF able to take damage?
	var/durability_loss_vulnerable = TRUE
	/// How many rewards points does the NIF currently have?
	var/rewards_points = 0

	//Software Variables
	///How many programs can the NIF store at once?
	var/max_nifsofts = 5
	///What programs are currently loaded onto the NIF?
	var/list/loaded_nifsofts = list()
	///What programs come already installed on the NIF?
	var/list/preinstalled_nifsofts = list(/datum/nifsoft/soul_poem)
	///What programs do we want to carry between rounds?
	var/list/persistent_nifsofts = list()
	///This shows up in the NIF settings screen as a way to ICly display lore.
	var/manufacturer_notes = "There is no data currently avalible for this product."

	//Appearance Variables
	///This is the sound that plays when doing most things!
	var/good_sound ='modular_skyrat/modules/modular_implants/sounds/default_good.ogg'
	///This is the sound that plays if there is an issue going on.
	var/bad_sound = 'modular_skyrat/modules/modular_implants/sounds/default_bad.ogg'
	///This is the sound that you would hear if you enable if you activate or enable something.
	var/click_sound = 'modular_skyrat/modules/modular_implants/sounds/default_click.ogg'
	///What icon does the NIF display in chat when sending out alerts? Icon states are stored in 'modular_skyrat/modules/modular_implants/icons/chat.dmi'
	var/chat_icon = "standard"

/obj/item/organ/cyberimp/brain/nif/Initialize(mapload)
	. = ..()

	durability = max_durability
	power_level = max_power_level

/obj/item/organ/cyberimp/brain/nif/Destroy()
	if(linked_mob)
		UnregisterSignal(linked_mob, COMSIG_LIVING_DEATH, PROC_REF(damage_on_death))

		var/found_component = linked_mob.GetComponent(/datum/component/nif_examine)
		if(found_component)
			qdel(found_component)

	linked_mob = null

	QDEL_LIST(loaded_nifsofts)
	return ..()

/obj/item/organ/cyberimp/brain/nif/on_mob_insert(mob/living/carbon/human/insertee, special = FALSE, movement_flags = DELETE_IF_REPLACED)
	. = ..()

	if(linked_mob && stored_ckey != insertee.ckey && theft_protection)
		insertee.audible_message(span_warning("[src] lets out a negative buzz before forcefully removing itself from [insertee]'s brain."))
		playsound(insertee, 'sound/machines/buzz/buzz-sigh.ogg', 30, TRUE)
		Remove(insertee)
		forceMove(get_turf(insertee))

		return FALSE

	linked_mob = insertee
	stored_ckey = linked_mob.ckey

	START_PROCESSING(SSobj, src)

	if(!is_calibrated)
		send_message("The calibration process is starting.")
		calibrating = TRUE

	linked_mob.AddComponent(/datum/component/nif_examine)
	RegisterSignal(linked_mob, COMSIG_LIVING_DEATH, PROC_REF(damage_on_death))

	if(preinstalled_nifsofts || persistent_nifsofts)
		send_message("Loading preinstalled and stored NIFSofts, please wait...")
		addtimer(CALLBACK(src, PROC_REF(install_preinstalled_nifsofts)), 3 SECONDS)

/obj/item/organ/cyberimp/brain/nif/on_mob_remove(mob/living/carbon/organ_owner, special = FALSE, movement_flags)
	. = ..()

	organ_owner.log_message("'s [src] was removed from [organ_owner]", LOG_GAME)
	STOP_PROCESSING(SSobj, src)

	var/found_component = organ_owner.GetComponent(/datum/component/nif_examine)
	if(found_component)
		qdel(found_component)

	if(linked_mob)
		UnregisterSignal(linked_mob, COMSIG_LIVING_DEATH, PROC_REF(damage_on_death))
		linked_mob = null

	QDEL_LIST(loaded_nifsofts)

///Installs preinstalled NIFSofts
/obj/item/organ/cyberimp/brain/nif/proc/install_preinstalled_nifsofts()
	if(isnull(linked_mob)) // BUBBER TODO: Fix the race condition here
		return FALSE
	if(!preinstalled_nifsofts)
		return FALSE

	for(var/datum/nifsoft/preinstalled_nifsoft as anything in preinstalled_nifsofts)
		new preinstalled_nifsoft(src)

	for(var/stored_nifsoft in persistent_nifsofts)
		var/datum/nifsoft/new_stored_nifsoft = new stored_nifsoft(src)
		new_stored_nifsoft.keep_installed = TRUE

	return TRUE

/obj/item/organ/cyberimp/brain/nif/process(seconds_per_tick)
	. = ..()

	if(!linked_mob || broken || HAS_TRAIT(linked_mob, TRAIT_STASIS))
		return FALSE

	if(calibrating)
		perform_calibration()
		return

	if(nutrition_drain && linked_mob.nutrition < minimum_nutrition) //Turns nutrition drain off if nutrition is lower than minimum
		toggle_nutrition_drain(TRUE)

	if(blood_drain && !blood_check()) //Disables blood draining if the mob fails the blood check
		toggle_blood_drain(TRUE)

	if(blood_drain)
		linked_mob.blood_volume -= blood_drain_rate

	if(power_usage > power_level)
		for(var/datum/nifsoft/nifsoft as anything in loaded_nifsofts)
			if(!nifsoft.active)
				continue

			nifsoft.activate()

	change_power_level(power_usage)

///Modifies power based off power_to_change. Negative numbers add charge, positive numbers remove charge
/obj/item/organ/cyberimp/brain/nif/proc/change_power_level(power_to_change)
	if(!power_to_change)
		return TRUE

	if((!power_level && (power_to_change > 0)) || ((power_to_change < 0) && (power_level >= max_power_level)))
		return FALSE

	if(power_to_change > 0)
		power_level = max((power_level - power_to_change), NIF_MINIMUM_POWER_LEVEL)
		return TRUE

	power_level = min((power_level - power_to_change), max_power_level)
	return TRUE

///Toggles nutrition drain as a power source on NIFs on/off. Bypass - Ignores the need to perform the nutirition_check() proc.
/obj/item/organ/cyberimp/brain/nif/proc/toggle_nutrition_drain(bypass = FALSE)
	if(!bypass && !nutrition_check())
		return FALSE

	var/hunger_modifier = linked_mob.physiology.hunger_mod

	if(nutrition_drain)
		hunger_modifier = nutrition_drain_rate
		power_usage += (nutrition_drain_rate * nutrition_conversion_rate)
		nutrition_drain = FALSE
		return TRUE

	hunger_modifier *= nutrition_drain_rate
	power_usage -= (nutrition_drain_rate * nutrition_conversion_rate)
	nutrition_drain = TRUE
	return TRUE

/// Checks to see if the mob has a nutrition that can be drain from
/obj/item/organ/cyberimp/brain/nif/proc/nutrition_check() //This is a seperate proc so that TGUI can perform this check on the menu
	if(!linked_mob || !linked_mob.nutrition)
		return FALSE

	if(HAS_TRAIT(linked_mob, TRAIT_NOHUNGER)) //Hemophages HATE this one simple check.
		return FALSE

	return linked_mob.nutrition >= minimum_nutrition

///Toggles Blood Drain. Bypasss -  Ignores the need to perform the blood_check proc.
/obj/item/organ/cyberimp/brain/nif/proc/toggle_blood_drain(bypass = FALSE)
	if(!bypass && !blood_check())
		return FALSE

	blood_drain = !blood_drain

	if(!blood_drain)
		power_usage += (blood_drain_rate * blood_conversion_rate)

		balloon_alert(linked_mob, "blood draining disabled")
		return TRUE

	power_usage -= (blood_drain_rate * blood_conversion_rate)
	balloon_alert(linked_mob, "blood draining enabled")
	return TRUE

///Checks if the NIF is able to draw blood as a power source?
/obj/item/organ/cyberimp/brain/nif/proc/blood_check()
	if(!linked_mob || !linked_mob.blood_volume || (linked_mob.blood_volume <= minimum_blood_level))
		return FALSE

	return TRUE

///Calibrates the Parent NIF, this is ran every time the parent NIF is first installed inside of someone.
/obj/item/organ/cyberimp/brain/nif/proc/perform_calibration()
	if(linked_mob.stat >= DEAD)
		return FALSE

	if(!calibration_duration)
		calibration_duration = world.time + calibration_time

	var/percentage_done = (world.time - (calibration_duration - (calibration_time))) / calibration_time
	switch(percentage_done)
		if(NIF_CALIBRATION_STAGE_1 to NIF_CALIBRATION_STAGE_1_END)
			linked_mob.become_blind(NIF_SETUP_BLINDNESS)

		if(NIF_CALIBRATION_STAGE_2 to NIF_CALIBRATION_STAGE_2_END)
			linked_mob.cure_blind(NIF_SETUP_BLINDNESS)
			var/random_ailment = rand(1, side_effect_risk)
			switch(random_ailment)
				if(1)
					to_chat(linked_mob, span_warning("You feel sick to your stomach!"))
					linked_mob.adjust_disgust(25)
				if(2)
					to_chat(linked_mob, span_warning("You feel a wave of fatigue roll over you!"))
					linked_mob.adjust_stamina_loss(50)

		if(NIF_CALIBRATION_STAGE_FINISHED to INFINITY)
			send_message("The calibration process is complete.")

			calibrating = FALSE
			is_calibrated = TRUE
			if(isnull(linked_mob.mind))
				return // BUBBER TODO - Refactor NIFs
			if(!linked_mob.save_individual_persistence())
				stack_trace("persistence was not saved for [linked_mob]!")

///Installs the loaded_nifsoft to the parent NIF.
/obj/item/organ/cyberimp/brain/nif/proc/install_nifsoft(datum/nifsoft/loaded_nifsoft)
	if(broken || calibrating) //NIFSofts can't be installed to a broken NIF
		return FALSE

	if(length(loaded_nifsofts) >= max_nifsofts)
		send_message("You cannot install any additional NIFSofts, please uninstall one to make room!", alert = TRUE)
		return FALSE

	if(!is_type_in_list(src, loaded_nifsoft.compatible_nifs))
		send_message("[loaded_nifsoft] is incompatible with your NIF!", TRUE)
		return FALSE

	for(var/datum/nifsoft/current_nifsoft as anything in loaded_nifsofts)
		if(loaded_nifsoft.single_install && (loaded_nifsoft.type == current_nifsoft.type))
			send_message("Multiple of [loaded_nifsoft] cannot be installed.", TRUE)
			return FALSE

		if(is_type_in_list(current_nifsoft, loaded_nifsoft.mutually_exclusive_programs))
			send_message("[current_nifsoft] is preventing [loaded_nifsoft] from being installed.", TRUE)
			return FALSE

	loaded_nifsofts += loaded_nifsoft
	loaded_nifsoft.parent_nif = WEAKREF(src)
	loaded_nifsoft.linked_mob = linked_mob
	rewards_points += (loaded_nifsoft.rewards_points_rate * loaded_nifsoft.purchase_price)

	rewards_points = min(rewards_points, MAX_NIF_REWARDS_POINTS)

	send_message("[loaded_nifsoft] has been added.")
	update_static_data_for_all_viewers()
	return TRUE

///Removes a NIFSoft from a NIF. Silent - determines whether or not alerts will be given to the owner of the NIF
/obj/item/organ/cyberimp/brain/nif/proc/remove_nifsoft(datum/nifsoft/removed_nifsoft, silent = FALSE)
	if(!is_type_in_list(removed_nifsoft, loaded_nifsofts) || broken)
		return FALSE

	if(!silent)
		send_message("[removed_nifsoft.name] has been removed", alert = TRUE)

	qdel(removed_nifsoft)
	update_static_data_for_all_viewers()

	return TRUE

///Adjusts the NIF based on the adjustment_amount. Positive values repair, negative values damage
/obj/item/organ/cyberimp/brain/nif/proc/adjust_durability(adjustment_amount)
	if(!adjustment_amount || ((adjustment_amount > 0) && (durability >= max_durability) || ((adjustment_amount < 0) && (durability <= NIF_MINIMUM_DURABILITY))))
		return FALSE

	if(adjustment_amount < 0)
		durability = max((durability + adjustment_amount), NIF_MINIMUM_DURABILITY)
		return TRUE

	durability = min((durability + adjustment_amount), max_durability)
	return TRUE

///Sends a message to the owner of the NIF. Typically used for messages from the NIF itself or from NIFSofts.
/obj/item/organ/cyberimp/brain/nif/proc/send_message(message_to_send, alert = FALSE)
	var/datum/asset/spritesheet_batched/sheet = get_asset_datum(/datum/asset/spritesheet_batched/chat)
	var/tag = sheet.icon_tag("nif-[chat_icon]")
	var/nif_icon = ""

	if(tag)
		nif_icon = tag

	if(alert)
		to_chat(linked_mob, span_warning("[nif_icon] <b>NIF Alert</b>: [message_to_send]"))
		linked_mob.playsound_local(linked_mob, bad_sound, 60, FALSE)
		return

	to_chat(linked_mob, span_cyan("[nif_icon] <b>NIF Message</b>: [message_to_send]"))
	linked_mob.playsound_local(linked_mob, good_sound, 60, FALSE)


///Changes the broken variable to be false. This does not relate to durability.
/obj/item/organ/cyberimp/brain/nif/proc/fix_nif()
	if(!broken)
		return FALSE

	broken = FALSE
	send_message("Your NIF is now in working condition!")
	return TRUE

///Re-enables the durability_loss_vulnerable variable, allowing the parent NIF to take durability damage again.
/obj/item/organ/cyberimp/brain/nif/proc/make_vulnerable()
	durability_loss_vulnerable = TRUE

//This is here so that a TGUI can't be opened by using the implant while it isn't implanted.
/obj/item/organ/cyberimp/brain/nif/attack_self(mob/user, modifiers)
	return FALSE

/obj/item/organ/cyberimp/brain/nif/emp_act(severity)
	if(!durability_loss_vulnerable)
		return FALSE

	durability_loss_vulnerable = FALSE

	if(!broken)
		broken = TRUE
		addtimer(CALLBACK(src, PROC_REF(fix_nif)), 30 SECONDS)

	addtimer(CALLBACK(src, PROC_REF(make_vulnerable)), 3 MINUTES)

	switch(severity)
		if(1)
			adjust_durability(-death_durability_loss)
		if(2)
			adjust_durability(-death_durability_loss / NIF_DURABILITY_LOSS_HALVED)

	for(var/datum/nifsoft/installed_nifsoft as anything in loaded_nifsofts)
		installed_nifsoft.on_emp(severity)

	send_message("<b>ELECTROMAGNETIC INTERFERENCE DETECTED.</b>", TRUE)

///Applies damage to the parent NIF whenever the user dies.
/obj/item/organ/cyberimp/brain/nif/proc/damage_on_death()
	SIGNAL_HANDLER

	if(!durability_loss_vulnerable)
		return FALSE

	adjust_durability(-death_durability_loss)
	durability_loss_vulnerable = FALSE

	addtimer(CALLBACK(src, PROC_REF(make_vulnerable)), 20 MINUTES) //Players should have a decent grace period on this.

/// Removes rewards points from the parent NIF. Returns FALSE if there are not enough points to remove, returns TRUE if the points have been succesfully removed.
/obj/item/organ/cyberimp/brain/nif/proc/remove_rewards_points(points_to_remove)
	if(points_to_remove > rewards_points)
		return FALSE

	rewards_points -= points_to_remove
	return TRUE

/datum/component/nif_examine
	///What text is shown when examining someone with NIF Examine text?
	var/nif_examine_text = "There's a certain spark to their eyes."

/datum/component/nif_examine/New()
	. = ..()
	if(!ishuman(parent))
		return COMPONENT_INCOMPATIBLE

	RegisterSignal(parent, COMSIG_ATOM_EXAMINE, PROC_REF(add_examine))

/datum/component/nif_examine/Destroy(force, silent)
	UnregisterSignal(parent, COMSIG_MOB_EXAMINATE)
	return ..()

///Adds and examine based on the nif_examine_text of the nif_user
/datum/component/nif_examine/proc/add_examine(mob/nif_user, mob/looker, list/examine_texts)
	SIGNAL_HANDLER

	examine_texts += span_purple("[EXAMINE_SECTION_BREAK][EXAMINE_HINT(nif_examine_text)]")

///Checks to see if a human with a NIF has the nifsoft_to_find type of NIFSoft installed?
/mob/living/carbon/human/proc/find_nifsoft(datum/nifsoft/nifsoft_to_find)
	var/obj/item/organ/cyberimp/brain/nif/installed_nif = get_organ_by_type(/obj/item/organ/cyberimp/brain/nif)
	var/list/nifsoft_list = installed_nif?.loaded_nifsofts

	if(!nifsoft_list)
		return FALSE

	var/datum/nifsoft/located_nifsoft =	locate(nifsoft_to_find) in nifsoft_list
	if(located_nifsoft)
		return located_nifsoft

	return FALSE

/datum/asset/spritesheet_batched/chat/create_spritesheets()
	. = ..()
	insert_all_icons("nif", 'modular_skyrat/modules/modular_implants/icons/chat.dmi')

/obj/item/autosurgeon/organ/nif
	starting_organ = /obj/item/organ/cyberimp/brain/nif/standard
	uses = 1

/obj/item/organ/cyberimp/brain/nif/debug
	is_calibrated = TRUE

/obj/item/autosurgeon/organ/nif/debug
	starting_organ = /obj/item/organ/cyberimp/brain/nif/debug
	uses = 1

/obj/item/storage/box/nif_ghost_box
	name = "\improper NIF Starter Kit"
	desc = "Contains a calibration-free NIF along with a variety of NIFSofts."
	illustration = "disk_kit"

/obj/item/storage/box/nif_ghost_box/PopulateContents()
	new /obj/item/autosurgeon/organ/nif/ghost_role(src)
	new /obj/item/disk/nifsoft_uploader/shapeshifter(src)
	new /obj/item/disk/nifsoft_uploader/summoner(src)
	new /obj/item/disk/nifsoft_uploader/dorms(src)
	new /obj/item/disk/nifsoft_uploader/dorms/hypnosis(src)
	new /obj/item/disk/nifsoft_uploader/soulcatcher(src)
	new /obj/item/disk/nifsoft_uploader/money_sense(src)

/obj/item/storage/box/nif_ghost_box/ghost_role/PopulateContents()
	. = ..()
	new /obj/item/disk/nifsoft_uploader/hivemind(src)

#undef NIF_CALIBRATION_STAGE_1
#undef NIF_CALIBRATION_STAGE_1_END
#undef NIF_CALIBRATION_STAGE_2
#undef NIF_CALIBRATION_STAGE_2_END
#undef NIF_CALIBRATION_STAGE_FINISHED
#undef NIF_DURABILITY_LOSS_HALVED
#undef NIF_MINIMUM_DURABILITY
#undef NIF_MINIMUM_POWER_LEVEL
#undef NIF_SETUP_BLINDNESS
#undef MAX_NIF_REWARDS_POINTS