注意, 目前最多支持2参数的方法 多余的参数将会传入null
12 | * 受到注解的方法必须为public的
13 | *
14 | * 例如
15 | *
16 | * {@code
17 | * 在群消息中发送任何一条含有aaa的消息就会被触发
18 | * @EventHandler(target = "aaa")
19 | * public void test1() { your code }
20 | * 在群消息中发送任何一条含有/aaa的消息就会被触发
21 | * @EventHandler(start = "/", target = "aaa")
22 | * public void test1(MessageEventPack eventPack) { your code }
23 | * 任何消息均会触发该事件处理器
24 | * !!此时消息中所有纯文本内容均会作为参数被解析后放入data.args中!
25 | * @EventHandler(isAny = true)
26 | * public void test1(MessageEventPack eventPack, PreProcessData data) { your code }
27 | * }
28 | *
29 | *
30 | * 关于PreProcessData的内容, 请参考{@link PreProcessorData}
31 | * @author diyigemt
32 | * @since 1.0.0
33 | */
34 | @Target({ElementType.METHOD})
35 | @Retention(RetentionPolicy.RUNTIME)
36 | @Documented
37 | public @interface EventHandler {
38 | /**
39 | * 是否为常驻消息处理器
40 | * 设置为true时无论如何该消息处理器均会触发
41 | * @since 1.0.0
42 | */
43 | boolean isAny() default false;
44 | /**
45 | * 匹配的指令开头 为空表示忽略
46 | * 如:@MessageEventFilter(start = "/")
47 | * 表示将从 '/'之后的内容获取指令名和参数
48 | * 如:消息 '一些乱七八糟的内容 /help asd'
49 | * 将会获取到指令 help 和参数 asd
50 | * 当配置文件中存在DEFAULT_COMMAND_START非空配置项且start值为空时 将使用配置项代替
51 | * @since 1.0.0
52 | */
53 | String start() default "";
54 | /**
55 | * 参数分隔符 用于获取指令之后的文本参数
56 | * 值为正则表达式 默认使用'\\s+' 即以至少1个空格为分割
57 | * 如:@EventHandler(start = "/", split = "\\s+")
58 | * 处理消息 '一些乱七八糟的内容 /help asd ssd'
59 | * 将会得到 指令 help 和参数 asd与ssd
60 | * @since 1.0.0
61 | */
62 | String split() default "\\s+";
63 | /**
64 | * 触发handler的指令 为空时取受到注解的方法名
65 | * @since 1.0.0
66 | */
67 | String target() default "";
68 | /**
69 | * 处理类型 群消息或好友消息或者全部
70 | * 更多消息类型在做了
71 | * @since 1.0.0
72 | * @see EventHandlerType
73 | */
74 | EventHandlerType[] type() default EventHandlerType.MESSAGE_HANDLER_ALL;
75 | }
76 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/annotation/EventHandlerComponent.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.annotation;
2 |
3 | import java.lang.annotation.*;
4 |
5 | /**
6 | * 在受到注解的类中扫描事件handler
7 | * 未受到该注解的类 扫描时将会被忽略
8 | * @author diyigemt
9 | * @since 1.0.0
10 | */
11 | @Target({ElementType.TYPE})
12 | @Retention(RetentionPolicy.RUNTIME)
13 | @Documented
14 | public @interface EventHandlerComponent {
15 | /**
16 | * 默认优先级
17 | * 当@CheckPermission的permissionIndex未设置(为默认的0)时取该值
18 | */
19 | int value() default 0;
20 | }
21 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/annotation/EventHandlerHelp.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.annotation;
2 |
3 |
4 | import java.lang.annotation.*;
5 |
6 | /**
7 | * 为指令添加帮助信息
8 | * 例如:
9 |
10 | {@code
11 | @EventHandler(target = "乐", start = "/")
12 | @EventHandlerHelp(value = 1, description = "发送一个乐子", detail = {/乐 随机发送一个乐子", "/乐 1 发送编号为1的乐子"})
13 | public void happy(MessageEventPack eventPack, PreProcessData data) { your code }
14 | }
15 |
16 | * 以上将会在用户发送"帮助"的时候回复:"可用指令:1./乐"
17 | * 在用户发送"帮助 1"的时候回复:"1./乐 随机发送一个乐子/n2./乐 1 发送编号为1的乐子"
18 | * @author diyigemt
19 | * @since 1.1.0
20 | */
21 | @Target({ElementType.METHOD})
22 | @Retention(RetentionPolicy.RUNTIME)
23 | @Documented
24 | public @interface EventHandlerHelp {
25 | /**
26 | * 是否录入帮助信息
27 | * 为false时不会将被注解的方法录入帮助信息中
28 | */
29 | boolean enable() default true;
30 | /**
31 | * 指令对应的帮助id
32 | * 会在用户输入 帮助 帮助id时，将id对应的指令具体触发方式描述打印
33 | */
34 | int value();
35 | /**
36 | * 指令功能描述
37 | */
38 | String description() default "";
39 |
40 | /**
41 | * 指令的具体触发方式描述
42 | */
43 | String[] detail() default {};
44 | }
45 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/annotation/ExceptionHandler.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.annotation;
2 |
3 |
4 | import java.lang.annotation.*;
5 |
6 | /**
7 | * 将收到注解的方法注册为异常Handler
8 | * 受到注解的方法至多有一个参数 即[Throwable]本身 多余的参数将会被传入null
9 | * 方法返回值可以为void 也可以为boolean 返回boolean且为true时将会阻止低优先级的异常处理器的触发
10 | * 返回值不正确的方法将不会被注册成异常处理器!!
11 | * 注解的value为空也不会被注册为异常处理器!!
12 | *
13 | * {@code
14 | * @ExceptionHandler
15 | * public void testHandler1(Throwable e) {your code}
16 | *
17 | * @ExceptionHandler(priority = 1)
18 | * public boolean testHandler2(Throwable e) {your code}
19 | *
20 | * @ExceptionHandler(priority = 1)
21 | * public boolean testHandler3() {your code}
22 | * }
23 | *
24 | */
25 | @Target({ElementType.METHOD})
26 | @Retention(RetentionPolicy.RUNTIME)
27 | @Documented
28 | public @interface ExceptionHandler {
29 | /**
30 | * 处理器名
31 | * 可根据该名字主动移除一个异常处理器
32 | */
33 | String name() default "";
34 | /**
35 | * 要处理的异常列表
36 | */
37 | Class value();
38 |
39 | /**
40 | * 处理优先级
41 | */
42 | int priority() default 0;
43 | }
44 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/annotation/ExceptionHandlerComponent.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.annotation;
2 |
3 |
4 | import java.lang.annotation.*;
5 |
6 | /**
7 | * 在受到注解的类中扫描异常handler
8 | * 未受到该注解的类 扫描时将会被忽略
9 | * @author diyigemt
10 | * @since 1.0.0
11 | */
12 | @Target({ElementType.TYPE})
13 | @Retention(RetentionPolicy.RUNTIME)
14 | @Documented
15 | public @interface ExceptionHandlerComponent {
16 | /**
17 | * 默认优先级
18 | * 当@ExceptionHandler的priority未设置(为默认的0)时取该值
19 | */
20 | int value() default 0;
21 | }
22 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/annotation/MessageFilter.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.annotation;
2 |
3 | import net.diyigemt.miraiboot.constant.MessageFilterMatchType;
4 | import net.diyigemt.miraiboot.entity.MessageFilterImp;
5 | import net.diyigemt.miraiboot.interfaces.IMessageFilter;
6 |
7 | import java.lang.annotation.*;
8 |
9 | /**
10 | * 消息事件过滤器
11 | * 在所有规则通过时EventHandler才会被触发
12 | * 其实和@EventHandler有点重复了
13 | * 可以重复注解 此时将会对所有进行匹配 (更多匹配规则在做了)
14 | * @author diyigemt
15 | * @since 1.0.0
16 | */
17 | @Target({ElementType.METHOD})
18 | @Retention(RetentionPolicy.RUNTIME)
19 | @Repeatable(MessageFilters.class)
20 | @Documented
21 | public @interface MessageFilter {
22 |
23 | /**
24 | * 匹配关键词的内容 为空表示忽略
25 | * @since 1.0.0
26 | */
27 | String value() default "";
28 | /**
29 | * 匹配类型 详见MessageEventFilterMatchType
30 | * @see MessageFilterMatchType
31 | * @since 1.0.0
32 | */
33 | MessageFilterMatchType matchType() default MessageFilterMatchType.NULL;
34 | /**
35 | * 匹配的账号列表
36 | * 若消息发送着不在列表内则不做响应
37 | * 为空则允许所有
38 | * @since 1.0.0
39 | */
40 | String[] accounts() default {};
41 | /**
42 | * 匹配的群列表
43 | * 若消息发送着不在列表内则不做响应
44 | * 为空则允许所有
45 | * @since 1.0.0
46 | */
47 | String[] groups() default {};
48 | /**
49 | * 匹配的机器人qq号列表
50 | * 若消息接受的bot不在列表内则不做响应
51 | * 为空则允许所有
52 | * @since 1.0.0
53 | */
54 | String[] bots() default {};
55 | /**
56 | * 是否当bot被at时才触发
57 | * 好友消息时忽略
58 | * @since 1.0.0
59 | */
60 | boolean isAt() default false;
61 | /**
62 | * 是否at全体时才触发
63 | * 好友消息时忽略
64 | * @since 1.0.0
65 | */
66 | boolean isAtAll() default false;
67 | /**
68 | * 是否有人被at时才触发
69 | * 不一定是bot被at
70 | * 好友消息时忽略
71 | * @since 1.0.0
72 | */
73 | boolean isAtAny() default false;
74 |
75 | /**
76 | * 自定义消息过滤器
77 | * 不为默认值时将与以上设置共存
78 | */
79 | Class filter() default MessageFilterImp.class;
80 | }
81 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/annotation/MessageFilters.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.annotation;
2 |
3 | import java.lang.annotation.*;
4 |
5 | /**
6 | * 消息事件过滤器组
7 | * @author diyigemt
8 | * @since 1.0.0
9 | * @see MessageFilter
10 | */
11 | @Target({ElementType.METHOD})
12 | @Retention(RetentionPolicy.RUNTIME)
13 | @Documented
14 | public @interface MessageFilters {
15 | MessageFilter[] value() default {};
16 | }
17 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/annotation/MessagePreProcessor.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.annotation;
2 |
3 | import net.diyigemt.miraiboot.constant.MessagePreProcessorMessageType;
4 | import net.diyigemt.miraiboot.entity.MessageProcessorImp;
5 | import net.diyigemt.miraiboot.entity.PreProcessorData;
6 | import net.diyigemt.miraiboot.interfaces.IMessagePreProcessor;
7 |
8 | import java.lang.annotation.*;
9 |
10 | /**
11 | * 消息事件预处理器
12 | * 对收到的消息事件进行预处理
13 | * 例如
14 | *
15 | * {@code
16 | * 以下配置将会响应 搜图 指令 并将消息中的图片过滤至 data.classified中
17 | * @EventHandler(target = "搜图")
18 | * @MessagePreProcessor(filterType = MessagePreProcessorMessageType.IMAGE)
19 | * public void searchImage(MessageEvent event, PreProcessData data) { your code }
20 | * }
21 | *
22 | * @author diyigemt
23 | * @since 1.0.0
24 | */
25 | @Target({ElementType.METHOD})
26 | @Retention(RetentionPolicy.RUNTIME)
27 | @Documented
28 | public @interface MessagePreProcessor {
29 | /**
30 | * 将所有纯文本消息提取出来
31 | * 保存在PreProcessorData.text中
32 | * 其实无论是true还是false此项均会起作用
33 | * @see PreProcessorData
34 | * @since 1.0.0
35 | */
36 | boolean textProcessor() default false;
37 | /**
38 | * 将对应类型的消息提取出来
39 | * 保存在PreProcessorData.classified中
40 | * @see PreProcessorData
41 | * @see MessagePreProcessorMessageType
42 | * @since 1.0.0
43 | */
44 | MessagePreProcessorMessageType[] filterType() default {};
45 |
46 | /**
47 | * 自定义消息过滤器, 默认使用内置实现
48 | * @since 1.0.5
49 | */
50 | Class> filter() default MessageProcessorImp.class;
51 | }
52 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/annotation/MiraiBootApplication.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.annotation;
2 |
3 | import java.lang.annotation.*;
4 |
5 | /**
6 | * 开玩笑的 没什么用
7 | * @author diyigemt
8 | * @since 1.0.0
9 | */
10 | @Target({ElementType.TYPE})
11 | @Retention(RetentionPolicy.RUNTIME)
12 | @Documented
13 | public @interface MiraiBootApplication {
14 | /**
15 | * 初始化时控制台打印信息 将会在banner后打印
16 | */
17 | String description() default "";
18 | }
19 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/annotation/MiraiBootComponent.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.annotation;
2 |
3 | import java.lang.annotation.*;
4 |
5 | /**
6 | * 将一个类标注为miraiboot组件
7 | * @author diyigemt
8 | */
9 | @Target({ElementType.TYPE})
10 | @Retention(RetentionPolicy.RUNTIME)
11 | @Documented
12 | public @interface MiraiBootComponent {
13 | }
14 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/autoconfig/GlobalLoader.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.autoconfig;
2 |
3 | import java.io.FileFilter;
4 | import java.io.File;
5 | import java.io.IOException;
6 | import java.lang.annotation.Annotation;
7 | import java.lang.reflect.Method;
8 | import java.net.JarURLConnection;
9 | import java.net.URL;
10 | import java.net.URLDecoder;
11 | import java.nio.charset.StandardCharsets;
12 | import java.util.ArrayList;
13 | import java.util.Enumeration;
14 | import java.util.List;
15 | import java.util.jar.JarEntry;
16 | import java.util.jar.JarFile;
17 |
18 | /**
19 | * 全局加载器 用于包扫描
20 | * @author diyigemt
21 | * @since 1.0.0
22 | */
23 | public class GlobalLoader {
24 | // private static final char SYSTEM_PATH_DIV = File.separatorChar;
25 | private static final char SYSTEM_PATH_DIV = '/';
26 |
27 | public static List> getClassHasAnnotation(Class annotation, String packageName) {
28 | List> classes = getClasses(packageName);
29 | if (classes.isEmpty()) return null;
30 | List> res = new ArrayList>();
31 | for (Class aClass : classes) {
32 | Method[] methods = aClass.getMethods();
33 | if (aClass.getAnnotation(annotation) != null) {
34 | res.add(aClass);
35 | }
36 | }
37 | return res;
38 | }
39 |
40 | /**
41 | * 从包package中获取所有的Class
42 | *
43 | * @param packageName 包名
44 | * @return CLass LIst
45 | */
46 | public static List> getClasses(String packageName) {
47 |
48 | //第一个class类的集合
49 | List> classes = new ArrayList>();
50 | //是否循环迭代
51 | boolean recursive = true;
52 | //获取包的名字 并进行替换
53 | String packageDirName = packageName.replace('.', SYSTEM_PATH_DIV);
54 | //定义一个枚举的集合 并进行循环来处理这个目录下的things
55 | Enumeration dirs;
56 | try {
57 | dirs = Thread.currentThread().getContextClassLoader().getResources(packageDirName);
58 | //循环迭代下去
59 | while (dirs.hasMoreElements()) {
60 | //获取下一个元素
61 | URL url = dirs.nextElement();
62 | //得到协议的名称
63 | String protocol = url.getProtocol();
64 | //如果是以文件的形式保存在服务器上
65 | if ("file".equals(protocol)) {
66 | //获取包的物理路径
67 | String filePath = URLDecoder.decode(url.getFile(), StandardCharsets.UTF_8);
68 | //以文件的方式扫描整个包下的文件 并添加到集合中
69 | findAndAddClassesInPackageByFile(packageName, filePath, recursive, classes);
70 | } else if ("jar".equals(protocol)) {
71 | //如果是jar包文件
72 | //定义一个JarFile
73 | JarFile jar;
74 | try {
75 | //获取jar
76 | jar = ((JarURLConnection) url.openConnection()).getJarFile();
77 | //从此jar包 得到一个枚举类
78 | Enumeration entries = jar.entries();
79 | //同样的进行循环迭代
80 | while (entries.hasMoreElements()) {
81 | //获取jar里的一个实体 可以是目录 和一些jar包里的其他文件 如META-INF等文件
82 | JarEntry entry = entries.nextElement();
83 | String name = entry.getName();
84 | //如果是以/开头的
85 | if (name.charAt(0) == SYSTEM_PATH_DIV) {
86 | //获取后面的字符串
87 | name = name.substring(1);
88 | }
89 | //如果前半部分和定义的包名相同
90 | if (name.startsWith(packageDirName)) {
91 | int idx = name.lastIndexOf(SYSTEM_PATH_DIV);
92 | //如果以"/"结尾 是一个包
93 | if (idx != -1) {
94 | //获取包名 把"/"替换成"."
95 | packageName = name.substring(0, idx).replace(SYSTEM_PATH_DIV, '.');
96 | }
97 | //如果可以迭代下去 并且是一个包
98 | if ((idx != -1) || recursive) {
99 | //如果是一个.class文件 而且不是目录
100 | if (name.endsWith(".class") && !entry.isDirectory()) {
101 | //去掉后面的".class" 获取真正的类名
102 | String className = name.substring(packageName.length() + 1, name.length() - 6);
103 | try {
104 | //添加到classes
105 | String clazzName = packageName + '.' + className;
106 | classes.add(Class.forName(clazzName.replace(SYSTEM_PATH_DIV, '.')));
107 | } catch (ClassNotFoundException e) {
108 | e.printStackTrace();
109 | }
110 | }
111 | }
112 | }
113 | }
114 | } catch (IOException e) {
115 | e.printStackTrace();
116 | }
117 | }
118 | }
119 | } catch (IOException e) {
120 | e.printStackTrace();
121 | }
122 |
123 | return classes;
124 | }
125 |
126 | /**
127 | * 以文件的形式来获取包下的所有Class
128 | *
129 | * @param packageName 全包名
130 | * @param packagePath 包路径
131 | * @param recursive 是否递归
132 | * @param classes 包列表
133 | */
134 | public static void findAndAddClassesInPackageByFile(String packageName, String packagePath, final boolean recursive, List> classes) {
135 | //获取此包的目录 建立一个File
136 | File dir = new File(packagePath);
137 | //如果不存在或者 也不是目录就直接返回
138 | if (!dir.exists() || !dir.isDirectory()) {
139 | return;
140 | }
141 | //如果存在 就获取包下的所有文件 包括目录
142 | //自定义过滤规则 如果可以循环(包含子目录) 或则是以.class结尾的文件(编译好的java类文件)
143 | File[] dirFiles = dir.listFiles(file -> (recursive && file.isDirectory()) || (file.getName().endsWith(".class")));
144 | if (dirFiles == null) return;
145 | //循环所有文件
146 | for (File file : dirFiles) {
147 | //如果是目录 则继续扫描
148 | if (file.isDirectory()) {
149 | findAndAddClassesInPackageByFile(packageName + "." + file.getName(),
150 | file.getAbsolutePath(),
151 | recursive,
152 | classes);
153 | } else {
154 | //如果是java类文件 去掉后面的.class 只留下类名
155 | String className = file.getName().substring(0, file.getName().length() - 6);
156 | try {
157 | //添加到集合中去
158 | String clazzName = packageName + '.' + className;
159 | classes.add(Class.forName(clazzName.replace(SYSTEM_PATH_DIV, '.')));
160 | } catch (ClassNotFoundException e) {
161 | e.printStackTrace();
162 | }
163 | }
164 | }
165 | }
166 | }
167 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/autoconfig/JarPluginLoader.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.autoconfig;
2 |
3 | import java.io.IOException;
4 | import java.net.URL;
5 | import java.net.URLClassLoader;
6 |
7 | /**
8 | * 插件ClassLoader
9 | * 每个插件都有一个该ClassLoader实例
10 | * @author Haythem
11 | */
12 | public class JarPluginLoader extends URLClassLoader {
13 |
14 | public JarPluginLoader(URL[] urls){
15 | super(urls);
16 | }
17 |
18 | public JarPluginLoader(ClassLoader parent){
19 | super(new URL[0], parent);
20 | }
21 |
22 | @Override
23 | public void close() throws IOException {
24 | super.close();
25 | }
26 | }
27 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/autoconfig/MiraiApplication.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.autoconfig;
2 |
3 | import net.diyigemt.miraiboot.annotation.*;
4 | import net.diyigemt.miraiboot.constant.ConstantGlobal;
5 | import net.diyigemt.miraiboot.constant.EventHandlerType;
6 | import net.diyigemt.miraiboot.constant.FunctionId;
7 | import net.diyigemt.miraiboot.core.MiraiBootConsole;
8 | import net.diyigemt.miraiboot.core.PluginMgr;
9 | import net.diyigemt.miraiboot.core.RegisterProcess;
10 | import net.diyigemt.miraiboot.dao.PermissionDAO;
11 | import net.diyigemt.miraiboot.entity.*;
12 | import net.diyigemt.miraiboot.function.Alias;
13 | import net.diyigemt.miraiboot.function.console.ConsoleExit;
14 | import net.diyigemt.miraiboot.listener.BotEventListener;
15 | import net.diyigemt.miraiboot.listener.MessageEventListener;
16 | import net.diyigemt.miraiboot.mirai.MiraiMain;
17 | import net.diyigemt.miraiboot.permission.AuthMgr;
18 | import net.diyigemt.miraiboot.permission.CheckPermission;
19 | import net.diyigemt.miraiboot.utils.*;
20 | import net.mamoe.mirai.Bot;
21 | import net.mamoe.mirai.BotFactory;
22 | import net.mamoe.mirai.event.events.BotEvent;
23 | import net.mamoe.mirai.event.events.MessageEvent;
24 | import org.yaml.snakeyaml.Yaml;
25 |
26 | import java.io.*;
27 | import java.lang.reflect.InvocationTargetException;
28 | import java.lang.reflect.Method;
29 | import java.nio.charset.StandardCharsets;
30 | import java.util.*;
31 | import java.util.concurrent.atomic.AtomicBoolean;
32 |
33 | /**
34 | * 主实现逻辑
35 | * @author diyigemt
36 | * @since 1.0.0
37 | */
38 | public class MiraiApplication {
39 |
40 | public static ConfigFile config;
41 | public static void run(Class mainClass, String... args) {
42 | // 打印banner
43 | InputStream banner = mainClass.getResourceAsStream("/banner.txt");
44 | if (banner != null) {
45 | BufferedReader bannerReader = new BufferedReader(new InputStreamReader(banner));
46 | String line;
47 | try {
48 | while ((line = bannerReader.readLine()) != null) {
49 | System.out.println(line);
50 | }
51 | } catch (IOException e) {
52 | e.printStackTrace();
53 | } finally {
54 | try {
55 | banner.close();
56 | } catch (IOException e) {
57 | e.printStackTrace();
58 | }
59 | }
60 | }
61 | // AutoConfiguration
62 | // 打印描述
63 | MiraiBootApplication miraiBootApplication = mainClass.getAnnotation(MiraiBootApplication.class);
64 | if (miraiBootApplication == null) {
65 | MiraiMain.logger.error("没有找到主类! 请将主类加上@MiraiBootApplication注解");
66 | return;
67 | }
68 | MiraiMain.logger.info(miraiBootApplication.description());
69 | // 初始化FileUtil
70 | FileUtil.init(mainClass);
71 | // 尝试读取配置文件
72 | MiraiMain.logger.info("开始读取配置文件");
73 | File configFile = FileUtil.getInstance().getConfigFile();
74 | if (configFile == null) {
75 | MiraiMain.logger.warning("未找到配置文件, 请自行在新创建的配置文件中修改");
76 | configFile = FileUtil.getInstance().createConfigFile();
77 | if (configFile == null) {
78 | MiraiMain.logger.error("配置文件创建失败");
79 | }
80 | return;
81 | }
82 | try {
83 | config = new Yaml().loadAs(new InputStreamReader(new FileInputStream(configFile), StandardCharsets.UTF_8), ConfigFile.class);
84 | } catch (FileNotFoundException e) {
85 | e.printStackTrace();
86 | MiraiMain.logger.error("配置文件读取出错");
87 | return;
88 | }
89 | // 注册全局配置
90 | GlobalConfig.getInstance().putAll(config.getMiraiboot().getConfigs());
91 | GlobalConfig.getInstance().init();
92 | MiraiMain.logger.info("配置文件读取成功");
93 | // 开始自动包扫描 注册event handler
94 | String packageName = mainClass.getPackageName();
95 | List> classes = GlobalLoader.getClasses(packageName);
96 | // 添加权限管理命令和别名命令
97 | classes.add(Alias.class);
98 | classes.add(AuthMgr.class);
99 | // 初始化permission数据库
100 | classes.add(PermissionDAO.class);
101 | // 初始化自带控制台指令
102 | List> consoleCommand = GlobalLoader.getClasses(ConsoleExit.class.getPackageName());
103 | classes.addAll(consoleCommand);
104 | classes.addAll(PluginLoader.getPluginClasses(mainClass));
105 | // 开始处理事件handler和autoInit
106 | List inits = RegisterProcess.AnnotationScanner(classes);
107 | //事件注册完成，释放所有List
108 | classes.clear();
109 | // 开始读取配置文件
110 | ConfigFileMain miraiboot = config.getMiraiboot();
111 | final boolean isNetwork = miraiboot.getLogger().isNetwork();
112 | // 设置事件监听的logger是否启用
113 | MessageEventListener.eventLoggerEnable = miraiboot.getLogger().isEventStatus();
114 | BotEventListener.eventLoggerEnable = miraiboot.getLogger().isEventStatus();
115 | // 注册Bot
116 | for (ConfigFileBot configFileBot : miraiboot.getBots()) {
117 | long account = configFileBot.getAccount();
118 | if (account == 123L) continue;
119 | String value = configFileBot.getPassword().getValue();
120 | if (value.equals("pwd")) continue;
121 | Bot bot = BotFactory.INSTANCE.newBot(account, value, botConfiguration -> {
122 | ConfigFileBotConfiguration configuration = configFileBot.getConfiguration();
123 | botConfiguration.fileBasedDeviceInfo(FileUtil.getInstance().getBotDeviceFilePath(account, configuration.getDevice()));
124 | botConfiguration.setProtocol(configuration.getProtocol());
125 | if (!isNetwork) {
126 | botConfiguration.noNetworkLog();
127 | }
128 | });
129 | // 注册统一的事件监听器
130 | bot.getEventChannel().subscribeAlways(BotEvent.class, new BotEventListener());
131 | bot.getEventChannel().subscribeAlways(MessageEvent.class, new MessageEventListener());
132 | BotManager.getInstance().register(configFileBot.getAccount(), bot);
133 | }
134 | // 注册配置文件中的指令别名
135 | EventHandlerManager.getInstance().registerAlias(miraiboot.getAlias());
136 | // 开始自动初始化
137 | Collections.sort(inits);
138 | for (AutoInitItem item : inits) {
139 | int parameterCount = item.getHandler().getParameterCount();
140 | Object[] param = null;
141 | if (parameterCount != 0) {
142 | param = new Object[parameterCount];
143 | param[0] = config;
144 | }
145 | try {
146 | if (parameterCount == 0) {
147 | item.getHandler().invoke(null);
148 | } else {
149 | item.getHandler().invoke(null, param);
150 | }
151 | } catch (IllegalAccessException | InvocationTargetException e) {
152 | e.printStackTrace();
153 | }
154 | }
155 | inits.clear();//初始化完成后释放所有记载
156 | System.gc();//清理插件加载时因各种失败而变得无用的class
157 | // 初始化完成 统一登录
158 | MiraiMain.logger.info("初始化完成 开始登录bot");
159 | BotManager.getInstance().loginAll();
160 | MiraiMain.logger.info("bot登录成功 系统启动完成");
161 | // 阻塞主线程
162 | // Map> text = PluginMgr.manifests;
163 | // int i = 0;
164 | MiraiBootConsole.getInstance().listenLoop();
165 | }
166 | }
167 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/autoconfig/MiraiBootPlugin.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.autoconfig;
2 |
3 | import java.util.ArrayList;
4 | import java.util.List;
5 | import java.util.jar.JarFile;
6 |
7 | /**
8 | * 插件类
9 | * 在主类上继承该类，输出的JAR即为插件
10 | * @author Haythem
11 | */
12 | public class MiraiBootPlugin {
13 |
14 | private List PluginDependencies = new ArrayList<>();
15 |
16 | /**
17 | * UEFI模式
18 | * 仅加载主类所在的包，其余的包均不加载，大幅提升加载速度
19 | * （前提是将必须的依赖打入插件包中）
20 | */
21 | public boolean UEFIMode = true;
22 |
23 | /**
24 | * 插件被加载前执行的方法
25 | */
26 | public void onLoad(){}
27 |
28 | /**
29 | * 单独加载依赖
30 | * 在onLoad方法中使用
31 | * 注:此方法并不是UEFI启动
32 | */
33 | protected void addDependencies(List file){
34 | PluginDependencies.addAll(file);
35 | PluginLoader.LoadPluginDependencies(PluginDependencies);
36 | }
37 | }
38 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/constant/ConstantException.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.constant;
2 |
3 | /**
4 | * 存储有关异常处理的常量
5 | * @since 1.0.5
6 | * @author diyigemt
7 | */
8 | public final class ConstantException {
9 | public static final int DEFAULT_PRIORITY = 0;
10 | public static final int MAX_PARAM_COUNT = 3;
11 | public static final String OTHER_EVENT_EXCEPTION_HANDLER_NAME = "other_exception_handler";
12 | }
13 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/constant/ConstantGlobal.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.constant;
2 |
3 | /**
4 | * 全局变量默认值以及变量名
5 | * @author diyigemt
6 | * @since 1.0.0
7 | */
8 | public class ConstantGlobal {
9 | /**
10 | * 配置文件中配置指令默认开头的字段名
11 | */
12 | public static final String DEFAULT_COMMAND_START = "DEFAULT_COMMAND_START";
13 | /**
14 | * 配置文件中配置上下文监听超时时间的字段名
15 | */
16 | public static final String DEFAULT_EVENT_NET_TIMEOUT = "DEFAULT_EVENT_NET_TIMEOUT";
17 | /**
18 | * 默认上下文监听超时时间 可在配置文件中更改
19 | */
20 | public static final long DEFAULT_EVENT_NET_TIMEOUT_TIME = 5 * 60 * 1000L;
21 | }
22 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/constant/ConstantHttp.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.constant;
2 |
3 | /**
4 | * HttpUtil用常量
5 | * @author Heythem723
6 | * @since 1.0.0
7 | */
8 | public class ConstantHttp {
9 | /**
10 | * 反爬header
11 | */
12 | public static final String HEADER_USER_AGENT = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/89.0.4389.72 Safari/537.36";
13 | }
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/constant/ConstantSQL.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.constant;
2 |
3 | /**
4 | * sqlDAO结果常量
5 | * @author diyigemt
6 | * @since 1.0.0
7 | */
8 | public class ConstantSQL {
9 | public static final int ERROR = -1;
10 | public static final int SUCCESS = 0;
11 | public static final int INSERT_CREATE = 1;
12 | public static final int INSERT_UPDATE = 2;
13 | }
14 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/constant/EventHandlerType.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.constant;
2 |
3 | import net.diyigemt.miraiboot.annotation.EventHandler;
4 |
5 | /**
6 | * 消息处理Handler类型
7 | * 目前仅有群消息 好友消息和全部
8 | * 更多消息类型在做了
9 | * @author diyigemt
10 | * @since 1.0.0
11 | * @see EventHandler
12 | */
13 | public enum EventHandlerType {
14 | /**
15 | * 处理好友消息
16 | * @since 1.0.0
17 | */
18 | MESSAGE_HANDLER_FRIEND,
19 |
20 | /**
21 | * 处理临时会话
22 | */
23 | MESSAGE_HANDLER_TEMP,
24 |
25 | /**
26 | * 处理群消息
27 | * @since 1.0.0
28 | */
29 | MESSAGE_HANDLER_GROUP,
30 |
31 | /**
32 | * 处理所有消息
33 | * @since 1.0.0
34 | */
35 | MESSAGE_HANDLER_ALL,
36 |
37 | /**
38 | * 处理其他事件
39 | * @since 1.0.0
40 | */
41 | OTHER_HANDLER;
42 | }
43 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/constant/EventType.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.constant;
2 |
3 | /**
4 | * 事件类型
5 | * @author diyigemt
6 | * @since 1.0.0
7 | */
8 |
9 | public enum EventType {
10 | /**
11 | * 群消息事件
12 | * @since 1.0.0
13 | */
14 | GROUP_MESSAGE_EVENT,
15 | /**
16 | * 好友消息事件
17 | * @since 1.0.0
18 | */
19 | FRIEND_MESSAGE_EVENT,
20 |
21 | /**
22 | * 通过群发送的临时消息事件
23 | * @since 1.0.0
24 | */
25 | GROUP_TMP_MESSAGE_EVENT,
26 |
27 | /**
28 | * 其他的事件
29 | * @since 1.0.0
30 | */
31 | OTHER_EVENT;
32 | }
33 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/constant/FunctionId.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.constant;
2 |
3 | import java.util.*;
4 |
5 | /**
6 | * 管理EventHandler在权限管理中对应的id
7 | * 所有功能都需要在这注册ID号，依次顺延
8 | * 建用 管理员功能使用负数ID
9 | * 对外开放功能使用正数ID
10 | * 极值±127（Tinyint），需要扩充请去修改permission表属性，记得注释也更新一下
11 | * (不用尝试枚举了，注解里不会生效的233)
12 | * ⬆大概是指注解里不能用枚举?
13 | * @author Haythem723
14 | * @author diyigemt
15 | */
16 |
17 | public class FunctionId {
18 |
19 | //用来查询ID
20 | public static final Map map = new HashMap<>();
21 |
22 | public static void put(String key, int value) {
23 | map.put(key, value);
24 | }
25 |
26 | /**
27 | *
28 | * @param target
29 | * 提供原始命令
30 | * @return 返回ID值
31 | */
32 | public static int getMap(String target){
33 | Integer integer = map.get(target);
34 | return integer == null ? 0 : integer;
35 | }
36 |
37 | public static String getKey(int value){
38 | Set set = map.entrySet();
39 | Iterator it = set.iterator();
40 | while(it.hasNext()) {
41 | Map.Entry entry = (Map.Entry) it.next();
42 | if(entry.getValue().equals(value)) {
43 | return entry.getKey().toString();
44 | }
45 | }
46 | return null;
47 | }
48 |
49 | public static void registerAlias(String target, String alias) {
50 | Integer permissionIndex = getMap(target);
51 | if (permissionIndex == null) return;
52 | map.put(alias, permissionIndex);
53 | }
54 |
55 | // 默认permissionIndex
56 | public static final int DEFAULT_INDEX = 0;
57 |
58 | /**
59 | * 给注解用的
60 | */
61 |
62 | //对外开放功能
63 | public static final int reply = 1;
64 |
65 | //管理员功能
66 | public static final int permit = -1;
67 | }
68 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/constant/MessageFilterMatchType.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.constant;
2 |
3 | import net.diyigemt.miraiboot.annotation.MessageFilter;
4 |
5 | /**
6 | * 消息过滤类型
7 | * @author diyigemt
8 | * @since 1.0.0
9 | * @see MessageFilter
10 | */
11 | public enum MessageFilterMatchType {
12 | /**
13 | * 忽略
14 | */
15 | NULL,
16 | /**
17 | * 相同匹配
18 | */
19 | EQUALS,
20 | /**
21 | * 忽略大小写的相同匹配
22 | */
23 | EQUALS_IGNORE_CASE,
24 | /**
25 | * 包含匹配
26 | */
27 | CONTAINS,
28 | /**
29 | * 开头匹配
30 | */
31 | STARTS_WITH,
32 | /**
33 | * 结尾匹配
34 | */
35 | ENDS_WITH,
36 | /**
37 | * 正则全文匹配
38 | */
39 | REGEX_MATCHES,
40 | /**
41 | * 正则查找匹配
42 | */
43 | REGEX_FIND
44 | }
45 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/constant/MessagePreProcessorMessageType.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.constant;
2 |
3 | import net.diyigemt.miraiboot.annotation.MessageFilter;
4 |
5 | /**
6 | * 消息预处理器过滤类型
7 | * 与mirai核心的消息类型完全对应
8 | * @author diyigemt
9 | * @since 1.0.0
10 | * @see MessageFilter
11 | */
12 | public enum MessagePreProcessorMessageType {
13 | /**
14 | * 纯文本
15 | * https://github.com/mamoe/mirai/blob/dev/mirai-core-api/src/commonMain/kotlin/message/data/PlainText.kt
16 | * @see net.mamoe.mirai.message.data.PlainText
17 | */
18 | PlainText,
19 | /**
20 | * 图片
21 | * 包括群图片 好友图片
22 | * https://github.com/mamoe/mirai/blob/dev/mirai-core-api/src/commonMain/kotlin/message/data/Image.kt
23 | * @see net.mamoe.mirai.message.data.GroupImage
24 | * @see net.mamoe.mirai.message.data.FriendImage
25 | */
26 | Image,
27 | /**
28 | * 艾特(@)人的消息
29 | * https://github.com/mamoe/mirai/blob/dev/mirai-core-api/src/commonMain/kotlin/message/data/At.kt
30 | * @see net.mamoe.mirai.message.data.At
31 | */
32 | At,
33 | /**
34 | * 艾特(@)全体的消息
35 | * https://github.com/mamoe/mirai/blob/dev/mirai-core-api/src/commonMain/kotlin/message/data/AtAll.kt
36 | * @see net.mamoe.mirai.message.data.AtAll
37 | */
38 | AtAll,
39 | /**
40 | * 自带的表情栏的表情
41 | * https://github.com/mamoe/mirai/blob/dev/mirai-core-api/src/commonMain/kotlin/message/data/Face.kt
42 | * @see net.mamoe.mirai.message.data.Face
43 | */
44 | Face,
45 | /**
46 | * 闪照
47 | * https://github.com/mamoe/mirai/blob/dev/mirai-core-api/src/commonMain/kotlin/message/data/FlashImage.kt
48 | * @see net.mamoe.mirai.message.data.FlashImage
49 | */
50 | FlashImage,
51 | /**
52 | * 戳一戳
53 | * https://github.com/mamoe/mirai/blob/dev/mirai-core-api/src/commonMain/kotlin/message/data/PokeMessage.kt
54 | * @see net.mamoe.mirai.message.data.PokeMessage
55 | */
56 | PokeMessage,
57 | /**
58 | * vip才能用的表情
59 | * https://github.com/mamoe/mirai/blob/dev/mirai-core-api/src/commonMain/kotlin/message/data/VipFace.kt
60 | * @see net.mamoe.mirai.message.data.VipFace
61 | */
62 | VipFace,
63 | /**
64 | * 快应用
65 | * https://github.com/mamoe/mirai/blob/dev/mirai-core-api/src/commonMain/kotlin/message/data/RichMessage.kt
66 | * @see net.mamoe.mirai.message.data.LightApp
67 | */
68 | LightApp,
69 | /**
70 | * 语音
71 | * https://github.com/mamoe/mirai/blob/dev/mirai-core-api/src/commonMain/kotlin/message/data/Voice.kt
72 | * @see net.mamoe.mirai.message.data.Audio
73 | */
74 | Audio,
75 | /**
76 | * 表情市场中的表情
77 | * https://github.com/mamoe/mirai/blob/dev/mirai-core-api/src/commonMain/kotlin/message/data/MarketFace.kt
78 | * @see net.mamoe.mirai.message.data.MarketFace
79 | */
80 | MarketFace,
81 | /**
82 | * 合并转发的消息
83 | * https://github.com/mamoe/mirai/blob/dev/mirai-core-api/src/commonMain/kotlin/message/data/ForwardMessage.kt
84 | * @see net.mamoe.mirai.message.data.ForwardMessage
85 | */
86 | ForwardMessage,
87 | /**
88 | * 另一种消息类型 不常用
89 | * https://github.com/mamoe/mirai/blob/dev/mirai-core-api/src/commonMain/kotlin/message/data/RichMessage.kt
90 | * @see net.mamoe.mirai.message.data.SimpleServiceMessage
91 | */
92 | SimpleServiceMessage,
93 | /**
94 | * 音乐分享
95 | * https://github.com/mamoe/mirai/blob/dev/mirai-core-api/src/commonMain/kotlin/message/data/MusicShare.kt
96 | * @see net.mamoe.mirai.message.data.MusicShare
97 | */
98 | MusicShare,
99 | /**
100 | * 骰子消息
101 | * https://github.com/mamoe/mirai/blob/dev/mirai-core-api/src/commonMain/kotlin/message/data/Dice.kt
102 | * @see net.mamoe.mirai.message.data.Dice
103 | */
104 | Dice,
105 | /**
106 | * 文件消息
107 | * https://github.com/mamoe/mirai/blob/dev/mirai-core-api/src/commonMain/kotlin/message/data/FileMessage.kt
108 | * @see net.mamoe.mirai.message.data.FileMessage
109 | */
110 | FileMessage
111 | }
112 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/core/MiraiBootConsole.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.core;
2 |
3 | import net.diyigemt.miraiboot.annotation.ConsoleCommand;
4 | import net.diyigemt.miraiboot.annotation.EventHandler;
5 | import net.diyigemt.miraiboot.entity.ConsoleHandlerItem;
6 | import net.diyigemt.miraiboot.entity.PluginItem;
7 | import net.diyigemt.miraiboot.interfaces.UnloadHandler;
8 | import net.diyigemt.miraiboot.mirai.MiraiMain;
9 | import net.diyigemt.miraiboot.utils.CommandUtil;
10 | import net.diyigemt.miraiboot.utils.ExceptionHandlerManager;
11 |
12 | import java.lang.annotation.Annotation;
13 | import java.lang.reflect.InvocationTargetException;
14 | import java.lang.reflect.Method;
15 | import java.util.*;
16 | import java.util.regex.Matcher;
17 | import java.util.regex.Pattern;
18 |
19 | /**
20 | * MiraiBoot控制台
21 | *
22 | * @author diyigemt
23 | * @author Haythem
24 | */
25 | public final class MiraiBootConsole implements UnloadHandler {
26 |
27 | private static final MiraiBootConsole INSTANCE = new MiraiBootConsole();
28 |
29 | private static final Map store = new HashMap<>();
30 |
31 | private static final Pattern EMPTY_SOURCE = Pattern.compile("^\\s+$");
32 |
33 | public static MiraiBootConsole getInstance() {
34 | return INSTANCE;
35 | }
36 |
37 | /**
38 | * 注册一条控制台指令
39 | *
40 | * @param target 指令名
41 | * @param invoker 操作类
42 | * @param handler 操作方法
43 | * @return 是否注册成功, 当存在同名指令时返回false
44 | */
45 | public boolean on(String target, Class invoker, Method handler) {
46 | ConsoleHandlerItem one = store.get(target);
47 | if (one != null) return false;
48 | String name = CommandUtil.getInstance().parseHandlerBaseName(invoker);
49 | ConsoleHandlerItem item = new ConsoleHandlerItem(name, invoker, handler);
50 | store.put(target, item);
51 | return true;
52 | }
53 |
54 | /**
55 | * 根据指令执行对应的方法
56 | * @param target 指令名
57 | * @param args 指令的参数
58 | * @return 有对应的处理方法且执行成功返回true
59 | */
60 | public boolean emit(String target, String... args) {
61 | List arg = Arrays.asList(args);
62 | ConsoleHandlerItem item = store.get(target);
63 | if (item == null) return false;
64 | Method method = item.getHandler();
65 | Class invoker = item.getInvoker();
66 | int count = method.getParameterCount();
67 | Object[] param = null;
68 | if (count != 0) {
69 | param = new Object[count];
70 | param[0] = arg;
71 | }
72 | try {
73 | if (count != 0) {
74 | method.invoke(invoker.getDeclaredConstructor().newInstance(), param);
75 | } else {
76 | method.invoke(invoker.getDeclaredConstructor().newInstance());
77 | }
78 | } catch (IllegalAccessException | InvocationTargetException | InstantiationException | NoSuchMethodException e) {
79 | boolean res = ExceptionHandlerManager.getInstance().emit(e, null, null);
80 | if (!res) e.printStackTrace();
81 | return false;
82 | }
83 | return true;
84 | }
85 |
86 | private String[] parsArgs(String source) {
87 | Matcher matcher = EMPTY_SOURCE.matcher(source);
88 | if (source.equals("") || matcher.matches()) return null;
89 | return source.split("\\s+");
90 | }
91 |
92 |
93 | /**
94 | * 控制台输入监听
95 | */
96 | public void listenLoop() {
97 | Scanner scanner = new Scanner(System.in);
98 | while (true) {
99 | String source = scanner.nextLine();
100 | String[] strings = parsArgs(source);
101 | if (strings == null) continue;
102 | boolean emit = emit(strings[0], Arrays.copyOfRange(strings, 1, strings.length, String[].class));
103 | if (!emit) {
104 | MiraiMain.logger.error("命令：" + "\"" + source + "\"" + " 执行失败或不是有效的MiraiBoot命令");
105 | }
106 | }
107 | }
108 |
109 | /**
110 | * 根据指令移除一个指令
111 | * @param command 指令
112 | * @return 被移除的指令
113 | */
114 | public ConsoleHandlerItem removeByCommand(String command) {
115 | return store.remove(command);
116 | }
117 |
118 | /**
119 | * 根据名字移除一个控制台指令
120 | * @param name 指令名
121 | * @return 被移除的指令
122 | */
123 | public ConsoleHandlerItem removeByClassName(String name) {
124 | if (store.isEmpty()) return null;
125 | String t = null;
126 | for (Map.Entry next : store.entrySet()) {
127 | if (next.getValue().getName().equals(name)) {
128 | t = next.getKey();
129 | break;
130 | }
131 | }
132 | if (t == null) return null;
133 | return store.remove(t);
134 | }
135 |
136 | @Override
137 | public void onUnload(List pluginItems) {
138 | for (PluginItem item : pluginItems) {
139 | Annotation annotationData = item.getAnnotationData();
140 | if (!(annotationData instanceof ConsoleCommand)) {
141 | continue;
142 | }
143 | ConsoleCommand annotation = (ConsoleCommand) annotationData;
144 | String value = annotation.value();
145 | removeByCommand(value);
146 | }
147 | }
148 | }
149 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/core/PluginMgr.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.core;
2 |
3 | import net.diyigemt.miraiboot.autoconfig.JarPluginLoader;
4 | import net.diyigemt.miraiboot.autoconfig.PluginLoader;
5 | import net.diyigemt.miraiboot.entity.PluginItem;
6 | import net.diyigemt.miraiboot.mirai.MiraiMain;
7 | import net.diyigemt.miraiboot.utils.EventHandlerManager;
8 | import net.diyigemt.miraiboot.utils.ExceptionHandlerManager;
9 |
10 | import java.io.IOException;
11 | import java.net.JarURLConnection;
12 | import java.util.*;
13 |
14 | public class PluginMgr {
15 |
16 | private static List Plugin_Cache = new ArrayList<>();
17 |
18 | /**
19 | * 加载器集合
20 | */
21 | private static List> loaders = new ArrayList<>();
22 |
23 | private static Map> manifests = new HashMap<>();
24 |
25 | // private static List unLoadControllers = new ArrayList<>();
26 |
27 | public static void addPluginConnection(JarURLConnection connection){
28 | Plugin_Cache.add(connection);
29 | }
30 |
31 | // public static void addUnloadController(Method UnloadController){
32 | // unLoadControllers.add(UnloadController);
33 | // }
34 |
35 | public static void addPluginManifest(String jarName, List pluginItemList){
36 | manifests.put(jarName, pluginItemList);
37 | }
38 |
39 | public static void addLoader(JarPluginLoader loader, String jarName){
40 | Map map = new HashMap<>();
41 | map.put(loader, jarName);
42 | }
43 |
44 | private static void RemoveLoader(String jarName) throws IOException {
45 | for(Map map : loaders){
46 | JarPluginLoader loader = map.get(jarName);
47 | if(loader != null){
48 | loader.close();//关闭URL链接
49 | loader = null;//释放该loader
50 | loaders.remove(map);//将loader从List中移除
51 | return;
52 | }
53 | }
54 | }
55 |
56 | public static List getPluginConnectionList(){
57 | List PluginList = new ArrayList<>();
58 | try{
59 | for(JarURLConnection connection : Plugin_Cache){
60 | PluginList.add(connection.getJarFile().getName());
61 | }
62 | }catch (IOException e){
63 | e.printStackTrace();
64 | }
65 |
66 | return PluginList;
67 | }
68 |
69 | public static void unLoadPlugin(String jarFileName){
70 | if(unloadProcess(jarFileName)){
71 | MiraiMain.logger.info("插件：" + jarFileName + " 卸载成功");
72 | }
73 | }
74 |
75 | public static void reLoadPlugin(String jarFileName){
76 | String path = Objects.requireNonNull(getConnection(jarFileName)).getJarFileURL().getPath();
77 | if(unloadProcess(jarFileName)){
78 | PluginLoader.LoadPlugin(path);
79 | }
80 |
81 | }
82 |
83 | private static JarURLConnection getConnection(String jarFileName){
84 | try{
85 | for(JarURLConnection connection : Plugin_Cache){
86 | if(connection.getJarFile().getName().contains(jarFileName)){
87 | return connection;
88 | }
89 | }
90 | }catch (IOException e){
91 | e.printStackTrace();
92 | }
93 |
94 | return null;
95 | }
96 |
97 | private static boolean unloadProcess(String jarFileName){
98 | JarURLConnection connection = getConnection(jarFileName);
99 | if(connection == null){
100 | MiraiMain.logger.error("未找到插件：" + jarFileName);
101 | return false;
102 | }
103 | try{
104 | List pluginItem = manifests.get(jarFileName);
105 | connection.getJarFile().close();//关闭插件jar打开状态
106 | RemoveLoader(jarFileName);//销毁该插件的ClassLoader
107 | //TODO: 释放该插件的实例化
108 | EventHandlerManager.getInstance().onUnload(pluginItem);//注销EventHandler
109 | ExceptionHandlerManager.getInstance().onUnload(pluginItem);//注销ExceptionHandler
110 | MiraiBootConsole.getInstance().onUnload(pluginItem);//注销控制台指令
111 | // List> list = loaders;
112 | System.gc();//让JVM启动垃圾回收
113 | Plugin_Cache.remove(connection);//从当前清单中除名
114 | }catch (IOException e){
115 | MiraiMain.logger.error("插件：" + jarFileName + " 卸载失败，请关闭本程序后手动删除插件");
116 | return false;
117 | }
118 |
119 | return true;
120 | }
121 | }
122 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/dao/PermissionDAO.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.dao;
2 |
3 | import com.j256.ormlite.dao.Dao;
4 | import com.j256.ormlite.dao.Dao.CreateOrUpdateStatus;
5 | import com.j256.ormlite.stmt.DeleteBuilder;
6 | import com.j256.ormlite.stmt.PreparedQuery;
7 | import com.j256.ormlite.stmt.QueryBuilder;
8 | import com.j256.ormlite.stmt.Where;
9 | import net.diyigemt.miraiboot.annotation.AutoInit;
10 | import net.diyigemt.miraiboot.constant.ConstantSQL;
11 | import net.diyigemt.miraiboot.entity.ConfigFile;
12 | import net.diyigemt.miraiboot.entity.PermissionItem;
13 | import net.diyigemt.miraiboot.sql.DatabaseHelper;
14 |
15 | import java.sql.SQLException;
16 | import java.util.HashMap;
17 | import java.util.List;
18 | import java.util.Map;
19 |
20 | /**
21 | * sqlite数据库权限管理的DAO
22 | * 一般并不需要关心
23 | * @author diyigemt
24 | * @author Heythem723
25 | * @since 1.0.0
26 | */
27 | @AutoInit
28 | public class PermissionDAO {
29 | private static final PermissionDAO INSTANCE = new PermissionDAO();
30 | private static Dao dao;
31 |
32 | public static void init(ConfigFile config) {
33 | dao = DatabaseHelper.getInstance().getDAO("permission", PermissionItem.class, Integer.class);
34 | }
35 |
36 | public static PermissionDAO getInstance() { return INSTANCE; }
37 |
38 | public QueryBuilder getQueryBuilder() {
39 | return dao.queryBuilder();
40 | }
41 |
42 | public PermissionItem selectById(int id) {
43 | try {
44 | return dao.queryForId(id);
45 | } catch (SQLException throwables) {
46 | throwables.printStackTrace();
47 | }
48 | return null;
49 | }
50 |
51 | public List selectAll() {
52 | try {
53 | return dao.queryForAll();
54 | } catch (SQLException throwables) {
55 | throwables.printStackTrace();
56 | }
57 | return null;
58 | }
59 |
60 | public List selectByPrepared(PreparedQuery query) {
61 | try {
62 | return dao.query(query);
63 | } catch (SQLException throwables) {
64 | throwables.printStackTrace();
65 | }
66 | return null;
67 | }
68 |
69 | public List selectForFieldValuesArgs(Map args) {
70 | try {
71 | return dao.queryForFieldValuesArgs(args);
72 | } catch (SQLException throwables) {
73 | throwables.printStackTrace();
74 | }
75 | return null;
76 | }
77 |
78 | public int insert(PermissionItem item) {
79 | int res = ConstantSQL.SUCCESS;
80 | QueryBuilder builder = dao.queryBuilder();
81 | Map args = new HashMap();
82 | long senderId = item.getSenderId();
83 | int commandId = item.getCommandId();
84 | int remain = item.getRemain();
85 | int permits = item.getPermits();
86 | args.put("sender_id", senderId);
87 | args.put("command_id", commandId);
88 | List permissionItems = selectForFieldValuesArgs(args);
89 | if (permissionItems != null && !permissionItems.isEmpty()) {
90 | item = permissionItems.get(0);
91 | item.setSenderId(String.valueOf(senderId));
92 | item.setCommandId(commandId);
93 | item.setRemain(remain);
94 | item.setPermits(permits);
95 | }
96 | try {
97 | CreateOrUpdateStatus status = dao.createOrUpdate(item);
98 | if (status.isCreated()) res = ConstantSQL.INSERT_CREATE;
99 | if (status.isUpdated()) res = ConstantSQL.INSERT_UPDATE;
100 | } catch (SQLException e) {
101 | res = ConstantSQL.ERROR;
102 | e.printStackTrace();
103 | }
104 | return res;
105 | }
106 |
107 | public int update(PermissionItem item) {
108 | int res = ConstantSQL.SUCCESS;
109 | try {
110 | res = dao.update(item);
111 | } catch (SQLException e) {
112 | e.printStackTrace();
113 | res = ConstantSQL.ERROR;
114 | }
115 | return res;
116 | }
117 |
118 | public int delete(PermissionItem item) {
119 | int res = ConstantSQL.SUCCESS;
120 | DeleteBuilder builder = dao.deleteBuilder();
121 | Where where = builder.where();
122 | try {
123 | Where eq = where.eq("sender_id", item.getSenderId()).and().eq("command_id", item.getCommandId());
124 | builder.setWhere(eq);
125 | res = dao.delete(builder.prepare());
126 | } catch (SQLException throwables) {
127 | throwables.printStackTrace();
128 | res = ConstantSQL.ERROR;
129 | }
130 | return res;
131 | }
132 |
133 | public int delete(int id) {
134 | int res = ConstantSQL.SUCCESS;
135 | try {
136 | res = dao.deleteById(id);
137 | } catch (SQLException e) {
138 | res = ConstantSQL.ERROR;
139 | e.printStackTrace();
140 | }
141 | return res;
142 | }
143 | }
144 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/entity/AutoInitItem.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.entity;
2 |
3 | import lombok.AllArgsConstructor;
4 | import lombok.Data;
5 | import lombok.NoArgsConstructor;
6 | import net.diyigemt.miraiboot.annotation.AutoInit;
7 | import org.jetbrains.annotations.NotNull;
8 |
9 | import java.lang.reflect.Method;
10 |
11 | /**
12 | * 自动初始化信息存储类
13 | * @author diyigem
14 | * @see AutoInit
15 | */
16 | @Data
17 | public class AutoInitItem implements Comparable {
18 | private final String name;
19 | private final int priority;
20 | private final Method handler;
21 |
22 | @Override
23 | public int compareTo(@NotNull AutoInitItem o) {
24 | return -Integer.compare(this.priority, o.priority);
25 | }
26 | }
27 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/entity/BaseEventPack.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.entity;
2 |
3 | /**
4 | * 用于泛型约束
5 | */
6 | public class BaseEventPack {
7 |
8 | }
9 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/entity/BotEventPack.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.entity;
2 |
3 | import net.mamoe.mirai.event.events.BotEvent;
4 |
5 | /**
6 | * 对Bot事件的简单封装
7 | * @author diyigemt
8 | * @since 1.0.0
9 | */
10 | public class BotEventPack extends BaseEventPack{
11 |
12 | private BotEvent event;
13 |
14 | public BotEventPack(BotEvent event) {
15 | this.event = event;
16 | }
17 |
18 | public BotEvent getEvent() {
19 | return event;
20 | }
21 |
22 | public void setEvent(BotEvent event) {
23 | this.event = event;
24 | }
25 | }
26 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/entity/ConfigFile.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.entity;
2 |
3 | import lombok.AllArgsConstructor;
4 | import lombok.Data;
5 | import lombok.NoArgsConstructor;
6 |
7 | /**
8 | * 配置文件存储类
9 | * @since 1.0.0
10 | * @author diyiegmt
11 | */
12 | @Data
13 | @NoArgsConstructor
14 | @AllArgsConstructor
15 | public class ConfigFile {
16 | private ConfigFileMain miraiboot;
17 | }
18 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/entity/ConfigFileBot.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.entity;
2 |
3 | import lombok.AllArgsConstructor;
4 | import lombok.Data;
5 |
6 | @Data
7 | @AllArgsConstructor
8 | public class ConfigFileBot {
9 | private long account;
10 | private ConfigFileBotPassword password;
11 | private ConfigFileBotConfiguration configuration;
12 |
13 | public ConfigFileBot() {
14 | this.password = new ConfigFileBotPassword();
15 | this.configuration = new ConfigFileBotConfiguration();
16 | }
17 | }
18 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/entity/ConfigFileBotConfiguration.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.entity;
2 |
3 | import lombok.AllArgsConstructor;
4 | import lombok.Data;
5 | import net.mamoe.mirai.utils.BotConfiguration;
6 |
7 | @Data
8 | @AllArgsConstructor
9 | public class ConfigFileBotConfiguration {
10 | private BotConfiguration.MiraiProtocol protocol;
11 | private String device;
12 |
13 | public ConfigFileBotConfiguration() {
14 | this.protocol = BotConfiguration.MiraiProtocol.ANDROID_PHONE;
15 | this.device = "device.json";
16 | }
17 | }
18 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/entity/ConfigFileBotPassword.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.entity;
2 |
3 | import lombok.AllArgsConstructor;
4 | import lombok.Data;
5 |
6 | @Data
7 | @AllArgsConstructor
8 | public class ConfigFileBotPassword {
9 | private PasswordKind kind;
10 | private String value;
11 |
12 | public ConfigFileBotPassword() {
13 | this.kind = PasswordKind.PLAIN;
14 | }
15 | public enum PasswordKind {
16 | PLAIN, MD5;
17 | }
18 | }
19 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/entity/ConfigFileLogger.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.entity;
2 |
3 | import lombok.AllArgsConstructor;
4 | import lombok.Data;
5 | import lombok.NoArgsConstructor;
6 |
7 | @Data
8 | @AllArgsConstructor
9 | public class ConfigFileLogger {
10 | private boolean network;
11 | private boolean eventStatus;
12 | private boolean debug;
13 | public ConfigFileLogger() {
14 | this.network = false;
15 | this.eventStatus = true;
16 | this.debug = false;
17 | }
18 | }
19 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/entity/ConfigFileMain.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.entity;
2 |
3 | import lombok.AllArgsConstructor;
4 | import lombok.Data;
5 | import lombok.NoArgsConstructor;
6 |
7 | import java.util.List;
8 | import java.util.Map;
9 |
10 | @Data
11 | @NoArgsConstructor
12 | @AllArgsConstructor
13 | public class ConfigFileMain {
14 | List bots;
15 | ConfigFileLogger logger;
16 | Map configs;
17 | Map alias;
18 | }
19 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/entity/ConsoleHandlerItem.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.entity;
2 |
3 | import lombok.Getter;
4 |
5 | import java.lang.reflect.Method;
6 |
7 | @Getter
8 | public final class ConsoleHandlerItem extends MiraiBootHandlerItem{
9 | public ConsoleHandlerItem(String name, Class invoker, Method handler) {
10 | super(name, invoker, handler);
11 | }
12 | }
13 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/entity/EnhancedMessageChain.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.entity;
2 |
3 | import lombok.Data;
4 | import net.mamoe.mirai.message.data.MessageChain;
5 | import org.jetbrains.annotations.NotNull;
6 |
7 | import java.util.ArrayList;
8 | import java.util.Iterator;
9 | import java.util.List;
10 |
11 |
12 | /**
13 | * 加强消息链
14 | * 相比MessageChain，该消息链可以添加多个语音，群文件等特殊Message
15 | * @author Haythem
16 | * @since 1.0.0
17 | */
18 | @Data
19 | public class EnhancedMessageChain implements Iterable{
20 |
21 | private List EnhancedMsgChain = new ArrayList<>();
22 |
23 | /**
24 | * 加强消息链添加消息方法
25 | * 支持以下类型输入:
26 | *
27 | * 1: MessageChain消息链
28 | * 2: EnhancedMessageChain加强的消息链
29 | *
30 | * @param messageChain 当前类型: MessageChain 消息链
31 | */
32 | public void append(MessageChain messageChain){
33 | this.EnhancedMsgChain.add(messageChain);
34 | }
35 |
36 | /**
37 | * 加强消息链添加消息方法
38 | * 支持以下类型输入:
39 | *
40 | * 1: MessageChain消息链
41 | * 2: EnhancedMessageChain加强的消息链
42 | *
43 | * @param messageChain 当前类型: EnhancedMessageChain 特殊消息链
44 | */
45 | public void append(EnhancedMessageChain messageChain){
46 | this.EnhancedMsgChain.addAll(messageChain.getEnhancedMsgChain());
47 | }
48 |
49 | /**
50 | * 迭代器，用于支持foreach循环
51 | */
52 | @NotNull
53 | @Override
54 | public Iterator iterator() {
55 | return new MsgChainIterator();
56 | }
57 |
58 | private class MsgChainIterator implements Iterator{
59 | int current;
60 |
61 | @Override
62 | public boolean hasNext() {
63 | return current != EnhancedMsgChain.size();
64 | }
65 |
66 | @Override
67 | public MessageChain next() {
68 | return EnhancedMsgChain.get(current++);
69 | }
70 | }
71 |
72 | public int size() { return EnhancedMsgChain.size(); }
73 | }
74 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/entity/EventHandlerItem.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.entity;
2 |
3 | import lombok.AllArgsConstructor;
4 | import lombok.Data;
5 | import lombok.Getter;
6 | import net.diyigemt.miraiboot.annotation.EventHandler;
7 | import net.diyigemt.miraiboot.constant.EventHandlerType;
8 |
9 | import java.lang.reflect.Method;
10 | import java.util.List;
11 |
12 | /**
13 | * EventHandler信息存储类
14 | * @see EventHandler
15 | * @author diyigemt
16 | * @since 1.0.0
17 | */
18 | @Getter
19 | public final class EventHandlerItem extends MiraiBootHandlerItem {
20 | private final EventHandlerType[] type;
21 | /**
22 | * 同一个类中处理异常的方法
23 | */
24 | private final List exceptionHandlers;
25 |
26 | public EventHandlerItem(String name, Class invoker, Method handler, EventHandlerType[] type, List list) {
27 | super(name, invoker, handler);
28 | this.type = type;
29 | this.exceptionHandlers = list;
30 | }
31 | }
32 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/entity/EventHandlerNextItem.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.entity;
2 |
3 | import lombok.Data;
4 | import net.diyigemt.miraiboot.constant.ConstantGlobal;
5 | import net.diyigemt.miraiboot.interfaces.EventHandlerNext;
6 | import net.mamoe.mirai.event.ListeningStatus;
7 |
8 | import java.util.Timer;
9 | import java.util.TimerTask;
10 |
11 | /**
12 | * 上下文监听信息存储类
13 | * 在计时中如果被触发 将会重新开始计时
14 | * @author diyigemt
15 | * @since 1.0.0
16 | */
17 | @Data
18 | public class EventHandlerNextItem> {
19 | /**
20 | * 超时时间
21 | * @see ConstantGlobal -> DEFAULT_EVENT_NET_TIMEOUT_TIME
22 | */
23 | private long timeOut;
24 | /**
25 | * 最高触发次数
26 | */
27 | private int triggerCount;
28 | /**
29 | * 存储信息的handler类
30 | * @see EventHandlerNext
31 | */
32 | private K handler;
33 | /**
34 | * timer用
35 | */
36 | private TimerTask task;
37 | /**
38 | * 计时的timer
39 | */
40 | private Timer timer;
41 | /**
42 | * 最后一次被触发时的event
43 | * 给超时函数 次数耗尽函数 和销毁函数用
44 | */
45 | private MessageEventPack lastEventPack;
46 | /**
47 | * 最后一次触发时的data内容
48 | * 给超时函数 次数耗尽函数 和销毁函数用
49 | */
50 | private PreProcessorData lastData;
51 |
52 | public EventHandlerNextItem() {
53 | this.timeOut = -1;
54 | this.triggerCount = -1;
55 | this.handler = null;
56 | this.task = null;
57 | }
58 |
59 | public EventHandlerNextItem(K handler) {
60 | this.handler = handler;
61 | }
62 |
63 | public EventHandlerNextItem(K handler, long timeOut, int triggerCount) {
64 | this(handler);
65 | this.timeOut = timeOut;
66 | this.triggerCount = triggerCount;
67 | if (timeOut != -1 || triggerCount != -1) this.timer = new Timer();
68 | }
69 |
70 | public ListeningStatus onNext(MessageEventPack eventPack, PreProcessorData data) {
71 | PreProcessorData cast = (PreProcessorData) data;
72 | this.lastEventPack = eventPack;
73 | this.lastData = cast;
74 | if (this.triggerCount != -1) data.setTriggerCount(this.triggerCount);
75 | return onNextSelf(eventPack, cast);
76 | }
77 |
78 | public ListeningStatus onNextSelf(MessageEventPack eventPack, PreProcessorData data) {
79 | try {
80 | return handler.onNext(eventPack, data);
81 | } catch (Throwable e) {
82 | return handlerException(e);
83 | }
84 | }
85 |
86 | public void onTimeOut() {
87 | try {
88 | handler.onTimeOut(this.lastEventPack, this.lastData);
89 | } catch (Throwable e) {
90 | handlerException(e);
91 | }
92 | }
93 |
94 | public void onTriggerOut() {
95 | try {
96 | handler.onTriggerOut(this.lastEventPack, this.lastData);
97 | } catch (Throwable e) {
98 | handlerException(e);
99 | }
100 | }
101 |
102 | public void onDestroy() {
103 | try {
104 | handler.onDestroy(this.lastEventPack, this.lastData);
105 | } catch (Throwable e) {
106 | handlerException(e);
107 | }
108 | }
109 |
110 | public boolean check() {
111 | if (this.triggerCount == -1 && this.timeOut == -1) return true;
112 | if (this.triggerCount != -1) this.triggerCount--;
113 | if (this.triggerCount == 0) {
114 | return false;
115 | }
116 | if (this.timeOut != -1) {
117 | this.timer.cancel();
118 | this.timer = new Timer();
119 | }
120 | return true;
121 | }
122 |
123 | public void start(TimerTask task) {
124 | if (this.timeOut != -1) {
125 | this.task = task;
126 | this.timer.schedule(task, this.timeOut);
127 | }
128 | }
129 |
130 | public void cancel() {
131 | this.timer.cancel();
132 | }
133 |
134 | private ListeningStatus handlerException(Throwable e) {
135 | return handler.onException(e, lastEventPack, lastData);
136 | }
137 | }
138 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/entity/ExceptionHandlerItem.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.entity;
2 |
3 | import lombok.AllArgsConstructor;
4 | import lombok.Data;
5 | import lombok.Getter;
6 | import org.jetbrains.annotations.NotNull;
7 |
8 | import java.lang.reflect.Method;
9 |
10 | /**
11 | * 异常处理信息存储类
12 | * 优先级越高越先触发!!
13 | * @author diyigemt
14 | * @since 1.0.0
15 | */
16 | @Getter
17 | public final class ExceptionHandlerItem extends MiraiBootHandlerItem implements Comparable {
18 | private final Class target;
19 | private final int priority;
20 |
21 | public ExceptionHandlerItem(String name, Class invoker, Method handler, Class target, int priority) {
22 | super(name, invoker, handler);
23 | this.target = target;
24 | this.priority = priority;
25 | }
26 |
27 | @Override
28 | public int compareTo(@NotNull ExceptionHandlerItem o) {
29 | return -Integer.compare(priority, o.priority);
30 | }
31 |
32 | public boolean check(Class t) {
33 | return target.isAssignableFrom(t);
34 | }
35 | }
36 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/entity/HttpProperties.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.entity;
2 |
3 | import lombok.Getter;
4 | import lombok.Setter;
5 |
6 | import java.util.LinkedHashMap;
7 | import java.util.Map;
8 |
9 | /**
10 | * 自定义HTTP属性
11 | * 实例化后在Builder中作为参数传入
12 | * 构造函数已为您添加：
13 | * 最大请求时间：3000
14 | * 请求模式：GET
15 | * 引擎伪装
16 | * 保持连接
17 | *
18 | * 如有需求，重写对应属性即可
19 | * @author Haythem
20 | * @since 1.0.0
21 | */
22 |
23 | public class HttpProperties {
24 |
25 | @Getter
26 | private Map RequestProperties = new LinkedHashMap<>();
27 |
28 | @Getter
29 | @Setter
30 | private int Timeout = 3000;
31 |
32 | @Setter
33 | @Getter
34 | private String RequestMethod = "GET";
35 |
36 | /**
37 | * 自定义HTTP属性
38 | * 实例化后在Builder中作为参数传入
39 | * 构造函数已为您添加保持连接和引擎伪装,如有需求重写对应参数即可覆盖
40 | */
41 | public HttpProperties(){
42 | RequestProperties.put("Connection", "keep-alive");
43 | RequestProperties.put("User-agent","Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/89.0.4389.72 Safari/537.36");
44 | }
45 |
46 | /**
47 | * 获取RequestProperties长度
48 | */
49 | public int getRequestPropertiesSize(){
50 | return RequestProperties.size();
51 | }
52 |
53 | /**
54 | * 添加HTTP属性
55 | * 提供属性键值对
56 | * 构造函数已为您添加保持连接和引擎伪装，如有需求重写对应参数即可覆盖
57 | * @param key 键
58 | * @param value 值
59 | */
60 | public void setRequestProperties(String key, String value){
61 | if(RequestProperties.containsKey(key)){
62 | RequestProperties.replace(key, value);
63 | }else {
64 | RequestProperties.put(key, value);
65 | }
66 | }
67 | }
68 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/entity/MessageFilterImp.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.entity;
2 |
3 | import net.diyigemt.miraiboot.interfaces.IMessageFilter;
4 |
5 | public class MessageFilterImp implements IMessageFilter {
6 | @Override
7 | public boolean check(String source, MessageEventPack eventPack, MessageFilterItem item) {
8 | return item.check(eventPack, source);
9 | }
10 | }
11 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/entity/MessageFilterItem.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.entity;
2 |
3 | import lombok.Data;
4 | import net.diyigemt.miraiboot.annotation.MessageFilter;
5 | import net.diyigemt.miraiboot.constant.EventType;
6 | import net.diyigemt.miraiboot.constant.MessageFilterMatchType;
7 | import net.mamoe.mirai.message.data.At;
8 | import net.mamoe.mirai.message.data.AtAll;
9 | import net.mamoe.mirai.message.data.SingleMessage;
10 |
11 | import java.util.Arrays;
12 | import java.util.HashSet;
13 | import java.util.Set;
14 | import java.util.regex.Pattern;
15 |
16 | /**
17 | * 用于保存MessageFilter的参数
18 | * @see MessageFilter
19 | * @author diyiegmt
20 | * @since 1.0.0
21 | */
22 | @Data
23 | public class MessageFilterItem {
24 | private String value;
25 | private MessageFilterMatchType matchType;
26 | private Set accounts;
27 | private Set groups;
28 | private Set bots;
29 | private boolean isAt;
30 | private boolean isAtAll;
31 | private boolean isAtAny;
32 |
33 | public MessageFilterItem() {
34 | this.accounts = new HashSet();
35 | this.groups = new HashSet();
36 | this.bots = new HashSet();
37 | }
38 |
39 | public MessageFilterItem(MessageFilter filter) {
40 | this();
41 | value = filter.value();
42 | matchType = filter.matchType();
43 | addAccounts(filter.accounts());
44 | addGroups(filter.groups());
45 | addBots(filter.bots());
46 | isAt = filter.isAt();
47 | isAtAll = filter.isAtAll();
48 | isAtAny = filter.isAtAny();
49 | }
50 |
51 | public void addAccounts(String[] accounts) {
52 | this.accounts.addAll(Arrays.asList(accounts));
53 | }
54 |
55 | public void addGroups(String[] groups) {
56 | this.groups.addAll(Arrays.asList(groups));
57 | }
58 |
59 | public void addBots(String[] bots) {
60 | this.bots.addAll(Arrays.asList(bots));
61 | }
62 |
63 | public boolean check(MessageEventPack eventPack, String source) {
64 | if (!bots.isEmpty() && !bots.contains(String.valueOf(eventPack.getBotId()))) return false;
65 | boolean res = false;
66 | switch (matchType) {
67 | case NULL:
68 | res = true;
69 | break;
70 | case EQUALS:
71 | res = source.equals(value);
72 | break;
73 | case EQUALS_IGNORE_CASE:
74 | res = source.equalsIgnoreCase(value);
75 | break;
76 | case CONTAINS:
77 | res = source.contains(value);
78 | break;
79 | case STARTS_WITH:
80 | res = source.startsWith(value);
81 | break;
82 | case ENDS_WITH:
83 | res = source.endsWith(value);
84 | break;
85 | case REGEX_MATCHES:
86 | Pattern p1 = Pattern.compile(value);
87 | res = p1.matcher(source).matches();
88 | break;
89 | case REGEX_FIND:
90 | Pattern p2 = Pattern.compile(value);
91 | res = p2.matcher(source).find();
92 | break;
93 | }
94 | if (!res) return false;
95 | String id = String.valueOf(eventPack.getSenderId());
96 | if (!accounts.isEmpty() && !accounts.contains(id)) return false;
97 | if (eventPack.getEventType() == EventType.GROUP_MESSAGE_EVENT) return checkGroup(eventPack);
98 | return true;
99 | }
100 |
101 | public boolean checkGroup(MessageEventPack pack) {
102 | String group = String.valueOf(pack.getSubject().getId());
103 | if (!groups.isEmpty() && !groups.contains(group)) return false;
104 | boolean at = false;
105 | boolean atAll = false;
106 | boolean atAny = false;
107 | for (SingleMessage message : pack.getMessage()) {
108 | if (message instanceof At) {
109 | atAny = true;
110 | if (at) continue;
111 | at = ((At) message).getTarget() == pack.getBotId();
112 | }
113 | if (message instanceof AtAll) {
114 | if (atAll) continue;
115 | atAll = true;
116 | }
117 | }
118 | if (isAt && !at) return false;
119 | if (isAtAll && !atAll) return false;
120 | if (isAtAny && !atAny) return false;
121 | return true;
122 | }
123 | }
124 |
--------------------------------------------------------------------------------
/src/main/java/net/diyigemt/miraiboot/entity/MessageProcessorImp.java:
--------------------------------------------------------------------------------
1 | package net.diyigemt.miraiboot.entity;
2 |
3 | import lombok.Data;
4 | import net.diyigemt.miraiboot.annotation.MessagePreProcessor;
5 | import net.diyigemt.miraiboot.constant.MessagePreProcessorMessageType;
6 | import net.diyigemt.miraiboot.interfaces.IMessagePreProcessor;
7 | import net.mamoe.mirai.message.data.*;
8 |
9 | import java.util.*;
10 |
11 | /**
12 | * 默认的消息预处理器
13 | * @see MessagePreProcessor
14 | * @author diyiegmt
15 | * @since 1.0.0
16 | */
17 | @Data
18 | public class MessageProcessorImp implements IMessagePreProcessor