Add ChorusFruitPlantType

Author: Alexander01998

src/main/java/net/wurstclient/hacks/autofarm/AutoFarmPlantTypeManager.java

@@ -24,6 +24,8 @@ public final class AutoFarmPlantTypeManager
 	public final BeetrootsPlantType beetrootsType = new BeetrootsPlantType();
 	public final CactusPlantType cactusType = new CactusPlantType();
 	public final CarrotsPlantType carrotsType = new CarrotsPlantType();
+	public final ChorusFruitPlantType chorusFruitType =
+		new ChorusFruitPlantType();
 	public final CocoaBeanPlantType cocoaBeanType = new CocoaBeanPlantType();
 	public final GlowBerryPlantType glowBerryType = new GlowBerryPlantType();
 	public final KelpPlantType kelpType = new KelpPlantType();
@@ -44,11 +46,12 @@ public final class AutoFarmPlantTypeManager
 		new WeepingVinesPlantType();
 	public final WheatPlantType wheatType = new WheatPlantType();
 
-	public final List<AutoFarmPlantType> plantTypes = List.of(amethystType,
-		bambooType, beetrootsType, cactusType, carrotsType, cocoaBeanType,
-		glowBerryType, kelpType, melonType, netherWartType, pitcherPlantType,
-		potatoesType, pumpkinType, sugarCaneType, sweetBerryPlantType,
-		torchflowerType, twistingVinesType, weepingVinesType, wheatType);
+	public final List<AutoFarmPlantType> plantTypes =
+		List.of(amethystType, bambooType, beetrootsType, cactusType,
+			carrotsType, chorusFruitType, cocoaBeanType, glowBerryType,
+			kelpType, melonType, netherWartType, pitcherPlantType, potatoesType,
+			pumpkinType, sugarCaneType, sweetBerryPlantType, torchflowerType,
+			twistingVinesType, weepingVinesType, wheatType);
 
 	public final ToggleAllPlantTypesSetting toggleAllSetting =
 		new ToggleAllPlantTypesSetting("All plant types",

src/main/java/net/wurstclient/hacks/autofarm/plants/ChorusFruitPlantType.java

@@ -0,0 +1,106 @@
+/*
+ * Copyright (c) 2014-2025 Wurst-Imperium and contributors.
+ *
+ * This source code is subject to the terms of the GNU General Public
+ * License, version 3. If a copy of the GPL was not distributed with this
+ * file, You can obtain one at: https://www.gnu.org/licenses/gpl-3.0.txt
+ */
+package net.wurstclient.hacks.autofarm.plants;
+
+import java.util.EnumMap;
+import java.util.HashSet;
+import java.util.Map;
+import java.util.Map.Entry;
+
+import com.google.common.collect.Maps;
+
+import net.minecraft.block.BlockState;
+import net.minecraft.block.Blocks;
+import net.minecraft.block.ChorusFlowerBlock;
+import net.minecraft.block.ConnectingBlock;
+import net.minecraft.item.Item;
+import net.minecraft.item.Items;
+import net.minecraft.state.property.BooleanProperty;
+import net.minecraft.util.math.BlockPos;
+import net.minecraft.util.math.Direction;
+import net.wurstclient.hacks.autofarm.AutoFarmPlantType;
+import net.wurstclient.settings.PlantTypeSetting;
+import net.wurstclient.util.BlockUtils;
+
+public final class ChorusFruitPlantType extends AutoFarmPlantType
+{
+	private static final EnumMap<Direction, BooleanProperty> CHORUS_GROWING_DIRECTIONS =
+		Maps.newEnumMap(Map.of(Direction.NORTH, ConnectingBlock.NORTH,
+			Direction.SOUTH, ConnectingBlock.SOUTH, Direction.WEST,
+			ConnectingBlock.WEST, Direction.EAST, ConnectingBlock.EAST,
+			Direction.UP, ConnectingBlock.UP));
+	
+	@Override
+	public final boolean isReplantingSpot(BlockPos pos, BlockState state)
+	{
+		return (state.isOf(Blocks.CHORUS_FLOWER)
+			|| state.isOf(Blocks.CHORUS_PLANT)) && hasPlantingSurface(pos);
+	}
+	
+	@Override
+	public final boolean hasPlantingSurface(BlockPos pos)
+	{
+		return BlockUtils.getState(pos.down()).isOf(Blocks.END_STONE);
+	}
+	
+	@Override
+	public Item getSeedItem()
+	{
+		return Items.CHORUS_FLOWER;
+	}
+	
+	@Override
+	public boolean shouldHarvestByMining(BlockPos pos, BlockState state)
+	{
+		if(state.isOf(Blocks.CHORUS_FLOWER)
+			&& state.get(ChorusFlowerBlock.AGE, 0) == ChorusFlowerBlock.MAX_AGE)
+			return true;
+		
+		if(!state.isOf(Blocks.CHORUS_PLANT))
+			return false;
+		
+		return !hasAttachedFlowers(pos, state, new HashSet<>());
+	}
+	
+	private boolean hasAttachedFlowers(BlockPos pos, BlockState state,
+		HashSet<BlockPos> visited)
+	{
+		// If the plant is unreasonably large, just assume there's a flower
+		// somewhere so we don't harvest it
+		if(visited.size() > 1000)
+			return true;
+		
+		if(!visited.add(pos))
+			return false;
+		
+		for(Entry<Direction, BooleanProperty> entry : CHORUS_GROWING_DIRECTIONS
+			.entrySet())
+		{
+			if(!state.get(entry.getValue(), false))
+				continue;
+			
+			BlockPos neighborPos = pos.offset(entry.getKey());
+			BlockState neighborState = BlockUtils.getState(neighborPos);
+			if(neighborState.isOf(Blocks.CHORUS_FLOWER))
+				return true;
+			
+			if(neighborState.isOf(Blocks.CHORUS_PLANT)
+				&& hasAttachedFlowers(neighborPos, neighborState, visited))
+				return true;
+		}
+		
+		return false;
+	}
+	
+	@Override
+	protected PlantTypeSetting createSetting()
+	{
+		return new PlantTypeSetting("Chorus Fruit", Items.CHORUS_FRUIT, true,
+			true);
+	}
+}