UNLOZE/projects-jenz
9
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

finnaly fixed sticker crashing and moved logic a bit around. also made it learn from other channels again to increase data collection

Browse Source
This commit is contained in:
christian
2021-12-05 01:22:40 +01:00
parent d6d15e87e0
commit 45308b5d65
3 changed files with 71 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ArtificialAutism/src/main/java/PresentationLayer/DiscordHandler.java
+68 -1
View File
@@ -6,6 +6,9 @@ import FunctionLayer.PipelineJMWESingleton;
import discord4j.core.DiscordClient;
import discord4j.core.GatewayDiscordClient;
import discord4j.core.event.domain.message.MessageCreateEvent;
import discord4j.core.object.entity.Sticker;
import discord4j.core.object.entity.User;
import discord4j.core.object.entity.channel.TextChannel;
import edu.stanford.nlp.pipeline.StanfordCoreNLP;
import java.io.IOException;
@@ -13,6 +16,7 @@ import java.io.UnsupportedEncodingException;
import java.net.*;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
/**
@@ -104,7 +108,70 @@ public class DiscordHandler {
}).start();
}
gateway.on(MessageCreateEvent.class).subscribe(event -> {
FunctionLayer.DoStuff.doStuff(event, usernameBot, datahandler, stanfordCoreNLP, stanfordCoreNLPSentiment);
String username = "";
try {
username = event.getMessage().getAuthor().get().getUsername();
} catch (java.util.NoSuchElementException e) {
username = null;
}
//System.out.println("past event.getMessage().getAuthor().get().getUsername()");
List<User> blockLast = event.getMessage().getUserMentions();
String content = event.getMessage().getContent();
if (blockLast != null) {
for (User user : blockLast) {
content = content.replace(user.getId().asString(), "");
}
}
//System.out.println("past event.getMessage().getContent()");
if (username != null && !username.equals(usernameBot) && !content.isEmpty()) {
TextChannel block = event.getMessage().getChannel().cast(TextChannel.class).block();
String name = block.getCategory().block().getName();
name = name.toLowerCase();
String channelName = block.getName().toLowerCase();
boolean channelpermissionsDenied = false;
if (channelName.contains("suggestion-box")) {
channelpermissionsDenied = true;
}
switch (name) {
case "public area":
case "information area": {
break;
}
default: {
channelpermissionsDenied = true;
break;
}
}
if (!channelpermissionsDenied) {
//System.out.println("past !channelpermissionsDenied");
boolean mentionedBot = false;
if (blockLast != null) {
for (User user : blockLast) {
if (user.getUsername().equals(usernameBot)) {
mentionedBot = true;
break;
}
}
}
if (mentionedBot || channelName.contains("general-autism")) {
String ResponseStr;
ResponseStr = datahandler.getResponseMsg(content, username, stanfordCoreNLP, stanfordCoreNLPSentiment,
false);
if (!ResponseStr.isEmpty()) {
System.out.print("\nResponseStr3: " + ResponseStr + "\n");
event.getMessage().getChannel().flatMap(chan -> chan.createMessage(ResponseStr)).block();
//event.getMessage().getChannel().block().createMessage(ResponseStr).block();
//System.out.print("past event.getMessage().getChannel()");
}
} else {
//System.out.println("pre else statement trimString");
String strF = datahandler.trimString(content);
//System.out.println("post else statement trimString. strf: " + strF);
datahandler.getResponseFutures(strF, stanfordCoreNLP, stanfordCoreNLPSentiment);
}
}
}
});
gateway.onDisconnect().block();
}