package moe.nea.firmament.events

import java.util.concurrent.CompletableFuture
import java.util.concurrent.Executor
import net.minecraft.resource.ReloadableResourceManagerImpl
import net.minecraft.resource.ResourceManager
import net.minecraft.resource.ResourceReloader

data class FinalizeResourceManagerEvent(
	val resourceManager: ReloadableResourceManagerImpl,
) : FirmamentEvent() {
	companion object : FirmamentEventBus<FinalizeResourceManagerEvent>()

	inline fun registerOnApply(name: String, crossinline function: () -> Unit) {
		resourceManager.registerReloader(object : ResourceReloader {
			override fun reload(
				store: ResourceReloader.Store,
				prepareExecutor: Executor,
				reloadSynchronizer: ResourceReloader.Synchronizer,
				applyExecutor: Executor
			): CompletableFuture<Void> {
				return CompletableFuture.completedFuture(Unit)
					.thenCompose(reloadSynchronizer::whenPrepared)
					.thenAcceptAsync({ function() }, applyExecutor)
			}

			override fun getName(): String {
				return name
			}
		})
	}
}