Search in sources :

Example 1 with Wallpaper

use of com.ivkos.wallhaven4j.models.wallpaper.Wallpaper in project Shadbot by Shadorc.

the class WallpaperCmd method execute.

@Override
public void execute(Context context) throws MissingArgumentException, IllegalCmdArgumentException {
    LoadingMessage loadingMsg = new LoadingMessage("Loading wallpaper...", context.getChannel());
    loadingMsg.send();
    if (wallhaven == null) {
        wallhaven = new Wallhaven(APIKeys.get(APIKey.WALLHAVEN_LOGIN), APIKeys.get(APIKey.WALLHAVEN_PASSWORD));
    }
    Options options = new Options();
    options.addOption("p", PURITY, true, FormatUtils.format(Purity.values(), purity -> purity.toString().toLowerCase(), ", "));
    options.addOption("c", CATEGORY, true, FormatUtils.format(Category.values(), cat -> cat.toString().toLowerCase(), ", "));
    Option ratioOpt = new Option("rat", RATIO, true, "image ratio");
    ratioOpt.setValueSeparator('x');
    options.addOption(ratioOpt);
    Option resOpt = new Option("res", RESOLUTION, true, "image resolution");
    resOpt.setValueSeparator('x');
    options.addOption(resOpt);
    Option keyOpt = new Option("k", KEYWORD, true, KEYWORD);
    keyOpt.setValueSeparator(',');
    options.addOption(keyOpt);
    CommandLine cmdLine;
    try {
        List<String> args = StringUtils.split(context.getArg());
        cmdLine = new DefaultParser().parse(options, args.toArray(new String[args.size()]));
    } catch (UnrecognizedOptionException | org.apache.commons.cli.MissingArgumentException err) {
        loadingMsg.delete();
        throw new IllegalCmdArgumentException(String.format("%s. Use `%shelp %s` for more information.", err.getMessage(), context.getPrefix(), this.getName()));
    } catch (ParseException err) {
        loadingMsg.delete();
        Utils.handle("getting a wallpaper", context, err);
        return;
    }
    Purity purity = this.parseEnum(loadingMsg, context, Purity.class, PURITY, cmdLine.getOptionValue(PURITY, Purity.SFW.toString()));
    if ((purity.equals(Purity.NSFW) || purity.equals(Purity.SKETCHY)) && !context.getChannel().isNSFW()) {
        loadingMsg.edit(TextUtils.mustBeNSFW(context.getPrefix()));
        return;
    }
    SearchQueryBuilder queryBuilder = new SearchQueryBuilder();
    queryBuilder.purity(purity);
    if (cmdLine.hasOption(CATEGORY)) {
        queryBuilder.categories(this.parseEnum(loadingMsg, context, Category.class, CATEGORY, cmdLine.getOptionValue(CATEGORY)));
    }
    if (cmdLine.hasOption(RATIO)) {
        Dimension dim = this.parseDim(loadingMsg, context, RATIO, cmdLine.getOptionValues(RATIO));
        queryBuilder.ratios(new Ratio((int) dim.getWidth(), (int) dim.getHeight()));
    }
    if (cmdLine.hasOption(RESOLUTION)) {
        Dimension dim = this.parseDim(loadingMsg, context, RESOLUTION, cmdLine.getOptionValues(RESOLUTION));
        queryBuilder.resolutions(new Resolution((int) dim.getWidth(), (int) dim.getHeight()));
    }
    if (cmdLine.hasOption(KEYWORD)) {
        queryBuilder.keywords(cmdLine.getOptionValues(KEYWORD));
    }
    try {
        List<Wallpaper> wallpapers = wallhaven.search(queryBuilder.pages(1).build());
        if (wallpapers.isEmpty()) {
            loadingMsg.edit(TextUtils.noResult(context.getMessage().getContent()));
            return;
        }
        Wallpaper wallpaper = wallpapers.get(ThreadLocalRandom.current().nextInt(wallpapers.size()));
        String tags = FormatUtils.format(wallpaper.getTags(), tag -> String.format("`%s`", StringUtils.remove(tag.toString(), "#")), " ");
        EmbedBuilder embed = EmbedUtils.getDefaultEmbed().withAuthorName("Wallpaper").withAuthorUrl(wallpaper.getUrl()).withImage(wallpaper.getImageUrl()).appendField("Resolution", wallpaper.getResolution().toString(), false).appendField("Tags", tags, false);
        loadingMsg.edit(embed.build());
    } catch (ConnectionException err) {
        loadingMsg.delete();
        Utils.handle("getting a wallpaper", context, err.getCause());
    }
}
Also used : HelpBuilder(me.shadorc.shadbot.utils.embed.HelpBuilder) Wallhaven(com.ivkos.wallhaven4j.Wallhaven) CommandCategory(me.shadorc.shadbot.core.command.CommandCategory) Options(org.apache.commons.cli.Options) Ratio(com.ivkos.wallhaven4j.models.misc.Ratio) Category(com.ivkos.wallhaven4j.models.misc.enums.Category) Purity(com.ivkos.wallhaven4j.models.misc.enums.Purity) DefaultParser(org.apache.commons.cli.DefaultParser) MissingArgumentException(me.shadorc.shadbot.exception.MissingArgumentException) StringUtils(me.shadorc.shadbot.utils.StringUtils) Command(me.shadorc.shadbot.core.command.annotation.Command) EmbedBuilder(sx.blah.discord.util.EmbedBuilder) TextUtils(me.shadorc.shadbot.utils.TextUtils) ThreadLocalRandom(java.util.concurrent.ThreadLocalRandom) CommandLine(org.apache.commons.cli.CommandLine) Option(org.apache.commons.cli.Option) Utils(me.shadorc.shadbot.utils.Utils) LoadingMessage(me.shadorc.shadbot.utils.object.LoadingMessage) Resolution(com.ivkos.wallhaven4j.models.misc.Resolution) RateLimited(me.shadorc.shadbot.core.command.annotation.RateLimited) UnrecognizedOptionException(org.apache.commons.cli.UnrecognizedOptionException) SearchQueryBuilder(com.ivkos.wallhaven4j.util.searchquery.SearchQueryBuilder) APIKeys(me.shadorc.shadbot.data.APIKeys) EmbedObject(sx.blah.discord.api.internal.json.objects.EmbedObject) FormatUtils(me.shadorc.shadbot.utils.FormatUtils) IllegalCmdArgumentException(me.shadorc.shadbot.exception.IllegalCmdArgumentException) CastUtils(me.shadorc.shadbot.utils.CastUtils) ConnectionException(com.ivkos.wallhaven4j.util.exceptions.ConnectionException) Dimension(java.awt.Dimension) List(java.util.List) Context(me.shadorc.shadbot.core.command.Context) ParseException(org.apache.commons.cli.ParseException) AbstractCommand(me.shadorc.shadbot.core.command.AbstractCommand) APIKey(me.shadorc.shadbot.data.APIKeys.APIKey) EmbedUtils(me.shadorc.shadbot.utils.embed.EmbedUtils) Wallpaper(com.ivkos.wallhaven4j.models.wallpaper.Wallpaper) Options(org.apache.commons.cli.Options) CommandCategory(me.shadorc.shadbot.core.command.CommandCategory) Category(com.ivkos.wallhaven4j.models.misc.enums.Category) Purity(com.ivkos.wallhaven4j.models.misc.enums.Purity) UnrecognizedOptionException(org.apache.commons.cli.UnrecognizedOptionException) IllegalCmdArgumentException(me.shadorc.shadbot.exception.IllegalCmdArgumentException) Ratio(com.ivkos.wallhaven4j.models.misc.Ratio) LoadingMessage(me.shadorc.shadbot.utils.object.LoadingMessage) DefaultParser(org.apache.commons.cli.DefaultParser) MissingArgumentException(me.shadorc.shadbot.exception.MissingArgumentException) Dimension(java.awt.Dimension) CommandLine(org.apache.commons.cli.CommandLine) EmbedBuilder(sx.blah.discord.util.EmbedBuilder) SearchQueryBuilder(com.ivkos.wallhaven4j.util.searchquery.SearchQueryBuilder) Option(org.apache.commons.cli.Option) Wallpaper(com.ivkos.wallhaven4j.models.wallpaper.Wallpaper) Wallhaven(com.ivkos.wallhaven4j.Wallhaven) ParseException(org.apache.commons.cli.ParseException) ConnectionException(com.ivkos.wallhaven4j.util.exceptions.ConnectionException) Resolution(com.ivkos.wallhaven4j.models.misc.Resolution)

Aggregations

Wallhaven (com.ivkos.wallhaven4j.Wallhaven)1 Ratio (com.ivkos.wallhaven4j.models.misc.Ratio)1 Resolution (com.ivkos.wallhaven4j.models.misc.Resolution)1 Category (com.ivkos.wallhaven4j.models.misc.enums.Category)1 Purity (com.ivkos.wallhaven4j.models.misc.enums.Purity)1 Wallpaper (com.ivkos.wallhaven4j.models.wallpaper.Wallpaper)1 ConnectionException (com.ivkos.wallhaven4j.util.exceptions.ConnectionException)1 SearchQueryBuilder (com.ivkos.wallhaven4j.util.searchquery.SearchQueryBuilder)1 Dimension (java.awt.Dimension)1 List (java.util.List)1 ThreadLocalRandom (java.util.concurrent.ThreadLocalRandom)1 AbstractCommand (me.shadorc.shadbot.core.command.AbstractCommand)1 CommandCategory (me.shadorc.shadbot.core.command.CommandCategory)1 Context (me.shadorc.shadbot.core.command.Context)1 Command (me.shadorc.shadbot.core.command.annotation.Command)1 RateLimited (me.shadorc.shadbot.core.command.annotation.RateLimited)1 APIKeys (me.shadorc.shadbot.data.APIKeys)1 APIKey (me.shadorc.shadbot.data.APIKeys.APIKey)1 IllegalCmdArgumentException (me.shadorc.shadbot.exception.IllegalCmdArgumentException)1 MissingArgumentException (me.shadorc.shadbot.exception.MissingArgumentException)1