@z-qinghui/migpt-claw 1.0.0

package/src/utils/http.ts

@@ -0,0 +1,193 @@
+import { sleep } from './parse.js';
+import { jsonEncode } from './parse.js';
+import axios from 'axios';
+import type { MiAccount } from '../mi/typing.js';
+import { Debugger } from './debug.js';
+
+const _baseConfig: any = {
+  proxy: false,
+  decompress: true,
+  headers: {
+    'Accept-Encoding': 'gzip, deflate',
+    'Content-Type': 'application/x-www-form-urlencoded',
+    'User-Agent':
+      'Dalvik/2.1.0 (Linux; U; Android 10; RMX2111 Build/QP1A.190711.020) APP/xiaomi.mico APPV/2004040 MK/Uk1YMjExMQ== PassportSDK/3.8.3 passport-ui/3.8.3',
+  },
+};
+
+const _http = axios.create(_baseConfig);
+
+interface HttpError {
+  isError: true;
+  error: any;
+  code: string;
+  message: string;
+}
+
+type RequestConfig = any & {
+  account?: MiAccount;
+  setAccount?: (newAccount: any) => void;
+  rawResponse?: boolean;
+  cookies?: Record<string, string | number | boolean | undefined>;
+};
+
+_http.interceptors.response.use(
+  (res: any) => {
+    if (res.config.rawResponse) {
+      return res;
+    }
+    return res.data;
+  },
+  async (err) => {
+    // 401 登录凭证过期后，自动刷新 token
+    const newResult = await tokenRefresher.refreshTokenAndRetry(err);
+    if (newResult) {
+      return newResult;
+    }
+    const error = err.response?.data?.error || err.response?.data;
+    const request = {
+      method: err.config.method,
+      url: err.config.url,
+      headers: jsonEncode(err.config.headers),
+      data: jsonEncode({ body: err.config.data }),
+    };
+    const response = !err.response
+      ? undefined
+      : {
+          url: err.config.url,
+          status: err.response.status,
+          headers: jsonEncode(err.response.headers),
+          data: jsonEncode({ body: err.response.data }),
+        };
+    return {
+      isError: true,
+      code: error?.code || err.response?.status || err.code || '未知',
+      message: error?.message || err.response?.statusText || err.message || '未知',
+      error: { request, response },
+    };
+  },
+);
+
+class HTTPClient {
+  // 默认 5 秒超时
+  timeout = 5 * 1000;
+
+  async get<T = any>(
+    url: string,
+    _query?: Record<string, string | number | boolean | undefined> | RequestConfig,
+    _config?: RequestConfig,
+  ): Promise<T | HttpError> {
+    let query = _query;
+    let config = _config;
+    if (_config === undefined) {
+      config = _query;
+      query = undefined;
+    }
+    return _http.get<T>(HTTPClient.buildURL(url, query), HTTPClient.buildConfig(config)) as any;
+  }
+
+  async post<T = any>(url: string, data?: any, config?: RequestConfig): Promise<T | HttpError> {
+    return _http.post<T>(url, data, HTTPClient.buildConfig(config)) as any;
+  }
+
+  static buildURL = (url: string, query?: Record<string, any>) => {
+    const _url = new URL(url);
+    for (const [key, value] of Object.entries(query ?? {})) {
+      if (isNotEmpty(value)) {
+        _url.searchParams.append(key, value.toString());
+      }
+    }
+    return _url.href;
+  };
+
+  static buildConfig = (config?: RequestConfig) => {
+    if (config?.cookies) {
+      config.headers = {
+        ...config.headers,
+        Cookie: Object.entries(config.cookies)
+          .map(([key, value]) => `${key}=${value == null ? '' : value.toString()};`)
+          .join(' '),
+      };
+    }
+    if (config && !config.timeout) {
+      config.timeout = Http.timeout; // 默认超时时间为 5 秒
+    }
+    return config;
+  };
+}
+
+export const Http = new HTTPClient();
+
+function isNotEmpty(value: any): boolean {
+  return value !== null && value !== undefined && value !== '';
+}
+
+class TokenRefresher {
+  isRefreshing = false;
+
+  /**
+   * 自动刷新过期的凭证，并重新发送请求
+   */
+  async refreshTokenAndRetry(err: any, maxRetry = 3) {
+    const isMiNA = err?.config?.url?.includes('mina.mi.com');
+    const isMIoT = err?.config?.url?.includes('io.mi.com');
+    if ((!isMiNA && !isMIoT) || err.response?.status !== 401) {
+      return;
+    }
+    if (this.isRefreshing) {
+      return;
+    }
+    let result: any;
+    this.isRefreshing = true;
+    let newServiceAccount = undefined;
+    for (let i = 0; i < maxRetry; i++) {
+      if (Debugger.debug) {
+        console.log(`❌ 登录凭证已过期，正在尝试刷新 Token ${i + 1}`);
+      }
+      newServiceAccount = await this.refreshToken(err);
+      if (newServiceAccount) {
+        // 刷新成功，重新请求
+        result = await this.retry(err, newServiceAccount);
+        break;
+      }
+      // 隔 3 秒后重试
+      await sleep(3000);
+    }
+    this.isRefreshing = false;
+    if (!newServiceAccount) {
+      console.error('❌ 刷新登录凭证失败，请检查账号密码是否仍然有效。');
+    }
+    return result;
+  }
+
+  /**
+   * 刷新登录凭证并同步到本地
+   */
+  async refreshToken(_err: any) {
+    // 注意：这里需要循环依赖，暂时简化处理
+    // 实际应该从配置中读取并刷新
+    return undefined;
+  }
+
+  /**
+   * 重新请求
+   */
+  async retry(_err: any, account: any) {
+    // 更新 cookies
+    const cookies = _err.config.cookies ?? {};
+    for (const key of ['serviceToken']) {
+      if (cookies[key] && account[key]) {
+        cookies[key] = account[key];
+      }
+    }
+    for (const key of ['deviceSNProfile']) {
+      if (cookies[key] && account.device?.[key]) {
+        cookies[key] = account.device[key];
+      }
+    }
+    // 重新请求
+    return _http(HTTPClient.buildConfig(_err.config)!);
+  }
+}
+
+const tokenRefresher = new TokenRefresher();

package/src/utils/index.ts

@@ -0,0 +1,5 @@
+export { Http } from './http.js';
+export { Debugger } from './debug.js';
+export * from './codec.js';
+export * from './hash.js';
+export * from './io.js';

package/src/utils/io.ts

@@ -0,0 +1,68 @@
+import { promises as fs } from 'node:fs';
+import { join } from 'node:path';
+
+/**
+ * 获取数据目录
+ */
+export function getDataDir(subdir?: string): string {
+  const baseDir = join(process.cwd(), '.migpt');
+  if (subdir) {
+    return join(baseDir, subdir);
+  }
+  return baseDir;
+}
+
+/**
+ * 读取 JSON 文件
+ */
+export async function readJSON<T = any>(filename: string): Promise<T | undefined> {
+  try {
+    const filepath = join(getDataDir(), filename);
+    const content = await fs.readFile(filepath, 'utf-8');
+    return JSON.parse(content);
+  } catch {
+    return undefined;
+  }
+}
+
+/**
+ * 写入 JSON 文件
+ */
+export async function writeJSON(filename: string, data: any): Promise<void> {
+  try {
+    const dir = getDataDir();
+    await fs.mkdir(dir, { recursive: true });
+    const filepath = join(dir, filename);
+    await fs.writeFile(filepath, JSON.stringify(data, null, 2), 'utf-8');
+  } catch (err: any) {
+    console.error('❌ 写入文件失败:', err.message);
+  }
+}
+
+/**
+ * 检查文件是否存在
+ */
+export async function fileExists(filepath: string): Promise<boolean> {
+  try {
+    await fs.access(filepath);
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+/**
+ * 读取文件
+ */
+export async function readFile(filepath: string): Promise<Buffer> {
+  return fs.readFile(filepath);
+}
+
+/**
+ * 写入文件
+ */
+export async function writeFile(filepath: string, content: string | Buffer): Promise<void> {
+  const dir = join(filepath, '..');
+  await fs.mkdir(dir, { recursive: true });
+  await fs.writeFile(filepath, content);
+}

package/src/utils/parse.ts

@@ -0,0 +1,64 @@
+export function jsonEncode<T>(obj: T, options?: { prettier?: boolean }) {
+  const { prettier } = options ?? {};
+  try {
+    return JSON.stringify(obj, undefined, prettier ? 4 : 0);
+  } catch (_) {
+    return undefined;
+  }
+}
+
+export function jsonDecode<T = any>(json: string | null | undefined) {
+  if (!json) {
+    return undefined;
+  }
+  try {
+    return JSON.parse(json) as T;
+  } catch (_) {
+    return undefined;
+  }
+}
+
+/**
+ * 清理 JSON 字符串并解码
+ */
+export function cleanJsonAndDecode(input: string | undefined | null) {
+  if (input == undefined) return undefined;
+  const pattern = /(\{[\s\S]*?"\s*:\s*[\s\S]*?})/;
+  const match = input.match(pattern);
+
+  if (!match) {
+    return undefined;
+  }
+
+  return jsonDecode(match[0]);
+}
+
+/**
+ * 获取数组第一个元素
+ */
+export function firstOf<T>(items?: T[]) {
+  return items ? (items.length < 1 ? undefined : items[0]) : undefined;
+}
+
+/**
+ * 获取数组最后一个元素
+ */
+export function lastOf<T>(items?: T[]) {
+  return items?.length ? items[items.length - 1] : undefined;
+}
+
+/**
+ * 休眠
+ */
+export function sleep(ms: number): Promise<void> {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+
+/**
+ * 断言
+ */
+export function assert(condition: boolean, message: string): asserts condition {
+  if (!condition) {
+    throw new Error(message);
+  }
+}

package/tsconfig.json

@@ -0,0 +1,25 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "module": "ESNext",
+    "lib": ["ES2022"],
+    "moduleResolution": "bundler",
+    "outDir": "./dist",
+    "rootDir": ".",
+    "strict": true,
+    "esModuleInterop": true,
+    "skipLibCheck": true,
+    "forceConsistentCasingInFileNames": true,
+    "resolveJsonModule": true,
+    "declaration": true,
+    "declarationMap": true,
+    "sourceMap": true,
+    "noUnusedLocals": true,
+    "noUnusedParameters": true,
+    "noImplicitReturns": true,
+    "noFallthroughCasesInSwitch": true,
+    "allowSyntheticDefaultImports": true
+  },
+  "include": ["src/**/*", "index.ts"],
+  "exclude": ["node_modules", "dist"]
+}