frontend/plugins/withXiaomiPush.js

const { withDangerousMod } = require('@expo/config-plugins');
const fs = require('fs');
const path = require('path');

// 小米厂商推送通道的 Expo config plugin.
//
// 客户端职责（依据极光官方文档 https://docs.jiguang.cn/jpush/client/Android/android_3rd_guide）：
//   1) 添加 cn.jiguang.sdk.plugin:xiaomi 依赖
//   2) 注入 manifestPlaceholders: XIAOMI_APPKEY / XIAOMI_APPID
//   3) 添加 Proguard 规则 (com.xiaomi.push.** 保留)
//
// 实现说明：用 withDangerousMod 直接读写磁盘文件。
// 原因：Expo SDK 56+ 的 withAppBuildGradle hook 链中，第二个以后的 plugin 写回
// modResults.contents 会被静默忽略（基于 withAndroidProjectBuildGradleBaseMod），
// 导致多个 vendor plugin 只能写入第一个的修改。withDangerousMod 走文件系统，
// 串行、可靠，幂等检查保证可重复执行 prebuild。
const withXiaomiPush = (config, options = {}) => {
  const {
    jpushVersion = '5.8.0',
    appId = '2882303761520539252',
    appKey = '5792053978252',
  } = options;

  config = withDangerousMod(config, [
    'android',
    async (config) => {
      const appBuildGradlePath = path.join(
        config.modRequest.platformProjectRoot,
        'app',
        'build.gradle'
      );
      const proguardFilePath = path.join(
        config.modRequest.platformProjectRoot,
        'app',
        'proguard-rules.pro'
      );

      // --- 1. 处理 build.gradle ---
      if (fs.existsSync(appBuildGradlePath)) {
        let gradle = fs.readFileSync(appBuildGradlePath, 'utf-8');
        let changed = false;

        // 1a. 注入 AAR 依赖
        if (!gradle.includes('cn.jiguang.sdk.plugin:xiaomi')) {
          const depsPattern = /dependencies\s*\{/;
          if (depsPattern.test(gradle)) {
            gradle = gradle.replace(
              depsPattern,
              `dependencies {\n    implementation 'cn.jiguang.sdk.plugin:xiaomi:${jpushVersion}'`
            );
            changed = true;
          }
        }

        // 1b. 注入 manifestPlaceholders
        const manifestPlaceholderRegex = /manifestPlaceholders\s*=\s*\[([\s\S]*?)\]/;
        const match = gradle.match(manifestPlaceholderRegex);

        const requiredPlaceholders = {
          XIAOMI_APPKEY: appKey,
          XIAOMI_APPID: appId,
        };

        if (match) {
          const existingBlock = match[1];
          const existingEntries = {};
          const entryRegex = /(\w+)\s*:\s*"([^"]*)"/g;
          let entryMatch;
          while ((entryMatch = entryRegex.exec(existingBlock)) !== null) {
            existingEntries[entryMatch[1]] = entryMatch[2];
          }

          let placeholdersChanged = false;
          for (const [key, value] of Object.entries(requiredPlaceholders)) {
            if (existingEntries[key] !== value) {
              existingEntries[key] = value;
              placeholdersChanged = true;
            }
          }

          if (placeholdersChanged) {
            const entriesStr = Object.entries(existingEntries)
              .map(([k, v]) => `${k}: "${v}"`)
              .join(',\n            ');
            const newBlock = `\n        manifestPlaceholders = [\n            ${entriesStr}\n        ]`;
            gradle = gradle.replace(manifestPlaceholderRegex, newBlock.trim());
            changed = true;
          }
        } else if (gradle.includes('REACT_NATIVE_RELEASE_LEVEL')) {
          // 没有 manifestPlaceholders block 时新建
          const entriesStr = Object.entries(requiredPlaceholders)
            .map(([k, v]) => `${k}: "${v}"`)
            .join(',\n            ');
          const block = `\n        manifestPlaceholders = [\n            ${entriesStr}\n        ]\n`;
          const lines = gradle.split('\n');
          for (let i = 0; i < lines.length; i++) {
            if (lines[i].includes('REACT_NATIVE_RELEASE_LEVEL')) {
              lines.splice(i + 1, 0, block);
              break;
            }
          }
          gradle = lines.join('\n');
          changed = true;
        }

        if (changed) {
          fs.writeFileSync(appBuildGradlePath, gradle);
          console.log('[withXiaomiPush] updated app/build.gradle (AAR + manifestPlaceholders)');
        }
      }

      // --- 2. 处理 proguard-rules.pro ---
      if (fs.existsSync(proguardFilePath)) {
        let proguard = fs.readFileSync(proguardFilePath, 'utf-8');
        if (!proguard.includes('-keep class com.xiaomi.push.**')) {
          const xiaomiRules = `
# JPush Xiaomi vendor channel — keep rules (https://docs.jiguang.cn/jpush/client/Android/android_3rd_guide)
-dontwarn com.xiaomi.push.**
-keep class com.xiaomi.push.** { *; }
`;
          fs.writeFileSync(proguardFilePath, proguard + xiaomiRules);
          console.log('[withXiaomiPush] appended Xiaomi Proguard rules to proguard-rules.pro');
        }
      }

      return config;
    },
  ]);

  return config;
};

module.exports = withXiaomiPush;