yuzu/Minikura
1
0
Fork 0
mirror of https://github.com/YuzuZensai/Minikura.git synced 2026-03-31 05:31:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: websocket auto detect server info change

Browse Source
This commit is contained in:
Nakiri
2025-06-01 16:27:34 +07:00
parent e843480f17
commit 367be7dc11
13 changed files with 248 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
plugins/MinikuraVelocity/src/main/kotlin/cafe/kirameki/minikuraVelocity/Main.kt
View File

@@ -41,8 +41,8 @@ class Main @Inject constructor(private val logger: Logger, private val server: P
private val servers: MutableMap<String, RegisteredServer> = HashMap()
private val client = OkHttpClient()
private val apiKey: String = System.getenv("MINIKURA_API_KEY") ?: ""
private val apiUrl: String = System.getenv("MINIKURA_API_URL") ?: "http://localhost:3000"
private val websocketUrl: String = System.getenv("MINIKURA_WEBSOCKET_URL") ?: "ws://localhost:3000/ws"
private val apiUrl: String = System.getenv("MINIKURA_API_URL") ?: "http://localhost:3000/api"
private val websocketUrl: String = System.getenv("MINIKURA_WEBSOCKET_URL") ?: "ws://localhost:3000/ws?apiKey=$apiKey"
private var acceptingTransfers = AtomicBoolean(false)
private val redisBungeeApi = RedisBungeeAPI.getRedisBungeeApi()
@@ -87,6 +87,10 @@ class Main @Inject constructor(private val logger: Logger, private val server: P
.plugin(this)
.build()
val getallServerCommandMeta: CommandMeta = commandManager.metaBuilder("getallserver")
.plugin(this)
.build()
// TODO: Rework this command and support <origin> and <destination> arguments
val migrateCommand = SimpleCommand { p ->
val source = p.source()
@@ -114,10 +118,15 @@ class Main @Inject constructor(private val logger: Logger, private val server: P
source.sendMessage(Component.text("Migrating players to server '$targetServerName'..."))
}
val getallServerCommand = SimpleCommand { p ->
getallServer()
}
commandManager.register(serversCommandMeta, ServerCommand.createServerCommand(server))
commandManager.register(endCommandMeta, EndCommand.createEndCommand(server))
commandManager.register(refreshCommandMeta, refreshCommand)
commandManager.register(migrateCommandMeta, migrateCommand)
commandManager.register(getallServerCommandMeta, getallServerCommand)
val connectionHandler = ProxyTransferHandler(servers, logger, acceptingTransfers)
server.eventManager.register(this, connectionHandler)
@@ -132,6 +141,22 @@ class Main @Inject constructor(private val logger: Logger, private val server: P
logger.info("Minikura-Velocity has been initialized.")
}
fun refreshServers() {
fetchServers()
fetchReverseProxyServers()
}
fun getallServer() {
synchronized(ServerDataStore) {
ServerDataStore.getServers().forEach { serverData ->
logger.info("Server ID: ${serverData.id}, Type: ${serverData.type}, Description: ${serverData.description}")
}
ServerDataStore.getReverseProxyServers().forEach { reverseProxyServerData ->
logger.info("Reverse Proxy Server ID: ${reverseProxyServerData.id}, External Address: ${reverseProxyServerData.external_address}, External Port: ${reverseProxyServerData.external_port}")
}
}
}
private fun fetchReverseProxyServers() {
val request = Request.Builder()
.url("$apiUrl/reverse_proxy_servers")
@@ -207,9 +232,9 @@ class Main @Inject constructor(private val logger: Logger, private val server: P
servers.clear()
for (data in serversData) {
val serverInfo = ServerInfo(data.name, InetSocketAddress(data.address, data.port))
val serverInfo = ServerInfo(data.id, InetSocketAddress("localhost", data.listen_port))
val registeredServer = server.createRawRegisteredServer(serverInfo)
servers[data.name] = registeredServer
servers[data.id] = registeredServer
this.server.registerServer(registeredServer.serverInfo)
}
}