github-repos/mirai
1
0
Fork 0
mirror of https://github.com/mamoe/mirai.git synced 2025-03-13 06:30:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

Extract enable and disable functions

Browse Source
This commit is contained in:
Him188 2020-01-19 00:13:19 +08:00
parent 37b280d491
commit 5779e077b8
1 changed files with 19 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
mirai-console/src/main/kotlin/net/mamoe/mirai/plugin/PluginBase.kt
View File

@ -1,6 +1,6 @@
package net.mamoe.mirai.plugin
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.*
import net.mamoe.mirai.utils.DefaultLogger
import net.mamoe.mirai.utils.io.encodeToString
import java.io.File
@ -11,8 +11,11 @@ import kotlin.coroutines.CoroutineContext
import kotlin.coroutines.EmptyCoroutineContext
abstract class PluginBase : CoroutineScope {
override val coroutineContext: CoroutineContext = EmptyCoroutineContext
abstract class PluginBase(coroutineContext: CoroutineContext) : CoroutineScope {
constructor() : this(EmptyCoroutineContext)
private val supervisorJob = SupervisorJob()
final override val coroutineContext: CoroutineContext = coroutineContext + supervisorJob
val dataFolder: File by lazy {
File(PluginManager.pluginsPath + pluginDescription.name).also { it.mkdir() }
@ -39,6 +42,15 @@ abstract class PluginBase : CoroutineScope {
}
internal fun enable() {
this.onEnable()
}
@JvmOverloads
internal fun disable(throwable: CancellationException? = null) {
this.coroutineContext[Job]!!.cancelChildren(throwable)
this.onDisable()
}
private lateinit var pluginDescription: PluginDescription
@ -233,19 +245,17 @@ object PluginManager {
}
nameToPluginBaseMap.values.forEach {
it.onEnable()
it.enable()
}
}
fun disableAllPlugins() {
@JvmOverloads
fun disableAllPlugins(throwable: CancellationException? = null) {
nameToPluginBaseMap.values.forEach {
it.onDisable()
it.disable(throwable)
}
}
}