NapCatQQ/src/common/utils/ConfigBase.ts

import path from 'node:path';
import fs from 'node:fs';
import { log, logDebug, logError } from '@/common/utils/log';
import { dirname } from 'node:path';
import { fileURLToPath } from 'node:url';
import { selfInfo } from '@/core/data';


const __filename = fileURLToPath(import.meta.url);
const __dirname = dirname(__filename);

const configDir = path.resolve(__dirname, 'config');
fs.mkdirSync(configDir, { recursive: true });


export class ConfigBase<T> {
  public name: string = 'default_config';
  private pathName: string | null = null; // 本次读取的文件路径
  constructor() {
  }

  protected getKeys(): string[] | null {
    // 决定 key 在json配置文件中的顺序
    return null;
  }

  getConfigDir() {
    const configDir = path.resolve(__dirname, '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(selfInfo.uin, false)) return this;
    // 尝试加载默认配置
    return this.read_from_file('', true);
  }
  read_from_file(pathName: string, createIfNotExist: boolean) {
    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));
        log(`配置文件${configPath}已创建\n如果修改此文件后需要重启 NapCat 生效`);
      }
      catch (e: any) {
        logError(`创建配置文件 ${configPath} 时发生错误:`, e.message);
      }
      return this;
    }

    try {
      const data = JSON.parse(fs.readFileSync(configPath, 'utf-8'));
      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) {
        logError(`配置文件 ${configPath} 格式错误，请检查配置文件:`, e.message);
      } else {
        logError(`读取配置文件 ${configPath} 时发生错误:`, e.message);
      }
      return this;
    }
  }

  save(config: T, overwrite: boolean = false) {
    Object.assign(this, config);
    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) {
      logError(`保存配置文件 ${configPath} 时发生错误:`, e.message);
    }
  }
}