api-amr/device_manager.ts

// device_manager.ts - 动态设备管理工具

interface DeviceConfig {
  manufacturer: string;
  serialNumber: string;
}

export class DeviceManager {
  private devicesFilePath: string;
  private devices: DeviceConfig[] = [];

  constructor(devicesFilePath = "./devices.json") {
    this.devicesFilePath = devicesFilePath;
  }

  /**
   * 加载现有设备列表
   */
  async loadDevices(): Promise<DeviceConfig[]> {
    try {
      const text = await Deno.readTextFile(this.devicesFilePath);
      this.devices = JSON.parse(text);
      console.log(`📱 已加载 ${this.devices.length} 个设备`);
      return this.devices;
    } catch (error) {
      if (error instanceof Deno.errors.NotFound) {
        console.log("📄 devices.json 不存在，创建新文件");
        this.devices = [];
        await this.saveDevices();
        return this.devices;
      }
      throw error;
    }
  }

  /**
   * 保存设备列表到文件
   */
  async saveDevices(): Promise<void> {
    const json = JSON.stringify(this.devices, null, 2);
    await Deno.writeTextFile(this.devicesFilePath, json);
    console.log(`💾 已保存 ${this.devices.length} 个设备到 ${this.devicesFilePath}`);
  }

  /**
   * 添加新设备
   */
  async addDevice(manufacturer: string, serialNumber: string): Promise<boolean> {
    const deviceKey = `${manufacturer}-${serialNumber}`;
    const exists = this.devices.some(d => 
      d.manufacturer === manufacturer && d.serialNumber === serialNumber
    );

    if (exists) {
      console.log(`⚠️ 设备 ${deviceKey} 已存在`);
      return false;
    }

    this.devices.push({ manufacturer, serialNumber });
    await this.saveDevices();
    console.log(`➕ 已添加设备: ${deviceKey}`);
    return true;
  }

  /**
   * 批量添加设备
   */
  async addDevices(newDevices: DeviceConfig[]): Promise<number> {
    let addedCount = 0;
    
    for (const device of newDevices) {
      const exists = this.devices.some(d => 
        d.manufacturer === device.manufacturer && d.serialNumber === device.serialNumber
      );
      
      if (!exists) {
        this.devices.push(device);
        addedCount++;
        console.log(`➕ 已添加设备: ${device.manufacturer}-${device.serialNumber}`);
      } else {
        console.log(`⚠️ 设备 ${device.manufacturer}-${device.serialNumber} 已存在，跳过`);
      }
    }

    if (addedCount > 0) {
      await this.saveDevices();
    }
    
    console.log(`✅ 批量添加完成: 新增 ${addedCount} 个设备`);
    return addedCount;
  }

  /**
   * 删除设备
   */
  async removeDevice(manufacturer: string, serialNumber: string): Promise<boolean> {
    const deviceKey = `${manufacturer}-${serialNumber}`;
    const initialLength = this.devices.length;
    
    this.devices = this.devices.filter(d => 
      !(d.manufacturer === manufacturer && d.serialNumber === serialNumber)
    );

    if (this.devices.length < initialLength) {
      await this.saveDevices();
      console.log(`➖ 已删除设备: ${deviceKey}`);
      return true;
    } else {
      console.log(`⚠️ 设备 ${deviceKey} 不存在`);
      return false;
    }
  }

  /**
   * 强制删除设备（即使配置文件中不存在）
   */
  async forceRemoveDevice(manufacturer: string, serialNumber: string): Promise<boolean> {
    const deviceKey = `${manufacturer}-${serialNumber}`;
    const initialLength = this.devices.length;
    
    // 从内存中移除
    this.devices = this.devices.filter(d => 
      !(d.manufacturer === manufacturer && d.serialNumber === serialNumber)
    );

    // 总是保存文件，确保一致性
    await this.saveDevices();
    
    if (this.devices.length < initialLength) {
      console.log(`🗑️ 已强制删除设备: ${deviceKey}`);
    } else {
      console.log(`🗑️ 强制删除设备: ${deviceKey} (配置文件中不存在，但已清理运行时状态)`);
    }
    
    return true;
  }

  /**
   * 列出所有设备
   */
  listDevices(): DeviceConfig[] {
    console.log(`📋 当前设备列表 (${this.devices.length} 个):`);
    this.devices.forEach((device, index) => {
      console.log(`  ${index + 1}. ${device.manufacturer}/${device.serialNumber}`);
    });
    return this.devices;
  }

  /**
   * 清空所有设备
   */
  async clearDevices(): Promise<void> {
    this.devices = [];
    await this.saveDevices();
    console.log("🗑️ 已清空所有设备");
  }

  /**
   * 生成测试设备
   */
  async generateTestDevices(count: number, prefix = "TEST"): Promise<void> {
    const testDevices: DeviceConfig[] = [];
    
    for (let i = 0; i < count; i++) {
      testDevices.push({
        manufacturer: "SEER",
        serialNumber: `${prefix}-${i}`
      });
    }

    await this.addDevices(testDevices);
    console.log(`🧪 已生成 ${count} 个测试设备`);
  }

  /**
   * 检查设备是否存在
   */
  deviceExists(manufacturer: string, serialNumber: string): boolean {
    return this.devices.some(d => 
      d.manufacturer === manufacturer && d.serialNumber === serialNumber
    );
  }

  /**
   * 获取设备数量
   */
  getDeviceCount(): number {
    return this.devices.length;
  }
}

// 导出便捷函数
export async function addDeviceToConfig(manufacturer: string, serialNumber: string): Promise<boolean> {
  const manager = new DeviceManager();
  await manager.loadDevices();
  return await manager.addDevice(manufacturer, serialNumber);
}

export async function removeDeviceFromConfig(manufacturer: string, serialNumber: string): Promise<boolean> {
  const manager = new DeviceManager();
  await manager.loadDevices();
  return await manager.removeDevice(manufacturer, serialNumber);
}

export async function listConfigDevices(): Promise<DeviceConfig[]> {
  const manager = new DeviceManager();
  await manager.loadDevices();
  return manager.listDevices();
}

// CLI 使用示例
if (import.meta.main) {
  const manager = new DeviceManager();
  await manager.loadDevices();

  const args = Deno.args;
  const command = args[0];

  switch (command) {
    case "add":
      if (args.length >= 3) {
        await manager.addDevice(args[1], args[2]);
      } else {
        console.log("用法: deno run device_manager.ts add <manufacturer> <serialNumber>");
      }
      break;

    case "remove":
      if (args.length >= 3) {
        await manager.removeDevice(args[1], args[2]);
      } else {
        console.log("用法: deno run device_manager.ts remove <manufacturer> <serialNumber>");
      }
      break;

    case "force-remove":
      if (args.length >= 3) {
        await manager.forceRemoveDevice(args[1], args[2]);
      } else {
        console.log("用法: deno run device_manager.ts force-remove <manufacturer> <serialNumber>");
      }
      break;

    case "list":
      manager.listDevices();
      break;

    case "clear":
      await manager.clearDevices();
      break;

    case "generate":
      const count = parseInt(args[1]) || 10;
      const prefix = args[2] || "TEST";
      await manager.generateTestDevices(count, prefix);
      break;

    default:
      console.log(`
🔧 设备管理工具

用法:
  deno run --allow-read --allow-write device_manager.ts <command> [args]

命令:
  add <manufacturer> <serialNumber>  - 添加设备
  remove <manufacturer> <serialNumber> - 删除设备
  force-remove <manufacturer> <serialNumber> - 强制删除设备
  list                                - 列出所有设备
  clear                              - 清空所有设备
  generate <count> [prefix]          - 生成测试设备

示例:
  deno run --allow-read --allow-write device_manager.ts add SEER AGV-001
  deno run --allow-read --allow-write device_manager.ts list
  deno run --allow-read --allow-write device_manager.ts generate 5 TEST
      `);
  }
}