From d232e2ceb36eabc382ad4d529e016e0f10402de6 Mon Sep 17 00:00:00 2001
From: Ilya Goncharov <ilya.goncharov@jetbrains.com>
Date: Mon, 3 Aug 2020 17:38:15 +0300
Subject: [PATCH] [Gradle, JS] Add ANTLR visitor for inverting of npm versions

^KT-41054 fixed
---
 .../js/npm/NpmVersionInvertedVisitor.kt       | 87 +++++++++++++++++++
 1 file changed, 87 insertions(+)
 create mode 100644 libraries/tools/kotlin-gradle-plugin/src/main/kotlin/org/jetbrains/kotlin/gradle/targets/js/npm/NpmVersionInvertedVisitor.kt

diff --git a/libraries/tools/kotlin-gradle-plugin/src/main/kotlin/org/jetbrains/kotlin/gradle/targets/js/npm/NpmVersionInvertedVisitor.kt b/libraries/tools/kotlin-gradle-plugin/src/main/kotlin/org/jetbrains/kotlin/gradle/targets/js/npm/NpmVersionInvertedVisitor.kt
new file mode 100644
index 00000000000..4084e01a893
--- /dev/null
+++ b/libraries/tools/kotlin-gradle-plugin/src/main/kotlin/org/jetbrains/kotlin/gradle/targets/js/npm/NpmVersionInvertedVisitor.kt
@@ -0,0 +1,87 @@
+/*
+ * Copyright 2010-2020 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * 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.gradle.targets.js.npm
+
+import com.github.gundy.semver4j.generated.grammar.NodeSemverExpressionBaseVisitor
+import com.github.gundy.semver4j.generated.grammar.NodeSemverExpressionParser
+import com.github.gundy.semver4j.model.Version
+import org.antlr.v4.runtime.tree.TerminalNode
+
+class NpmVersionInvertedVisitor : NodeSemverExpressionBaseVisitor<String>() {
+    override fun visitEmptyRange(ctx: NodeSemverExpressionParser.EmptyRangeContext): String =
+        WILDCARD_VERSION
+
+    override fun visitWildcard(ctx: NodeSemverExpressionParser.WildcardContext): String =
+        NONE_VERSION
+
+    override fun visitWildcardRange(ctx: NodeSemverExpressionParser.WildcardRangeContext): String =
+        inverseVersion(ctx.partialWildcardSemver().text)
+
+    override fun visitOperator(ctx: NodeSemverExpressionParser.OperatorContext): String {
+        val eq: TerminalNode? = ctx.unaryOperator().EQ()
+        val gt: TerminalNode? = ctx.unaryOperator().GT()
+        val lt: TerminalNode? = ctx.unaryOperator().LT()
+
+        val version = ctx.partialWildcardSemver().text
+
+        if (gt == null && lt == null) {
+            return inverseVersion(version)
+        }
+
+        val invertedEqToken = if (eq != null) EQ else ""
+
+        if (lt != null) {
+            return "$GT$invertedEqToken$version"
+        }
+
+        return "$LT$invertedEqToken$version"
+    }
+
+    override fun visitWildcardOperator(ctx: NodeSemverExpressionParser.WildcardOperatorContext): String =
+        NONE_VERSION
+
+    override fun visitTildeRange(ctx: NodeSemverExpressionParser.TildeRangeContext): String {
+        val version = Version.fromString(ctx.fullSemver().text)
+        val nextVersion = version
+            .incrementMinor()
+
+        return "$LT$version $OR $GTEQ$nextVersion"
+    }
+
+    override fun visitCaretRange(ctx: NodeSemverExpressionParser.CaretRangeContext): String {
+        val version = Version.fromString(ctx.fullSemver().text)
+        val nextVersion = version
+            .incrementMajor()
+
+        return "$LT$version $OR $GTEQ$nextVersion"
+    }
+
+    override fun visitFullySpecifiedSemver(ctx: NodeSemverExpressionParser.FullySpecifiedSemverContext): String =
+        inverseVersion(ctx.fullSemver().text)
+
+    override fun visitLogicalAndOfSimpleExpressions(ctx: NodeSemverExpressionParser.LogicalAndOfSimpleExpressionsContext): String =
+        ctx.simple()
+            .joinToString(OR) { visit(it) }
+
+    override fun visitLogicalOrOfMultipleRanges(ctx: NodeSemverExpressionParser.LogicalOrOfMultipleRangesContext): String =
+        ctx.basicRange()
+            .joinToString(AND) { visitBasicRange(it) }
+
+    private fun inverseVersion(version: String): String =
+        "$GT$version $OR $LT$version"
+
+
+    companion object {
+        const val EQ = "="
+        const val GT = ">"
+        const val LT = "<"
+        const val GTEQ = ">="
+        const val LTEQ = "<="
+
+        const val AND = " "
+        const val OR = "||"
+    }
+}
\ No newline at end of file