Move builtins generators to their own source set

Make them depend on bootstrap stdlib. This is to prevent recompiling the entire project when we need just to generate builtins sources.
2018-04-26 00:07:17 +03:00
parent a704a1b6d0
commit 7438d12ad7
14 changed files with 51 additions and 32 deletions
@@ -0,0 +1,87 @@
+/*
+ * Copyright 2010-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license
+ * that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.generators.builtins.generateBuiltIns
+
+import org.jetbrains.kotlin.generators.builtins.arrayIterators.GenerateArrayIterators
+import org.jetbrains.kotlin.generators.builtins.arrays.GenerateArrays
+import org.jetbrains.kotlin.generators.builtins.functions.GenerateFunctions
+import org.jetbrains.kotlin.generators.builtins.iterators.GenerateIterators
+import org.jetbrains.kotlin.generators.builtins.progressionIterators.GenerateProgressionIterators
+import org.jetbrains.kotlin.generators.builtins.progressions.GenerateProgressions
+import org.jetbrains.kotlin.generators.builtins.ranges.GeneratePrimitives
+import org.jetbrains.kotlin.generators.builtins.ranges.GenerateRanges
+import org.xml.sax.InputSource
+import java.io.File
+import java.io.PrintWriter
+import javax.xml.xpath.XPathFactory
+
+fun assertExists(file: File) {
+    if (!file.exists()) error("Output dir does not exist: ${file.absolutePath}")
+}
+
+val BUILT_INS_NATIVE_DIR = File("core/builtins/native/")
+val BUILT_INS_SRC_DIR = File("core/builtins/src/")
+val RUNTIME_JVM_DIR = File("libraries/stdlib/jvm/runtime/")
+
+abstract class BuiltInsSourceGenerator(val out: PrintWriter) {
+    protected abstract fun generateBody(): Unit
+
+    protected open fun getPackage(): String = "kotlin"
+
+    enum class Language {
+        KOTLIN,
+        JAVA
+    }
+
+    fun generate() {
+        out.println(readCopyrightNoticeFromProfile(File(".idea/copyright/apache.xml")))
+        // Don't include generator class name in the message: these are built-in sources,
+        // and we don't want to scare users with any internal information about our project
+        out.println("// Auto-generated file. DO NOT EDIT!")
+        out.println()
+        out.print("package ${getPackage()}")
+        out.println()
+        out.println()
+
+        generateBody()
+    }
+}
+
+fun readCopyrightNoticeFromProfile(copyrightProfile: File): String {
+    val template = copyrightProfile.reader().use { reader ->
+        XPathFactory.newInstance().newXPath().evaluate("/component/copyright/option[@name='notice']/@value", InputSource(reader))
+    }
+    val yearTemplate = "&#36;today.year"
+    val year = java.time.LocalDate.now().year.toString()
+    assert(yearTemplate in template)
+
+    return template.replace(yearTemplate, year).lines().joinToString("", prefix = "/*\n", postfix = " */\n") { " * $it\n" }
+}
+
+fun generateBuiltIns(generate: (File, (PrintWriter) -> BuiltInsSourceGenerator) -> Unit) {
+    assertExists(BUILT_INS_NATIVE_DIR)
+    assertExists(BUILT_INS_SRC_DIR)
+    assertExists(RUNTIME_JVM_DIR)
+
+    generate(File(RUNTIME_JVM_DIR, "kotlin/jvm/functions/Functions.kt")) { GenerateFunctions(it) }
+    generate(File(BUILT_INS_NATIVE_DIR, "kotlin/Arrays.kt")) { GenerateArrays(it) }
+    generate(File(BUILT_INS_NATIVE_DIR, "kotlin/Primitives.kt")) { GeneratePrimitives(it) }
+    generate(File(BUILT_INS_SRC_DIR, "kotlin/Iterators.kt")) { GenerateIterators(it) }
+    generate(File(RUNTIME_JVM_DIR, "kotlin/jvm/internal/ArrayIterators.kt")) { GenerateArrayIterators(it) }
+    generate(File(BUILT_INS_SRC_DIR, "kotlin/ProgressionIterators.kt")) { GenerateProgressionIterators(it) }
+    generate(File(BUILT_INS_SRC_DIR, "kotlin/Progressions.kt")) { GenerateProgressions(it) }
+    generate(File(BUILT_INS_SRC_DIR, "kotlin/Ranges.kt")) { GenerateRanges(it) }
+}
+
+fun main(args: Array<String>) {
+    generateBuiltIns { file, generator ->
+        println("generating $file")
+        file.parentFile?.mkdirs()
+        PrintWriter(file).use {
+            generator(it).generate()
+        }
+    }
+}