50 lines
1.9 KiB
Java
50 lines
1.9 KiB
Java
// Created by Fynn Godau 2019, licensed GNU GPL version 3 or later
|
|
|
|
package org.schabi.newpipe.extractor.services.bandcamp.extractors;
|
|
|
|
import static org.schabi.newpipe.extractor.services.bandcamp.extractors.BandcampExtractorHelper.BASE_API_URL;
|
|
|
|
import com.grack.nanojson.JsonObject;
|
|
import com.grack.nanojson.JsonParser;
|
|
import com.grack.nanojson.JsonParserException;
|
|
|
|
import org.schabi.newpipe.extractor.NewPipe;
|
|
import org.schabi.newpipe.extractor.StreamingService;
|
|
import org.schabi.newpipe.extractor.downloader.Downloader;
|
|
import org.schabi.newpipe.extractor.exceptions.ExtractionException;
|
|
import org.schabi.newpipe.extractor.suggestion.SuggestionExtractor;
|
|
|
|
import java.io.IOException;
|
|
import java.net.URLEncoder;
|
|
import java.util.Collections;
|
|
import java.util.List;
|
|
import java.util.stream.Collectors;
|
|
|
|
public class BandcampSuggestionExtractor extends SuggestionExtractor {
|
|
|
|
private static final String AUTOCOMPLETE_URL = BASE_API_URL + "/fuzzysearch/1/autocomplete?q=";
|
|
public BandcampSuggestionExtractor(final StreamingService service) {
|
|
super(service);
|
|
}
|
|
|
|
@Override
|
|
public List<String> suggestionList(final String query) throws IOException, ExtractionException {
|
|
final Downloader downloader = NewPipe.getDownloader();
|
|
|
|
try {
|
|
final JsonObject fuzzyResults = JsonParser.object().from(downloader
|
|
.get(AUTOCOMPLETE_URL + URLEncoder.encode(query, "UTF-8")).responseBody());
|
|
|
|
return fuzzyResults.getObject("auto").getArray("results").stream()
|
|
.filter(JsonObject.class::isInstance)
|
|
.map(JsonObject.class::cast)
|
|
.map(jsonObject -> jsonObject.getString("name"))
|
|
.distinct()
|
|
.collect(Collectors.toList());
|
|
} catch (final JsonParserException e) {
|
|
return Collections.emptyList();
|
|
}
|
|
|
|
}
|
|
}
|