frontend/plugins/withVivoPush.js

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

// vivo 厂商推送通道的 Expo config plugin.
//
// 客户端职责（依据极光官方文档 https://docs.jiguang.cn/jpush/client/Android/android_3rd_guide）：
//   1) 添加 cn.jiguang.sdk.plugin:vivo 依赖
//   2) 注入 manifestPlaceholders: VIVO_APPKEY / VIVO_APPID（AAR 自带 vivo Receiver）
//   3) 添加 Proguard 规则 (com.vivo.push.** + com.vivo.vms.** 保留)
//
// 注意事项：
//   - 极光 JPUSH_CHANNEL 是「APK 分发渠道」统计字段，不是厂商通道开关，**不应覆盖**。
//   - 客户端仅需要 VIVO_APPKEY / VIVO_APPID 两个占位符。
//     VIVO_APP_SECRET 只在极光控制台「集成设置」页面填写，客户端不需要。
//   - vivo AAR 已发布到 Maven Central，不需要额外 Maven 仓库。
//   - vivo 平台限制：若应用未上架 vivo 商店，测试推送时需在 vivo 推送平台添加测试设备，
//     且需通过 API 指定 push_mode=1（测试推送）下发。
//
// 实现说明：用 withDangerousMod 直接读写磁盘文件。
// 原因：Expo SDK 56+ 的 withAppBuildGradle hook 链中，第二个以后的 plugin 写回
// modResults.contents 会被静默忽略，导致多个 vendor plugin 只能写入第一个的修改。
// withDangerousMod 走文件系统，串行、可靠，幂等检查保证可重复执行 prebuild。
const withVivoPush = (config, options = {}) => {
  const {
    jpushVersion = '6.1.0',
    appId = '106100295',
    appKey = '2110cb5f6dc5bd7abea82c38f8d0a1ec',
  } = 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:vivo')) {
          const depsPattern = /dependencies\s*\{/;
          if (depsPattern.test(gradle)) {
            gradle = gradle.replace(
              depsPattern,
              `dependencies {\n    implementation 'cn.jiguang.sdk.plugin:vivo:${jpushVersion}'`
            );
            changed = true;
          }
        }

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

        const requiredPlaceholders = {
          VIVO_APPKEY: appKey,
          VIVO_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('[withVivoPush] 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.vivo.push.**')) {
          const vivoRules = `
# JPush vivo vendor channel — keep rules (https://docs.jiguang.cn/jpush/client/Android/android_3rd_guide)
-dontwarn com.vivo.push.**
-keep class com.vivo.push.** { *; }
-keep class com.vivo.vms.** { *; }
`;
          fs.writeFileSync(proguardFilePath, proguard + vivoRules);
          console.log('[withVivoPush] appended vivo Proguard rules to proguard-rules.pro');
        }
      }

      return config;
    },
  ]);

  return config;
};

module.exports = withVivoPush;