Examples with NSFWSettings tk.ardentbot.rethink.models.NSFWSettings used on opensource projects

Example 1 with NSFWSettings

use of tk.ardentbot.rethink.models.NSFWSettings in project Ardent by adamint.

the class NSFW method getSettings.

private static NSFWSettings getSettings(Guild guild) {
    NSFWSettings settings = asPojo(r.table("nsfw_settings").get(guild.getId()).run(connection), NSFWSettings.class);
    if (settings == null) {
        settings = new NSFWSettings(guild.getId());
        r.table("nsfw_settings").insert(r.json(getStaticGson().toJson(settings))).run(connection);
    }
    return settings;
}

Example 2 with NSFWSettings

use of tk.ardentbot.rethink.models.NSFWSettings in project Ardent by adamint.

the class NSFW method canSendNSFW.

public static boolean canSendNSFW(User user, MessageChannel mc, Guild guild, Command command) {
    TextChannel channel = (TextChannel) mc;
    NSFWSettings settings = getSettings(guild);
    if (!settings.isGlobal()) {
        if (!mc.getName().startsWith("nsfw")) {
            channel.sendMessage("You can't use NSFW commands in this channel - it must be a channel name prefixed with nsfw").queue();
            return false;
        }
    }
    if (settings.isNeedNsfwRole()) {
        final boolean[] hasRole = { false };
        guild.getMember(user).getRoles().forEach(role -> {
            if (role.getName().equalsIgnoreCase("nsfw"))
                hasRole[0] = true;
        });
        if (!hasRole[0]) {
            channel.sendMessage("In order to send NSFW commands, you need the NSFW role!").queue();
            return false;
        }
    }
    return true;
}

Example 3 with NSFWSettings

use of tk.ardentbot.rethink.models.NSFWSettings in project Ardent by adamint.

the class NSFW method setupSubcommands.

@Override
public void setupSubcommands() throws Exception {
    subcommands.add(new Subcommand("Set whether people can send NSFW channels in all channels", "alloweverywhere [true/false]", "alloweverywhere") {

        @Override
        public void onCall(Guild guild, MessageChannel channel, User user, Message message, String[] args) throws Exception {
            if (guild.getMember(user).hasPermission(Permission.MANAGE_SERVER)) {
                if (args.length == 3) {
                    try {
                        boolean allowEverywhere = Boolean.parseBoolean(args[2]);
                        getSettings(guild);
                        if (allowEverywhere)
                            sendTranslatedMessage("You can now use NSFW commands in all channels", channel, user);
                        else
                            sendTranslatedMessage("You can now use NSFW commands only in specified channels", channel, user);
                        r.table("nsfw_settings").get(guild.getId()).update(r.hashMap("global", allowEverywhere)).run(connection);
                    } catch (Exception e) {
                        sendTranslatedMessage("You need to specify true or false", channel, user);
                    }
                } else
                    sendTranslatedMessage("Invalid arguments ¯\\_(ツ)_/¯", channel, user);
            } else
                sendTranslatedMessage("You need the Manage Server permission to use this command", channel, user);
        }
    });
    subcommands.add(new Subcommand("Set whether you need a role called NSFW to send NSFW commands", "neednsfwrole [true/false]", "neednsfwrole") {

        @Override
        public void onCall(Guild guild, MessageChannel channel, User user, Message message, String[] args) throws Exception {
            if (guild.getMember(user).hasPermission(Permission.MANAGE_SERVER)) {
                if (args.length == 3) {
                    try {
                        boolean needs = Boolean.parseBoolean(args[2]);
                        getSettings(guild);
                        if (needs)
                            sendTranslatedMessage("You now need the NSFW role to send NSFW commands", channel, user);
                        else
                            sendTranslatedMessage("Anyone can now send NSFW commands", channel, user);
                        r.table("nsfw_settings").get(guild.getId()).update(r.hashMap("needNsfwRole", needs)).run(connection);
                    } catch (Exception e) {
                        sendTranslatedMessage("You need to specify true or false", channel, user);
                    }
                } else
                    sendTranslatedMessage("Invalid arguments ¯\\_(ツ)_/¯", channel, user);
            } else
                sendTranslatedMessage("You need the Manage Server permission to use this command", channel, user);
        }
    });
    subcommands.add(new Subcommand("View the current NSFW settings for your server", "settings", "settings") {

        @Override
        public void onCall(Guild guild, MessageChannel channel, User user, Message message, String[] args) throws Exception {
            String nsfwSettings = "NSFW Settings";
            NSFWSettings settings = getSettings(guild);
            EmbedBuilder embedBuilder = MessageUtils.getDefaultEmbed(user);
            embedBuilder.setAuthor(nsfwSettings, getShard().url, getShard().bot.getAvatarUrl());
            StringBuilder description = new StringBuilder();
            description.append("**" + nsfwSettings + "**");
            description.append("\n\nAble to use NSFW commands in all channels: " + settings.isGlobal());
            description.append("\n\nNeed the NSFW role: " + settings.isNeedNsfwRole());
            if (!settings.isGlobal()) {
                description.append("\n\nCan only use these command in discord NSFFW channels");
            }
            embedBuilder.setDescription(description.toString());
            sendEmbed(embedBuilder, channel, user);
        }
    });
}

Example 4 with NSFWSettings

use of tk.ardentbot.rethink.models.NSFWSettings in project Ardent by adamint.

the class NSFW method csn.

public static boolean csn(User user, MessageChannel mc, Guild guild) {
    TextChannel channel = (TextChannel) mc;
    NSFWSettings settings = getSettings(guild);
    if (!settings.isGlobal()) {
        if (!settings.getNsfwChannels().contains(channel.getId())) {
            return false;
        }
    }
    if (settings.isNeedNsfwRole()) {
        final boolean[] hasRole = { false };
        guild.getMember(user).getRoles().forEach(role -> {
            if (role.getName().equalsIgnoreCase("nsfw"))
                hasRole[0] = true;
        });
        if (!hasRole[0]) {
            return false;
        }
    }
    return true;
}