NapCatQQ

src/onebot11/config.ts

@@ -0,0 +1,80 @@
+import fs from 'node:fs';
+import path from 'node:path';
+import { selfInfo } from '@/common/data';
+
+export interface OB11Config {
+  httpPort: number;
+  httpPostUrls: string[];
+  httpSecret: string;
+  wsPort: number;
+  wsReverseUrls: string[];
+  enableHttp: boolean;
+  enableHttpPost: boolean;
+  enableWs: boolean;
+  enableWsReverse: boolean;
+  messagePostFormat: 'array' | 'string';
+  reportSelfMessage: boolean;
+  enableLocalFile2Url: boolean;
+  debug: boolean;
+  heartInterval: number;
+  token: string;
+
+  read(): OB11Config;
+
+  save(config: OB11Config): void;
+}
+
+const ob11ConfigDir = path.resolve(__dirname, 'config');
+fs.mkdirSync(ob11ConfigDir, { recursive: true });
+
+class Config implements OB11Config {
+  httpPort: number = 3000;
+  httpPostUrls: string[] = [];
+  httpSecret = '';
+  wsPort = 3001;
+  wsReverseUrls: string[] = [];
+  enableHttp = false;
+  enableHttpPost = false;
+  enableWs = false;
+  enableWsReverse = false;
+  messagePostFormat: 'array' | 'string' = 'array';
+  reportSelfMessage = false;
+  debug = false;
+  enableLocalFile2Url = true;
+  heartInterval = 30000;
+  token = '';
+
+  constructor() {
+  }
+
+  getConfigPath() {
+    return path.join(ob11ConfigDir, `onebot11_${selfInfo.uin}.json`);
+  }
+
+  read() {
+    const ob11ConfigPath = this.getConfigPath();
+    if (!fs.existsSync(ob11ConfigPath)) {
+      console.log(`onebot11配置文件 ${ob11ConfigPath} 不存在, 现已创建，请修改配置文件后重启NapCat`);
+      this.save();
+      return this;
+    }
+    const data = fs.readFileSync(ob11ConfigPath, 'utf-8');
+    try {
+      const jsonData = JSON.parse(data);
+      console.log('get config', jsonData);
+      Object.assign(this, jsonData);
+      // eslint-disable-next-line
+    } catch (e) {
+    }
+    return this;
+  }
+
+  save(newConfig: OB11Config | null = null) {
+    if (newConfig) {
+      Object.assign(this, newConfig);
+    }
+    fs.writeFileSync(this.getConfigPath(), JSON.stringify(this, null, 4));
+  }
+}
+
+export const ob11Config = new Config();