aboutsummaryrefslogtreecommitdiff
path: root/src/main/kotlin/repo/ItemCache.kt
blob: 23c5ffb873bc5fd52e4a839dd5d5f8ab65b050e3 (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
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
package moe.nea.firmament.repo

import com.mojang.serialization.Dynamic
import io.github.moulberry.repo.IReloadable
import io.github.moulberry.repo.NEURepository
import io.github.moulberry.repo.data.NEUItem
import io.github.notenoughupdates.moulconfig.xml.Bind
import java.text.NumberFormat
import java.util.UUID
import java.util.concurrent.ConcurrentHashMap
import org.apache.logging.log4j.LogManager
import kotlinx.coroutines.Job
import kotlinx.coroutines.launch
import kotlin.jvm.optionals.getOrNull
import net.minecraft.SharedConstants
import net.minecraft.client.resource.language.I18n
import net.minecraft.component.DataComponentTypes
import net.minecraft.component.type.NbtComponent
import net.minecraft.datafixer.Schemas
import net.minecraft.datafixer.TypeReferences
import net.minecraft.item.ItemStack
import net.minecraft.item.Items
import net.minecraft.nbt.NbtCompound
import net.minecraft.nbt.NbtElement
import net.minecraft.nbt.NbtOps
import net.minecraft.nbt.NbtString
import net.minecraft.text.Text
import moe.nea.firmament.Firmament
import moe.nea.firmament.gui.config.HudMeta
import moe.nea.firmament.gui.config.HudPosition
import moe.nea.firmament.gui.hud.MoulConfigHud
import moe.nea.firmament.repo.RepoManager.initialize
import moe.nea.firmament.util.LegacyTagParser
import moe.nea.firmament.util.MC
import moe.nea.firmament.util.SkyblockId
import moe.nea.firmament.util.TestUtil
import moe.nea.firmament.util.mc.FirmamentDataComponentTypes
import moe.nea.firmament.util.mc.appendLore
import moe.nea.firmament.util.mc.modifyLore
import moe.nea.firmament.util.mc.setCustomName
import moe.nea.firmament.util.mc.setSkullOwner

object ItemCache : IReloadable {
	private val cache: MutableMap<String, ItemStack> = ConcurrentHashMap()
	private val df = Schemas.getFixer()
	val logger = LogManager.getLogger("${Firmament.logger.name}.ItemCache")
	var isFlawless = true
		private set

	private fun NEUItem.get10809CompoundTag(): NbtCompound = NbtCompound().apply {
		put("tag", LegacyTagParser.parse(nbttag))
		putString("id", minecraftItemId)
		putByte("Count", 1)
		putShort("Damage", damage.toShort())
	}

	private fun NbtCompound.transformFrom10809ToModern(): NbtCompound? =
		try {
			df.update(
				TypeReferences.ITEM_STACK,
				Dynamic(NbtOps.INSTANCE, this),
				-1,
				SharedConstants.getGameVersion().saveVersion.id
			).value as NbtCompound
		} catch (e: Exception) {
			isFlawless = false
			logger.error("Could not data fix up $this", e)
			null
		}

	val ItemStack.isBroken
		get() = get(FirmamentDataComponentTypes.IS_BROKEN) ?: false

	fun ItemStack.withFallback(fallback: ItemStack?): ItemStack {
		if (isBroken && fallback != null) return fallback
		return this
	}

	fun brokenItemStack(neuItem: NEUItem?, idHint: SkyblockId? = null): ItemStack {
		return ItemStack(Items.PAINTING).apply {
			setCustomName(Text.literal(neuItem?.displayName ?: idHint?.neuItem ?: "null"))
			appendLore(
				listOf(
					Text.stringifiedTranslatable(
						"firmament.repo.brokenitem",
						(neuItem?.skyblockItemId ?: idHint)
					)
				)
			)
			set(DataComponentTypes.CUSTOM_DATA, NbtComponent.of(NbtCompound().apply {
				put("ID", NbtString.of(neuItem?.skyblockItemId ?: idHint?.neuItem ?: "null"))
			}))
			set(FirmamentDataComponentTypes.IS_BROKEN, true)
		}
	}

	private fun NEUItem.asItemStackNow(): ItemStack {
		try {
			val oldItemTag = get10809CompoundTag()
			val modernItemTag = oldItemTag.transformFrom10809ToModern()
				?: return brokenItemStack(this)
			val itemInstance =
				ItemStack.fromNbt(MC.defaultRegistries, modernItemTag).getOrNull() ?: return brokenItemStack(this)
			val extraAttributes = oldItemTag.getCompound("tag").getCompound("ExtraAttributes")
			if (extraAttributes != null)
				itemInstance.set(DataComponentTypes.CUSTOM_DATA, NbtComponent.of(extraAttributes))
			return itemInstance
		} catch (e: Exception) {
			e.printStackTrace()
			return brokenItemStack(this)
		}
	}

	fun NEUItem?.asItemStack(idHint: SkyblockId? = null, loreReplacements: Map<String, String>? = null): ItemStack {
		if (this == null) return brokenItemStack(null, idHint)
		var s = cache[this.skyblockItemId]
		if (s == null) {
			s = asItemStackNow()
			cache[this.skyblockItemId] = s
		}
		if (!loreReplacements.isNullOrEmpty()) {
			s = s.copy()!!
			s.applyLoreReplacements(loreReplacements)
			s.setCustomName(s.name.applyLoreReplacements(loreReplacements))
		}
		return s
	}

	fun ItemStack.applyLoreReplacements(loreReplacements: Map<String, String>) {
		modifyLore { lore ->
			lore.map {
				it.applyLoreReplacements(loreReplacements)
			}
		}
	}

	fun Text.applyLoreReplacements(loreReplacements: Map<String, String>): Text {
		assert(this.siblings.isEmpty())
		var string = this.string
		loreReplacements.forEach { (find, replace) ->
			string = string.replace("{$find}", replace)
		}
		return Text.literal(string).styled { this.style }
	}

	var job: Job? = null

	object ReloadProgressHud : MoulConfigHud(
		"repo_reload", HudMeta(HudPosition(0.0, 0.0, 1F), Text.literal("Repo Reload"), 180, 18)) {


		var isEnabled = false
		override fun shouldRender(): Boolean {
			return isEnabled
		}

		@get:Bind("current")
		var current: Double = 0.0

		@get:Bind("label")
		var label: String = ""

		@get:Bind("max")
		var max: Double = 0.0

		fun reportProgress(label: String, current: Int, max: Int) {
			this.label = label
			this.current = current.toDouble()
			this.max = max.toDouble()
		}
	}

	override fun reload(repository: NEURepository) {
		val j = job
		if (j != null && j.isActive) {
			j.cancel()
		}
		cache.clear()
		isFlawless = true
		if (TestUtil.isInTest) return
		job = Firmament.coroutineScope.launch {
			val items = repository.items?.items
			if (items == null) {
				ReloadProgressHud.isEnabled = false
				return@launch
			}
			val recacheItems = I18n.translate("firmament.repo.cache")
			ReloadProgressHud.reportProgress(recacheItems, 0, items.size)
			ReloadProgressHud.isEnabled = true
			var i = 0
			items.values.forEach {
				it.asItemStack() // Rebuild cache
				ReloadProgressHud.reportProgress(recacheItems, i++, items.size)
			}
			ReloadProgressHud.isEnabled = false
		}
	}

	fun coinItem(coinAmount: Int): ItemStack {
		var uuid = UUID.fromString("2070f6cb-f5db-367a-acd0-64d39a7e5d1b")
		var texture =
			"http://textures.minecraft.net/texture/538071721cc5b4cd406ce431a13f86083a8973e1064d2f8897869930ee6e5237"
		if (coinAmount >= 100000) {
			uuid = UUID.fromString("94fa2455-2881-31fe-bb4e-e3e24d58dbe3")
			texture =
				"http://textures.minecraft.net/texture/c9b77999fed3a2758bfeaf0793e52283817bea64044bf43ef29433f954bb52f6"
		}
		if (coinAmount >= 10000000) {
			uuid = UUID.fromString("0af8df1f-098c-3b72-ac6b-65d65fd0b668")
			texture =
				"http://textures.minecraft.net/texture/7b951fed6a7b2cbc2036916dec7a46c4a56481564d14f945b6ebc03382766d3b"
		}
		val itemStack = ItemStack(Items.PLAYER_HEAD)
		itemStack.setCustomName(Text.literal("§r§6" + NumberFormat.getInstance().format(coinAmount) + " Coins"))
		itemStack.setSkullOwner(uuid, texture)
		return itemStack
	}

	init {
		if (TestUtil.isInTest) {
			initialize()
		}
	}

}


operator fun NbtCompound.set(key: String, value: String) {
	putString(key, value)
}

operator fun NbtCompound.set(key: String, value: NbtElement) {
	put(key, value)
}