502647092/YumCore
1
0
Fork 0
mirror of https://e.coding.net/circlecloud/YumCore.git synced 2024-11-21 01:38:51 +00:00
Code Issues Releases Activity

feat: 添加参数自定义限制

Browse Source 
Signed-off-by: 502647092 <admin@yumc.pw>
This commit is contained in:
502647092 2016-10-10 01:39:32 +08:00
parent aa7608fba2
commit bf8e6b55a4
1 changed files with 137 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

154
src/main/java/pw/yumc/YumCore/commands/CommandParse.java
View File

@ -1,6 +1,10 @@
package pw.yumc.YumCore.commands;
import java.lang.annotation.Annotation;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.LinkedList;
@ -33,15 +37,28 @@ public class CommandParse {
public List<Object> parsed = new LinkedList<>();
public CommandParse(final Class[] classes, final Annotation[][] annons) {
for (final Class classe : classes) {
final Class clazz = classe;
for (int i = 0; i < classes.length; i++) {
final Class clazz = classes[i];
final Annotation[] annotations = annons[i];
if (clazz.isAssignableFrom(CommandSender.class)) {
continue;
}
if (!allparses.containsKey(clazz)) {
throw new CommandParseException(String.format("无法解析的参数类型 %s !", clazz.getName()));
}
parse.add(allparses.get(clazz));
final Parse parse = allparses.get(clazz).clone();
for (final Annotation annotation : annotations) {
if (annotation.annotationType() == Default.class) {
parse.setAttr("default", ((Default) annotation).value());
} else if (annotation.annotationType() == Limit.class) {
parse.setAttr("min", ((Limit) annotation).min());
parse.setAttr("max", ((Limit) annotation).max());
} else if (annotation.annotationType() == KeyValue.class) {
final KeyValue kv = (KeyValue) annotation;
parse.setAttr(kv.key(), kv.value());
}
}
this.parse.add(parse);
}
}
@ -66,13 +83,13 @@ public class CommandParse {
pobjs.add(parse.get(i).parse(args[i]));
}
} catch (final Exception e) {
throw new CommandException(String.format(e.getMessage(), i + 1));
throw new CommandException(String.format("第 %s 个参数 ", i + 1) + e.getMessage());
}
}
return pobjs.toArray();
}
public static class BooleanParse implements Parse<Boolean> {
public static class BooleanParse extends Parse<Boolean> {
public BooleanParse() {
allparses.put(Boolean.class, this);
allparses.put(boolean.class, this);
@ -83,12 +100,24 @@ public class CommandParse {
try {
return Boolean.parseBoolean(arg);
} catch (final Exception e) {
throw new CommandParseException("第 %s 个参数必须为True或者False!");
throw new CommandParseException("必须为True或者False!");
}
}
}
public static class IntegerParse implements Parse<Integer> {
/**
* 默认参数
*
* @since 2016年7月23日 上午9:00:27
* @author 喵♂呜
*/
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface Default {
String value();
}
public static class IntegerParse extends Parse<Integer> {
public IntegerParse() {
allparses.put(Integer.class, this);
allparses.put(int.class, this);
@ -97,14 +126,58 @@ public class CommandParse {
@Override
public Integer parse(final String arg) {
try {
return Integer.parseInt(arg);
final int result = Integer.parseInt(arg);
if (min > result || result > max) {
throwRange();
}
return result;
} catch (final Exception e) {
throw new CommandParseException("第 %s 个参数必须为数字!");
throw new CommandParseException("必须为数字!");
}
}
}
public static class LongParse implements Parse<Long> {
/**
* 自定义参数
*
* @since 2016年7月23日 上午9:00:27
* @author 喵♂呜
*/
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface KeyValue {
/**
* @return 键
*/
String key();
/**
* @return 值
*/
String value() default "";
}
/**
* 参数限制
*
* @since 2016年7月23日 上午9:00:27
* @author 喵♂呜
*/
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface Limit {
/**
* @return 最大长度(最大值)
*/
int max() default 255;
/**
* @return 最小长度(或最小值)
*/
int min();
}
public static class LongParse extends Parse<Long> {
public LongParse() {
allparses.put(Long.class, this);
allparses.put(long.class, this);
@ -113,35 +186,82 @@ public class CommandParse {
@Override
public Long parse(final String arg) {
try {
return Long.parseLong(arg);
final long result = Long.parseLong(arg);
if (min > result || result > max) {
throwRange();
}
return result;
} catch (final Exception e) {
throw new CommandParseException("第 %s 个参数必须为数字!");
throw new CommandParseException("必须为数字!");
}
}
}
public static interface Parse<RT> {
public RT parse(String arg) throws CommandParseException;
public static abstract class Parse<RT> implements Cloneable {
protected Object def;
protected Map<String, Object> attrs = new HashMap<>();
protected int min;
protected int max;
@Override
public Parse<RT> clone() {
try {
return (Parse<RT>) super.clone();
} catch (final CloneNotSupportedException e) {
return null;
}
}
public Object getDefault() {
return def;
}
public void load() {
def = attrs.get("default");
min = (int) attrs.get("min");
max = (int) attrs.get("max");
}
public abstract RT parse(String arg) throws CommandParseException;
public void setAttr(final String name, final Object value) {
attrs.put(name, value);
}
public void throwRange() {
throwRange(null);
}
public void throwRange(final String str) {
throw new CommandException(String.format(str == null ? "必须在 %s 到 %s之间!" : str, min, max));
}
}
public static class PlayerParse implements Parse<Player> {
public static class PlayerParse extends Parse<Player> {
public PlayerParse() {
allparses.put(Player.class, this);
}
@Override
public Player parse(final String arg) {
return Bukkit.getPlayerExact(arg);
final Player p = Bukkit.getPlayerExact(arg);
if (attrs.containsKey("check") && p == null) {
throw new CommandParseException("玩家 " + arg + "不存在或不在线!");
}
return p;
}
}
public static class StringParse implements Parse<String> {
public static class StringParse extends Parse<String> {
public StringParse() {
allparses.put(String.class, this);
}
@Override
public String parse(final String arg) {
if (min > arg.length() || arg.length() > max) {
throwRange("长度必须在 %s 和 %s 之间!");
}
return arg;
}
}