[JS IR BE] support CharSequence and String methods

compiler/ir/backend.js/src/org/jetbrains/kotlin/ir/backend/js/JsIntrinsics.kt

@@ -9,9 +9,12 @@ import org.jetbrains.kotlin.builtins.KotlinBuiltIns
 import org.jetbrains.kotlin.builtins.PrimitiveType
 import org.jetbrains.kotlin.ir.backend.js.ir.JsIrBuilder
 import org.jetbrains.kotlin.ir.backend.js.utils.Namer
+import org.jetbrains.kotlin.ir.declarations.IrFunction
+import org.jetbrains.kotlin.ir.declarations.IrProperty
 import org.jetbrains.kotlin.ir.declarations.impl.IrExternalPackageFragmentImpl
 import org.jetbrains.kotlin.ir.descriptors.IrBuiltIns
 import org.jetbrains.kotlin.ir.symbols.impl.IrExternalPackageFragmentSymbolImpl
+import org.jetbrains.kotlin.ir.util.getPropertyGetter
 import org.jetbrains.kotlin.name.FqName
 import org.jetbrains.kotlin.name.Name
 import org.jetbrains.kotlin.psi2ir.findSingleFunction
@@ -219,6 +222,20 @@ class JsIntrinsics(private val irBuiltIns: IrBuiltIns, val context: JsIrBackendC
 
     val jsArraySlice = unOp("slice")
 
+    // TODO move to IntrinsifyCallsLowering
+    val doNotIntrinsifyAnnotationSymbol = context.symbolTable.referenceClass(context.getInternalClass("DoNotIntrinsify"))
+
+    // TODO move CharSequence-related stiff to IntrinsifyCallsLowering
+    val charSequenceClassSymbol = context.symbolTable.referenceClass(context.getClass(FqName("kotlin.CharSequence")))
+    val charSequenceLengthPropertyGetterSymbol = charSequenceClassSymbol.owner.declarations.filterIsInstance<IrProperty>().first { it.name.asString() == "length" }.getter!!.symbol
+    val charSequenceGetFunctionSymbol = charSequenceClassSymbol.owner.declarations.filterIsInstance<IrFunction>().single { it.name.asString() == "get"}.symbol
+    val charSequenceSubSequenceFunctionSymbol = charSequenceClassSymbol.owner.declarations.filterIsInstance<IrFunction>().single { it.name.asString() == "subSequence"}.symbol
+
+
+    val jsCharSequenceGet = getInternalFunction("charSequenceGet")
+    val jsCharSequenceLength = getInternalFunction("charSequenceLength")
+    val jsCharSequenceSubSequence = getInternalFunction("charSequenceSubSequence")
+
     // Helpers:
 
     private fun getInternalFunction(name: String) =

compiler/ir/backend.js/src/org/jetbrains/kotlin/ir/backend/js/lower/IntrinsicifyCallsLowering.kt

@@ -162,6 +162,13 @@ class IntrinsicifyCallsLowering(private val context: JsIrBackendContext) : FileL
             }
 
             add(context.irBuiltIns.stringClass.lengthProperty, context.intrinsics.jsArrayLength, true)
+            add(context.irBuiltIns.stringClass.getFunction, intrinsics.jsCharSequenceGet.owner, true)
+            add(context.irBuiltIns.stringClass.owner.declarations.filterIsInstance<IrFunction>().single { it.name.asString() == "subSequence"}.symbol,
+                intrinsics.jsCharSequenceSubSequence.owner, true)
+
+            add(intrinsics.charSequenceLengthPropertyGetterSymbol, intrinsics.jsCharSequenceLength.owner, true)
+            add(intrinsics.charSequenceGetFunctionSymbol, intrinsics.jsCharSequenceGet.owner, true)
+            add(intrinsics.charSequenceSubSequenceFunctionSymbol, intrinsics.jsCharSequenceSubSequence.owner, true)
         }
 
         memberToTransformer.run {
@@ -352,6 +359,10 @@ class IntrinsicifyCallsLowering(private val context: JsIrBackendContext) : FileL
                     val symbol = call.symbol
                     val declaration = symbol.owner
 
+                    if (declaration.annotations.any { it.superQualifierSymbol == intrinsics.doNotIntrinsifyAnnotationSymbol }) {
+                        return call
+                    }
+
                     if (declaration.isDynamic() || declaration.isEffectivelyExternal()) {
                         when (call.origin) {
                             IrStatementOrigin.GET_PROPERTY -> {

compiler/testData/codegen/box/controlStructures/forInCharSequenceMut.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // WITH_RUNTIME
 
 fun box(): String {

compiler/testData/codegen/box/controlStructures/forInCharSequenceWithIndex/forInCharSequenceWithIndex.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // WITH_RUNTIME
 
 val cs: CharSequence = "abcd"

compiler/testData/codegen/box/controlStructures/forInCharSequenceWithIndex/forInEmptyStringWithIndex.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // WITH_RUNTIME
 
 fun box(): String {

compiler/testData/codegen/box/controlStructures/forInCharSequenceWithIndex/forInStringWithIndex.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // WITH_RUNTIME
 
 fun box(): String {

compiler/testData/codegen/box/controlStructures/forInCharSequenceWithIndex/forInStringWithIndexNoElementVar.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // WITH_RUNTIME
 
 val xs = "abcd"

compiler/testData/codegen/box/controlStructures/forInCharSequenceWithIndex/forInStringWithIndexNoIndexVar.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // WITH_RUNTIME
 
 val xs = "abcd"

compiler/testData/codegen/box/controlStructures/forInCharSequenceWithIndex/forInStringWithIndexWithExplicitlyTypedIndexVariable.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // WITH_RUNTIME
 
 val xs = "abcd"

compiler/testData/codegen/box/coroutines/featureIntersection/tailrec/realStringEscape.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // IGNORE_BACKEND: JVM_IR
 // WITH_RUNTIME
 // WITH_COROUTINES

compiler/testData/codegen/box/coroutines/mergeNullAndString.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // IGNORE_BACKEND: JVM_IR
 // WITH_RUNTIME
 // WITH_COROUTINES

compiler/testData/codegen/box/coroutines/suspendFunctionTypeCall/localVal.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // IGNORE_BACKEND: JVM_IR
 // WITH_RUNTIME
 // WITH_COROUTINES

compiler/testData/codegen/box/coroutines/suspendFunctionTypeCall/suspendModifier.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // IGNORE_BACKEND: JVM_IR
 // WITH_RUNTIME
 // COMMON_COROUTINES_TEST

compiler/testData/codegen/box/diagnostics/functions/tailRecursion/realStringEscape.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // DONT_RUN_GENERATED_CODE: JS
 
 fun escapeChar(c : Char) : String? = when (c) {

compiler/testData/codegen/box/extensionFunctions/simple.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 fun StringBuilder.first() = this.get(0)
 
 fun foo() = StringBuilder("foo").first()

compiler/testData/codegen/box/intrinsics/stringFromCollection.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 fun box(): String {
     val list = ArrayList<String>()
     list.add("0")

compiler/testData/codegen/box/primitiveTypes/kt4098.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 fun box(): String {
     val c: Char? = '0'
     c!!.toInt()

compiler/testData/codegen/box/primitiveTypes/kt4210.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 fun box(): String {
     val s: String? = "abc"
     val c = s?.get(0)!! - 'b'

compiler/testData/codegen/box/primitiveTypes/kt518.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 
 fun foo(i : Int?, a : Any?) {
     i?.plus(1)

compiler/testData/codegen/box/ranges/contains/generated/charSequenceIndices.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // Auto-generated by GenerateInRangeExpressionTestData. Do not edit!
 // WITH_RUNTIME
 

compiler/testData/codegen/box/ranges/expression/simpleRangeWithNonConstantEnds.kt

@@ -1,9 +1,6 @@
 // TODO: muted automatically, investigate should it be ran for JVM_IR or not
 // IGNORE_BACKEND: JVM_IR
 
-// TODO: muted automatically, investigate should it be ran for JS_IR or not
-// IGNORE_BACKEND: JS_IR
-
 // Auto-generated by org.jetbrains.kotlin.generators.tests.GenerateRangesCodegenTestData. DO NOT EDIT!
 // WITH_RUNTIME
 

compiler/testData/codegen/box/ranges/forInIndices/forInCharSequenceIndices.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // WITH_RUNTIME
 
 fun test(s: CharSequence): Int {

compiler/testData/codegen/box/ranges/forInIndices/kt13241_CharSequence.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // WITH_RUNTIME
 
 import kotlin.test.assertEquals

compiler/testData/codegen/box/ranges/forInReversed/forInReversedCharSequenceIndices.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // WITH_RUNTIME
 import kotlin.test.*
 

compiler/testData/codegen/box/ranges/forInStringVarUpdatedInLoopBody.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // WITH_RUNTIME
 
 fun box(): String {

compiler/testData/codegen/box/ranges/literal/simpleRangeWithNonConstantEnds.kt

@@ -1,9 +1,6 @@
 // TODO: muted automatically, investigate should it be ran for JVM_IR or not
 // IGNORE_BACKEND: JVM_IR
 
-// TODO: muted automatically, investigate should it be ran for JS_IR or not
-// IGNORE_BACKEND: JS_IR
-
 // Auto-generated by org.jetbrains.kotlin.generators.tests.GenerateRangesCodegenTestData. DO NOT EDIT!
 // WITH_RUNTIME
 

compiler/testData/codegen/box/strings/forInString.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // WITH_RUNTIME
 
 fun foo(): Int {

compiler/testData/codegen/box/strings/kt3571.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 class Thing(delegate: CharSequence) : CharSequence by delegate
   
 fun box(): String {

compiler/testData/codegen/box/strings/kt5389_stringBuilderGet.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 fun box(): String {
     val sb = StringBuilder("OK")
     return "${sb.get(0)}${sb[1]}"

compiler/testData/codegen/box/strings/kt5956.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // KT-5956 java.lang.AbstractMethodError: test.Thing.subSequence(II)Ljava/lang/CharSequence
 
 class Thing(val delegate: CharSequence) : CharSequence {

compiler/tests/org/jetbrains/kotlin/generators/tests/GenerateRangesCodegenTestData.java

@@ -122,8 +122,7 @@ public class GenerateRangesCodegenTestData {
     private static final List<String> IGNORED_FOR_JS_BACKEND = Collections.emptyList();
 
     private static final List<String> IGNORED_FOR_JS_IR_BACKEND = Arrays.asList("inexactDownToMinValue.kt",
-                                                                                "inexactToMaxValue.kt",
+                                                                                "inexactToMaxValue.kt");
-                                                                                "simpleRangeWithNonConstantEnds.kt");
 
     private static final List<String> IGNORED_FOR_NATIVE_BACKEND = Collections.emptyList();
 

js/js.translator/testData/box/expression/for/forIteratesOverLiteralRange.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // EXPECTED_REACHABLE_NODES: 1230
 package foo
 

js/js.translator/testData/box/expression/stringClass/extensionMethods.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // EXPECTED_REACHABLE_NODES: 1250
 package foo
 

js/js.translator/testData/box/inlineEvaluationOrder/continueInExtractedDoWhile.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // EXPECTED_REACHABLE_NODES: 1286
 package foo
 

js/js.translator/testData/box/inlineEvaluationOrder/whileConditionExtracted.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // EXPECTED_REACHABLE_NODES: 1284
 // See KT-8005
 package foo

js/js.translator/testData/box/inlineMultiFile/tryCatch.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // EXPECTED_REACHABLE_NODES: 1300
 /*
  * Copy of JVM-backend test

js/js.translator/testData/box/inlineMultiFile/tryCatch2.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // EXPECTED_REACHABLE_NODES: 1303
 /*
  * Copy of JVM-backend test

js/js.translator/testData/box/nameClashes/extensionFunctionAndProperty.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // EXPECTED_REACHABLE_NODES: 1524
 package foo
 

js/js.translator/testData/box/nameClashes/localFunctions.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JS_IR
 // EXPECTED_REACHABLE_NODES: 1288
 var log = ""
 

libraries/stdlib/js/irRuntime/charSequence.kt

@@ -0,0 +1,38 @@
+/*
+ * 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 kotlin.js
+
+internal annotation class DoNotIntrinsify
+
+@PublishedApi
+@DoNotIntrinsify
+internal fun charSequenceGet(a: CharSequence, index: Int): Char {
+    return  if (a is String) {
+        Char(a.asDynamic().charCodeAt(index).unsafeCast<Int>())
+    } else {
+        a[index]
+    }
+}
+
+@PublishedApi
+@DoNotIntrinsify
+internal fun charSequenceLength(a: CharSequence): Int {
+    return if (a is String) {
+        a.asDynamic().length.unsafeCast<Int>()
+    } else {
+        a.length
+    }
+}
+
+@PublishedApi
+@DoNotIntrinsify
+internal fun charSequenceSubSequence(a: CharSequence, startIndex: Int, endIndex: Int): CharSequence {
+    return if (a is String) {
+        a.asDynamic().substring(startIndex, endIndex).unsafeCast<String>()
+    } else {
+        a.subSequence(startIndex, endIndex)
+    }
+}