241 lines
8.5 KiB
Java
241 lines
8.5 KiB
Java
package net.Broken;
|
|
|
|
import net.Broken.Commands.Move;
|
|
import net.Broken.DB.Entity.GuildPreferenceEntity;
|
|
import net.Broken.DB.Entity.UserEntity;
|
|
import net.Broken.DB.Repository.GuildPreferenceRepository;
|
|
import net.Broken.DB.Repository.UserRepository;
|
|
import net.Broken.Tools.AntiSpam;
|
|
import net.Broken.Tools.Command.CommandParser;
|
|
import net.Broken.Tools.EmbedMessageUtils;
|
|
import net.Broken.Tools.Moderateur;
|
|
import net.Broken.Tools.PrivateMessage;
|
|
import net.Broken.Tools.UserManager.Stats.UserStatsUtils;
|
|
import net.Broken.audio.AudioM;
|
|
import net.dv8tion.jda.core.EmbedBuilder;
|
|
import net.dv8tion.jda.core.entities.*;
|
|
import net.dv8tion.jda.core.events.ReadyEvent;
|
|
import net.dv8tion.jda.core.events.guild.GuildJoinEvent;
|
|
import net.dv8tion.jda.core.events.guild.member.GuildMemberJoinEvent;
|
|
import net.dv8tion.jda.core.events.guild.member.GuildMemberRoleRemoveEvent;
|
|
import net.dv8tion.jda.core.events.guild.voice.GuildVoiceJoinEvent;
|
|
import net.dv8tion.jda.core.events.guild.voice.GuildVoiceLeaveEvent;
|
|
import net.dv8tion.jda.core.events.message.MessageReceivedEvent;
|
|
import net.dv8tion.jda.core.hooks.ListenerAdapter;
|
|
import net.dv8tion.jda.core.managers.GuildManager;
|
|
import org.apache.logging.log4j.LogManager;
|
|
import org.apache.logging.log4j.Logger;
|
|
import org.springframework.context.ApplicationContext;
|
|
|
|
import java.awt.*;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
|
|
|
|
/**
|
|
* Bot Listener
|
|
*/
|
|
public class BotListener extends ListenerAdapter {
|
|
private AntiSpam antispam=new AntiSpam();
|
|
private Moderateur modo = new Moderateur();
|
|
|
|
private GuildPreferenceRepository guildPreferenceRepository;
|
|
private UserRepository userRepository;
|
|
|
|
private Logger logger = LogManager.getLogger();
|
|
|
|
public BotListener() {
|
|
|
|
ApplicationContext context = SpringContext.getAppContext();
|
|
guildPreferenceRepository = (GuildPreferenceRepository) context.getBean("guildPreferenceRepository");
|
|
userRepository = (UserRepository) context.getBean("userRepository");
|
|
|
|
}
|
|
|
|
@Override
|
|
public void onReady(ReadyEvent event) {
|
|
logger.info("Connection succees");
|
|
}
|
|
|
|
|
|
@Override
|
|
public void onGuildMemberJoin(GuildMemberJoinEvent event) {
|
|
|
|
GuildPreferenceEntity guildPref = getPreference(event.getGuild());
|
|
|
|
if(guildPref.isDefaultRole()){
|
|
|
|
logger.info(event.getUser().getName()+ "join the guild, move it!");
|
|
|
|
List<Role> roles = new ArrayList<>();
|
|
roles.add(event.getGuild().getRoleById(guildPref.getDefaultRoleId()));
|
|
|
|
|
|
new Move().exc(event.getMember(), roles,false,event.getGuild(),event.getGuild().getManager());
|
|
}
|
|
|
|
|
|
if(guildPref.isWelcome()){
|
|
|
|
TextChannel chanel = event.getGuild().getTextChannelById(guildPref.getWelcomeChanelID());
|
|
if(chanel != null){
|
|
String message = guildPref.getWelcomeMessage();
|
|
message = message.replaceAll("@name", event.getMember().getAsMention());
|
|
logger.debug(message);
|
|
chanel.sendMessage(message).complete();
|
|
}
|
|
|
|
MainBot.roleFlag = false;
|
|
}
|
|
|
|
|
|
}
|
|
|
|
@Override
|
|
public void onGuildMemberRoleRemove(GuildMemberRoleRemoveEvent event) {
|
|
|
|
GuildPreferenceEntity guildPref = getPreference(event.getGuild());
|
|
if(guildPref.isDefaultRole()){
|
|
|
|
if(!MainBot.roleFlag){
|
|
|
|
if(event.getMember().getRoles().size() == 0){
|
|
|
|
logger.info(event.getUser().getName()+ "have no roles, move it!");
|
|
List<Role> roles = new ArrayList<>();
|
|
roles.add(event.getGuild().getRoleById(guildPref.getDefaultRoleId()));
|
|
|
|
|
|
new Move().exc(event.getMember(), roles,false,event.getGuild(),event.getGuild().getManager());
|
|
MainBot.roleFlag = false;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
logger.debug("ignore it");
|
|
MainBot.roleFlag = false;
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
@Override
|
|
public void onGuildVoiceJoin(GuildVoiceJoinEvent event) {
|
|
super.onGuildVoiceJoin(event);
|
|
if(!event.getMember().getUser().isBot()){
|
|
UserStatsUtils userStatsUtils = UserStatsUtils.getINSTANCE();
|
|
if(!userStatsUtils.runningCounters.containsKey(event.getMember())){
|
|
UserStatsUtils.VoicePresenceCounter temp = new UserStatsUtils.VoicePresenceCounter(event.getMember());
|
|
temp.start();
|
|
userStatsUtils.runningCounters.put(event.getMember(), temp);
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void onGuildVoiceLeave(GuildVoiceLeaveEvent event) {
|
|
super.onGuildVoiceLeave(event);
|
|
if(event.getGuild().getAudioManager().isConnected())
|
|
{
|
|
logger.trace("User disconnected from voice channel.");
|
|
|
|
if(event.getGuild().getAudioManager().getConnectedChannel().getMembers().size() == 1){
|
|
logger.debug("I'm alone, close audio connection.");
|
|
|
|
AudioM.getInstance(event.getGuild()).stop();
|
|
}
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void onMessageReceived(MessageReceivedEvent event) {
|
|
if(!event.getAuthor().isBot()){
|
|
UserStatsUtils.getINSTANCE().addMessageCount(event.getMember());
|
|
}
|
|
try{
|
|
if (event.getMessage().getContentRaw().startsWith("//") && !event.getMessage().getAuthor().getId().equals(event.getJDA().getSelfUser().getId())) {
|
|
//On a detecter que c'etait une commande
|
|
//System.out.println(event.getMessage().getContent());
|
|
List<UserEntity> users = userRepository.findByJdaId(event.getAuthor().getId());
|
|
UserEntity user = users.size() == 0 ? null : users.get(0);
|
|
MainBot.handleCommand(new CommandParser().parse(event.getMessage().getContentRaw(), event), user);
|
|
|
|
}
|
|
else if (!event.getMessage().getAuthor().getId().equals(event.getJDA().getSelfUser().getId()))
|
|
{
|
|
|
|
if(!event.isFromType(ChannelType.PRIVATE)) {
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Guild serveur = event.getGuild();
|
|
GuildPreferenceEntity guildPref = getPreference(serveur);
|
|
|
|
if(!guildPref.isAntiSpam())
|
|
return;
|
|
|
|
GuildManager guildManager = serveur.getManager();
|
|
Member user = event.getMember();
|
|
|
|
// appel de la methode d'analyse de message de "Moderateur"
|
|
if (event.getMessage().getContentRaw().length() > 0) {
|
|
|
|
if (modo.analyse(user, serveur, guildManager, event) == 1) {
|
|
antispam.extermine(user, serveur, guildManager, true, event);
|
|
}
|
|
} else if (event.getMessage().getContentRaw().length() == 0)
|
|
logger.error("Image detected, ignoring it.");
|
|
}
|
|
|
|
|
|
}
|
|
}catch (Exception e){
|
|
logger.catching(e);
|
|
|
|
if(event.isFromType(ChannelType.PRIVATE))
|
|
PrivateMessage.send(event.getAuthor(), EmbedMessageUtils.getInternalError(), logger);
|
|
else
|
|
event.getTextChannel().sendMessage(EmbedMessageUtils.getInternalError()).queue();
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
@Override
|
|
public void onGuildJoin(GuildJoinEvent event) {
|
|
logger.info("Join new guild! (" + event.getGuild().getName() + " " +event.getGuild().getMembers().size() + " Members)");
|
|
super.onGuildJoin(event);
|
|
getPreference(event.getGuild());
|
|
EmbedBuilder eb = new EmbedBuilder().setColor(Color.GREEN)
|
|
.setTitle("Hello there !")
|
|
.setDescription("Allow me to introduce myself -- I am a CL4P-TP the discord bot, but my friends call me Claptrap ! Or they would, if any of them were real...\n"+
|
|
"\nYou can access to my web UI with: " + MainBot.url)
|
|
.setImage("https://i.imgur.com/Anf1Srg.gif");
|
|
|
|
event.getGuild().getDefaultChannel().sendMessage(EmbedMessageUtils.buildStandar(eb)).complete();
|
|
}
|
|
|
|
private GuildPreferenceEntity getPreference(Guild guild){
|
|
List<GuildPreferenceEntity> guildPrefList = guildPreferenceRepository.findByGuildId(guild.getId());
|
|
GuildPreferenceEntity guildPref;
|
|
if(guildPrefList.isEmpty()){
|
|
logger.info("Generate default pref");
|
|
guildPref = GuildPreferenceEntity.getDefault(guild);
|
|
guildPreferenceRepository.save(guildPref);
|
|
}
|
|
else
|
|
guildPref = guildPrefList.get(0);
|
|
return guildPref;
|
|
}
|
|
|
|
}
|