114 lines
4.1 KiB
Java
114 lines
4.1 KiB
Java
package org.schabi.newpipe.extractor.services.youtube.linkHandler;
|
|
|
|
import static org.schabi.newpipe.extractor.utils.Utils.isNullOrEmpty;
|
|
|
|
import org.schabi.newpipe.extractor.exceptions.ParsingException;
|
|
import org.schabi.newpipe.extractor.linkhandler.SearchQueryHandlerFactory;
|
|
|
|
import java.io.UnsupportedEncodingException;
|
|
import java.net.URLEncoder;
|
|
import java.nio.charset.StandardCharsets;
|
|
import java.util.List;
|
|
|
|
import javax.annotation.Nonnull;
|
|
|
|
public final class YoutubeSearchQueryHandlerFactory extends SearchQueryHandlerFactory {
|
|
|
|
public static final String ALL = "all";
|
|
public static final String VIDEOS = "videos";
|
|
public static final String CHANNELS = "channels";
|
|
public static final String PLAYLISTS = "playlists";
|
|
|
|
public static final String MUSIC_SONGS = "music_songs";
|
|
public static final String MUSIC_VIDEOS = "music_videos";
|
|
public static final String MUSIC_ALBUMS = "music_albums";
|
|
public static final String MUSIC_PLAYLISTS = "music_playlists";
|
|
public static final String MUSIC_ARTISTS = "music_artists";
|
|
|
|
private static final String SEARCH_URL = "https://www.youtube.com/results?search_query=";
|
|
private static final String MUSIC_SEARCH_URL = "https://music.youtube.com/search?q=";
|
|
|
|
@Nonnull
|
|
public static YoutubeSearchQueryHandlerFactory getInstance() {
|
|
return new YoutubeSearchQueryHandlerFactory();
|
|
}
|
|
|
|
@Override
|
|
public String getUrl(final String searchString,
|
|
@Nonnull final List<String> contentFilters,
|
|
final String sortFilter) throws ParsingException {
|
|
try {
|
|
if (!contentFilters.isEmpty()) {
|
|
final String contentFilter = contentFilters.get(0);
|
|
switch (contentFilter) {
|
|
case ALL:
|
|
default:
|
|
break;
|
|
case VIDEOS:
|
|
return SEARCH_URL + URLEncoder.encode(searchString,
|
|
StandardCharsets.UTF_8.name())
|
|
+ "&sp=EgIQAQ%253D%253D";
|
|
case CHANNELS:
|
|
return SEARCH_URL + URLEncoder.encode(searchString,
|
|
StandardCharsets.UTF_8.name())
|
|
+ "&sp=EgIQAg%253D%253D";
|
|
case PLAYLISTS:
|
|
return SEARCH_URL + URLEncoder.encode(searchString,
|
|
StandardCharsets.UTF_8.name())
|
|
+ "&sp=EgIQAw%253D%253D";
|
|
case MUSIC_SONGS:
|
|
case MUSIC_VIDEOS:
|
|
case MUSIC_ALBUMS:
|
|
case MUSIC_PLAYLISTS:
|
|
case MUSIC_ARTISTS:
|
|
return MUSIC_SEARCH_URL + URLEncoder.encode(searchString,
|
|
StandardCharsets.UTF_8.name());
|
|
}
|
|
}
|
|
|
|
return SEARCH_URL + URLEncoder.encode(searchString, StandardCharsets.UTF_8.name());
|
|
} catch (final UnsupportedEncodingException e) {
|
|
throw new ParsingException("Could not encode query", e);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public String[] getAvailableContentFilter() {
|
|
return new String[]{
|
|
ALL,
|
|
VIDEOS,
|
|
CHANNELS,
|
|
PLAYLISTS,
|
|
MUSIC_SONGS,
|
|
MUSIC_VIDEOS,
|
|
MUSIC_ALBUMS,
|
|
MUSIC_PLAYLISTS
|
|
// MUSIC_ARTISTS
|
|
};
|
|
}
|
|
|
|
@Nonnull
|
|
public static String getSearchParameter(final String contentFilter) {
|
|
if (isNullOrEmpty(contentFilter)) {
|
|
return "";
|
|
}
|
|
|
|
switch (contentFilter) {
|
|
case VIDEOS:
|
|
return "EgIQAQ%3D%3D";
|
|
case CHANNELS:
|
|
return "EgIQAg%3D%3D";
|
|
case PLAYLISTS:
|
|
return "EgIQAw%3D%3D";
|
|
case ALL:
|
|
case MUSIC_SONGS:
|
|
case MUSIC_VIDEOS:
|
|
case MUSIC_ALBUMS:
|
|
case MUSIC_PLAYLISTS:
|
|
case MUSIC_ARTISTS:
|
|
default:
|
|
return "";
|
|
}
|
|
}
|
|
}
|