aboutsummaryrefslogtreecommitdiff
path: root/src/main/kotlin/features/items/BlockZapperOverlay.kt
blob: a853012710fd685fbcfe10c19996f6cd19d0bfca (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
package moe.nea.firmament.features.items

import io.github.notenoughupdates.moulconfig.ChromaColour
import java.util.LinkedList
import net.minecraft.block.Block
import net.minecraft.block.BlockState
import net.minecraft.block.Blocks
import net.minecraft.util.hit.BlockHitResult
import net.minecraft.util.hit.HitResult
import net.minecraft.util.math.BlockPos
import moe.nea.firmament.annotations.Subscribe
import moe.nea.firmament.events.WorldKeyboardEvent
import moe.nea.firmament.events.WorldRenderLastEvent
import moe.nea.firmament.util.MC
import moe.nea.firmament.util.data.Config
import moe.nea.firmament.util.data.ManagedConfig
import moe.nea.firmament.util.render.RenderInWorldContext
import moe.nea.firmament.util.skyBlockId
import moe.nea.firmament.util.skyblock.SkyBlockItems

object BlockZapperOverlay {
	val identifier: String
		get() = "block-zapper-overlay"

	@Config
	object TConfig : ManagedConfig(identifier, Category.ITEMS) {
		var blockZapperOverlay by toggle("block-zapper-overlay") { false }
		val color by colour("color") { ChromaColour.fromStaticRGB(160, 0, 0, 60) }
		var undoKey by keyBindingWithDefaultUnbound("undo-key")
	}

	val bannedZapper: List<Block> = listOf<Block>(
		Blocks.WHEAT,
		Blocks.CARROTS,
		Blocks.POTATOES,
		Blocks.PUMPKIN,
		Blocks.PUMPKIN_STEM,
		Blocks.MELON,
		Blocks.MELON_STEM,
		Blocks.CACTUS,
		Blocks.SUGAR_CANE,
		Blocks.NETHER_WART,
		Blocks.TALL_GRASS,
		Blocks.SUNFLOWER,
		Blocks.FARMLAND,
		Blocks.BREWING_STAND,
		Blocks.SNOW,
		Blocks.RED_MUSHROOM,
		Blocks.BROWN_MUSHROOM,
	)

	private val zapperOffsets: List<BlockPos> = listOf(
		BlockPos(0, 0, -1),
		BlockPos(0, 0, 1),
		BlockPos(-1, 0, 0),
		BlockPos(1, 0, 0),
		BlockPos(0, 1, 0),
		BlockPos(0, -1, 0)
	)

	// Skidded from NEU
	// Credit: https://github.com/NotEnoughUpdates/NotEnoughUpdates/blob/9b1fcfebc646e9fb69f99006327faa3e734e5f51/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/CustomItemEffects.java#L1281-L1355 (Modified)
	@Subscribe
	fun renderBlockZapperOverlay(event: WorldRenderLastEvent) {
		if (!TConfig.blockZapperOverlay) return
		val player = MC.player ?: return
		val world = player.world ?: return
		val heldItem = MC.stackInHand
		if (heldItem.skyBlockId != SkyBlockItems.BLOCK_ZAPPER) return
		val hitResult = MC.instance.crosshairTarget ?: return

		val zapperBlocks: HashSet<BlockPos> = HashSet()
		val returnablePositions = LinkedList<BlockPos>()

		if (hitResult is BlockHitResult && hitResult.type == HitResult.Type.BLOCK) {
			var pos: BlockPos = hitResult.blockPos
			val firstBlockState: BlockState = world.getBlockState(pos)
			val block = firstBlockState.block

			val initialAboveBlock = world.getBlockState(pos.up()).block
			if (!bannedZapper.contains(initialAboveBlock) && !bannedZapper.contains(block)) {
				var i = 0
				while (i < 164) {
					zapperBlocks.add(pos)
					returnablePositions.remove(pos)

					val availableNeighbors: MutableList<BlockPos> = ArrayList()

					for (offset in zapperOffsets) {
						val newPos = pos.add(offset)

						if (zapperBlocks.contains(newPos)) continue

						val state: BlockState? = world.getBlockState(newPos)
						if (state != null && state.block === block) {
							val above = newPos.up()
							val aboveBlock = world.getBlockState(above).block
							if (!bannedZapper.contains(aboveBlock)) {
								availableNeighbors.add(newPos)
							}
						}
					}

					if (availableNeighbors.size >= 2) {
						returnablePositions.add(pos)
						pos = availableNeighbors[0]
					} else if (availableNeighbors.size == 1) {
						pos = availableNeighbors[0]
					} else if (returnablePositions.isEmpty()) {
						break
					} else {
						i--
						pos = returnablePositions.last()
					}

					i++
				}
			}

			RenderInWorldContext.renderInWorld(event) {
				if (MC.player?.isSneaking ?: false) {
					zapperBlocks.forEach {
						block(it, TConfig.color.getEffectiveColourRGB())
					}
				} else {
					sharedVoxelSurface(zapperBlocks, TConfig.color.getEffectiveColourRGB())
				}
			}
		}
	}

	@Subscribe
	fun onWorldKeyboard(it: WorldKeyboardEvent) {
		if (!TConfig.undoKey.isBound) return
		if (!it.matches(TConfig.undoKey)) return
		if (MC.stackInHand.skyBlockId != SkyBlockItems.BLOCK_ZAPPER) return
		MC.sendCommand("undozap")
	}
}