From bc71b31be28ed558e022babcf40ec1fd3bff923a Mon Sep 17 00:00:00 2001
From: Simon Ogorodnik <Simon.Ogorodnik@jetbrains.com>
Date: Mon, 27 Feb 2017 20:08:31 +0300
Subject: Platform and version requirements for packages. Fixed tests.

Post review.
---
 .../main/kotlin/Formats/StructuredFormatService.kt   | 20 +++++++++++++++-----
 core/src/test/kotlin/format/MarkdownFormatTest.kt    |  8 ++++++--
 2 files changed, 21 insertions(+), 7 deletions(-)

(limited to 'core/src')

diff --git a/core/src/main/kotlin/Formats/StructuredFormatService.kt b/core/src/main/kotlin/Formats/StructuredFormatService.kt
index 79ebabe0..a7b696eb 100644
--- a/core/src/main/kotlin/Formats/StructuredFormatService.kt
+++ b/core/src/main/kotlin/Formats/StructuredFormatService.kt
@@ -263,6 +263,7 @@ abstract class StructuredOutputBuilder(val to: StringBuilder,
                     val packageName = if (singleNode.name.isEmpty()) "<root>" else singleNode.name
                     appendHeader(2) { appendText("Package $packageName") }
                 }
+                singleNode.appendPlatforms()
                 appendContent(singleNode.content)
             } else {
                 val breakdownByName = nodes.groupBy { node -> node.name }
@@ -354,13 +355,25 @@ abstract class StructuredOutputBuilder(val to: StringBuilder,
         }
 
         private fun DocumentationNode.appendPlatforms() {
-            val platforms = platformsToShow.ifEmpty { return }
+            val platforms = if (isModuleOrPackage())
+                platformsToShow.toSet() + platformsOfItems(members)
+            else
+                platformsToShow
+
+            if(platforms.isEmpty()) return
+
             appendParagraph {
                 appendStrong { to.append("Platform and version requirements:") }
                 to.append(" " + platforms.joinToString())
             }
         }
 
+        protected fun platformsOfItems(items: List<DocumentationNode>): Set<String> =
+                items.foldRight(items.first().platformsToShow.toSet()) {
+                    node, platforms ->
+                    platforms.intersect(node.platformsToShow)
+                }
+
         val DocumentationNode.platformsToShow: List<String>
             get() = platforms.let { if (it.containsAll(impliedPlatforms)) it - impliedPlatforms else it }
 
@@ -501,10 +514,7 @@ abstract class StructuredOutputBuilder(val to: StringBuilder,
         }
 
         private fun platformsOfItems(items: List<DocumentationNode>, omitSamePlatforms: Boolean = true): Set<String> {
-            val platforms = items.foldRight(items.first().platformsToShow.toSet()) {
-                node, platforms ->
-                platforms.intersect(node.platformsToShow)
-            }
+            val platforms = platformsOfItems(items)
             if (platforms.isNotEmpty() && (platforms != node.platformsToShow.toSet() || !omitSamePlatforms)) {
                 return platforms
             }
diff --git a/core/src/test/kotlin/format/MarkdownFormatTest.kt b/core/src/test/kotlin/format/MarkdownFormatTest.kt
index e87811ee..d21528b5 100644
--- a/core/src/test/kotlin/format/MarkdownFormatTest.kt
+++ b/core/src/test/kotlin/format/MarkdownFormatTest.kt
@@ -284,15 +284,19 @@ class MarkdownFormatTest {
         }
     }
 
-    @Test fun multiplePlatformsPackagePlatformFromMembers() {
+    @Test fun multiplePlatformsPackagePlatformFromMembersIndex() {
         val module = buildMultiplePlatforms("multiplatform/packagePlatformsFromMembers")
         verifyModelOutput(module, ".md", "testdata/format/multiplatform/packagePlatformsFromMembers/multiplatform.index.kt") {
             model, output ->
             MarkdownFormatService(InMemoryLocationService, KotlinLanguageService(), listOf())
-                    .createOutputBuilder(output, tempLocation).appendNodes(model.members)
+                    .createOutputBuilder(output, tempLocation).appendNodes(listOf(model))
         }
     }
 
+    @Test fun multiplePlatformsPackagePlatformFromMembers() {
+        verifyMultiplatformPackage(buildMultiplePlatforms("multiplatform/packagePlatformsFromMembers"), "multiplatform/packagePlatformsFromMembers")
+    }
+
     private fun buildMultiplePlatforms(path: String): DocumentationModule {
         val module = DocumentationModule("test")
         val options = DocumentationOptions("", "html", generateIndexPages = false)
-- 
cgit