NapCatQQ source snapshot

packages/napcat-core/helper/config-base.ts

@@ -0,0 +1,84 @@
+import path from 'node:path';
+import fs from 'node:fs';
+import type { NapCatCore } from '@/napcat-core';
+import json5 from 'json5';
+import { TypeCompiler, TypeCheck } from '@sinclair/typebox/compiler';
+import { Value } from '@sinclair/typebox/value';
+import { TSchema } from '@sinclair/typebox';
+
+export abstract class ConfigBase<T> {
+  name: string;
+  core: NapCatCore;
+  configPath: string;
+  configData: T = {} as T;
+  configSchema: TSchema;
+  validate: TypeCheck<TSchema>;
+
+  protected constructor (name: string, core: NapCatCore, configPath: string, ConfigSchema: TSchema) {
+    this.name = name;
+    this.core = core;
+    this.configPath = configPath;
+    this.configSchema = ConfigSchema;
+    this.validate = TypeCompiler.Compile(ConfigSchema);
+    fs.mkdirSync(this.configPath, { recursive: true });
+    this.read();
+  }
+
+  getConfigPath (pathName?: string): string {
+    const filename = pathName ? `${this.name}_${pathName}.json` : `${this.name}.json`;
+    return path.join(this.configPath, filename);
+  }
+
+  read (): T {
+    const configPath = this.getConfigPath(this.core.selfInfo.uin);
+    const defaultConfigPath = this.getConfigPath();
+    if (!fs.existsSync(configPath)) {
+      if (fs.existsSync(defaultConfigPath)) {
+        this.configData = this.loadConfig(defaultConfigPath);
+      }
+      this.save();
+      return this.configData;
+    }
+    return this.loadConfig(configPath);
+  }
+
+  private loadConfig (configPath: string): T {
+    try {
+      const newConfigData = json5.parse(fs.readFileSync(configPath, 'utf-8'));
+      let data: unknown = newConfigData;
+      data = Value.Parse(this.configSchema, data);
+      if (!this.validate.Check(data)) {
+        throw new Error([...this.validate.Errors(data)].map(e => e.message).join(', '));
+      }
+      this.configData = data as T;
+      this.core.context.logger.logDebug(`[Core] [Config] 配置文件${configPath}加载`, this.configData);
+      return this.configData;
+    } catch (e: unknown) {
+      this.handleError(e, '读取配置文件时发生错误');
+      return {} as T;
+    }
+  }
+
+  save (newConfigData: T = this.configData): void {
+    const configPath = this.getConfigPath(this.core.selfInfo.uin);
+    let data: unknown = newConfigData;
+    data = Value.Parse(this.configSchema, data);
+    if (!this.validate.Check(data)) {
+      throw new Error([...this.validate.Errors(data)].map(e => e.message).join(', '));
+    }
+    this.configData = data as T;
+    try {
+      fs.writeFileSync(configPath, JSON.stringify(this.configData, null, 2));
+    } catch (e: unknown) {
+      this.handleError(e, `保存配置文件 ${configPath} 时发生错误:`);
+    }
+  }
+
+  private handleError (e: unknown, message: string): void {
+    if (e instanceof SyntaxError) {
+      this.core.context.logger.logError('[Core] [Config] 操作配置文件格式错误，请检查配置文件:', e.message);
+    } else {
+      this.core.context.logger.logError(`[Core] [Config] ${message}:`, (e as Error).message);
+    }
+  }
+}