public class LightningDragonBoneNunchaku extends BaseNunchakuItem {

	public LightningDragonBoneNunchaku(Tier tier, Properties properties, float attackDamage, float attackSpeed) {
		super(tier, properties, attackDamage, attackSpeed);
	}

	@Override
	public boolean hurtEnemy(ItemStack stack, LivingEntity target, LivingEntity source) {
		boolean modloaded = ModList.get().isLoaded("iceandfire");
		boolean flag = super.hurtEnemy(stack, target, source);
		if (source instanceof Player player) {
			// Since i dont know how to apply InF lightning effects i will just apply the
			// default minecraft one
			LightningBolt lightningBolt = EntityType.LIGHTNING_BOLT.create(player.level());
			if (lightningBolt != null) {
				lightningBolt.moveTo(target.getX(), target.getY(), target.getZ());
				lightningBolt.setVisualOnly(true); // This makes it not cause damage/fire.
				player.level().addFreshEntity(lightningBolt);
			}
			applyChainLightning(target, player, modloaded);
			target.knockback(1F, source.getX() - target.getX(), source.getZ() - target.getZ());
		}
		return flag;
	}

	
	public void applyChainLightning(LivingEntity target, Player player, boolean modloaded) {
	    // Maximum number of additional targets in the chain (main target + 4 additional = 5 total)
		
	    final int maxChainCount = 4;
	    // Maximum chain range (in blocks)
	    final double chainRange = 5.0D;
	    List<LivingEntity> chainTargets = new ArrayList<>();
	    chainTargets.add(target);
	    LivingEntity current = target;
	    for (int i = 0; i < maxChainCount; i++) {
	        // Define a search area centered at the current target
	        AABB searchBox = new AABB(
	            current.getX() - chainRange, current.getY() - chainRange, current.getZ() - chainRange,
	            current.getX() + chainRange, current.getY() + chainRange, current.getZ() + chainRange
	        );
	        List<LivingEntity> candidates = player.level().getEntitiesOfClass(LivingEntity.class,
	                searchBox,
	                e -> e != player && e.isAlive() && !chainTargets.contains(e));
	        if (candidates.isEmpty()) {
	            break;
	        }
	        LivingEntity nextTarget = candidates.get(player.getRandom().nextInt(candidates.size()));
	        chainTargets.add(nextTarget);
	        current = nextTarget;
	    }
	    float bonus = 0;
	    ResourceKey<DamageType> LIGHTNING_DAMAGE_KEY = ResourceKey.create(Registries.DAMAGE_TYPE, new ResourceLocation(NunchakusConfig.lightningDamage));
	    Holder.Reference<DamageType> customDamageType = player.level()
                .registryAccess()
                .registryOrThrow(Registries.DAMAGE_TYPE)
                .getHolderOrThrow(LIGHTNING_DAMAGE_KEY);
	    DamageSource lightningDamage = new DamageSource(customDamageType, player);
		int[] hpLosses = {5, 4, 3, 2, 1 };
		int[] slowDurations = {10, 8, 6, 4, 2 };
		// Loop over each extra target and apply the damage and effect
		LivingEntity last = target;
		for (int i = 0; i < chainTargets.size(); i++) {
			LivingEntity extra = chainTargets.get(i);
			if (modloaded) {
				if (extra instanceof EntityFireDragon || extra instanceof EntityIceDragon) {
					bonus = 4;
				} else {
					bonus = 0;
				}
			}
			extra.hurt(lightningDamage, hpLosses[i] + bonus);
			extra.addEffect(
					new MobEffectInstance(NunchakusConfig.lightningMobEffect, slowDurations[i], 4, false, false));
			last = extra;
		}
	}

}

@SubscribeEvent
	public static void onGemDropped(ItemTossEvent event) {
		Player entity = event.getPlayer();
		ItemStack itemstack = event.getEntity().getItem();
		if (EnchantmentHelper.getItemEnchantmentLevel(JlmeModEnchantments.CURSE_OF_POSSESSION.get(), itemstack) != 0 && !entity.level().isClientSide()) {
			if (entity.getMainHandItem().isEmpty()) {
				JlmeMod.LOGGER.info("Stack dropped is: " + itemstack);
				ItemStack _setstack = itemstack.copy();
				entity.setItemInHand(InteractionHand.MAIN_HAND, _setstack);
				JlmeMod.LOGGER.info("Stack in mainhand after calling method: " + entity.getMainHandItem());
				JlmeMod.LOGGER.info("Stack copy is: " + _setstack);
			} else {
				ItemStack _setstack = itemstack.copy();
				ItemHandlerHelper.giveItemToPlayer(entity, _setstack);
			}
			event.setCanceled(true);
		}
	}

@Nullable
   public static ItemEntity onPlayerTossEvent(@NotNull Player player, @NotNull ItemStack item, boolean includeName)
   {
       player.captureDrops(Lists.newArrayList());
       ItemEntity ret = player.drop(item, false, includeName);
       player.captureDrops(null);

       if (ret == null)
           return null;

       ItemTossEvent event = new ItemTossEvent(ret, player);
       if (MinecraftForge.EVENT_BUS.post(event))
           return null;

       if (!player.level().isClientSide)
           player.getCommandSenderWorld().addFreshEntity(event.getEntity());
       return event.getEntity();
   }

ItemTossEvent event = new ItemTossEvent(ret, player);
        if (MinecraftForge.EVENT_BUS.post(event))
            return null;

            ItemStack equippedStack = p.getMainHandItem();
                if (ItemStack.EMPTY != equippedStack) {
                    //Since the event is meant to only avoid the creation of an ItemEntity and not to avoid the removal of the ItemStack
                    //This kinda of makes sense
                    entity.setItemInHand(InteractionHand.MAIN_HAND, ItemStack.EMPTY);
                    ItemTossEvent event = new ItemTossEvent(new ItemEntity(p.level(), p.getX(), p.getY(), p.getZ(), equippedStack), player);
                    if (!MinecraftForge.EVENT_BUS.post(event)) {
                        // drop may be cancelled by forge event hook, but if it was, restoring the lost item
                        // is the responsibility of that outside party
                        drop.setPickUpDelay(50);
                        mob.level().playSound(null, mob.blockPosition(), SoundEvents.SLIME_ATTACK, SoundSource.HOSTILE, 1.0F + mob.getRandom().nextFloat(), mob.getRandom().nextFloat() * 0.7F + 0.3F);
                    }
                }