Examples with VideoSelection fredboat.audio.player.VideoSelectionCache.VideoSelection used on opensource projects

Example 1 with VideoSelection

use of fredboat.audio.player.VideoSelectionCache.VideoSelection in project FredBoat by Frederikam.

the class PlayCommand method searchForVideos.

private void searchForVideos(CommandContext context) {
    // Now remove all punctuation
    String query = context.rawArgs.replaceAll(TrackSearcher.PUNCTUATION_REGEX, "");
    context.reply(context.i18n("playSearching").replace("{q}", query), outMsg -> {
        AudioPlaylist list;
        try {
            list = trackSearcher.searchForTracks(query, searchProviders);
        } catch (TrackSearcher.SearchingException e) {
            context.reply(context.i18n("playYoutubeSearchError"));
            log.error("YouTube search exception", e);
            return;
        }
        if (list == null || list.getTracks().isEmpty()) {
            CentralMessaging.editMessage(outMsg, context.i18n("playSearchNoResults").replace("{q}", query));
        } else {
            // Get at most 5 tracks
            List<AudioTrack> selectable = list.getTracks().subList(0, Math.min(TrackSearcher.MAX_RESULTS, list.getTracks().size()));
            VideoSelection oldSelection = videoSelectionCache.remove(context.invoker);
            if (oldSelection != null) {
                oldSelection.deleteMessage();
            }
            MessageBuilder builder = CentralMessaging.getClearThreadLocalMessageBuilder();
            builder.append(context.i18nFormat("playSelectVideo", TextUtils.escapeMarkdown(context.getPrefix())));
            int i = 1;
            for (AudioTrack track : selectable) {
                builder.append("\n**").append(String.valueOf(i)).append(":** ").append(TextUtils.escapeAndDefuse(track.getInfo().title)).append(" (").append(TextUtils.formatTime(track.getInfo().length)).append(")");
                i++;
            }
            CentralMessaging.editMessage(outMsg, builder.build());
            videoSelectionCache.put(context.invoker, selectable, outMsg);
        }
    });
}

Example 2 with VideoSelection

use of fredboat.audio.player.VideoSelectionCache.VideoSelection in project FredBoat by Frederikam.

the class SelectCommand method select.

static void select(CommandContext context, VideoSelectionCache videoSelectionCache) {
    Member invoker = context.invoker;
    VideoSelection selection = videoSelectionCache.get(invoker);
    if (selection == null) {
        context.reply(context.i18n("selectSelectionNotGiven"));
        return;
    }
    try {
        // Step 1: Parse the issued command for numbers
        // LinkedHashSet to handle order of choices + duplicates
        LinkedHashSet<Integer> requestChoices = new LinkedHashSet<>();
        // Combine all args and the command trigger if it is numeric
        String commandOptions = context.rawArgs;
        if (StringUtils.isNumeric(context.trigger)) {
            commandOptions = (context.trigger + " " + commandOptions).trim();
        }
        if (StringUtils.isNumeric(commandOptions)) {
            requestChoices.add(Integer.valueOf(commandOptions));
        } else if (TextUtils.isSplitSelect(commandOptions)) {
            requestChoices.addAll(TextUtils.getSplitSelect(commandOptions));
        }
        // Step 2: Use only valid numbers (usually 1-5)
        ArrayList<Integer> validChoices = new ArrayList<>();
        // Only include valid values which are 1 to <size> of the offered selection
        for (Integer value : requestChoices) {
            if (1 <= value && value <= selection.choices.size()) {
                validChoices.add(value);
                Metrics.selectionChoiceChosen.labels(value.toString()).inc();
            }
        }
        // any valid choices at all?
        if (validChoices.isEmpty()) {
            throw new NumberFormatException();
        } else {
            if (validChoices.size() > 1) {
                Metrics.multiSelections.labels(Integer.toString(validChoices.size())).inc();
            }
            AudioTrack[] selectedTracks = new AudioTrack[validChoices.size()];
            StringBuilder outputMsgBuilder = new StringBuilder();
            GuildPlayer player = Launcher.getBotController().getPlayerRegistry().getOrCreate(context.guild);
            for (int i = 0; i < validChoices.size(); i++) {
                selectedTracks[i] = selection.choices.get(validChoices.get(i) - 1);
                String msg = context.i18nFormat("selectSuccess", validChoices.get(i), TextUtils.escapeAndDefuse(selectedTracks[i].getInfo().title), TextUtils.formatTime(selectedTracks[i].getInfo().length));
                if (i < validChoices.size()) {
                    outputMsgBuilder.append("\n");
                }
                outputMsgBuilder.append(msg);
                player.queue(new AudioTrackContext(Launcher.getBotController().getJdaEntityProvider(), selectedTracks[i], invoker));
            }
            videoSelectionCache.remove(invoker);
            TextChannel tc = Launcher.getBotController().getJdaEntityProvider().getTextChannelById(selection.channelId);
            if (tc != null) {
                CentralMessaging.editMessage(tc, selection.outMsgId, CentralMessaging.from(outputMsgBuilder.toString()));
            }
            player.setPause(false);
            context.deleteMessage();
        }
    } catch (NumberFormatException | ArrayIndexOutOfBoundsException e) {
        context.reply(context.i18nFormat("selectInterval", selection.choices.size()));
    }
}