From 5ebdf7974e49cde2d2e59e02d5b80193e2ac5a21 Mon Sep 17 00:00:00 2001
From: Khyber Sen <kkysen@gmail.com>
Date: Tue, 24 Sep 2024 02:39:57 -0700
Subject: [PATCH] rewriter: warn on multiple compile commands per src and
 truncate to first one

---
 tools/rewriter/SourceRewriter.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/tools/rewriter/SourceRewriter.cpp b/tools/rewriter/SourceRewriter.cpp
index 9804df0e2..d8914b07f 100644
--- a/tools/rewriter/SourceRewriter.cpp
+++ b/tools/rewriter/SourceRewriter.cpp
@@ -1029,7 +1029,14 @@ std::optional<Pkey> pkey_from_commands(std::function<std::optional<std::vector<C
     return {};
   }
 
+  if (comp_cmds.size() != 1) {
+    llvm::errs() << "warning: multiple compile commands for a single source file" << '\n'
+                 << "         truncating to first compile command" << '\n'
+                 << "         may not work properly" << '\n';
+    comp_cmds.resize(1);
+  }
   assert(comp_cmds.size() == 1);
+  
   auto cc_cmd = *comp_cmd_with_pkey;
 
   auto pkey_define = std::find_if(cc_cmd.CommandLine.begin(),