From d6329ea328bc355cce688e5404c81b2e783a4f83 Mon Sep 17 00:00:00 2001
From: Fangrui Song <i@maskray.me>
Date: Sun, 13 Jan 2019 18:33:18 +0800
Subject: [PATCH] completion: if preamble size changes, rebuild it

Fix #190

If a new header is added, the preamble size changes. Language clients may cache completion results, thus we rebuild preamble to avoid inaccurate results.
---
 src/sema_manager.cc | 18 +++++++++++++-----
 src/sema_manager.hh |  9 +++++----
 2 files changed, 18 insertions(+), 9 deletions(-)

diff --git a/src/sema_manager.cc b/src/sema_manager.cc
index 91fae5e3..c352f0f4 100644
--- a/src/sema_manager.cc
+++ b/src/sema_manager.cc
@@ -423,7 +423,9 @@ void *PreambleMain(void *manager_) {
             BuildCompilerInvocation(task.path, session->file.args, FS))
       BuildPreamble(*session, *CI, FS, task, std::move(stat_cache));
 
-    if (task.from_diag) {
+    if (task.comp_task) {
+      manager->comp_tasks.PushBack(std::move(task.comp_task));
+    } else if (task.from_diag) {
       manager->ScheduleDiag(task.path, 0);
     } else {
       int debounce =
@@ -474,12 +476,18 @@ void *CompletionMain(void *manager_) {
     DiagnosticConsumer DC;
     std::string content = manager->wfiles->GetContent(task->path);
     auto Buf = llvm::MemoryBuffer::getMemBuffer(content);
+    PreambleBounds Bounds =
+        ComputePreambleBounds(*CI->getLangOpts(), Buf.get(), 0);
     bool in_preamble =
         GetOffsetForPosition({task->position.line, task->position.character},
-                             content) <
-        ComputePreambleBounds(*CI->getLangOpts(), Buf.get(), 0).Size;
-    if (in_preamble)
+                             content) < (int)Bounds.Size;
+    if (in_preamble) {
       preamble.reset();
+    } else if (preamble && Bounds.Size != preamble->Preamble.getBounds().Size) {
+      manager->preamble_tasks.PushBack({task->path, std::move(task), false},
+                                       true);
+      continue;
+    }
     auto Clang = BuildCompilerInstance(*session, std::move(CI), FS, DC,
                                        preamble.get(), task->path, Buf);
     if (!Clang)
@@ -556,7 +564,7 @@ void *DiagnosticMain(void *manager_) {
           }
       }
       if (rebuild) {
-        manager->preamble_tasks.PushBack({task.path, true}, true);
+        manager->preamble_tasks.PushBack({task.path, nullptr, true}, true);
         continue;
       }
     }
diff --git a/src/sema_manager.hh b/src/sema_manager.hh
index 5d573349..0e1584a1 100644
--- a/src/sema_manager.hh
+++ b/src/sema_manager.hh
@@ -114,10 +114,6 @@ struct SemaManager {
       std::function<void(clang::CodeCompleteConsumer *OptConsumer)>;
   using OnDropped = std::function<void(RequestId request_id)>;
 
-  struct PreambleTask {
-    std::string path;
-    bool from_diag = false;
-  };
   struct CompTask {
     CompTask(const RequestId &id, const std::string &path,
              const Position &position,
@@ -138,6 +134,11 @@ struct SemaManager {
     int64_t wait_until;
     int64_t debounce;
   };
+  struct PreambleTask {
+    std::string path;
+    std::unique_ptr<CompTask> comp_task;
+    bool from_diag = false;
+  };
 
   SemaManager(Project *project, WorkingFiles *wfiles,
                     OnDiagnostic on_diagnostic, OnDropped on_dropped);