feat: 前后端联调 — 任务创建 + 账户信息 + 预估花费 (0.0.18)
- task: device_id 改为必填参数，API 层 Omit 封装 + getDeviceId() 自动注入 - pricing.ts: Python 后端 calculate_cost 完整移植，六种定价模式 + 6 个 Bug 修复 (浮点冗余/维度匹配不一致/null 守卫/matrix 数据提取/unit_scale 等) - ModelInputForm: 三层兜底预估消费展示 + React Hooks 顺序修复 - AccountInfo: VIP 等级列表匹配，套餐详情展示（说明/周期/席位/到期） - 模型列表加密缓存 (safe-storage, stale-while-revalidate) - VIP API 模块 (激活码/等级/订单/模拟支付) - device_id 统一使用 UUID v4 持久化 (与 login/register 对齐)
This commit is contained in:
279
electron/preload/device.ts
Normal file
279
electron/preload/device.ts
Normal file
@@ -0,0 +1,279 @@
|
||||
/**
|
||||
* 设备指纹采集 — TypeScript 实现
|
||||
*
|
||||
* 与 device.py 逻辑一致，保留原 Python 方法不变。
|
||||
* 通过采集 UUID / CPU / HDD 三项硬件 ID，拼接后做 SHA256 得到 device_id。
|
||||
*
|
||||
* 适用环境：
|
||||
* - Node.js / Electron（Windows）：通过 child_process 执行 wmic / powershell 采集硬件 ID
|
||||
* - 浏览器：仅提供 SHA256 哈希工具函数，硬件采集需由上层注入
|
||||
*/
|
||||
|
||||
import * as crypto from "node:crypto";
|
||||
import { spawnSync } from "node:child_process";
|
||||
import { platform } from "node:os";
|
||||
|
||||
// ============================================================
|
||||
// 类型定义
|
||||
// ============================================================
|
||||
|
||||
/** 可采集的硬件组件类型 */
|
||||
export type Component = "uuid" | "cpu" | "hdd";
|
||||
|
||||
/** 查询方式 */
|
||||
export type QueryMethod = "auto" | "ps" | "powershell" | "wmic";
|
||||
|
||||
/** 内部解析后的方法 */
|
||||
type ResolvedMethod = "auto" | "powershell" | "wmic";
|
||||
|
||||
// ============================================================
|
||||
// 查询配置（对应 Python _COMPONENT_QUERIES）
|
||||
// ============================================================
|
||||
|
||||
interface ComponentQuery {
|
||||
powershell: string;
|
||||
wmicArgs: string[];
|
||||
headerTokens: string[];
|
||||
}
|
||||
|
||||
const COMPONENT_QUERIES: Record<Component, ComponentQuery> = {
|
||||
uuid: {
|
||||
powershell:
|
||||
"Get-CimInstance -Class Win32_ComputerSystemProduct | " +
|
||||
"Select-Object -ExpandProperty UUID",
|
||||
wmicArgs: ["wmic", "csproduct", "get", "uuid"],
|
||||
headerTokens: ["uuid"],
|
||||
},
|
||||
cpu: {
|
||||
powershell:
|
||||
"Get-CimInstance -Class Win32_Processor | " +
|
||||
"Select-Object -ExpandProperty ProcessorId",
|
||||
wmicArgs: ["wmic", "cpu", "get", "processorid"],
|
||||
headerTokens: ["processorid"],
|
||||
},
|
||||
hdd: {
|
||||
powershell:
|
||||
"Get-CimInstance -Class Win32_DiskDrive | " +
|
||||
"Select-Object -First 1 -ExpandProperty SerialNumber",
|
||||
wmicArgs: ["wmic", "diskdrive", "get", "serialnumber"],
|
||||
headerTokens: ["serialnumber"],
|
||||
},
|
||||
};
|
||||
|
||||
// ============================================================
|
||||
// 工具函数
|
||||
// ============================================================
|
||||
|
||||
/** 对应 Python: _normalize_identifier(value) -> str */
|
||||
function normalizeIdentifier(value: string): string {
|
||||
if (!value) return "";
|
||||
return value.replace(/[^A-Za-z0-9]/g, "").toUpperCase();
|
||||
}
|
||||
|
||||
/** 对应 Python: _first_value_from_output(raw_output, header_tokens) -> str */
|
||||
function firstValueFromOutput(
|
||||
rawOutput: string,
|
||||
headerTokens: readonly string[],
|
||||
): string {
|
||||
if (!rawOutput) return "";
|
||||
const headers = new Set(headerTokens.map((t) => t.toLowerCase()));
|
||||
for (const line of rawOutput.split(/\r?\n/)) {
|
||||
const candidate = line.trim();
|
||||
if (!candidate) continue;
|
||||
if (headers.has(candidate.toLowerCase())) continue;
|
||||
return candidate;
|
||||
}
|
||||
return "";
|
||||
}
|
||||
|
||||
/** 对应 Python: _resolve_method(method) -> _ResolvedMethod */
|
||||
function resolveMethod(method: QueryMethod): ResolvedMethod {
|
||||
if (method === "ps") return "powershell";
|
||||
return method;
|
||||
}
|
||||
|
||||
/** 计算 SHA256 哈希（hex 大写），对应 Python hashlib.sha256(...).hexdigest().upper() */
|
||||
function sha256HexUpper(input: string): string {
|
||||
return crypto.createHash("sha256").update(input, "utf-8").digest("hex").toUpperCase();
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 环境检测
|
||||
// ============================================================
|
||||
|
||||
function isWindows(): boolean {
|
||||
return platform() === "win32";
|
||||
}
|
||||
|
||||
function isWmicAvailable(): boolean {
|
||||
if (!isWindows()) return false;
|
||||
const result = spawnSync("where", ["wmic"], {
|
||||
timeout: 3000,
|
||||
stdio: "ignore",
|
||||
windowsHide: true,
|
||||
});
|
||||
return result.status === 0;
|
||||
}
|
||||
|
||||
function isPowershellAvailable(): boolean {
|
||||
if (!isWindows()) return false;
|
||||
const result = spawnSync("where", ["powershell"], {
|
||||
timeout: 3000,
|
||||
stdio: "ignore",
|
||||
windowsHide: true,
|
||||
});
|
||||
return result.status === 0;
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 进程执行
|
||||
// ============================================================
|
||||
|
||||
/**
|
||||
* 执行命令并返回 stdout（去除首尾空白）。
|
||||
* 对应 Python: _run_process(args, timeout=5.0) -> str
|
||||
*/
|
||||
function runProcess(args: string[], timeoutSec: number = 5.0): string {
|
||||
const command = args[0];
|
||||
const cmdArgs = args.slice(1);
|
||||
try {
|
||||
const result = spawnSync(command, cmdArgs, {
|
||||
timeout: Math.round(timeoutSec * 1000),
|
||||
encoding: "utf-8",
|
||||
stdio: ["ignore", "pipe", "pipe"],
|
||||
windowsHide: true, // 对应 Python STARTF_USESHOWWINDOW
|
||||
});
|
||||
if (result.status !== 0) return "";
|
||||
return (result.stdout ?? "").trim();
|
||||
} catch {
|
||||
return "";
|
||||
}
|
||||
}
|
||||
|
||||
/** 对应 Python: _run_powershell(command) -> str */
|
||||
function runPowershell(command: string): string {
|
||||
if (!isPowershellAvailable()) return "";
|
||||
const args = ["powershell", "-NoProfile", "-NonInteractive", "-Command", command];
|
||||
return runProcess(args);
|
||||
}
|
||||
|
||||
/** 对应 Python: _run_wmic(args) -> str */
|
||||
function runWmic(args: readonly string[]): string {
|
||||
if (!isWmicAvailable()) return "";
|
||||
return runProcess([...args]);
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// DeviceFingerprint 类
|
||||
// ============================================================
|
||||
|
||||
export class DeviceFingerprint {
|
||||
// private wmicAvailableCache: boolean | null = null;
|
||||
// private powershellAvailableCache: boolean | null = null;
|
||||
|
||||
// ---------- 组件 ID 查询 ----------
|
||||
|
||||
/**
|
||||
* 查询单个硬件组件 ID。
|
||||
* 对应 Python: get_component_id(component, method="auto") -> str
|
||||
*/
|
||||
getComponentId(component: Component, method: QueryMethod = "auto"): string {
|
||||
const query = COMPONENT_QUERIES[component];
|
||||
if (!query) return "";
|
||||
|
||||
const resolvedMethod = resolveMethod(method);
|
||||
|
||||
const readViaPowershell = (): string => {
|
||||
const rawOutput = runPowershell(query.powershell);
|
||||
return firstValueFromOutput(rawOutput, query.headerTokens);
|
||||
};
|
||||
|
||||
const readViaWmic = (): string => {
|
||||
const rawOutput = runWmic(query.wmicArgs);
|
||||
return firstValueFromOutput(rawOutput, query.headerTokens);
|
||||
};
|
||||
|
||||
let rawValue = "";
|
||||
if (resolvedMethod === "powershell") {
|
||||
rawValue = readViaPowershell();
|
||||
} else if (resolvedMethod === "wmic") {
|
||||
rawValue = readViaWmic();
|
||||
if (!rawValue) {
|
||||
rawValue = readViaPowershell();
|
||||
}
|
||||
} else {
|
||||
// auto: wmic 优先，回退 powershell
|
||||
rawValue = readViaWmic() || readViaPowershell();
|
||||
}
|
||||
|
||||
return normalizeIdentifier(rawValue);
|
||||
}
|
||||
|
||||
/**
|
||||
* 查询单个硬件组件（别名）。
|
||||
* 对应 Python: get_component(component_type, method="auto") -> str
|
||||
*/
|
||||
getComponent(component: Component, method: QueryMethod = "auto"): string {
|
||||
return this.getComponentId(component, method);
|
||||
}
|
||||
|
||||
// ---------- 机器指纹 ----------
|
||||
|
||||
/**
|
||||
* 生成设备指纹 SHA256。
|
||||
* 采集 UUID + HDD + CPU，拼接后做 SHA256 哈希。
|
||||
* 对应 Python: fingerprint_sha256(method="auto") -> str | None
|
||||
*/
|
||||
fingerprintSha256(method: QueryMethod = "auto"): string | null {
|
||||
const resolvedMethod = resolveMethod(method);
|
||||
const wmicOk = isWmicAvailable();
|
||||
const psOk = isPowershellAvailable();
|
||||
|
||||
if (resolvedMethod === "powershell") {
|
||||
if (!psOk) return null;
|
||||
} else {
|
||||
if (!wmicOk && !psOk) return null;
|
||||
}
|
||||
|
||||
const uuidId = this.getComponentId("uuid", method);
|
||||
const diskId = this.getComponentId("hdd", method);
|
||||
const cpuId = this.getComponentId("cpu", method);
|
||||
|
||||
if (!uuidId && !diskId && !cpuId) {
|
||||
return null;
|
||||
}
|
||||
|
||||
const combined = `${uuidId}|${diskId}|${cpuId}`;
|
||||
const normalized = normalizeIdentifier(combined);
|
||||
return sha256HexUpper(normalized);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取机器码（指纹的别名）。
|
||||
* 对应 Python: get_machine_code(method="auto") -> str | None
|
||||
*/
|
||||
getMachineCode(method: QueryMethod = "auto"): string | null {
|
||||
return this.fingerprintSha256(method);
|
||||
}
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 便捷函数（无需实例化）
|
||||
// ============================================================
|
||||
|
||||
const defaultFingerprint = new DeviceFingerprint();
|
||||
|
||||
/** 获取机器码（单例便捷调用） */
|
||||
export function getMachineCode(method: QueryMethod = "auto"): string | null {
|
||||
return defaultFingerprint.getMachineCode(method);
|
||||
}
|
||||
|
||||
/** 获取设备指纹 SHA256（单例便捷调用） */
|
||||
export function fingerprintSha256(method: QueryMethod = "auto"): string | null {
|
||||
return defaultFingerprint.fingerprintSha256(method);
|
||||
}
|
||||
|
||||
/** 查询单个硬件组件 ID（单例便捷调用） */
|
||||
export function getComponentId(component: Component, method: QueryMethod = "auto"): string {
|
||||
return defaultFingerprint.getComponentId(component, method);
|
||||
}
|
||||
200
electron/preload/device_service.ts
Normal file
200
electron/preload/device_service.ts
Normal file
@@ -0,0 +1,200 @@
|
||||
/**
|
||||
* 设备信息服务 — TypeScript 实现
|
||||
*
|
||||
* 与 device_service.py 逻辑一致，保留原 Python 方法不变。
|
||||
* 提供带缓存的机器码、用户 IP、MAC 地址查询。
|
||||
*
|
||||
* 适用环境：Node.js / Electron
|
||||
*/
|
||||
|
||||
import * as os from "node:os";
|
||||
import * as dgram from "node:dgram";
|
||||
import { DeviceFingerprint } from "./device";
|
||||
import type { QueryMethod } from "./device";
|
||||
|
||||
// ============================================================
|
||||
// 工具函数
|
||||
// ============================================================
|
||||
|
||||
/** 对应 Python: _is_valid_ipv4(value) -> bool */
|
||||
function isValidIPv4(value: string): boolean {
|
||||
const parts = value.split(".");
|
||||
if (parts.length !== 4) return false;
|
||||
for (const part of parts) {
|
||||
const num = parseInt(part, 10);
|
||||
if (isNaN(num) || num < 0 || num > 255) return false;
|
||||
if (part.length > 1 && part.startsWith("0")) return false;
|
||||
}
|
||||
// 排除环回地址和 0.0.0.0
|
||||
if (value.startsWith("127.") || value === "0.0.0.0") return false;
|
||||
return true;
|
||||
}
|
||||
|
||||
/** 对应 Python: _resolve_user_ip() -> Optional[str] */
|
||||
function resolveUserIp(): string | null {
|
||||
// 优先通过 UDP "伪连接" 获取本机出站 IP
|
||||
try {
|
||||
const sock = dgram.createSocket("udp4");
|
||||
// 同步方式不可行;使用 UDP connect + getsockname 的异步模式
|
||||
// 但在 Node.js 中这本质是同步的 (connect 不发送数据)
|
||||
sock.connect(80, "8.8.8.8");
|
||||
const address = sock.address();
|
||||
sock.close();
|
||||
if (address && typeof address.address === "string") {
|
||||
const ip = address.address;
|
||||
if (isValidIPv4(ip)) return ip;
|
||||
}
|
||||
} catch {
|
||||
// 忽略
|
||||
}
|
||||
|
||||
// 回退: 遍历网络接口查找第一个有效的非环回 IPv4
|
||||
const interfaces = os.networkInterfaces();
|
||||
for (const ifaceList of Object.values(interfaces)) {
|
||||
if (!ifaceList) continue;
|
||||
for (const iface of ifaceList) {
|
||||
if (iface.family === "IPv4") {
|
||||
const ip = iface.address;
|
||||
if (isValidIPv4(ip)) return ip;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
|
||||
/** 对应 Python: _resolve_user_mac_address() -> Optional[str] */
|
||||
function resolveUserMacAddress(): string | null {
|
||||
const interfaces = os.networkInterfaces();
|
||||
for (const ifaceList of Object.values(interfaces)) {
|
||||
if (!ifaceList) continue;
|
||||
for (const iface of ifaceList) {
|
||||
// 跳过环回接口
|
||||
if (iface.internal) continue;
|
||||
if (iface.mac && iface.mac !== "00:00:00:00:00:00") {
|
||||
// 对应 Python: 检查多播位
|
||||
const parts = iface.mac.split(":");
|
||||
if (parts.length === 6) {
|
||||
const firstByte = parseInt(parts[0], 16);
|
||||
// uuid.getnode 的多播位检查: (node >> 40) & 1
|
||||
if ((firstByte & 1) === 0) {
|
||||
return iface.mac.toUpperCase();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 未解析标记
|
||||
// ============================================================
|
||||
|
||||
const UNRESOLVED = Symbol("unresolved");
|
||||
|
||||
// ============================================================
|
||||
// DeviceService 类
|
||||
// ============================================================
|
||||
|
||||
export class DeviceService {
|
||||
private fingerprint = new DeviceFingerprint();
|
||||
|
||||
private cachedMachineCode: string | null | typeof UNRESOLVED = UNRESOLVED;
|
||||
private cachedUserIp: string | null | typeof UNRESOLVED = UNRESOLVED;
|
||||
private cachedUserMacAddress: string | null | typeof UNRESOLVED = UNRESOLVED;
|
||||
|
||||
private warmupStarted = false;
|
||||
|
||||
/**
|
||||
* 获取机器码（带缓存）。
|
||||
* 对应 Python: get_machine_code() -> Optional[str]
|
||||
*/
|
||||
getMachineCode(method: QueryMethod = "auto"): string | null {
|
||||
if (this.cachedMachineCode !== UNRESOLVED) {
|
||||
return this.cachedMachineCode;
|
||||
}
|
||||
|
||||
const machineCode = this.fingerprint.getMachineCode(method);
|
||||
if (this.cachedMachineCode === UNRESOLVED) {
|
||||
this.cachedMachineCode = machineCode;
|
||||
}
|
||||
return machineCode;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取用户本机 IP（带缓存）。
|
||||
* 对应 Python: get_user_ip() -> Optional[str]
|
||||
*/
|
||||
getUserIp(): string | null {
|
||||
if (this.cachedUserIp !== UNRESOLVED) {
|
||||
return this.cachedUserIp;
|
||||
}
|
||||
|
||||
const value = resolveUserIp();
|
||||
if (this.cachedUserIp === UNRESOLVED && value) {
|
||||
this.cachedUserIp = value;
|
||||
}
|
||||
return value;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取用户 MAC 地址（带缓存）。
|
||||
* 对应 Python: get_user_mac_address() -> Optional[str]
|
||||
*/
|
||||
getUserMacAddress(): string | null {
|
||||
if (this.cachedUserMacAddress !== UNRESOLVED) {
|
||||
return this.cachedUserMacAddress;
|
||||
}
|
||||
|
||||
const value = resolveUserMacAddress();
|
||||
if (this.cachedUserMacAddress === UNRESOLVED) {
|
||||
this.cachedUserMacAddress = value;
|
||||
}
|
||||
return value;
|
||||
}
|
||||
|
||||
/**
|
||||
* 异步预热机器码（后台线程）。
|
||||
* 对应 Python: warm_up_machine_code() -> None
|
||||
*/
|
||||
warmUpMachineCode(): void {
|
||||
if (this.cachedMachineCode !== UNRESOLVED) return;
|
||||
if (this.warmupStarted) return;
|
||||
this.warmupStarted = true;
|
||||
|
||||
// 使用 setImmediate 模拟后台线程，不阻塞当前事件循环
|
||||
setImmediate(() => {
|
||||
const machineCode = this.fingerprint.getMachineCode();
|
||||
if (this.cachedMachineCode === UNRESOLVED) {
|
||||
this.cachedMachineCode = machineCode;
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 便捷函数（单例）
|
||||
// ============================================================
|
||||
|
||||
const defaultService = new DeviceService();
|
||||
|
||||
/** 获取机器码（单例，带缓存） */
|
||||
export function getMachineCode(method: QueryMethod = "auto"): string | null {
|
||||
return defaultService.getMachineCode(method);
|
||||
}
|
||||
|
||||
/** 获取用户本机 IP（单例，带缓存） */
|
||||
export function getUserIp(): string | null {
|
||||
return defaultService.getUserIp();
|
||||
}
|
||||
|
||||
/** 获取用户 MAC 地址（单例，带缓存） */
|
||||
export function getUserMacAddress(): string | null {
|
||||
return defaultService.getUserMacAddress();
|
||||
}
|
||||
|
||||
/** 异步预热机器码 */
|
||||
export function warmUpMachineCode(): void {
|
||||
defaultService.warmUpMachineCode();
|
||||
}
|
||||
Reference in New Issue
Block a user