SenMeshNet/electron/bridges/inspection/adapterRegistry.cjs

"use strict";

/**
 * 采集器注册表
 *
 * 自动注册 adapters 目录下的所有采集器，供调度器使用。
 *
 * 使用方式：
 *   const adapterRegistry = require("./adapterRegistry.cjs");
 *   const adapter = adapterRegistry.createAdapter("LinuxServerAdapter", connInfo, options);
 */

const fs = require("fs");
const path = require("path");
const BaseAdapter = require("./adapters/baseAdapter.cjs");

const ADAPTERS_DIR = path.join(__dirname, "adapters");
const registry = {};

/**
 * 扫描 adapters 目录，自动注册所有采集器类
 * （只注册已实现的文件，不存在的适配器不报错，留作后续扩展）
 */
function loadAdapters() {
  if (!fs.existsSync(ADAPTERS_DIR)) return;

  const files = fs.readdirSync(ADAPTERS_DIR);

  for (const file of files) {
    if (!file.endsWith(".cjs")) continue;
    if (file === "baseAdapter.cjs") continue;

    try {
      const modulePath = path.join(ADAPTERS_DIR, file);
      const AdapterClass = require(modulePath);

      // 验证是否继承自 BaseAdapter
      if (
        typeof AdapterClass === "function" &&
        AdapterClass.prototype instanceof BaseAdapter
      ) {
        console.log(`[Inspection] 注册采集器: ${AdapterClass.name} (${file})`);
        registry[AdapterClass.name] = AdapterClass;
      } else {
        console.warn(`[Inspection] ${file} 不是 BaseAdapter 子类，导出类型: ${typeof AdapterClass}`);
      }
    } catch (e) {
      // 加载失败跳过（比如引用了不存在的模块）
      console.error(`[Inspection] 加载采集器失败: ${file}`, e.message, e.stack);
    }
  }
}

// 启动时自动加载
loadAdapters();

/**
 * 获取已注册的采集器列表
 * @returns {string[]} 采集器类名列表
 */
function getRegisteredAdapters() {
  return Object.keys(registry);
}

/**
 * 检查采集器是否已注册
 * @param {string} className
 * @returns {boolean}
 */
function hasAdapter(className) {
  return !!registry[className];
}

/**
 * 创建采集器实例
 * @param {string} className - 采集器类名（如 "LinuxServerAdapter"）
 * @param {object} connectionInfo - SSH 连接信息
 * @param {object} [options] - 配置项
 * @returns {BaseAdapter|null}
 */
function createAdapter(className, connectionInfo, options = {}) {
  const AdapterClass = registry[className];
  if (!AdapterClass) {
    console.warn(`[Inspection] 未找到采集器: ${className}`);
    return null;
  }
  return new AdapterClass(connectionInfo, options);
}

module.exports = {
  loadAdapters,
  getRegisteredAdapters,
  hasAdapter,
  createAdapter,
};