sciwhiz12/Janitor
1
0
Fork 0
mirror of https://github.com/sciwhiz12/Janitor.git synced 2024-11-10 03:21:26 +00:00
Code Issues Releases Wiki Activity

Add snowflake ids to user argument

Browse Source
This commit is contained in:
Arnold Alejo Nunag 2020-09-15 01:27:00 +08:00
parent 7f91f78a90
commit fb10a70891
Signed by: sciwhiz12
GPG Key ID: 622CF446534317E1
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/main/java/sciwhiz12/janitor/commands/arguments/UserArgument.java
View File

@ -38,6 +38,17 @@ public class UserArgument implements ArgumentType<UserArgument.IUserProvider> {
return new NumericalProvider(Long.parseLong(matcher.group(1)));
}
}
reader.setCursor(startCursor);
CommandSyntaxException idReadException = null;
if (StringReader.isAllowedNumber(reader.peek())) {
try {
long value = reader.readLong();
return new NumericalProvider(value);
} catch (CommandSyntaxException e) {
idReadException = e;
}
}
if (idReadException != null) throw idReadException;
throw UNKNOWN_USER_IDENTIFIER.create();
}
@ -46,6 +57,10 @@ public class UserArgument implements ArgumentType<UserArgument.IUserProvider> {
return ImmutableList.of("<@!607058472709652501>", "<@750291676764962816>");
}
private boolean isNumericalCharacter(char c) {
return c >= '0' && c <= '9';
}
public interface IUserProvider {
CompletableFuture<User> getUsers(DiscordApi api);
}