如何用Clang Rewriter删除含预处理条件的单语句包裹IfStmt？

这个问题的核心痛点在于Clang的AST是预处理后的产物，但Rewriter操作的是原始源文件的字符——你拿到的IfStmt::getSourceRange()只覆盖了if( someCondition )这部分语法节点，而后面嵌套的#if/#else/#endif属于预处理指令，不在AST节点的范围里，删除IfStmt后自然留下了孤立的预处理块，导致语法错误。

下面是几个通用的解决思路，按实用性排序：

1. 合并IfStmt与关联预处理指令的范围后再删除 #

这是最直接的解决方案，核心是找到IfStmt对应的完整预处理块范围，把两者合并后一次性删除：

• 步骤1：跟踪预处理指令的位置与嵌套层级
  借助Clang的Preprocessor和自定义PPCallbacks，在预处理阶段记录所有#if、#else、#endif的源位置和嵌套层级（比如遇到#if计数器+1，#endif计数器-1）。
• 步骤2：匹配IfStmt对应的预处理块
  当遍历AST找到目标IfStmt后，从它的结束位置向后扫描，找到第一个#if的起始点，然后继续扫描直到计数器回到0（即匹配的#endif），把整个从if(...)到#endif的区间作为完整的删除范围。
• 步骤3：执行删除
  用Rewriter的RemoveText()方法删除这个合并后的范围，确保不会残留任何预处理指令。

举个实际操作的代码片段思路（伪代码）：

// 自定义PPCallbacks记录预处理指令
class PPRecorder : public PPCallbacks {
public:
  std::vector<std::pair<SourceLocation, int>> ppDirectives; // 位置+层级
  int currentLevel = 0;

  void If(SourceLocation Loc, SourceRange ConditionRange, bool IsDefined) override {
    ppDirectives.emplace_back(Loc, currentLevel++);
  }
  void Endif(SourceLocation Loc) override {
    ppDirectives.emplace_back(Loc, --currentLevel);
  }
  // 还需处理#elif、#else等指令的记录
};

// 在AST遍历中找到目标IfStmt后
SourceRange ifRange = ifStmt->getSourceRange();
SourceLocation start = ifRange.getBegin();
SourceLocation end = ifRange.getEnd();

// 向后查找对应的#if和#endif
int level = 0;
SourceLocation ppEnd;
for (auto& dir : ppDirectives) {
  if (dir.first.isAfter(end) && dir.second == 1) { // 找到IfStmt后的第一个#if
    level = 1;
  }
  if (level > 0 && dir.second == 0) { // 找到匹配的#endif
    // 定位到#endif指令的末尾
    ppEnd = dir.first.getLocWithOffset(strlen("#endif"));
    break;
  }
}

// 合并范围并删除
SourceRange fullRange(start, ppEnd);
rewriter.RemoveText(fullRange);

2. 先预处理展开，再操作AST #

如果你的场景允许丢失原始预处理结构，可以先对源文件进行预处理展开（用Clang的-E选项），得到没有预处理指令的纯代码，再对这个展开后的文件进行AST分析和Rewriter修改。

这种方法的优点是简单，不需要处理预处理指令的嵌套问题；缺点是修改后的代码会丢失原始的#if逻辑，只保留当前编译配置下展开后的代码，适合不需要保留预处理分支的场景。

3. 替换IfStmt为预处理展开后的分支代码 #

如果想保留预处理后的逻辑（而不是删除整个块），可以先通过Clang的Preprocessor获取当前编译配置下#if SOMEOTHERCONDITION的展开结果，然后把IfStmt替换成对应的分支代码（比如doSth1();或doSth2();），同时删除掉预处理指令。

这个方法需要你提前知道编译配置（比如定义了哪些宏），适合需要保留有效代码、只删除IfStmt判断的场景。

内容的提问来源于stack exchange，提问作者Daniel