pingex
/
DiscordBot
Archived
1
0
Fork 0
Code Issues 3 Releases Wiki Activity

Assert on /perm commands. Closes #9.

Browse Source
master
Pingex aka Raphaël 9 years ago
parent c4887b1cd9
commit 2a21b44fcf
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File

11
src/main/java/net/pingex/discordbot/PermissionsModule.java
Unescape Escape View File

@ -75,6 +75,8 @@ class PermissionsModule extends AbstractModule
@Command(description = "Returns the Unique Identifier of the target user.", permission = DefaultPermission.ANY_OWNER)
public String getUID(MessageReceivedEvent event, String name, String discriminator)
{
if(!discriminator.matches("^\\d{4}$")) return "Invalid discriminator.";
for(IUser i : event.getMessage().getGuild().getUsers())
if(i.getName().equals(name) && i.getDiscriminator().equals(discriminator))
return i.getID();
@ -187,6 +189,9 @@ class PermissionsModule extends AbstractModule
@Command(description = "Sets an user permission.", permission = DefaultPermission.ANY_OWNER, required = {true, true, false})
public String setUser(MessageReceivedEvent event, String command, String user, Boolean target)
{
if(!command.matches("^\\w+:\\w+$")) return "Invalid command.";
if(!user.matches("^\\d{18}$")) return "Invalid user.";
if(!permissions.containsKey(event.getMessage().getGuild().getID()))
permissions.put(event.getMessage().getGuild().getID(), new Guild(new HashMap<>()));
Guild g = permissions.get(event.getMessage().getGuild().getID());
@ -211,6 +216,9 @@ class PermissionsModule extends AbstractModule
@Command(description = "Sets a role permission.", permission = DefaultPermission.ANY_OWNER, required = {true, true, false})
public String setRole(MessageReceivedEvent event, String command, String group, Boolean target)
{
if(!command.matches("^\\w+:\\w+$")) return "Invalid command.";
if(!group.matches("^\\d{18}$")) return "Invalid group.";
if(!permissions.containsKey(event.getMessage().getGuild().getID()))
permissions.put(event.getMessage().getGuild().getID(), new Guild(new HashMap<>()));
Guild g = permissions.get(event.getMessage().getGuild().getID());
@ -273,8 +281,9 @@ class PermissionsModule extends AbstractModule
@Command(description = "Says if the target user can run said command. Ignores default behavior.", permission = DefaultPermission.ANY_OWNER)
public String canRun(MessageReceivedEvent event, String user, String discriminator, String command)
{
IUser target = null;
if(!discriminator.matches("^\\d{4}$")) return "Invalid discriminator.";
IUser target = null;
for(IUser i : event.getMessage().getGuild().getUsers())
if(i.getName().equals(user) && i.getDiscriminator().equals(discriminator))
{