[AST] Clear module lookup cache in SourceFile::addTopLevelDecl

Completion unfortunately relies on this to add a top-level decl
after the lookup table has been built, make sure the decl gets added
to it. We ought to fix completion to avoid needing to mutate the AST
like this after-the-fact, but this should work around the issue for now.

Author: hamishknight

include/swift/AST/Module.h

@@ -920,7 +920,7 @@ class ModuleDecl
   ///
   /// FIXME: Refactor main file parsing to not pump the parser incrementally.
   /// FIXME: Remove the integrated REPL.
-  void clearLookupCache();
+  void clearLookupCache(bool includingImports = true);
 
   /// Finds all class members defined in this module.
   ///

lib/AST/Module.cpp

@@ -3239,8 +3239,9 @@ bool SourceFile::shouldCrossImport() const {
          getASTContext().LangOpts.EnableCrossImportOverlays;
 }
 
-void ModuleDecl::clearLookupCache() {
-  getASTContext().getImportCache().clear();
+void ModuleDecl::clearLookupCache(bool includingImports) {
+  if (includingImports)
+    getASTContext().getImportCache().clear();
 
   setIsObjCNameLookupCachePopulated(false);
   ObjCNameLookupCache.clear();
@@ -3495,6 +3496,8 @@ void SourceFile::addTopLevelDecl(Decl *d) {
   auto &ctx = getASTContext();
   auto *mutableThis = const_cast<SourceFile *>(this);
   ctx.evaluator.clearCachedOutput(ParseTopLevelDeclsRequest{mutableThis});
+
+  getParentModule()->clearLookupCache(/*includingImports*/isa<ImportDecl>(d));
 }
 
 void SourceFile::prependTopLevelDecl(Decl *d) {

test/IDE/pr83369.swift

@@ -0,0 +1,9 @@
+// RUN: %batch-code-completion -module-name main
+
+extension Int {}
+
+// Make sure we can resolve P here.
+protocol P<X> where X: main.#^COMPLETE^# {
+  associatedtype X
+}
+// COMPLETE: Decl[Protocol]/CurrModule: P[#P#]; name=P