@daimazun/hardware-info 1.0.0

package/lib/ohm-daemon.js

@@ -0,0 +1,208 @@
+const { spawn } = require('child_process');
+const path = require('path');
+const readline = require('readline');
+
+const SCRIPT = path.join(__dirname, '..', 'scripts', 'ohm-daemon.ps1');
+
+class CpuTempDaemon {
+  constructor(options = {}) {
+    this.options = options;
+    this.process = null;
+    this.ready = false;
+    this.readyPromise = null;
+    this.queue = [];
+    this.currentRequest = null;
+    this.closed = false;
+    this.restartCount = 0;
+    this.maxRestarts = options.maxRestarts || 3;
+  }
+
+  /**
+   * 启动常驻进程，等待 READY 信号
+   */
+  async start() {
+    if (this.process) {
+      if (this.ready) return;
+      return this.readyPromise;
+    }
+
+    this.readyPromise = new Promise((resolve, reject) => {
+      this.process = spawn('powershell', [
+        '-NoProfile',
+        '-ExecutionPolicy', 'Bypass',
+        '-File', SCRIPT
+      ], {
+        stdio: ['pipe', 'pipe', 'pipe']
+      });
+
+      const rl = readline.createInterface({
+        input: this.process.stdout,
+        terminal: false
+      });
+
+      let readyReceived = false;
+
+      rl.on('line', (line) => {
+        line = line.trim();
+        if (!line) return;
+
+        if (!readyReceived) {
+          if (line === 'READY') {
+            readyReceived = true;
+            this.ready = true;
+            resolve();
+          } else {
+            // 第一行不是 READY，可能是错误
+            reject(new Error(`Daemon failed to start: ${line}`));
+            this._cleanup();
+          }
+          return;
+        }
+
+        // 正常响应
+        this._handleResponse(line);
+      });
+
+      this.process.stderr.on('data', (data) => {
+        const msg = data.toString().trim();
+        if (msg) {
+          // stderr 输出，记录但不中断
+          if (this.options.verbose) console.error('[daemon stderr]', msg);
+        }
+      });
+
+      this.process.on('exit', (code) => {
+        this.ready = false;
+        this.process = null;
+        rl.close();
+
+        // 拒绝所有 pending 请求
+        const err = new Error(`Daemon exited with code ${code}`);
+        this._rejectAllPending(err);
+
+        if (!this.closed && this.restartCount < this.maxRestarts) {
+          this.restartCount++;
+          if (this.options.verbose) console.error(`[daemon] exited, restarting (${this.restartCount}/${this.maxRestarts})`);
+          this.start().catch(() => {});
+        }
+      });
+
+      this.process.on('error', (err) => {
+        if (!readyReceived) {
+          reject(err);
+        }
+        this._cleanup();
+      });
+    });
+
+    return this.readyPromise;
+  }
+
+  /**
+   * 获取一次 CPU 温度数据
+   */
+  async getCpuTemp() {
+    if (this.closed) {
+      throw new Error('Daemon is closed');
+    }
+    if (!this.process || !this.ready) {
+      await this.start();
+    }
+
+    return new Promise((resolve, reject) => {
+      this.queue.push({ resolve, reject, timestamp: Date.now() });
+      this._processQueue();
+    });
+  }
+
+  /**
+   * 关闭常驻进程
+   */
+  async close() {
+    this.closed = true;
+    if (this.process) {
+      try {
+        this.process.stdin.write('exit\n');
+      } catch (e) {
+        // ignore
+      }
+      // 等 2 秒，不退出就强杀
+      await new Promise((resolve) => {
+        const timer = setTimeout(() => {
+          if (this.process) {
+            this.process.kill();
+          }
+          resolve();
+        }, 2000);
+        this.process.on('exit', () => {
+          clearTimeout(timer);
+          resolve();
+        });
+      });
+    }
+    this._cleanup();
+  }
+
+  _processQueue() {
+    if (this.currentRequest || this.queue.length === 0) return;
+    if (!this.process || !this.ready) return;
+
+    this.currentRequest = this.queue.shift();
+    try {
+      this.process.stdin.write('get\n');
+    } catch (e) {
+      this.currentRequest.reject(e);
+      this.currentRequest = null;
+      this._processQueue();
+    }
+  }
+
+  _handleResponse(line) {
+    if (!this.currentRequest) {
+      // 没有等待中的请求，忽略
+      return;
+    }
+
+    try {
+      const data = JSON.parse(line);
+      if (data.error) {
+        this.currentRequest.reject(new Error(data.error));
+      } else {
+        this.currentRequest.resolve(data);
+      }
+    } catch (e) {
+      this.currentRequest.reject(new Error(`Failed to parse daemon response: ${e.message}`));
+    }
+
+    this.currentRequest = null;
+    this._processQueue();
+  }
+
+  _rejectAllPending(err) {
+    if (this.currentRequest) {
+      this.currentRequest.reject(err);
+      this.currentRequest = null;
+    }
+    while (this.queue.length > 0) {
+      const req = this.queue.shift();
+      req.reject(err);
+    }
+  }
+
+  _cleanup() {
+    this.ready = false;
+    this.process = null;
+    this.currentRequest = null;
+  }
+}
+
+/**
+ * 创建一个常驻 CPU 温度监控实例
+ */
+async function createCpuTempDaemon(options = {}) {
+  const daemon = new CpuTempDaemon(options);
+  await daemon.start();
+  return daemon;
+}
+
+module.exports = { CpuTempDaemon, createCpuTempDaemon };

package/lib/ohm.js

@@ -0,0 +1,41 @@
+const { spawn } = require('child_process');
+const path = require('path');
+
+const SCRIPT = path.join(__dirname, '..', 'scripts', 'get-lhm-temp.ps1');
+
+function readOhmTemp() {
+  return new Promise((resolve, reject) => {
+    const ps = spawn('powershell', [
+      '-NoProfile',
+      '-ExecutionPolicy', 'Bypass',
+      '-File', SCRIPT
+    ]);
+
+    let out = '';
+    let err = '';
+    ps.stdout.on('data', d => (out += d));
+    ps.stderr.on('data', d => (err += d));
+    ps.on('close', code => {
+      if (code !== 0) {
+        return reject(new Error(`OHM exited ${code}: ${err || out}`));
+      }
+      try {
+        const data = JSON.parse(out.trim());
+        if (data.error) {
+          return reject(new Error(`OHM: ${data.error}`));
+        }
+        const cores = data.temperature?.cores;
+        const hasCores = Array.isArray(cores) && cores.length > 0;
+        const hasPackage = data.temperature?.package != null;
+        if (!data.temperature || (!hasCores && !hasPackage)) {
+          return reject(new Error('OHM: no temperature sensors (likely no admin privileges)'));
+        }
+        resolve(data);
+      } catch (e) {
+        reject(new Error(`OHM parse error: ${e.message}`));
+      }
+    });
+  });
+}
+
+module.exports = { readOhmTemp };

package/lib/perfctr.js

@@ -0,0 +1,43 @@
+const { spawn } = require('child_process');
+
+const PS_CMD = [
+  '-NoProfile',
+  '-Command',
+  'Get-CimInstance Win32_PerfFormattedData_Counters_ThermalZoneInformation -Namespace "root\\cimv2" | Select-Object Name, HighPrecisionTemperature, Temperature | ConvertTo-Json'
+];
+
+function readPerfCounterTemp() {
+  return new Promise((resolve, reject) => {
+    const ps = spawn('powershell', PS_CMD);
+    let out = '';
+    let err = '';
+    ps.stdout.on('data', d => (out += d));
+    ps.stderr.on('data', d => (err += d));
+    ps.on('close', code => {
+      if (code !== 0) {
+        return reject(new Error(`PerfCounter exited ${code}: ${err || out}`));
+      }
+      try {
+        const trimmed = out.trim();
+        if (!trimmed) return reject(new Error('PerfCounter: empty output'));
+        const data = JSON.parse(trimmed);
+        const list = Array.isArray(data) ? data : [data];
+        const zones = list.map(item => ({
+          zone: item.Name,
+          celsius: Math.round(((item.HighPrecisionTemperature / 10) - 273.15) * 100) / 100
+        }));
+        if (zones.length === 0) return reject(new Error('PerfCounter: no thermal zones'));
+        resolve({
+          source: 'performance_counter',
+          isCoreTemp: false,
+          zones,
+          maxZone: Math.round(Math.max(...zones.map(z => z.celsius)) * 100) / 100
+        });
+      } catch (e) {
+        reject(new Error(`PerfCounter parse error: ${e.message}`));
+      }
+    });
+  });
+}
+
+module.exports = { readPerfCounterTemp };

package/lib/process-icon.js

@@ -0,0 +1,76 @@
+/**
+ * process-icon.js — 进程图标提取
+ *
+ * 用 PowerShell System.Drawing 提取 exe 关联图标，输出 base64 PNG。
+ * 按需调用，不进高频路径（图标是静态属性，建议调用方缓存）。
+ *
+ * 支持批量提取：路径列表走 stdin，无命令行长度限制。
+ */
+
+const { spawn } = require('child_process');
+const { existsSync } = require('fs');
+
+const PS_SCRIPT = [
+  'Add-Type -AssemblyName System.Drawing;',
+  '$paths = [Console]::In.ReadToEnd() | ConvertFrom-Json;',
+  '$out = @{};',
+  'foreach ($p in $paths) {',
+  '  $b64 = \'\';',
+  '  try {',
+  '    $ico = [System.Drawing.Icon]::ExtractAssociatedIcon($p);',
+  '    $bmp = $ico.ToBitmap();',
+  '    $ms = New-Object System.IO.MemoryStream;',
+  '    $bmp.Save($ms, [System.Drawing.Imaging.ImageFormat]::Png);',
+  '    $b64 = [Convert]::ToBase64String($ms.ToArray());',
+  '    $ms.Dispose(); $bmp.Dispose(); $ico.Dispose();',
+  '  } catch { };',
+  '  $out[$p] = $b64;',
+  '};',
+  'Write-Output (ConvertTo-Json -InputObject $out -Compress);'
+].join(' ');
+
+/**
+ * 批量提取进程图标
+ * @param {string[]} paths - exe 路径数组
+ * @returns {Promise<Record<string, string>>} 路径 → base64 PNG（提取失败为空串）
+ */
+function getProcessIcons(paths) {
+  const valid = paths.filter((p) => typeof p === 'string' && p && existsSync(p));
+  if (!valid.length) return Promise.resolve({});
+
+  return new Promise((resolve, reject) => {
+    const ps = spawn('powershell', ['-NoProfile', '-ExecutionPolicy', 'Bypass', '-Command', PS_SCRIPT], {
+      stdio: ['pipe', 'pipe', 'ignore']
+    });
+
+    let out = '';
+    ps.stdout.setEncoding('utf8');
+    ps.stdout.on('data', (c) => { out += c; });
+    ps.on('error', reject);
+    ps.on('close', (code) => {
+      if (code !== 0) {
+        reject(new Error(`PowerShell exited ${code}`));
+        return;
+      }
+      try {
+        const map = JSON.parse(out.trim());
+        resolve(map);
+      } catch (e) {
+        reject(new Error('图标数据解析失败: ' + e.message));
+      }
+    });
+    ps.stdin.end(JSON.stringify(valid));
+  });
+}
+
+/**
+ * 提取单个进程图标
+ * @param {string} exePath - exe 路径
+ * @returns {Promise<string>} base64 PNG（提取失败为空串）
+ */
+async function getProcessIcon(exePath) {
+  const map = await getProcessIcons([exePath]);
+  return map[exePath] || '';
+}
+
+module.exports = { getProcessIcon, getProcessIcons };

package/lib/process-ops.js

@@ -0,0 +1,218 @@
+/**
+ * process-ops.js — 进程操作
+ *
+ *   - killProcess: fkill（跨平台成熟包，force+tree，Windows 内部调 taskkill）
+ *   - showItemInFolder: explorer /select,<path>
+ */
+
+const { execFile, exec } = require('child_process');
+const { existsSync } = require('fs');
+const { sampleProcesses } = require('./win32-procs');
+
+// fkill v10 是 ESM only，动态 import 并缓存
+let fkillPromise = null;
+function getFkill() {
+  if (!fkillPromise) {
+    fkillPromise = import('fkill').then((m) => m.default);
+  }
+  return fkillPromise;
+}
+
+/**
+ * 查找占用指定端口的进程 PID（Windows netstat，零依赖）
+ * @param {number} port - 端口号
+ * @param {'tcp'|'udp'|'all'} [protocol='all'] - 协议类型
+ * @returns {Promise<number[]>} PID 数组（去重）
+ */
+function findPidsByPort(port, protocol = 'all') {
+  return new Promise((resolve, reject) => {
+    const p = Number(port);
+    if (!p || p < 1 || p > 65535) {
+      resolve([]);
+      return;
+    }
+    // netstat -ano 同时包含 TCP 和 UDP，输出最后一列是 PID
+    exec('netstat -ano', { windowsHide: true }, (err, stdout) => {
+      if (err) {
+        reject(err);
+        return;
+      }
+      const pids = new Set();
+      const lines = stdout.split('\n');
+      for (const line of lines) {
+        const trimmed = line.trim();
+        if (!trimmed) continue;
+        // 跳过表头
+        if (trimmed.startsWith('Proto') || trimmed.startsWith('活动连接')) continue;
+        const parts = trimmed.split(/\s+/);
+        if (parts.length < 5) continue;
+        const proto = parts[0].toUpperCase(); // TCP / UDP
+        const localAddr = parts[1]; // 0.0.0.0:3000 或 [::]:3000
+        const pid = parseInt(parts[parts.length - 1], 10);
+        if (!pid || pid <= 0) continue;
+        // 协议过滤
+        if (protocol === 'tcp' && proto !== 'TCP') continue;
+        if (protocol === 'udp' && proto !== 'UDP') continue;
+        // 本地端口匹配（支持 IPv4 和 IPv6）
+        const portMatch = localAddr.match(/:(\d+)$/);
+        if (portMatch && parseInt(portMatch[1], 10) === p) {
+          pids.add(pid);
+        }
+      }
+      resolve([...pids]);
+    });
+  });
+}
+
+/**
+ * 按进程名查找 PID（koffi 进程枚举，不区分大小写，支持 .exe 后缀）
+ * @param {string|string[]} name - 进程名或进程名数组
+ * @returns {number[]} PID 数组（去重）
+ */
+function findPidsByName(name) {
+  const names = (Array.isArray(name) ? name : [name])
+    .map((n) => String(n).toLowerCase().replace(/\.exe$/i, ''));
+  if (!names.length) return [];
+  const all = sampleProcesses();
+  const pids = new Set();
+  for (const p of all) {
+    const procName = p.name.toLowerCase().replace(/\.exe$/i, '');
+    if (names.includes(procName)) {
+      pids.add(p.pid);
+    }
+  }
+  return [...pids];
+}
+
+/**
+ * 结束进程（含子进程树），支持按 PID、端口号或进程名
+ * @param {number|number[]|string|string[]|Object} input - 杀进程目标
+ *   - number: 按 PID 杀
+ *   - number[]: 批量按 PID 杀
+ *   - string: 按进程名杀（如 'chrome'、'notepad'）
+ *   - string[]: 批量按进程名杀
+ *   - { pid: number|number[] }: 按 PID 杀
+ *   - { port: number|number[], protocol?: 'tcp'|'udp'|'all' }: 按端口杀
+ *   - { name: string|string[] }: 按进程名杀
+ * @returns {Promise<{ok: boolean, killed: number, failed: number, pids?: number[], message?: string}>}
+ */
+async function killProcess(input) {
+  let pids = [];
+  let ports = [];
+  let names = [];
+  let protocol = 'all';
+
+  // 解析输入
+  if (typeof input === 'number') {
+    pids = [input];
+  } else if (typeof input === 'string') {
+    names = [input];
+  } else if (Array.isArray(input)) {
+    // 数组元素可能是 PID（数字）或进程名（字符串）
+    for (const item of input) {
+      if (typeof item === 'number') pids.push(item);
+      else if (typeof item === 'string') names.push(item);
+    }
+  } else if (input && typeof input === 'object') {
+    if (input.pid !== undefined) {
+      pids = Array.isArray(input.pid) ? input.pid : [input.pid];
+    }
+    if (input.name !== undefined) {
+      names = Array.isArray(input.name) ? input.name : [input.name];
+    }
+    if (input.port !== undefined) {
+      ports = Array.isArray(input.port) ? input.port : [input.port];
+      if (input.protocol) protocol = input.protocol;
+    }
+  } else {
+    return { ok: false, killed: 0, failed: 0, message: '无效的参数（需要 PID、端口号或进程名）' };
+  }
+
+  // 按进程名查找 PID
+  if (names.length) {
+    const found = findPidsByName(names);
+    pids.push(...found);
+  }
+
+  // 按端口查找 PID
+  if (ports.length) {
+    for (const port of ports) {
+      const found = await findPidsByPort(port, protocol);
+      pids.push(...found);
+    }
+  }
+
+  // 去重 + 过滤无效 PID
+  const list = [...new Set(pids.map(Number).filter((n) => n > 0))];
+  if (!list.length) {
+    const reason = ports.length ? '未找到占用指定端口的进程' : names.length ? '未找到匹配进程名的进程' : '无效的 PID';
+    return { ok: false, killed: 0, failed: 0, message: reason };
+  }
+
+  // 拒绝自杀（Node 主进程）和父进程
+  if (list.includes(process.pid)) {
+    return { ok: false, killed: 0, failed: list.length, message: '不能结束当前进程自身' };
+  }
+  if (process.ppid && list.includes(process.ppid)) {
+    return { ok: false, killed: 0, failed: list.length, message: '不能结束当前进程的父进程' };
+  }
+
+  const fkill = await getFkill();
+  let killed = 0;
+  let failed = 0;
+
+  try {
+    // fkill 支持一次传 PID 数组，内部并行处理
+    await fkill(list, { force: true, tree: true });
+    killed = list.length;
+  } catch (e) {
+    // fkill 失败时抛 AggregateError，errors 数组包含每个 PID 的具体错误
+    if (e.name === 'AggregateError' && Array.isArray(e.errors)) {
+      for (const err of e.errors) {
+        const msg = String(err).toLowerCase();
+        // 进程已不存在，视为达成目标
+        if (msg.includes("doesn't exist") || msg.includes('not found') || msg.includes('不存在')) {
+          killed++;
+        } else {
+          failed++;
+        }
+      }
+    } else {
+      failed = list.length;
+    }
+  }
+
+  if (killed > 0) {
+    return { ok: true, killed, failed, pids: list };
+  }
+  return {
+    ok: false,
+    killed: 0,
+    failed,
+    pids: list,
+    message: '结束失败：权限不足或进程受系统保护，请以管理员身份运行'
+  };
+}
+
+/**
+ * 在资源管理器中打开并定位到文件
+ * @param {string} filePath - 文件完整路径
+ * @returns {{ok: boolean, message?: string}}
+ */
+function showItemInFolder(filePath) {
+  if (typeof filePath !== 'string' || !filePath) {
+    return { ok: false, message: '路径无效' };
+  }
+  if (!existsSync(filePath)) {
+    return { ok: false, message: '文件不存在（进程可能已退出）' };
+  }
+  try {
+    // explorer /select, 路径 — 打开资源管理器并选中该文件
+    execFile('explorer', ['/select,', filePath], () => {});
+    return { ok: true };
+  } catch (e) {
+    return { ok: false, message: e.message };
+  }
+}
+
+module.exports = { killProcess, showItemInFolder, findPidsByPort, findPidsByName };