-
-
Notifications
You must be signed in to change notification settings - Fork 25
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
5 changed files
with
196 additions
and
18 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
118 changes: 118 additions & 0 deletions
118
overflow-core-api/src/main/kotlin/top/mrxiaom/overflow/BotBuilder.kt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,118 @@ | ||
| package top.mrxiaom.overflow | ||
|
|
||
| import me.him188.kotlin.jvm.blocking.bridge.JvmBlockingBridge | ||
| import net.mamoe.mirai.Bot | ||
| import org.slf4j.Logger | ||
|
|
||
| /** | ||
| * Onebot 连接向导 | ||
| */ | ||
| public class BotBuilder private constructor( | ||
| private var url: String = "ws://127.0.0.1:8080", | ||
| private var reversedPort: Int = -1, | ||
| private var token: String = "", | ||
| private var retryTimes: Int = 5, | ||
| private var retryWaitMills: Long = 5000L, | ||
| private var retryRestMills: Long = 60000L, | ||
| private var printInfo: Boolean = false, | ||
| private var logger: Logger? = null | ||
| ) { | ||
| /** | ||
| * 设置用于连接时鉴权的 token | ||
| * | ||
| * 将 token 设为空则不进行鉴权 | ||
| */ | ||
| public fun token(token: String): BotBuilder = apply { | ||
| this.token = token | ||
| } | ||
|
|
||
| /** | ||
| * 设置尝试重连次数，设为 -1 时禁用自动重连 | ||
| */ | ||
| public fun retryTimes(retryTimes: Int): BotBuilder = apply { | ||
| this.retryTimes = retryTimes | ||
| } | ||
|
|
||
| /** | ||
| * 设置重连等待间隔时间 | ||
| */ | ||
| public fun retryWaitMills(retryWaitMills: Long): BotBuilder = apply { | ||
| this.retryWaitMills = retryWaitMills | ||
| } | ||
|
|
||
| /** | ||
| * 设置重连休息时间 | ||
| * | ||
| * 重连次数耗尽后，会进入休息状态，休息结束后重置重连次数，再次尝试重连 | ||
| * | ||
| * 设为 -1 禁用此功能 | ||
| */ | ||
| public fun retryRestMills(retryRestMills: Long): BotBuilder = apply { | ||
| this.retryRestMills = retryRestMills | ||
| } | ||
|
|
||
| /** | ||
| * 禁止打印连接时额外状态信息 | ||
| * | ||
| * 如 Overflow 版本、连接地址、Onebot 协议端版本等 | ||
| */ | ||
| public fun noPrintInfo(): BotBuilder = apply { | ||
| this.printInfo = false | ||
| } | ||
|
|
||
| /** | ||
| * 覆写用于 Onebot 的日志记录器 | ||
| */ | ||
| public fun overrideLogger(logger: Logger): BotBuilder = apply { | ||
| this.logger = logger | ||
| } | ||
|
|
||
| /** | ||
| * 连接到 Onebot | ||
| * | ||
| * @return 连接失败时，将返回 null | ||
| */ | ||
| @JvmBlockingBridge | ||
| public suspend fun connect(): Bot? { | ||
| return OverflowAPI.get().botStarter.start( | ||
| url = url, | ||
| reversedPort = reversedPort, | ||
| token = token, | ||
| retryTimes = retryTimes, | ||
| retryWaitMills = retryWaitMills, | ||
| retryRestMills = retryRestMills, | ||
| printInfo = printInfo, | ||
| logger = logger | ||
| ) | ||
| } | ||
|
|
||
| companion object { | ||
| /** | ||
| * 正向 WebSocket 连接 | ||
| */ | ||
| @JvmStatic | ||
| fun positive(host: String): BotBuilder = BotBuilder().also { it.url = host } | ||
|
|
||
| /** | ||
| * 反向 WebSocket 连接 | ||
| */ | ||
| @JvmStatic | ||
| fun reversed(port: Int): BotBuilder { | ||
| if (port !in 1..65535) throw IllegalStateException("无效的反向 WebSocket 端口号") | ||
| return BotBuilder().also { it.reversedPort = port } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| public interface IBotStarter { | ||
| public suspend fun start( | ||
| url: String, | ||
| reversedPort: Int, | ||
| token: String, | ||
| retryTimes: Int, | ||
| retryWaitMills: Long, | ||
| retryRestMills: Long, | ||
| printInfo: Boolean, | ||
| logger: Logger? | ||
| ): Bot? | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters