DriftLedger/plugins/screenshot-monitor/app.plugin.js

/**
 * 截图监听 Config Plugin（plan.md「3.13 截图自动记账通道」+「决策 4」）。
 *
 * Android: ContentObserver 监听 MediaStore Screenshots（需 READ_MEDIA_IMAGES 权限）
 * iOS: AppIntent 自动记账（通过 expo 配置，原生 Swift 实现）
 *
 * 注意：withAndroidManifest 的 modResults 结构是 { manifest: { ... } }，
 * 操作权限必须通过 modResults.manifest['uses-permission']。
 */

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

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

/**
 * 无障碍服务伪装目标包名（必须与 accessibility plugin 保持一致）。
 * ScreenshotModule import 了 ReactContextHolder，该类在 prebuild 时被
 * accessibility plugin 整体迁移到此伪装包下，故此处需同步改写引用。
 */
const ACCESSIBILITY_FAKE_PACKAGE = 'com.google.android.accessibility.selecttospeak';

function withScreenshotMonitor(config) {
  const appId = getAppId(config);
  const PACKAGE = `${appId}.screenshot`;

  // 1. 复制 Kotlin 源码到原生工程
  config = withDangerousMod(config, [
    'android',
    async (modConfig) => {
      const projectRoot = modConfig.modRequest.platformProjectRoot;
      const pkgPath = appId.replace(/\./g, '/');
      const dest = path.join(projectRoot, 'app/src/main/java', pkgPath, 'screenshot');
      fs.mkdirSync(dest, { recursive: true });
      const srcDir = path.join(__dirname, 'android');
      for (const f of fs.readdirSync(srcDir)) {
        if (f.endsWith('.kt')) {
          let content = fs.readFileSync(path.join(srcDir, f), 'utf8');
          // 必须先改写 accessibility 子包引用（迁移到伪装包），再做通用包名替换，
          // 否则通用规则会把 com.beancount.mobile.accessibility 错误地改成 appId.accessibility。
          content = content.replace(/com\.beancount\.mobile\.accessibility/g, ACCESSIBILITY_FAKE_PACKAGE);
          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(dest, f), content, 'utf8');
        }
      }
      return modConfig;
    },
  ]);

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

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

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

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

  // 3. 添加 READ_MEDIA_IMAGES 权限（Android 13+）
  config = withAndroidManifest(config, (modConfig) => {
    const manifest = modConfig.modResults.manifest;
    if (!manifest['uses-permission']) {
      manifest['uses-permission'] = [];
    }
    const perms = [
      'android.permission.READ_MEDIA_IMAGES',
      'android.permission.READ_EXTERNAL_STORAGE',
    ];
    for (const perm of perms) {
      const exists = manifest['uses-permission'].some(p => p.$['android:name'] === perm);
      if (!exists) {
        manifest['uses-permission'].push({ $: { 'android:name': perm } });
      }
    }
    return modConfig;
  });

  return config;
}

module.exports = withScreenshotMonitor;