120 lines
5.1 KiB
Java
120 lines
5.1 KiB
Java
package plugin;
|
|
|
|
import net.mamoe.mirai.console.plugin.jvm.JavaPlugin;
|
|
import net.mamoe.mirai.console.plugin.jvm.JvmPluginDescriptionBuilder;
|
|
import net.mamoe.mirai.event.GlobalEventChannel;
|
|
import net.mamoe.mirai.event.events.FriendMessageEvent;
|
|
import net.mamoe.mirai.event.events.GroupMessageEvent;
|
|
import net.mamoe.mirai.event.events.MessageEvent;
|
|
import plugin.constant.ChatConstant;
|
|
import plugin.listener.FriendMessageListener;
|
|
import plugin.listener.GroupMessageListener;
|
|
import plugin.listener.OwnerMessageListener;
|
|
import plugin.pojo.Config;
|
|
import plugin.utils.ConfigUtil;
|
|
|
|
import java.io.IOException;
|
|
import java.util.HashMap;
|
|
import java.util.List;
|
|
|
|
|
|
/**
|
|
* @author SLHAF
|
|
*/
|
|
public final class App extends JavaPlugin {
|
|
public static final App INSTANCE = new App();
|
|
|
|
private App() {
|
|
super(new JvmPluginDescriptionBuilder("com.plugin.chatAI-InGroup", "0.1.0")
|
|
.name("ChatAI-InGroup")
|
|
.author("SLHAF")
|
|
.build());
|
|
}
|
|
|
|
|
|
@Override
|
|
public void onEnable() {
|
|
String owner, bot;
|
|
HashMap<String, String> customCommands;
|
|
List<Long> blacklist;
|
|
//加载配置
|
|
try {
|
|
ConfigUtil.load();
|
|
Thread.sleep(1500);
|
|
Config config = ConfigUtil.getConfig();
|
|
owner = config.getOwner().substring(1);
|
|
bot = config.getBot().substring(1);
|
|
customCommands = config.getCustomCommands();
|
|
blacklist = config.getBlacklist();
|
|
} catch (IOException | ClassNotFoundException | InterruptedException e) {
|
|
throw new RuntimeException(e);
|
|
}
|
|
getLogger().info("ChatAI-InGroup loaded!");
|
|
|
|
//群聊监听器
|
|
GlobalEventChannel.INSTANCE.filterIsInstance(GroupMessageEvent.class)
|
|
.filter(event -> {
|
|
String msg = event.getMessage().contentToString();
|
|
long groupId = event.getGroup().getId();
|
|
return ((msg.startsWith(".") && msg.length() != 1) || msg.startsWith("@" + bot) || customCommands.containsKey(msg.split(" ")[0] + ChatConstant.BLANK)) && !blacklist.contains(groupId);
|
|
}).registerListenerHost(new GroupMessageListener());
|
|
|
|
//私聊监听器
|
|
GlobalEventChannel.INSTANCE.filterIsInstance(FriendMessageEvent.class)
|
|
.filter(event -> {
|
|
String msg = event.getMessage().contentToString();
|
|
String sender = String.valueOf(event.getFriend().getId());
|
|
return !(msg.startsWith(ChatConstant.SET) && sender.equals(owner)) && !msg.equals(ChatConstant.HELP);
|
|
}).registerListenerHost(new FriendMessageListener());
|
|
|
|
//所有者监听
|
|
GlobalEventChannel.INSTANCE.filterIsInstance(MessageEvent.class)
|
|
.filter(event -> {
|
|
String msg = event.getMessage().contentToString();
|
|
String sender = String.valueOf(event.getSender().getId());
|
|
return msg.startsWith(ChatConstant.SET) && sender.equals(owner);
|
|
}).registerListenerHost(new OwnerMessageListener());
|
|
|
|
//帮助监听
|
|
GlobalEventChannel.INSTANCE
|
|
.filterIsInstance(MessageEvent.class)
|
|
.filter(event -> event.getMessage().contentToString().equals(ChatConstant.HELP))
|
|
.subscribeAlways(MessageEvent.class, event -> {
|
|
synchronized (customCommands) {
|
|
final String[] helpMsg = {"""
|
|
————<群聊命令>————
|
|
|
|
@<bot> <content> (仅限群聊)
|
|
/<command> <content>
|
|
|
|
例：
|
|
@机器人 你好
|
|
/c 你好
|
|
|
|
————<控制命令>————
|
|
|
|
$ clearAll (仅限群聊)
|
|
$ shutUp (仅限群聊)
|
|
$ speak (仅限群聊)
|
|
$ 添加预设|<预设指令>|<模型名称/null>|<预设内容>
|
|
$ 切换模型|<预设指令>|<模型名称>
|
|
$ 更改预设|<预设指令>|<预设内容>
|
|
$ 删除预设|<预设指令>
|
|
|
|
例：
|
|
$ 添加预设|/c|glm-4-flash|你是一只猫娘...
|
|
|
|
"""};
|
|
helpMsg[0] += "————<预设列表>————";
|
|
customCommands.forEach((s, s2) -> helpMsg[0] += "\r\n\r\n" + s + "-> \r\n" + s2);
|
|
event.getSubject().sendMessage(helpMsg[0]);
|
|
}
|
|
});
|
|
|
|
}
|
|
|
|
@Override
|
|
public void onDisable() {
|
|
getLogger().info("ChatAI-InGroup disabled!");
|
|
}
|
|
} |