pending references logic moved from DocumentationBuilder to model

1 files changed, 4 insertions, 2 deletions

diff --git a/src/main.kt b/src/main.kt
index 44c435d2..535eee5c 100644
--- a/src/main.kt
+++ b/src/main.kt
@@ -195,7 +195,8 @@ fun buildDocumentationModule(environment: AnalysisEnvironment,
         val fragments = fragmentFiles.map { session.getPackageFragment(it.getPackageFqName()) }.filterNotNull().distinct()
 
         val moduleContent = Content()
-        val documentationBuilder = DocumentationBuilder(session, options, logger)
+        val pendingReferences = arrayListOf<PendingDocumentationReference>()
+        val documentationBuilder = DocumentationBuilder(session, options, pendingReferences, logger)
         for (include in includes) {
             val file = File(include)
             if (file.exists()) {
@@ -211,13 +212,14 @@ fun buildDocumentationModule(environment: AnalysisEnvironment,
 
         with(documentationBuilder) {
             documentationModule.appendFragments(fragments)
-            documentationBuilder.resolveReferences()
         }
 
         val javaFiles = environment.getJavaSourceFiles().filter(filesToDocumentFilter)
         val javaDocumentationBuilder = JavaDocumentationBuilder(options)
         javaFiles.map { javaDocumentationBuilder.appendFile(it, documentationModule) }
 
+        pendingReferences.forEach { it.resolve() }
+
         documentationModule
     }