com.blr19c.falowp.bot.system.plugin.hook.HookJoinPoint.kt Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of falowp-bot-system Show documentation

FalowpBot system infrastructure

There is a newer version: 2.0.0-RC2

package com.blr19c.falowp.bot.system.plugin.hook

import com.blr19c.falowp.bot.system.api.BotApi
import com.blr19c.falowp.bot.system.plugin.Plugin
import com.blr19c.falowp.bot.system.plugin.PluginInfo
import java.util.concurrent.atomic.AtomicReference

/**
 * hook接入点
 */
class HookJoinPoint(
    private val pluginInfo: PluginInfo?,
    private val hook: Plugin.Listener.Hook,
    private val hooks: ArrayDeque,
) {

    private val terminate = AtomicReference(false)
    private val currentHookProcess = AtomicReference(null)

    /**
     * 执行原目标方法/或者执行下一个
     */
    suspend fun process() {
        if (terminate.get()) return
        val process = hooks.removeFirstOrNull() ?: return
        currentHookProcess.set(process)
        process.process(this, hook)
    }

    /**
     * 终止执行
     */
    fun terminate() {
        terminate.set(true)
    }

    /**
     * 当前HookProcess的botApi
     */
    fun botApi(): BotApi = currentHookProcess.get().botApi()

    fun pluginInfo() = pluginInfo
}