@@ -20,6 +20,8 @@ import java.io.PrintWriter;
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.shell.command.CommandRegistration;
|
||||
import org.springframework.shell.standard.ShellComponent;
|
||||
import org.springframework.shell.standard.ShellMethod;
|
||||
import org.springframework.shell.standard.ShellOption;
|
||||
|
||||
/**
|
||||
* Commands used for e2e test.
|
||||
@@ -62,4 +64,169 @@ public class OptionTypeCommands extends BaseE2ECommands {
|
||||
.build();
|
||||
}
|
||||
|
||||
//
|
||||
// String
|
||||
//
|
||||
|
||||
@ShellMethod(key = LEGACY_ANNO + "option-type-string", group = GROUP)
|
||||
public String optionTypeStringAnnotation(
|
||||
@ShellOption(help = "Desc arg1") String arg1
|
||||
) {
|
||||
return "Hello " + arg1;
|
||||
}
|
||||
|
||||
@Bean
|
||||
public CommandRegistration optionTypeStringRegistration() {
|
||||
return CommandRegistration.builder()
|
||||
.command(REG, "option-type-string")
|
||||
.group(GROUP)
|
||||
.withOption()
|
||||
.longNames("arg1")
|
||||
.type(String.class)
|
||||
.required()
|
||||
.and()
|
||||
.withTarget()
|
||||
.function(ctx -> {
|
||||
String arg1 = ctx.getOptionValue("arg1");
|
||||
return "Hello " + arg1;
|
||||
})
|
||||
.and()
|
||||
.build();
|
||||
}
|
||||
|
||||
//
|
||||
// Boolean
|
||||
//
|
||||
|
||||
@ShellMethod(key = LEGACY_ANNO + "option-type-boolean", group = GROUP)
|
||||
public String optionTypeBooleanAnnotation(
|
||||
@ShellOption() boolean arg1,
|
||||
@ShellOption(defaultValue = "true") boolean arg2,
|
||||
@ShellOption(defaultValue = "false") boolean arg3,
|
||||
@ShellOption() Boolean arg4,
|
||||
@ShellOption(defaultValue = "true") Boolean arg5,
|
||||
@ShellOption(defaultValue = "false") Boolean arg6
|
||||
) {
|
||||
return String.format("Hello arg1=%s arg2=%s arg3=%s arg4=%s arg5=%s arg6=%s", arg1, arg2, arg3, arg4, arg5,
|
||||
arg6);
|
||||
}
|
||||
|
||||
@Bean
|
||||
public CommandRegistration optionTypeBooleanRegistration() {
|
||||
return CommandRegistration.builder()
|
||||
.command(REG, "option-type-boolean")
|
||||
.group(GROUP)
|
||||
.withOption()
|
||||
.longNames("arg1")
|
||||
.type(boolean.class)
|
||||
.and()
|
||||
.withOption()
|
||||
.longNames("arg2")
|
||||
.type(boolean.class)
|
||||
.defaultValue("true")
|
||||
.and()
|
||||
.withOption()
|
||||
.longNames("arg3")
|
||||
.type(boolean.class)
|
||||
.defaultValue("false")
|
||||
.and()
|
||||
.withOption()
|
||||
.longNames("arg4")
|
||||
.type(Boolean.class)
|
||||
.and()
|
||||
.withOption()
|
||||
.longNames("arg5")
|
||||
.type(Boolean.class)
|
||||
.defaultValue("true")
|
||||
.and()
|
||||
.withOption()
|
||||
.longNames("arg6")
|
||||
.type(Boolean.class)
|
||||
.defaultValue("false")
|
||||
.and()
|
||||
.withTarget()
|
||||
.function(ctx -> {
|
||||
boolean arg1 = ctx.hasMappedOption("arg1") ? ctx.getOptionValue("arg1") : false;
|
||||
boolean arg2 = ctx.getOptionValue("arg2");
|
||||
boolean arg3 = ctx.getOptionValue("arg3");
|
||||
Boolean arg4 = ctx.getOptionValue("arg4");
|
||||
Boolean arg5 = ctx.getOptionValue("arg5");
|
||||
Boolean arg6 = ctx.getOptionValue("arg6");
|
||||
return String.format("Hello arg1=%s arg2=%s arg3=%s arg4=%s arg5=%s arg6=%s", arg1, arg2, arg3,
|
||||
arg4, arg5, arg6);
|
||||
})
|
||||
.and()
|
||||
.build();
|
||||
}
|
||||
|
||||
//
|
||||
// Integer
|
||||
//
|
||||
|
||||
@ShellMethod(key = LEGACY_ANNO + "option-type-integer", group = GROUP)
|
||||
public String optionTypeIntegerAnnotation(
|
||||
@ShellOption int arg1,
|
||||
@ShellOption Integer arg2
|
||||
) {
|
||||
return String.format("Hello '%s' '%s'", arg1, arg2);
|
||||
}
|
||||
|
||||
@Bean
|
||||
public CommandRegistration optionTypeIntegerRegistration() {
|
||||
return CommandRegistration.builder()
|
||||
.command(REG, "option-type-integer")
|
||||
.group(GROUP)
|
||||
.withOption()
|
||||
.longNames("arg1")
|
||||
.type(int.class)
|
||||
.required()
|
||||
.and()
|
||||
.withOption()
|
||||
.longNames("arg2")
|
||||
.type(Integer.class)
|
||||
.required()
|
||||
.and()
|
||||
.withTarget()
|
||||
.function(ctx -> {
|
||||
int arg1 = ctx.getOptionValue("arg1");
|
||||
Integer arg2 = ctx.getOptionValue("arg2");
|
||||
return String.format("Hello '%s' '%s'", arg1, arg2);
|
||||
})
|
||||
.and()
|
||||
.build();
|
||||
}
|
||||
|
||||
//
|
||||
// Enum
|
||||
//
|
||||
|
||||
public static enum OptionTypeEnum {
|
||||
ONE,TWO,THREE
|
||||
}
|
||||
|
||||
@ShellMethod(key = LEGACY_ANNO + "option-type-enum", group = GROUP)
|
||||
public String optionTypeEnumAnnotation(
|
||||
@ShellOption(help = "Desc arg1") OptionTypeEnum arg1
|
||||
) {
|
||||
return "Hello " + arg1;
|
||||
}
|
||||
|
||||
@Bean
|
||||
public CommandRegistration optionTypeEnumRegistration() {
|
||||
return CommandRegistration.builder()
|
||||
.command(REG, "option-type-enum")
|
||||
.group(GROUP)
|
||||
.withOption()
|
||||
.longNames("arg1")
|
||||
.type(OptionTypeEnum.class)
|
||||
.required()
|
||||
.and()
|
||||
.withTarget()
|
||||
.function(ctx -> {
|
||||
OptionTypeEnum arg1 = ctx.getOptionValue("arg1");
|
||||
return "Hello " + arg1;
|
||||
})
|
||||
.and()
|
||||
.build();
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user