mirror of https://github.com/bvn13/JIrcBot.git
306 lines
12 KiB
Java
306 lines
12 KiB
Java
package ru.bvn13.jircbot.listeners;
|
|
|
|
import org.pircbotx.hooks.events.JoinEvent;
|
|
import org.pircbotx.hooks.events.MessageEvent;
|
|
import org.pircbotx.hooks.events.PrivateMessageEvent;
|
|
import org.pircbotx.hooks.types.GenericEvent;
|
|
import org.pircbotx.hooks.types.GenericMessageEvent;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.stereotype.Component;
|
|
import ru.bvn13.jircbot.bot.ImprovedListenerAdapter;
|
|
import ru.bvn13.jircbot.bot.JircBot;
|
|
import ru.bvn13.jircbot.config.JircBotConfiguration;
|
|
import ru.bvn13.jircbot.database.entities.ChannelSettings;
|
|
import ru.bvn13.jircbot.database.services.ChannelSettingsService;
|
|
import ru.bvn13.jircbot.model.Config;
|
|
|
|
import java.util.concurrent.atomic.AtomicReference;
|
|
|
|
/**
|
|
* Created by bvn13 on 27.03.2018.
|
|
*/
|
|
@Component
|
|
public class AdminListener extends ImprovedListenerAdapter {
|
|
|
|
private static final Logger logger = LoggerFactory.getLogger(AdminListener.class);
|
|
|
|
private static final String COMMAND = "?";
|
|
|
|
@Autowired
|
|
private JircBotConfiguration configuration;
|
|
|
|
@Autowired
|
|
private ChannelSettingsService channelSettingsService;
|
|
|
|
@Override
|
|
public void onJoin(JoinEvent event) throws Exception {
|
|
if (event.getChannel().getName().startsWith("#")) {
|
|
if (event.getUser().getNick().equals(event.getBot().getNick())) {
|
|
event.getBot().sendRaw().rawLineNow("MODE " + event.getBot().getUserBot().getNick() + " +B");
|
|
try {
|
|
channelSettingsService.createChannelSettings(JircBot.extractServer(event.getBot().getServerHostname()), event.getChannel().getName());
|
|
} catch (Exception e) {
|
|
logger.error("Could not create channel settings for channel "+event.getChannel().getName()+" at "+event.getBot().getServerHostname(), e);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
public void onMessage(MessageEvent event) throws Exception {
|
|
Config config = getBotConfig(event);
|
|
if (config == null) {
|
|
return;
|
|
}
|
|
|
|
if (event.getUser().isVerified()
|
|
&& !config.getMasterNick().isEmpty()
|
|
&& config.getMasterNick().equals(event.getUser().getNick())) {
|
|
|
|
if (event.getMessage().startsWith(COMMAND)) {
|
|
String command = event.getMessage().substring(COMMAND.length());
|
|
String commands[] = command.trim().split(" ", 2);
|
|
|
|
if (commands[0].startsWith("+") || commands[0].startsWith("-")) {
|
|
boolean isApply = commands[0].startsWith("+");
|
|
command = commands[0].substring(1);
|
|
|
|
switch (command.toLowerCase()) {
|
|
case "op" :
|
|
if (commands.length == 1) {
|
|
event.getBot().sendRaw().rawLine("PRIVMSG chanserv :" + (!isApply ? "deop" : "op") + " " + event.getChannel().getName() + " " + event.getUser().getNick());
|
|
} else {
|
|
event.getBot().sendRaw().rawLine("PRIVMSG chanserv :" + (!isApply ? "deop" : "op") + " " + event.getChannel().getName() + " " + commands[1]);
|
|
}
|
|
break;
|
|
case "v" :
|
|
case "voice":
|
|
if (commands.length == 1) {
|
|
event.getBot().sendRaw().rawLine("MODE " + event.getChannel().getName() + " " + (isApply ? "+" : "-") + "v " + event.getUser().getNick());
|
|
} else {
|
|
event.getBot().sendRaw().rawLine("MODE " + event.getChannel().getName() + " " + (isApply ? "+" : "-") + "v " + commands[1]);
|
|
}
|
|
break;
|
|
}
|
|
} else {
|
|
switch (command.toLowerCase()) {
|
|
case "inv" :
|
|
case "invite" :
|
|
if (command.length() > 1) {
|
|
event.getBot().sendRaw().rawLine("INVITE " + event.getChannel().getName() + " " + commands[1]);
|
|
}
|
|
break;
|
|
case "kick" :
|
|
if (command.length() > 1) {
|
|
event.getBot().sendRaw().rawLine("KICK " + event.getChannel().getName() + " " + commands[1] + (commands.length > 2 ? " "+commands[2] : ""));
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void onPrivateMessage(PrivateMessageEvent event) throws Exception {
|
|
|
|
Config config = getBotConfig(event);
|
|
if (config == null) {
|
|
return;
|
|
}
|
|
|
|
if (event.getUser().isVerified()
|
|
&& !config.getMasterNick().isEmpty()
|
|
&& config.getMasterNick().equals(event.getUser().getNick())) {
|
|
|
|
if (event.getMessage().startsWith(COMMAND)) {
|
|
String command = event.getMessage().substring(COMMAND.length());
|
|
String commands[] = command.trim().split(" ", 2);
|
|
|
|
if (commands.length != 2) {
|
|
event.respondPrivateMessage("Wrong command");
|
|
return;
|
|
}
|
|
|
|
String args[] = null;
|
|
switch (commands[0].toLowerCase()) {
|
|
case "restart":
|
|
event.getBot().stopBotReconnect(); break;
|
|
case "join" :
|
|
event.getBot().sendIRC().joinChannel(commands[1]); event.respondPrivateMessage("done"); break;
|
|
case "leave" :
|
|
event.getBot().sendRaw().rawLine("PART "+commands[1]); event.respondPrivateMessage("done"); break;
|
|
case "privmsg" :
|
|
event.getBot().sendRaw().rawLine("PRIVMSG "+commands[1]); event.respondPrivateMessage("done"); break;
|
|
case "cmd" :
|
|
args = commands[1].split(" ", 2);
|
|
event.getBot().sendRaw().rawLine(args[1]); event.respondPrivateMessage("done"); break;
|
|
case "set" :
|
|
try {
|
|
args = commands[1].split(" ", 4); // set, channel, mode/hello-message
|
|
if (args.length == 3) {
|
|
changeSettings(JircBot.extractServer(event.getBot().getServerHostname()), args[0], args[1], args[2]); //server, channel, set, mode
|
|
event.respondPrivateMessage("done");
|
|
} else {
|
|
event.respondPrivateMessage("SYNTAX: ?set <channel> <option> on/off");
|
|
}
|
|
} catch (Exception e) {
|
|
event.respondPrivateMessage(e.toString());
|
|
logger.error("ERROR", e);
|
|
}
|
|
break;
|
|
case "op" :
|
|
case "deop" :
|
|
String cmd = commands[0].toLowerCase();
|
|
args = commands[1].split(" ", 2);
|
|
if (args.length == 1) {
|
|
event.getBot().sendRaw().rawLine("PRIVMSG chanserv :"+cmd+" "+args[0]+" "+event.getUser().getNick());
|
|
event.respondPrivateMessage("done");
|
|
} else if (args.length == 2) {
|
|
event.getBot().sendRaw().rawLine("PRIVMSG chanserv :"+cmd+" "+args[0]+" "+args[1]);
|
|
event.respondPrivateMessage("done");
|
|
} else {
|
|
event.respondPrivateMessage("wrong arguments");
|
|
}
|
|
break;
|
|
case "kick" :
|
|
args = commands[1].split(" ", 3);
|
|
if (args.length == 2) {
|
|
event.getBot().sendRaw().rawLine("KICK "+args[0]+" "+args[1]);
|
|
event.respondPrivateMessage("done");
|
|
} else if (args.length == 3) {
|
|
event.getBot().sendRaw().rawLine("KICK "+args[0]+" "+args[1]+" "+args[2]);
|
|
event.respondPrivateMessage("done");
|
|
} else {
|
|
event.respondPrivateMessage("wrong arguments");
|
|
}
|
|
break;
|
|
default:
|
|
event.respondPrivateMessage("Command not supported");
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
private boolean sameServer(String s1, String s2) {
|
|
String d1[] = s1.split("[\\.]");
|
|
String d2[] = s2.split("[\\.]");
|
|
|
|
if (d1.length >= 2 && d2.length >= 2) {
|
|
if (d1[d1.length - 1].equals(d2[d2.length - 1])
|
|
&& d1[d1.length - 2].equals(d2[d2.length - 2])) {
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
private Config getBotConfig(GenericMessageEvent event) {
|
|
AtomicReference<Config> aConfig = new AtomicReference<>(null);
|
|
|
|
configuration.getConnections().forEach(c -> {
|
|
|
|
if (sameServer(event.getBot().getServerHostname(), c.getServer())) {
|
|
aConfig.set(c);
|
|
}
|
|
|
|
});
|
|
|
|
if (aConfig.get() == null) {
|
|
event.respondPrivateMessage("sorry, bot not found!");
|
|
return null;
|
|
}
|
|
|
|
Config config = aConfig.get();
|
|
return config;
|
|
}
|
|
|
|
private void changeSettings(String serverHost, String channelName, String set, String modeStr) {
|
|
if (set.equals("hello-message") || set.equals("hello-msg")) {
|
|
ChannelSettings settings = channelSettingsService.getChannelSettings(serverHost, channelName);
|
|
settings.setOnJoinMessage(modeStr);
|
|
channelSettingsService.save(settings);
|
|
} else {
|
|
if (!modeStr.equals("off") && !modeStr.equals("on")) {
|
|
throw new RuntimeException("Wrong mode. Variants: on, off");
|
|
}
|
|
|
|
Boolean mode = modeStr.equals("on");
|
|
ChannelSettings settings = channelSettingsService.getChannelSettings(serverHost, channelName);
|
|
|
|
switch (set.toLowerCase()) {
|
|
case "autorejoin":
|
|
case "auto-rejoin":
|
|
settings.setAutoRejoinEnabled(mode);
|
|
break;
|
|
case "bash":
|
|
case "bashorg":
|
|
settings.setBashOrgEnabled(mode);
|
|
break;
|
|
case "defferedmessages":
|
|
case "deffered-messages":
|
|
case "tell" :
|
|
settings.setDeferredMessagesEnabled(mode);
|
|
break;
|
|
case "gs":
|
|
case "googlesearch":
|
|
case "google-search":
|
|
settings.setGoogleSearchEnabled(mode);
|
|
break;
|
|
case "grammar":
|
|
case "grammarcorrection":
|
|
case "grammar-correction":
|
|
settings.setGrammarCorrectionEnabled(mode);
|
|
break;
|
|
case "hello":
|
|
case "helloonjoin":
|
|
case "hello-on-join":
|
|
settings.setHelloOnJoinEnabled(mode);
|
|
break;
|
|
case "links":
|
|
case "linkpreview":
|
|
case "links-preview":
|
|
settings.setLinkPreviewEnabled(mode);
|
|
break;
|
|
case "logging":
|
|
case "log":
|
|
settings.setLoggingEnabled(mode);
|
|
break;
|
|
case "regex":
|
|
case "regexp":
|
|
case "regexchecker":
|
|
case "regexpchecker":
|
|
case "regex-checker":
|
|
case "regexp-checker":
|
|
settings.setRegexCheckerEnabled(mode);
|
|
break;
|
|
case "advice":
|
|
case "advices":
|
|
settings.setAdvicesEnabled(mode);
|
|
break;
|
|
case "calc":
|
|
case "calculator":
|
|
settings.setCalculatorEnabled(mode);
|
|
break;
|
|
case "quiz":
|
|
settings.setQuizEnabled(mode);
|
|
break;
|
|
default:
|
|
throw new RuntimeException("Setting " + set + " not exist");
|
|
}
|
|
|
|
channelSettingsService.save(settings);
|
|
}
|
|
}
|
|
}
|