Sheldan/abstracto
1
0
Fork 0
mirror of https://github.com/Sheldan/abstracto.git synced 2026-04-12 19:18:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

enabling auto complete for youtube command

Browse Source
This commit is contained in:
Sheldan
2022-07-27 22:13:48 +02:00
parent 51edb50eee
commit f9b85feeaf
17 changed files with 238 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

84
abstracto-application/abstracto-modules/webservices/webservices-impl/src/main/java/dev/sheldan/abstracto/webservices/common/service/SuggestQueriesServiceBean.java Normal file
View File

@@ -0,0 +1,84 @@
package dev.sheldan.abstracto.webservices.common.service;
import com.google.gson.Gson;
import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonParser;
import dev.sheldan.abstracto.core.exception.AbstractoRunTimeException;
import dev.sheldan.abstracto.webservices.common.exception.SuggestQueriesException;
import lombok.extern.slf4j.Slf4j;
import okhttp3.OkHttpClient;
import okhttp3.Request;
import okhttp3.Response;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.cache.annotation.CacheConfig;
import org.springframework.cache.annotation.Cacheable;
import org.springframework.stereotype.Component;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
@Component
@Slf4j
@CacheConfig(cacheNames = "general-use-cache")
public class SuggestQueriesServiceBean implements SuggestQueriesService {
@Autowired
private OkHttpClient okHttpClient;
@Value("${abstracto.feature.webservices.common.suggestionsURL}")
private String suggestionUrl;
@Autowired
private Gson gson;
@Autowired
private SuggestQueriesServiceBean self;
private List<String> getSuggestionsFromResponse(String response) {
JsonElement rootJson = JsonParser.parseString(response);
if(!rootJson.isJsonArray()) {
return new ArrayList<>();
}
JsonArray mainArray = rootJson.getAsJsonArray();
if(mainArray.size() < 2 || !mainArray.get(1).isJsonArray() || mainArray.get(1).getAsJsonArray().size() == 0) {
return new ArrayList<>();
}
JsonArray suggestionArray = mainArray.get(1).getAsJsonArray();
return Arrays.asList(gson.fromJson(suggestionArray, String[].class));
}
@Override
@Cacheable(key = "{#query, #service}")
public List<String> getSuggestionsForQuery(String query, String service) {
Request request = new Request.Builder()
.url(String.format(suggestionUrl, service, query))
.get()
.build();
Response response;
try {
response = okHttpClient.newCall(request).execute();
if(!response.isSuccessful()) {
if(log.isDebugEnabled()) {
log.error("Failed to retrieve suggestions. Response had code {} with body {}.",
response.code(), response.body());
}
throw new SuggestQueriesException(response.code());
}
return getSuggestionsFromResponse(response.body().string());
} catch (IOException e) {
throw new AbstractoRunTimeException(e);
}
}
@Override
public List<String> getYoutubeSuggestionsForQuery(String query) {
if(query == null || "".equals(query)) {
return new ArrayList<>();
}
return self.getSuggestionsForQuery(query, "yt");
}
}

5
abstracto-application/abstracto-modules/webservices/webservices-impl/src/main/java/dev/sheldan/abstracto/webservices/urban/service/UrbanServiceBean.java
View File

@@ -31,7 +31,10 @@ public class UrbanServiceBean implements UrbanService {
@Override
public UrbanDefinition getUrbanDefinition(String query) throws IOException {
Request request = new Request.Builder().url(String.format(requestUrl, query)).get().build();
Request request = new Request.Builder()
.url(String.format(requestUrl, query))
.get()
.build();
Response response = okHttpClient.newCall(request).execute();
if(!response.isSuccessful()) {
if(log.isDebugEnabled()) {

19
abstracto-application/abstracto-modules/webservices/webservices-impl/src/main/java/dev/sheldan/abstracto/webservices/youtube/command/YoutubeVideoSearch.java
View File

@@ -8,6 +8,7 @@ import dev.sheldan.abstracto.core.command.config.Parameter;
import dev.sheldan.abstracto.core.interaction.slash.SlashCommandConfig;
import dev.sheldan.abstracto.core.command.execution.CommandContext;
import dev.sheldan.abstracto.core.command.execution.CommandResult;
import dev.sheldan.abstracto.core.interaction.slash.parameter.SlashCommandAutoCompleteService;
import dev.sheldan.abstracto.core.interaction.slash.parameter.SlashCommandParameterService;
import dev.sheldan.abstracto.core.config.FeatureDefinition;
import dev.sheldan.abstracto.core.interaction.InteractionService;
@@ -16,12 +17,14 @@ import dev.sheldan.abstracto.core.service.FeatureModeService;
import dev.sheldan.abstracto.core.templating.model.MessageToSend;
import dev.sheldan.abstracto.core.templating.service.TemplateService;
import dev.sheldan.abstracto.core.utils.FutureUtils;
import dev.sheldan.abstracto.webservices.common.service.SuggestQueriesService;
import dev.sheldan.abstracto.webservices.config.WebServicesSlashCommandNames;
import dev.sheldan.abstracto.webservices.config.WebserviceFeatureDefinition;
import dev.sheldan.abstracto.webservices.youtube.config.YoutubeWebServiceFeatureMode;
import dev.sheldan.abstracto.webservices.youtube.model.YoutubeVideo;
import dev.sheldan.abstracto.webservices.youtube.model.command.YoutubeVideoSearchCommandModel;
import dev.sheldan.abstracto.webservices.youtube.service.YoutubeSearchService;
import net.dv8tion.jda.api.events.interaction.command.CommandAutoCompleteInteractionEvent;
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
@@ -57,6 +60,12 @@ public class YoutubeVideoSearch extends AbstractConditionableCommand {
@Autowired
private InteractionService interactionService;
@Autowired
private SlashCommandAutoCompleteService slashCommandAutoCompleteService;
@Autowired
private SuggestQueriesService suggestQueriesService;
@Override
public CompletableFuture<CommandResult> executeAsync(CommandContext commandContext) {
String query = (String) commandContext.getParameters().getParameters().get(0);
@@ -100,6 +109,15 @@ public class YoutubeVideoSearch extends AbstractConditionableCommand {
.thenApply(o -> CommandResult.fromSuccess());
}
@Override
public List<String> performAutoComplete(CommandAutoCompleteInteractionEvent event) {
if(slashCommandAutoCompleteService.matchesParameter(event.getFocusedOption(), SEARCH_QUERY_PARAMETER)) {
return suggestQueriesService.getYoutubeSuggestionsForQuery(event.getFocusedOption().getValue());
} else {
return new ArrayList<>();
}
}
@Override
public CommandConfiguration getConfiguration() {
List<Parameter> parameters = new ArrayList<>();
@@ -108,6 +126,7 @@ public class YoutubeVideoSearch extends AbstractConditionableCommand {
.name(SEARCH_QUERY_PARAMETER)
.type(String.class)
.remainder(true)
.supportsAutoComplete(true)
.templated(true)
.build();
parameters.add(queryParameter);

2
abstracto-application/abstracto-modules/webservices/webservices-impl/src/main/java/dev/sheldan/abstracto/webservices/youtube/service/YoutubeSearchServiceBean.java
View File

@@ -6,6 +6,7 @@ import com.google.api.services.youtube.model.SearchResult;
import dev.sheldan.abstracto.webservices.youtube.exception.YoutubeAPIException;
import dev.sheldan.abstracto.webservices.youtube.exception.YoutubeVideoNotFoundException;
import dev.sheldan.abstracto.webservices.youtube.model.YoutubeVideo;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
@@ -13,6 +14,7 @@ import java.io.IOException;
import java.util.List;
@Component
@Slf4j
public class YoutubeSearchServiceBean implements YoutubeSearchService {
@Autowired

2
abstracto-application/abstracto-modules/webservices/webservices-impl/src/main/resources/webservices-config.properties
View File

@@ -1,3 +1,5 @@
abstracto.feature.webservices.common.suggestionsURL=http://suggestqueries.google.com/complete/search?client=firefox&ds=%s&q=%s
abstracto.featureFlags.youtube.featureName=youtube
abstracto.featureFlags.youtube.enabled=false

28
abstracto-application/abstracto-modules/webservices/webservices-int/src/main/java/dev/sheldan/abstracto/webservices/common/exception/SuggestQueriesException.java Normal file
View File

@@ -0,0 +1,28 @@
package dev.sheldan.abstracto.webservices.common.exception;
import dev.sheldan.abstracto.core.exception.AbstractoRunTimeException;
import dev.sheldan.abstracto.core.templating.Templatable;
import dev.sheldan.abstracto.webservices.common.model.exception.SuggestQueriesExceptionModel;
public class SuggestQueriesException extends AbstractoRunTimeException implements Templatable {
private final SuggestQueriesExceptionModel model;
public SuggestQueriesException(Integer responseCode) {
super(String.format("Request failure towards suggest queries %s.", responseCode));
this.model = SuggestQueriesExceptionModel
.builder()
.responseCode(responseCode)
.build();
}
@Override
public String getTemplateName() {
return "suggest_queries_request_exception";
}
@Override
public Object getTemplateModel() {
return model;
}
}

14
abstracto-application/abstracto-modules/webservices/webservices-int/src/main/java/dev/sheldan/abstracto/webservices/common/model/exception/SuggestQueriesExceptionModel.java Normal file
View File

@@ -0,0 +1,14 @@
package dev.sheldan.abstracto.webservices.common.model.exception;
import lombok.Builder;
import lombok.Getter;
import lombok.Setter;
import java.io.Serializable;
@Getter
@Setter
@Builder
public class SuggestQueriesExceptionModel implements Serializable {
private Integer responseCode;
}

8
abstracto-application/abstracto-modules/webservices/webservices-int/src/main/java/dev/sheldan/abstracto/webservices/common/service/SuggestQueriesService.java Normal file
View File

@@ -0,0 +1,8 @@
package dev.sheldan.abstracto.webservices.common.service;
import java.util.List;
public interface SuggestQueriesService {
List<String> getSuggestionsForQuery(String query, String service);
List<String> getYoutubeSuggestionsForQuery(String query);
}