chore config

This commit is contained in:
手瓜一十雪 2024-08-09 18:44:14 +08:00
parent b50556802c
commit 06d2884a88
3 changed files with 163 additions and 5 deletions

View File

@ -0,0 +1,84 @@
import path from 'node:path';
import fs from 'node:fs';
import { NapCatCore } from '@/core';
export class ConfigBase<T> {
public name: string = 'default_config';
private pathName: string | null = null; // 本次读取的文件路径
coreContext: NapCatCore;
configPath:string;
constructor(coreContext: NapCatCore,configPath:string) {
this.coreContext = coreContext;
this.configPath = configPath;
}
protected getKeys(): string[] | null {
// 决定 key 在json配置文件中的顺序
return null;
}
getConfigDir() {
const configDir = path.resolve(this.configPath, 'config');
fs.mkdirSync(configDir, { recursive: true });
return configDir;
}
getConfigPath(pathName: string | null): string {
const suffix = pathName ? `_${pathName}` : '';
const filename = `${this.name}${suffix}.json`;
return path.join(this.getConfigDir(), filename);
}
read() {
// 尝试加载当前账号配置
if (this.read_from_file(this.coreContext.selfInfo.uin, false)) return this;
// 尝试加载默认配置
return this.read_from_file('', true);
}
read_from_file(pathName: string, createIfNotExist: boolean) {
const logger = this.coreContext.context.logger;
const configPath = this.getConfigPath(pathName);
if (!fs.existsSync(configPath)) {
if (!createIfNotExist) return null;
this.pathName = pathName; // 记录有效的设置文件
try {
fs.writeFileSync(configPath, JSON.stringify(this, this.getKeys(), 2));
logger.log(`配置文件${configPath}已创建\n如果修改此文件后需要重启 NapCat 生效`);
}
catch (e: any) {
logger.logError(`创建配置文件 ${configPath} 时发生错误:`, e.message);
}
return this;
}
try {
const data = JSON.parse(fs.readFileSync(configPath, 'utf-8'));
logger.logDebug(`配置文件${configPath}已加载`, data);
Object.assign(this, data);
// eslint-disable-next-line @typescript-eslint/ban-ts-comment
// @ts-expect-error
this.save(this); // 保存一次,让新版本的字段写入
return this;
} catch (e: any) {
if (e instanceof SyntaxError) {
logger.logError(`配置文件 ${configPath} 格式错误,请检查配置文件:`, e.message);
} else {
logger.logError(`读取配置文件 ${configPath} 时发生错误:`, e.message);
}
return this;
}
}
save(config: T, overwrite: boolean = false) {
Object.assign(this, config);
const logger = this.coreContext.context.logger;
const selfInfo = this.coreContext.selfInfo;
if (overwrite) {
// 用户要求强制写入,则变更当前文件为目标文件
this.pathName = `${selfInfo.uin}`;
}
const configPath = this.getConfigPath(this.pathName);
try {
fs.writeFileSync(configPath, JSON.stringify(this, this.getKeys(), 2));
} catch (e: any) {
logger.logError(`保存配置文件 ${configPath} 时发生错误:`, e.message);
}
}
}

View File

@ -0,0 +1,76 @@
import fs from 'node:fs';
import path from 'node:path';
import { ConfigBase } from '@/common/utils/ConfigBase';
// export interface OB11Config {
// http: {
// enable: boolean;
// host: string;
// port: number;
// secret: string;
// enableHeart: boolean;
// enablePost: boolean;
// postUrls: string[];
// };
// ws: {
// enable: boolean;
// host: string;
// port: number;
// };
// reverseWs: {
// enable: boolean;
// urls: string[];
// };
// debug: boolean;
// heartInterval: number;
// messagePostFormat: 'array' | 'string';
// enableLocalFile2Url: boolean;
// musicSignUrl: string;
// reportSelfMessage: boolean;
// token: string;
// GroupLocalTime: {
// Record: boolean,
// RecordList: Array<string>
// },
// read(): OB11Config;
// save(config: OB11Config): void;
// }
export class OB11Config extends ConfigBase<OB11Config> {
name: string = 'onebot11';
http = {
enable: false,
host: '',
port: 3000,
secret: '',
enableHeart: false,
enablePost: false,
postUrls: [],
};
ws = {
enable: false,
host: '',
port: 3001,
};
reverseWs = {
enable: false,
urls: [],
};
debug = false;
heartInterval = 30000;
messagePostFormat: 'array' | 'string' = 'array';
enableLocalFile2Url = true;
musicSignUrl = '';
reportSelfMessage = false;
token = '';
GroupLocalTime = {
Record: false,
RecordList: [] as Array<string>
};
protected getKeys(): string[] | null {
return null;
}
}

View File

@ -68,8 +68,6 @@ export function postWsEvent(event: QuickActionEvent) {
} }
export function postOB11Event(msg: QuickActionEvent, reportSelf = false, postWs = true, coreContext: NapCatCore) { export function postOB11Event(msg: QuickActionEvent, reportSelf = false, postWs = true, coreContext: NapCatCore) {
const config = ob11Config;
// 判断msg是否是event // 判断msg是否是event
if (!config.reportSelfMessage && !reportSelf) { if (!config.reportSelfMessage && !reportSelf) {
if (msg.post_type === 'message' && (msg as OB11Message).user_id.toString() == coreContext.selfInfo.uin) { if (msg.post_type === 'message' && (msg as OB11Message).user_id.toString() == coreContext.selfInfo.uin) {
@ -105,7 +103,7 @@ export function postOB11Event(msg: QuickActionEvent, reportSelf = false, postWs
return; return;
} }
try { try {
handleQuickOperation(msg as QuickActionEvent, resJson,coreContext).then().catch(logError); handleQuickOperation(msg as QuickActionEvent, resJson, coreContext).then().catch(coreContext.context.logger.logError);
} catch (e: any) { } catch (e: any) {
coreContext.context.logger.logError('新消息事件HTTP上报返回快速操作失败', e); coreContext.context.logger.logError('新消息事件HTTP上报返回快速操作失败', e);
} }
@ -157,8 +155,8 @@ async function handleMsg(msg: OB11Message, quickAction: QuickAction, coreContext
} }
} }
replyMessage = replyMessage.concat(normalize(reply, quickAction.auto_escape)); replyMessage = replyMessage.concat(normalize(reply, quickAction.auto_escape));
const { sendElements, deleteAfterSentFiles } = await createSendElements(replyMessage, peer); const { sendElements, deleteAfterSentFiles } = await createSendElements(coreContext, replyMessage, peer);
sendMsg(peer, sendElements, deleteAfterSentFiles, false).then().catch(logError); sendMsg(coreContext, peer, sendElements, deleteAfterSentFiles, false).then().catch(coreContext.context.logger.logError);
} }
} }
async function handleGroupRequest(request: OB11GroupRequestEvent, quickAction: QuickActionGroupRequest, coreContext: NapCatCore) { async function handleGroupRequest(request: OB11GroupRequestEvent, quickAction: QuickActionGroupRequest, coreContext: NapCatCore) {