lionarius/SkinRestorer
1
0
Fork 0
mirror of https://github.com/Suiranoil/SkinRestorer.git synced 2026-01-16 04:42:12 +00:00
Code Wiki Activity

add refresh subcommand

Browse Source
This commit is contained in:
Lionarius
2024-07-01 23:05:24 +03:00
parent 0294661375
commit f7e486cd24
1 changed files with 75 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

138
common/src/main/java/net/lionarius/skinrestorer/command/SkinCommand.java
View File

@@ -40,7 +40,8 @@ public final class SkinCommand {
.executes(context -> resetAction(context.getSource())) .executes(context -> resetAction(context.getSource()))
.then(makeTargetsArgument( .then(makeTargetsArgument(
(context, profiles) -> resetAction(context.getSource(), profiles, true) (context, profiles) -> resetAction(context.getSource(), profiles, true)
))); )))
.then(literal("refresh").executes(context -> refreshAction(context.getSource())));
LiteralArgumentBuilder<CommandSourceStack> set = literal("set"); LiteralArgumentBuilder<CommandSourceStack> set = literal("set");
@@ -53,69 +54,15 @@ public final class SkinCommand {
dispatcher.register(base); dispatcher.register(base);
} }
private static LiteralArgumentBuilder<CommandSourceStack> buildSetAction(String name, SkinProvider provider) { private static int refreshAction(
LiteralArgumentBuilder<CommandSourceStack> action = literal(name); CommandSourceStack src
) {
var player = src.getPlayer();
if (player == null)
return 0;
if (provider.hasVariantSupport()) { var context = SkinRestorer.getSkinStorage().getSkin(player.getUUID()).toProviderContext();
for (SkinVariant variant : SkinVariant.values()) { return SkinCommand.setAction(src, Collections.singleton(player.getGameProfile()), context, false);
action.then(
literal(variant.toString())
.then(buildSetArgument(
argument(provider.getArgumentName(), StringArgumentType.string()),
context -> {
var argument = StringArgumentType.getString(context, provider.getArgumentName());
return new SkinProviderContext(name, argument, variant);
}
))
);
}
} else {
action.then(
buildSetArgument(
argument(provider.getArgumentName(), StringArgumentType.string()),
context -> {
var argument = StringArgumentType.getString(context, provider.getArgumentName());
return new SkinProviderContext(name, argument, null);
}
)
);
}
return action;
}
private static ArgumentBuilder<CommandSourceStack, LiteralArgumentBuilder<CommandSourceStack>> buildSetAction(
String name,
Supplier<SkinProviderContext> supplier
) {
return buildSetArgument(literal(name), context -> supplier.get());
}
private static <T extends ArgumentBuilder<CommandSourceStack, T>> ArgumentBuilder<CommandSourceStack, T> buildSetArgument(
ArgumentBuilder<CommandSourceStack, T> argument,
Function<CommandContext<CommandSourceStack>, SkinProviderContext> provider
) {
return argument
.executes(context -> setAction(
context.getSource(),
provider.apply(context)
))
.then(makeTargetsArgument(
(context, targets) -> setAction(
context.getSource(),
targets,
provider.apply(context),
true
)
));
}
private static RequiredArgumentBuilder<CommandSourceStack, GameProfileArgument.Result> makeTargetsArgument(
BiFunction<CommandContext<CommandSourceStack>, Collection<GameProfile>, Integer> consumer
) {
return argument("targets", GameProfileArgument.gameProfile())
.requires(source -> source.hasPermission(2))
.executes(context -> consumer.apply(context, GameProfileArgument.getGameProfiles(context, "targets")));
} }
private static int resetAction( private static int resetAction(
@@ -208,4 +155,69 @@ public final class SkinCommand {
), true); ), true);
} }
} }
private static LiteralArgumentBuilder<CommandSourceStack> buildSetAction(String name, SkinProvider provider) {
LiteralArgumentBuilder<CommandSourceStack> action = literal(name);
if (provider.hasVariantSupport()) {
for (SkinVariant variant : SkinVariant.values()) {
action.then(
literal(variant.toString())
.then(buildSetArgument(
argument(provider.getArgumentName(), StringArgumentType.string()),
context -> {
var argument = StringArgumentType.getString(context, provider.getArgumentName());
return new SkinProviderContext(name, argument, variant);
}
))
);
}
} else {
action.then(
buildSetArgument(
argument(provider.getArgumentName(), StringArgumentType.string()),
context -> {
var argument = StringArgumentType.getString(context, provider.getArgumentName());
return new SkinProviderContext(name, argument, null);
}
)
);
}
return action;
}
private static ArgumentBuilder<CommandSourceStack, LiteralArgumentBuilder<CommandSourceStack>> buildSetAction(
String name,
Supplier<SkinProviderContext> supplier
) {
return buildSetArgument(literal(name), context -> supplier.get());
}
private static <T extends ArgumentBuilder<CommandSourceStack, T>> ArgumentBuilder<CommandSourceStack, T> buildSetArgument(
ArgumentBuilder<CommandSourceStack, T> argument,
Function<CommandContext<CommandSourceStack>, SkinProviderContext> provider
) {
return argument
.executes(context -> setAction(
context.getSource(),
provider.apply(context)
))
.then(makeTargetsArgument(
(context, targets) -> setAction(
context.getSource(),
targets,
provider.apply(context),
true
)
));
}
private static RequiredArgumentBuilder<CommandSourceStack, GameProfileArgument.Result> makeTargetsArgument(
BiFunction<CommandContext<CommandSourceStack>, Collection<GameProfile>, Integer> consumer
) {
return argument("targets", GameProfileArgument.gameProfile())
.requires(source -> source.hasPermission(2))
.executes(context -> consumer.apply(context, GameProfileArgument.getGameProfiles(context, "targets")));
}
} }