DriftLedger/plugins/accessibility/app.plugin.js

/**
 * 无障碍服务 Config Plugin（plan.md「3.6 无障碍服务」+「决策 4」）。
 *
 * 在 expo prebuild 时注册 Android 无障碍服务（manifest service + xml 配置）。
 * 参考 AutoAccounting 的 SelectToSpeakService 伪装机制（侧载保留，plan.md 决策 3）。
 *
 * 注意：withAndroidManifest 的 modResults 结构是 { manifest: { ... } }，
 * 操作 application 必须通过 modResults.manifest.application。
 */

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

/**
 * 从 src/domain/constants.ts 的 PAYMENT_PACKAGES 中提取包名列表。
 * 这是唯一的真相来源，Kotlin 端通过此函数自动同步。
 */
function readPaymentPackagesFromConstants() {
  const constantsPath = path.resolve(__dirname, '../../src/domain/constants.ts');
  if (!fs.existsSync(constantsPath)) return null;
  const content = fs.readFileSync(constantsPath, 'utf8');
  // 匹配 PAYMENT_PACKAGES = { ... } 中的 key（冒号前的字符串）
  const match = content.match(/PAYMENT_PACKAGES:\s*Record<string,\s*string>\s*=\s*\{([\s\S]*?)\n\};/);
  if (!match) return null;
  const keys = [];
  const keyRegex = /^(\s*)'([^']+)'\s*:/gm;
  let m;
  while ((m = keyRegex.exec(match[1])) !== null) {
    keys.push(m[2]);
  }
  return keys.length > 0 ? keys : null;
}

/**
 * 生成 Kotlin setOf("...", "...", ...) 代码。
 */
function generateKotlinPaymentPackages(packages) {
  const items = packages.map(pkg => `            "${pkg}"`).join(',\n');
  return `        val PAYMENT_PACKAGES = setOf(\n${items}\n        )`;
}

/** 递归复制目录。 */
function copyDir(src, dest) {
  if (!fs.existsSync(src)) return;
  fs.mkdirSync(dest, { recursive: true });
  for (const entry of fs.readdirSync(src)) {
    const s = path.join(src, entry);
    const d = path.join(dest, entry);
    if (fs.statSync(s).isDirectory()) copyDir(s, d);
    else fs.copyFileSync(s, d);
  }
}

function getAppId(config) {
  return config.android?.package || 'com.example.driftledger';
}

function withAccessibilityService(config) {
  const appId = getAppId(config);
  const PACKAGE = `${appId}.accessibility`;

  // 1. 复制 Kotlin 源码 + res/xml 资源到原生工程
  config = withDangerousMod(config, [
    'android',
    async (modConfig) => {
      const projectRoot = modConfig.modRequest.platformProjectRoot;
      // Kotlin 源码（只复制 .kt 文件，不含 res/ 子目录）
      const pkgPath = appId.replace(/\./g, '/');
      const ktDest = path.join(projectRoot, 'app/src/main/java', pkgPath, 'accessibility');
      fs.mkdirSync(ktDest, { recursive: true });
      const androidDir = path.join(__dirname, 'android');
      for (const f of fs.readdirSync(androidDir)) {
        if (f.endsWith('.kt')) {
          let content = fs.readFileSync(path.join(androidDir, f), 'utf8');
          content = content.replace(/package\s+com\.beancount\.mobile/g, `package ${appId}`);
          content = content.replace(/import\s+com\.beancount\.mobile/g, `import ${appId}`);
          fs.writeFileSync(path.join(ktDest, f), content, 'utf8');
        }
      }
      // res/xml 资源
      const resSrc = path.join(__dirname, 'android/res');
      if (fs.existsSync(resSrc)) {
        copyDir(resSrc, path.join(projectRoot, 'app/src/main/res'));
      }

      // 从 constants.ts 读取包名列表，自动同步到 Kotlin
      const packages = readPaymentPackagesFromConstants();
      if (packages) {
        const ktFile = path.join(ktDest, 'BillingAccessibilityService.kt');
        if (fs.existsSync(ktFile)) {
          let ktContent = fs.readFileSync(ktFile, 'utf8');
          // 替换 PAYMENT_PACKAGES 定义（匹配 val PAYMENT_PACKAGES = setOf( ... )）
          const regex = /val PAYMENT_PACKAGES = setOf\([\s\S]*?\)/;
          if (regex.test(ktContent)) {
            ktContent = ktContent.replace(regex, generateKotlinPaymentPackages(packages));
            fs.writeFileSync(ktFile, ktContent, 'utf8');
            console.log(`[accessibility plugin] 已同步 ${packages.length} 个支付 App 包名到 Kotlin`);
          }
        }
      }

      // 确保有 accessibility_service_description 字符串资源
      const stringsXmlPath = path.join(projectRoot, 'app/src/main/res/values/strings.xml');
      if (fs.existsSync(stringsXmlPath)) {
        let content = fs.readFileSync(stringsXmlPath, 'utf8');
        if (!content.includes('accessibility_service_description')) {
          content = content.replace(
            /<\/resources>/,
            '  <string name="accessibility_service_description">自动识别支付账单页面，辅助快速记账</string>\n</resources>',
          );
          fs.writeFileSync(stringsXmlPath, content, 'utf8');
        }
      }
      return modConfig;
    },
  ]);

  // 2. 注册 AccessibilityBridgePackage 到 MainApplication
  config = withMainApplication(config, (modConfig) => {
    let content = modConfig.modResults.contents;

    // 2a. 注入 import（AccessibilityBridgePackage）
    content = content.replace(/^import\s+[\w.]+\.AccessibilityBridgePackage\s*$/gm, '');
    content = content.replace(
      /^(package\s+[\w.]+;?\s*)$/m,
      `$1\nimport ${PACKAGE}.AccessibilityBridgePackage`,
    );

    // 2c. 在 getPackages() 的 .apply {} 块里注入 add(AccessibilityBridgePackage())
    if (!content.includes('add(AccessibilityBridgePackage())')) {
      if (/PackageList\(this\)\.packages\.apply\s*\{/.test(content)) {
        content = content.replace(
          /(PackageList\(this\)\.packages\.apply\s*\{)/,
          `$1\n              add(AccessibilityBridgePackage())`,
        );
      } else if (/PackageList\(this\)\.packages\b/.test(content)) {
        content = content.replace(
          /PackageList\(this\)\.packages\b/,
          `PackageList(this).packages.apply { add(AccessibilityBridgePackage()) }`,
        );
      }
    }

    modConfig.modResults.contents = content;
    return modConfig;
  });

  // 3. 注册权限 + 服务到 AndroidManifest
  config = withAndroidManifest(config, (modConfig) => {
    const manifest = modConfig.modResults.manifest;

    // 0. 添加 SYSTEM_ALERT_WINDOW 权限（悬浮窗必需）
    if (!manifest['uses-permission']) {
      manifest['uses-permission'] = [];
    }
    const overlayPerm = 'android.permission.SYSTEM_ALERT_WINDOW';
    const overlayPermExists = manifest['uses-permission'].some(p => p.$['android:name'] === overlayPerm);
    if (!overlayPermExists) {
      manifest['uses-permission'].push({ $: { 'android:name': overlayPerm } });
    }

    // 1. 添加无障碍服务声明
    const serviceNode = {
      $: {
        'android:name': `${PACKAGE}.BillingAccessibilityService`,
        'android:permission': 'android.permission.BIND_ACCESSIBILITY_SERVICE',
        'android:label': '浮记-账单识别',
        'android:exported': 'false',
      },
      'intent-filter': [{
        action: [{ $: { 'android:name': 'android.accessibilityservice.AccessibilityService' } }],
      }],
      'meta-data': [{
        $: {
          'android:name': 'android.accessibilityservice',
          'android:resource': '@xml/accessibility_service_config',
        },
      }],
    };

    // 2. 确保 application[0] 存在
    if (!Array.isArray(manifest.application) || manifest.application.length === 0) {
      manifest.application = [{ $: {} }];
    }
    const app = manifest.application[0];
    if (!app.service) {
      app.service = [];
    }
    const exists = app.service.some(
      s => s.$['android:name'] === `${PACKAGE}.BillingAccessibilityService`
    );
    if (!exists) {
      app.service.push(serviceNode);
    }

    // 3. 添加 OcrTileService 声明（快速设置磁贴，plan.md「3.11」）
    const tileServiceNode = {
      $: {
        'android:name': `${PACKAGE}.OcrTileService`,
        'android:label': 'OCR 记账',
        'android:icon': '@android:drawable/ic_menu_camera',
        'android:permission': 'android.permission.BIND_QUICK_SETTINGS_TILE',
        'android:exported': 'true',
      },
      'intent-filter': [{
        action: [{ $: { 'android:name': 'android.service.quicksettings.action.QS_TILE' } }],
      }],
    };
    const tileExists = app.service.some(
      s => s.$['android:name'] === `${PACKAGE}.OcrTileService`
    );
    if (!tileExists) {
      app.service.push(tileServiceNode);
    }

    return modConfig;
  });

  return config;
}

module.exports = withAccessibilityService;