tech.carcadex.kotlinbukkitkit.architecture.ThirdPartyAware.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kotlinbukkitkit-architecture Show documentation

kotlinbukkitkit-architecture

The newest version!

/*
ORIGINAL PACKAGE: package br.com.devsrsouza.kotlinbukkitapi.architecture
ORIGINAL REPOSITORY: https://github.com/DevSrSouza/KotlinBukkitAPI
AUTHOR: https://github.com/DevSrSouza

Thanks DevSrSouza for KotlinBukkitAPI
 */
package tech.carcadex.kotlinbukkitkit.architecture

import tech.carcadex.kotlinbukkitkit.architecture.extensions.WithPlugin
import org.bukkit.Bukkit
import org.bukkit.event.EventHandler
import org.bukkit.event.HandlerList
import org.bukkit.event.Listener
import org.bukkit.event.server.PluginDisableEvent
import org.bukkit.plugin.Plugin
import java.util.WeakHashMap

/**
 * A holder for all instance of a class [T] that is bound a specific plugin.
 * It unregister when the plugin is disable.
 */
public abstract class PluginDisableAwareController {

    public abstract val factory: (plugin: Plugin) -> T

    private val instances: WeakHashMap = WeakHashMap()
    private val listeners: WeakHashMap = WeakHashMap()

    public fun get(plugin: Plugin): T {
        return instances.getOrPut(plugin) {
            listeners.put(
                plugin,
                DisableListener(plugin, ::disable).apply {
                    Bukkit.getServer().pluginManager.registerEvents(this, plugin)
                },
            )
            factory(plugin)
        }
    }

    public fun ensureInitialized(plugin: Plugin) {
        get(plugin)
    }

    private fun disable(plugin: Plugin) {
        instances.remove(plugin)?.onDisable()
        listeners.remove(plugin)?.also(HandlerList::unregisterAll)
    }
}

public interface PluginDisableAware : WithPlugin {
    public fun onDisable()
}

private class DisableListener(
    val plugin: Plugin,
    val onDisable: (Plugin) -> Unit,
) : Listener {
    @EventHandler
    public fun pluginDisableEvent(event: PluginDisableEvent) {
        if (event.plugin.name == plugin.name) {
            onDisable(plugin)
        }
    }
}