diff --git a/src/main/java/moe/yushi/authlibinjector/AuthlibInjector.java b/src/main/java/moe/yushi/authlibinjector/AuthlibInjector.java index 8ff4a55..6e5e8c6 100644 --- a/src/main/java/moe/yushi/authlibinjector/AuthlibInjector.java +++ b/src/main/java/moe/yushi/authlibinjector/AuthlibInjector.java @@ -52,6 +52,7 @@ import moe.yushi.authlibinjector.httpd.URLFilter; import moe.yushi.authlibinjector.httpd.URLProcessor; import moe.yushi.authlibinjector.transform.ClassTransformer; import moe.yushi.authlibinjector.transform.DumpClassListener; +import moe.yushi.authlibinjector.transform.support.AccountTypeTransformer; import moe.yushi.authlibinjector.transform.support.AuthServerNameInjector; import moe.yushi.authlibinjector.transform.support.AuthlibLogInterceptor; import moe.yushi.authlibinjector.transform.support.BungeeCordAllowedCharactersTransformer; @@ -295,6 +296,7 @@ public final class AuthlibInjector { config.getDecodedPublickey().ifPresent(YggdrasilKeyTransformUnit.PUBLIC_KEYS::add); transformer.units.add(new VelocityProfileKeyTransformUnit()); transformer.units.add(new BungeeCordProfileKeyTransformUnit()); + MainArgumentsTransformer.getArgumentsListeners().add(new AccountTypeTransformer()::transform); return transformer; } diff --git a/src/main/java/moe/yushi/authlibinjector/transform/support/AccountTypeTransformer.java b/src/main/java/moe/yushi/authlibinjector/transform/support/AccountTypeTransformer.java new file mode 100644 index 0000000..01a32fa --- /dev/null +++ b/src/main/java/moe/yushi/authlibinjector/transform/support/AccountTypeTransformer.java @@ -0,0 +1,39 @@ +/* + * Copyright (C) 2023 Haowei Wen and contributors + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU Affero General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Affero General Public License for more details. + * + * You should have received a copy of the GNU Affero General Public License + * along with this program. If not, see . + */ +package moe.yushi.authlibinjector.transform.support; + +import static moe.yushi.authlibinjector.util.Logging.log; +import static moe.yushi.authlibinjector.util.Logging.Level.INFO; + +public class AccountTypeTransformer { + + public String[] transform(String[] args) { + boolean userTypeMatched = false; + for (int i = 0; i < args.length; i++) { + String arg = args[i]; + if ("--userType".equals(arg)) { + userTypeMatched = true; + } else if (userTypeMatched && "mojang".equals(arg)) { + args[i] = "msa"; + log(INFO, "Setting accountType to msa"); + break; + } + } + return args; + } + +}