mirror of https://github.com/GeyserMC/Geyser.git
261 lines
9.3 KiB
Java
261 lines
9.3 KiB
Java
/*
|
|
* Copyright (c) 2019-2022 GeyserMC. http://geysermc.org
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in
|
|
* all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
* THE SOFTWARE.
|
|
*
|
|
* @author GeyserMC
|
|
* @link https://github.com/GeyserMC/Geyser
|
|
*/
|
|
|
|
package org.geysermc.geyser.platform.velocity;
|
|
|
|
import com.google.inject.Inject;
|
|
import com.velocitypowered.api.command.CommandSource;
|
|
import com.velocitypowered.api.event.Subscribe;
|
|
import com.velocitypowered.api.event.proxy.ListenerBoundEvent;
|
|
import com.velocitypowered.api.event.proxy.ProxyInitializeEvent;
|
|
import com.velocitypowered.api.event.proxy.ProxyShutdownEvent;
|
|
import com.velocitypowered.api.network.ListenerType;
|
|
import com.velocitypowered.api.network.ProtocolVersion;
|
|
import com.velocitypowered.api.plugin.Plugin;
|
|
import com.velocitypowered.api.plugin.PluginContainer;
|
|
import com.velocitypowered.api.proxy.ProxyServer;
|
|
import lombok.Getter;
|
|
import org.checkerframework.checker.nullness.qual.NonNull;
|
|
import org.checkerframework.checker.nullness.qual.Nullable;
|
|
import org.geysermc.geyser.GeyserBootstrap;
|
|
import org.geysermc.geyser.GeyserImpl;
|
|
import org.geysermc.geyser.api.util.PlatformType;
|
|
import org.geysermc.geyser.command.CommandRegistry;
|
|
import org.geysermc.geyser.command.CommandSourceConverter;
|
|
import org.geysermc.geyser.command.GeyserCommandSource;
|
|
import org.geysermc.geyser.configuration.GeyserConfiguration;
|
|
import org.geysermc.geyser.dump.BootstrapDumpInfo;
|
|
import org.geysermc.geyser.network.GameProtocol;
|
|
import org.geysermc.geyser.ping.GeyserLegacyPingPassthrough;
|
|
import org.geysermc.geyser.ping.IGeyserPingPassthrough;
|
|
import org.geysermc.geyser.platform.velocity.command.VelocityCommandSource;
|
|
import org.geysermc.geyser.text.GeyserLocale;
|
|
import org.geysermc.geyser.util.FileUtils;
|
|
import org.incendo.cloud.CommandManager;
|
|
import org.incendo.cloud.execution.ExecutionCoordinator;
|
|
import org.incendo.cloud.velocity.VelocityCommandManager;
|
|
import org.slf4j.Logger;
|
|
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.net.SocketAddress;
|
|
import java.nio.file.Path;
|
|
import java.nio.file.Paths;
|
|
import java.util.UUID;
|
|
|
|
@Plugin(id = "geyser", name = GeyserImpl.NAME + "-Velocity", version = GeyserImpl.VERSION, url = "https://geysermc.org", authors = "GeyserMC")
|
|
public class GeyserVelocityPlugin implements GeyserBootstrap {
|
|
@Inject
|
|
private Logger logger;
|
|
|
|
@Inject
|
|
private ProxyServer proxyServer;
|
|
|
|
@Inject
|
|
private PluginContainer container;
|
|
|
|
private CommandRegistry commandRegistry;
|
|
private GeyserVelocityConfiguration geyserConfig;
|
|
private GeyserVelocityInjector geyserInjector;
|
|
private GeyserVelocityLogger geyserLogger;
|
|
private IGeyserPingPassthrough geyserPingPassthrough;
|
|
|
|
private GeyserImpl geyser;
|
|
|
|
@Getter
|
|
private final Path configFolder = Paths.get("plugins/" + GeyserImpl.NAME + "-Velocity/");
|
|
|
|
@Override
|
|
public void onGeyserInitialize() {
|
|
GeyserLocale.init(this);
|
|
|
|
if (!ProtocolVersion.isSupported(GameProtocol.getJavaProtocolVersion())) {
|
|
logger.error(" / \\");
|
|
logger.error(" / \\");
|
|
logger.error(" / | \\");
|
|
logger.error(" / | \\ " + GeyserLocale.getLocaleStringLog("geyser.bootstrap.unsupported_proxy", proxyServer.getVersion().getName()));
|
|
logger.error(" / \\ " + GeyserLocale.getLocaleStringLog("geyser.may_not_work_as_intended_all_caps"));
|
|
logger.error(" / o \\");
|
|
logger.error("/_____________\\");
|
|
}
|
|
|
|
if (!loadConfig()) {
|
|
return;
|
|
}
|
|
this.geyserLogger = new GeyserVelocityLogger(logger, geyserConfig.isDebugMode());
|
|
GeyserConfiguration.checkGeyserConfiguration(geyserConfig, geyserLogger);
|
|
|
|
this.geyser = GeyserImpl.load(PlatformType.VELOCITY, this);
|
|
this.geyserInjector = new GeyserVelocityInjector(proxyServer);
|
|
}
|
|
|
|
@Override
|
|
public void onGeyserEnable() {
|
|
if (GeyserImpl.getInstance().isReloading()) {
|
|
if (!loadConfig()) {
|
|
return;
|
|
}
|
|
this.geyserLogger.setDebug(geyserConfig.isDebugMode());
|
|
GeyserConfiguration.checkGeyserConfiguration(geyserConfig, geyserLogger);
|
|
} else {
|
|
var sourceConverter = new CommandSourceConverter<>(
|
|
CommandSource.class,
|
|
id -> proxyServer.getPlayer(id).orElse(null),
|
|
proxyServer::getConsoleCommandSource,
|
|
VelocityCommandSource::new
|
|
);
|
|
CommandManager<GeyserCommandSource> cloud = new VelocityCommandManager<>(
|
|
container,
|
|
proxyServer,
|
|
ExecutionCoordinator.simpleCoordinator(),
|
|
sourceConverter
|
|
);
|
|
this.commandRegistry = new CommandRegistry(geyser, cloud);
|
|
}
|
|
|
|
GeyserImpl.start();
|
|
|
|
if (geyserConfig.isLegacyPingPassthrough()) {
|
|
this.geyserPingPassthrough = GeyserLegacyPingPassthrough.init(geyser);
|
|
} else {
|
|
this.geyserPingPassthrough = new GeyserVelocityPingPassthrough(proxyServer);
|
|
}
|
|
|
|
// No need to re-register events
|
|
if (!GeyserImpl.getInstance().isReloading()) {
|
|
proxyServer.getEventManager().register(this, new GeyserVelocityUpdateListener());
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void onGeyserDisable() {
|
|
if (geyser != null) {
|
|
geyser.disable();
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void onGeyserShutdown() {
|
|
if (geyser != null) {
|
|
geyser.shutdown();
|
|
}
|
|
if (geyserInjector != null) {
|
|
geyserInjector.shutdown();
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public GeyserVelocityConfiguration getGeyserConfig() {
|
|
return geyserConfig;
|
|
}
|
|
|
|
@Override
|
|
public GeyserVelocityLogger getGeyserLogger() {
|
|
return geyserLogger;
|
|
}
|
|
|
|
@Override
|
|
public CommandRegistry getCommandRegistry() {
|
|
return this.commandRegistry;
|
|
}
|
|
|
|
@Override
|
|
public IGeyserPingPassthrough getGeyserPingPassthrough() {
|
|
return geyserPingPassthrough;
|
|
}
|
|
|
|
@Subscribe
|
|
public void onInit(ProxyInitializeEvent event) {
|
|
this.onGeyserInitialize();
|
|
}
|
|
|
|
@Subscribe
|
|
public void onShutdown(ProxyShutdownEvent event) {
|
|
this.onGeyserShutdown();
|
|
}
|
|
|
|
@Subscribe
|
|
public void onProxyBound(ListenerBoundEvent event) {
|
|
if (event.getListenerType() == ListenerType.MINECRAFT) {
|
|
// Once listener is bound, do our startup process
|
|
this.onGeyserEnable();
|
|
|
|
if (geyserInjector != null) {
|
|
// After this bound, we know that the channel initializer cannot change without it being ineffective for Velocity, too
|
|
geyserInjector.initializeLocalChannel(this);
|
|
}
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public BootstrapDumpInfo getDumpInfo() {
|
|
return new GeyserVelocityDumpInfo(proxyServer);
|
|
}
|
|
|
|
@Nullable
|
|
@Override
|
|
public SocketAddress getSocketAddress() {
|
|
return this.geyserInjector.getServerSocketAddress();
|
|
}
|
|
|
|
@NonNull
|
|
@Override
|
|
public String getServerBindAddress() {
|
|
return proxyServer.getBoundAddress().getHostString();
|
|
}
|
|
|
|
@Override
|
|
public int getServerPort() {
|
|
return proxyServer.getBoundAddress().getPort();
|
|
}
|
|
|
|
@Override
|
|
public boolean testFloodgatePluginPresent() {
|
|
var floodgate = proxyServer.getPluginManager().getPlugin("floodgate");
|
|
if (floodgate.isPresent()) {
|
|
geyserConfig.loadFloodgate(this, proxyServer, configFolder.toFile());
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
@SuppressWarnings("BooleanMethodIsAlwaysInverted")
|
|
private boolean loadConfig() {
|
|
try {
|
|
if (!configFolder.toFile().exists())
|
|
//noinspection ResultOfMethodCallIgnored
|
|
configFolder.toFile().mkdirs();
|
|
File configFile = FileUtils.fileOrCopiedFromResource(configFolder.resolve("config.yml").toFile(),
|
|
"config.yml", (x) -> x.replaceAll("generateduuid", UUID.randomUUID().toString()), this);
|
|
this.geyserConfig = FileUtils.loadConfig(configFile, GeyserVelocityConfiguration.class);
|
|
} catch (IOException ex) {
|
|
logger.error(GeyserLocale.getLocaleStringLog("geyser.config.failed"), ex);
|
|
ex.printStackTrace();
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
}
|