feat: 开放小爱音箱补丁固件构建源代码和教程

2025-04-12 23:49:07 +08:00
parent 0585ddab72
commit 2cf6623db9
21 changed files with 1794 additions and 0 deletions
@@ -0,0 +1,261 @@
+import { getMiNA } from "@mi-gpt/miot";
+import { createHash } from "node:crypto";
+import * as fs from "node:fs";
+import * as path from "node:path";
+import * as stream from "node:stream";
+import { promisify } from "node:util";
+
+const kSupportedDevices = [
+  "LX06", // 小爱音箱 Pro
+  "OH2P", // Xiaomi 智能音箱 Pro
+];
+
+// 获取 OTA 信息
+async function getOTA(channel: "release" | "current" | "stable" = "release") {
+  const MiNA = await getMiNA({
+    userId: process.env.MI_USER!,
+    password: process.env.MI_PASS!,
+  });
+  if (!MiNA) {
+    console.log(`❌ 登录失败，请检查小米账号密码是否正确：`);
+    console.log(`当前账号：${process.env.MI_USER}`);
+    console.log(`当前密码：${process.env.MI_PASS}`);
+    return;
+  }
+
+  const devices = await MiNA.getDevices();
+  const speaker = devices?.find(
+    (e: any) =>
+      e.name === process.env.MI_DID || e.miotDID === process.env.MI_DID
+  );
+  if (!speaker) {
+    console.log(`❌ 找不到设备: ${process.env.MI_DID}`);
+    console.log(`可用设备列表：`);
+    console.log(
+      JSON.stringify(
+        devices.map((e: any) => ({
+          name: e.name,
+          did: e.miotDID,
+        })),
+        null,
+        4
+      )
+    );
+    return;
+  }
+
+  if (!kSupportedDevices.includes(speaker.hardware)) {
+    console.log(
+      `❌ 暂不支持当前设备型号: ${speaker.hardware}（${speaker.name}）`
+    );
+    console.log(`可用设备列表：`);
+    console.log(
+      JSON.stringify(
+        devices
+          .filter((e: any) => kSupportedDevices.includes(e.hardware))
+          .map((e: any) => ({
+            name: e.name,
+            did: e.miotDID,
+          })),
+        null,
+        4
+      )
+    );
+    return;
+  }
+
+  const sn = speaker.serialNumber;
+  const model = speaker.hardware;
+  const version = speaker.romVersion;
+  const time = new Date().getTime();
+  const otaInfo = `channel=${channel}&filterID=${sn}&locale=zh_CN&model=${model}&time=${time}&version=${version}&8007236f-a2d6-4847-ac83-c49395ad6d65`;
+  const base64Str = Buffer.from(otaInfo).toString("base64");
+  const code = createHash("md5").update(base64Str).digest("hex");
+
+  return {
+    sn,
+    model,
+    version,
+    url: `http://api.miwifi.com/rs/grayupgrade/v2/${model}?model=${model}&version=${version}&channel=${channel}&filterID=${sn}&locale=zh_CN&time=${time}&s=${code}`,
+  };
+}
+
+async function main() {
+  console.log(`🔥 正在获取设备信息...`);
+  const ota = await getOTA();
+  if (!ota) {
+    console.log(`❌ 获取设备信息失败`);
+    process.exit(1);
+  }
+  console.log(`🔥 正在获取 OTA 信息...`);
+  const res = await fetch(ota.url);
+  const data = await res.json();
+  if (data.code === "0" && data.data) {
+    if (data.data.currentInfo) {
+      console.log("\n=== 当前版本固件 ===\n");
+      const filePath = await downloadFirmware(data.data.currentInfo);
+      if (filePath) {
+        fs.writeFileSync(
+          path.join(process.cwd(), "assets", ".model"),
+          ota.model
+        );
+        fs.writeFileSync(
+          path.join(process.cwd(), "assets", ".version"),
+          ota.version
+        );
+      }
+    }
+  } else {
+    console.log(`❌ 获取固件信息失败: ${data.code || "未知错误"}`);
+    process.exit(1);
+  }
+}
+
+main();
+
+// 下载固件
+async function downloadFirmware(firmware: {
+  link: string;
+  hash: string;
+  toVersion?: string;
+  size?: number;
+}): Promise<string | undefined> {
+  if (!firmware || !firmware.link) {
+    console.log("❌ 无效的固件信息");
+    return;
+  }
+
+  const assetsDir = path.join(process.cwd(), "assets");
+
+  // 从链接中提取文件名
+  const url = new URL(firmware.link);
+  const filename = path.basename(url.pathname);
+
+  // 打印固件信息
+  console.log(`- 版本: ${firmware.toVersion || "未知"}`);
+  console.log(
+    `- 大小: ${
+      firmware.size ? (firmware.size / 1024 / 1024).toFixed(2) + "MB" : "未知"
+    }`
+  );
+  console.log(`- 文件: ${filename}`);
+  console.log(`- MD5: ${firmware.hash || "未知"}\n`);
+
+  try {
+    const filePath = await downloadFile(firmware.link, assetsDir, filename);
+    return filePath;
+  } catch (error) {
+    console.error(`❌ 下载固件失败: ${error}`);
+    return;
+  }
+}
+
+async function ensureDir(dirPath: string): Promise<void> {
+  try {
+    await fs.promises.access(dirPath);
+  } catch (error) {
+    await fs.promises.mkdir(dirPath, { recursive: true });
+  }
+}
+
+async function downloadFile(
+  url: string,
+  destDir: string,
+  filename: string
+): Promise<string> {
+  await ensureDir(destDir);
+
+  const destPath = path.join(destDir, filename);
+
+  // 检查文件是否已存在
+  try {
+    await fs.promises.access(destPath);
+    console.log(`ℹ️ 文件已存在: ${destPath}`);
+    return destPath;
+  } catch (error) {
+    // 文件不存在，下载它
+    console.log(`⬇️ 开始下载: ${url}`);
+
+    const response = await fetch(url);
+    if (!response.ok) {
+      throw new Error(`下载失败: ${response.status} ${response.statusText}`);
+    }
+
+    // 获取文件大小
+    const contentLength = response.headers.get("content-length");
+    const totalSize = contentLength ? parseInt(contentLength, 10) : 0;
+
+    // 创建文件写入流
+    const fileStream = fs.createWriteStream(destPath);
+
+    if (!response.body) {
+      throw new Error("下载失败: 无法获取响应体");
+    }
+
+    // 创建进度显示变量
+    let downloadedBytes = 0;
+    let lastLoggedPercent = -1;
+    let startTime = Date.now();
+
+    // 使用 TransformStream 来跟踪下载进度
+    const progressStream = new stream.Transform({
+      transform(chunk, encoding, callback) {
+        downloadedBytes += chunk.length;
+
+        if (totalSize) {
+          const percent = Math.floor((downloadedBytes / totalSize) * 100);
+
+          // 确保每增加1%才打印一次，避免日志过多
+          if (percent > lastLoggedPercent) {
+            lastLoggedPercent = percent;
+
+            // 格式化输出
+            const downloaded = (downloadedBytes / 1024 / 1024).toFixed(2);
+            const total = (totalSize / 1024 / 1024).toFixed(2);
+
+            // 使用\r使光标回到行首，实现同行更新
+            process.stdout.write(
+              `\r下载进度: ${percent}% | ${downloaded}MB/${total}MB`
+            );
+
+            // 如果下载完成，打印换行符
+            if (downloadedBytes === totalSize) {
+              process.stdout.write("\n");
+            }
+          }
+        } else {
+          // 无法获取总大小时，只显示已下载大小
+          if (downloadedBytes % (1024 * 1024) === 0) {
+            // 每1MB打印一次
+            const downloaded = (downloadedBytes / 1024 / 1024).toFixed(2);
+            process.stdout.write(`\r已下载: ${downloaded}MB`);
+          }
+        }
+
+        this.push(chunk);
+        callback();
+      },
+    });
+
+    // 完成后清理并打印最终结果
+    progressStream.on("end", () => {
+      const totalTime = ((Date.now() - startTime) / 1000).toFixed(2);
+      console.log(`\n✅ 下载完成: ${destPath}`);
+      console.log(
+        `   总大小: ${(downloadedBytes / 1024 / 1024).toFixed(
+          2
+        )}MB, 用时: ${totalTime}秒`
+      );
+    });
+
+    // 使用pipeline连接流
+    await promisify(stream.pipeline)(
+      // @ts-ignore - response.body 在 Node.js 中是 ReadableStream
+      response.body,
+      progressStream,
+      fileStream
+    );
+
+    return destPath;
+  }
+}