2019-12-22 13:28:02 +01:00
|
|
|
// Created by Fynn Godau 2019, licensed GNU GPL version 3 or later
|
|
|
|
|
|
|
|
package org.schabi.newpipe.extractor.services.bandcamp.extractors;
|
|
|
|
|
2022-03-18 17:50:25 +01:00
|
|
|
import static org.schabi.newpipe.extractor.services.bandcamp.extractors.BandcampExtractorHelper.BASE_API_URL;
|
|
|
|
|
2020-03-19 11:18:29 +01:00
|
|
|
import com.grack.nanojson.JsonObject;
|
|
|
|
import com.grack.nanojson.JsonParser;
|
|
|
|
import com.grack.nanojson.JsonParserException;
|
2022-03-18 17:50:25 +01:00
|
|
|
|
2019-12-22 13:28:02 +01:00
|
|
|
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;
|
2021-02-16 20:25:48 +01:00
|
|
|
import java.util.Collections;
|
2019-12-22 13:28:02 +01:00
|
|
|
import java.util.List;
|
2022-03-17 17:13:25 +01:00
|
|
|
import java.util.stream.Collectors;
|
2019-12-22 13:28:02 +01:00
|
|
|
|
|
|
|
public class BandcampSuggestionExtractor extends SuggestionExtractor {
|
|
|
|
|
2021-02-21 12:15:45 +01:00
|
|
|
private static final String AUTOCOMPLETE_URL = BASE_API_URL + "/fuzzysearch/1/autocomplete?q=";
|
2020-11-24 09:41:40 +01:00
|
|
|
public BandcampSuggestionExtractor(final StreamingService service) {
|
2019-12-22 13:28:02 +01:00
|
|
|
super(service);
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
2020-11-24 09:41:40 +01:00
|
|
|
public List<String> suggestionList(final String query) throws IOException, ExtractionException {
|
|
|
|
final Downloader downloader = NewPipe.getDownloader();
|
2019-12-22 13:28:02 +01:00
|
|
|
|
2020-03-19 11:18:29 +01:00
|
|
|
try {
|
2022-03-18 17:50:25 +01:00
|
|
|
final JsonObject fuzzyResults = JsonParser.object().from(downloader
|
|
|
|
.get(AUTOCOMPLETE_URL + URLEncoder.encode(query, "UTF-8")).responseBody());
|
2019-12-22 13:28:02 +01:00
|
|
|
|
2022-03-17 17:13:25 +01:00
|
|
|
return fuzzyResults.getObject("auto").getArray("results").stream()
|
|
|
|
.filter(JsonObject.class::isInstance)
|
|
|
|
.map(JsonObject.class::cast)
|
|
|
|
.map(jsonObject -> jsonObject.getString("name"))
|
|
|
|
.distinct()
|
|
|
|
.collect(Collectors.toList());
|
2020-11-24 09:41:40 +01:00
|
|
|
} catch (final JsonParserException e) {
|
2021-02-16 20:25:48 +01:00
|
|
|
return Collections.emptyList();
|
2020-03-19 11:18:29 +01:00
|
|
|
}
|
2019-12-22 13:28:02 +01:00
|
|
|
|
|
|
|
}
|
|
|
|
}
|