From c097f326fbe4cfe7d6480799b693498d594f8ea8 Mon Sep 17 00:00:00 2001
From: Ilya Gorbunov <ilya.gorbunov@jetbrains.com>
Date: Sat, 3 Dec 2016 08:13:52 +0300
Subject: [PATCH] Generate common headers of stdlib functions. What constitutes
 the header is yet to be decided.

---
 .../src/generators/GenerateStandardLib.kt     | 12 ++++++++--
 .../src/templates/engine/Engine.kt            | 23 ++++++++++++++-----
 2 files changed, 27 insertions(+), 8 deletions(-)

diff --git a/libraries/tools/kotlin-stdlib-gen/src/generators/GenerateStandardLib.kt b/libraries/tools/kotlin-stdlib-gen/src/generators/GenerateStandardLib.kt
index e39e18e98a7..07899c2992e 100644
--- a/libraries/tools/kotlin-stdlib-gen/src/generators/GenerateStandardLib.kt
+++ b/libraries/tools/kotlin-stdlib-gen/src/generators/GenerateStandardLib.kt
@@ -28,6 +28,7 @@ fun main(args: Array<String>) {
 
     generateCollectionsAPI(outDir)
     generateCollectionsJsAPI(jsCoreDir)
+    generateCommonAPI(baseDir.resolve("out/src/stdlib/common").apply { mkdirs() })
 
 }
 
@@ -61,10 +62,17 @@ fun generateCollectionsJsAPI(outDir: File) {
             .groupByFileAndWrite(outDir, Platform.JS, { "_${it.name.capitalize()}Js.kt"})
 }
 
+fun generateCommonAPI(outDir: File) {
+    (commonGenerators + ::specialJVM + ::specialJS).flatMap { it().sortedBy { it.signature }.asSequence() }
+            .groupByFileAndWrite(outDir, platform = null)
+}
+
+
+
 
 private fun Sequence<GenericFunction>.groupByFileAndWrite(
         outDir: File,
-        platform: Platform,
+        platform: Platform?,
         fileNameBuilder: (SourceFile) -> String = { "_${it.name.capitalize()}.kt" }
 ) {
     val groupedConcreteFunctions = map { it.instantiate(platform) }.flatten().groupBy { it.sourceFile }
@@ -75,7 +83,7 @@ private fun Sequence<GenericFunction>.groupByFileAndWrite(
     }
 }
 
-private fun List<ConcreteFunction>.writeTo(file: File, sourceFile: SourceFile, platform: Platform) {
+private fun List<ConcreteFunction>.writeTo(file: File, sourceFile: SourceFile, platform: Platform?) {
     println("Generating file: $file")
 
     FileWriter(file).use { writer ->
diff --git a/libraries/tools/kotlin-stdlib-gen/src/templates/engine/Engine.kt b/libraries/tools/kotlin-stdlib-gen/src/templates/engine/Engine.kt
index f73efac78a3..bbeafb4675b 100644
--- a/libraries/tools/kotlin-stdlib-gen/src/templates/engine/Engine.kt
+++ b/libraries/tools/kotlin-stdlib-gen/src/templates/engine/Engine.kt
@@ -215,22 +215,22 @@ class GenericFunction(val signature: String, val keyword: String = "fun") {
         buildPrimitives.addAll(p.toList())
     }
 
-    fun instantiate(platform: Platform, vararg families: Family = Family.values()): List<ConcreteFunction> {
+    fun instantiate(platform: Platform?, vararg families: Family = Family.values()): List<ConcreteFunction> {
         return families
                 .sortedBy { it.ordinal }
                 .filter { buildFamilies.contains(it) }
                 .filter { platform == Platform.JVM || jvmOnly[it] != true  }
-                .flatMap { family -> instantiate(family) }
+                .flatMap { family -> instantiate(family, platform == null) }
     }
 
-    fun instantiate(f: Family): List<ConcreteFunction> {
+    fun instantiate(f: Family, headerOnly: Boolean): List<ConcreteFunction> {
         val onlyPrimitives = buildFamilyPrimitives[f]
 
         if (f.isPrimitiveSpecialization || onlyPrimitives != null) {
             return (onlyPrimitives ?: buildPrimitives).sortedBy { it.ordinal }
-                    .map { primitive -> ConcreteFunction( { build(it, f, primitive) }, sourceFileFor(f) ) }
+                    .map { primitive -> ConcreteFunction( { build(it, f, primitive, headerOnly) }, sourceFileFor(f) ) }
         } else {
-            return listOf(ConcreteFunction( { build(it, f, null) }, sourceFileFor(f) ))
+            return listOf(ConcreteFunction( { build(it, f, null, headerOnly) }, sourceFileFor(f) ))
         }
     }
 
@@ -248,6 +248,7 @@ class GenericFunction(val signature: String, val keyword: String = "fun") {
         Primitives, Generic -> SourceFile.Misc
     }
 
+/*
     fun build(vararg families: Family = Family.values()): String {
         val builder = StringBuilder()
         for (family in families.sortedBy { it.name }) {
@@ -266,8 +267,9 @@ class GenericFunction(val signature: String, val keyword: String = "fun") {
             build(builder, f, null)
         }
     }
+*/
 
-    fun build(builder: Appendable, f: Family, primitive: PrimitiveType?) {
+    fun build(builder: Appendable, f: Family, primitive: PrimitiveType?, headerOnly: Boolean) {
         val returnType = returns[f] ?: throw RuntimeException("No return type specified for $signature")
 
         fun renderType(expression: String, receiver: String, self: String): String {
@@ -423,6 +425,9 @@ class GenericFunction(val signature: String, val keyword: String = "fun") {
         }
 
         builder.append(visibility[f] ?: "public").append(' ')
+        if (headerOnly) {
+            builder.append("header ")
+        }
         if (external[f] == true)
             builder.append("external ")
         if (inline[f]?.isInline() == true)
@@ -444,6 +449,12 @@ class GenericFunction(val signature: String, val keyword: String = "fun") {
         builder.append(receiverType)
         if (receiverType.isNotEmpty()) builder.append('.')
         builder.append("${(customSignature[f] ?: signature).renderType()}: ${returnType.renderType()}")
+
+        if (headerOnly) {
+            builder.append("\n\n")
+            return
+        }
+
         if (keyword == "fun") builder.append(" {")
 
         val body = (