Merge pull request #861 from Kotlin/readJsonStr-annotation

Add Language annotations for better IDE experience

Author: koperagen

core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/split.kt

@@ -1,5 +1,6 @@
 package org.jetbrains.kotlinx.dataframe.api
 
+import org.intellij.lang.annotations.Language
 import org.jetbrains.kotlinx.dataframe.AnyFrame
 import org.jetbrains.kotlinx.dataframe.ColumnsSelector
 import org.jetbrains.kotlinx.dataframe.DataColumn
@@ -111,7 +112,9 @@ internal fun <T, C, R> Split<T, C>.by(
 
 // region match
 
-public fun <T, C : String?> Split<T, C>.match(regex: String): SplitWithTransform<T, C, String?> = match(regex.toRegex())
+public fun <T, C : String?> Split<T, C>.match(
+    @Language("RegExp") regex: String,
+): SplitWithTransform<T, C, String?> = match(regex.toRegex())
 
 public fun <T, C : String?> Split<T, C>.match(regex: Regex): SplitWithTransform<T, C, String?> =
     by {

core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/io/json.kt

@@ -4,6 +4,7 @@ import kotlinx.serialization.ExperimentalSerializationApi
 import kotlinx.serialization.json.Json
 import kotlinx.serialization.json.JsonElement
 import kotlinx.serialization.json.decodeFromStream
+import org.intellij.lang.annotations.Language
 import org.jetbrains.kotlinx.dataframe.AnyFrame
 import org.jetbrains.kotlinx.dataframe.AnyRow
 import org.jetbrains.kotlinx.dataframe.DataFrame
@@ -254,7 +255,7 @@ public fun DataRow.Companion.readJson(
 @Refine
 @Interpretable("ReadJsonStr")
 public fun DataFrame.Companion.readJsonStr(
-    text: String,
+    @Language("json") text: String,
     header: List<String> = emptyList(),
     keyValuePaths: List<JsonPath> = emptyList(),
     typeClashTactic: TypeClashTactic = ARRAY_AND_VALUE_COLUMNS,
@@ -269,7 +270,7 @@ public fun DataFrame.Companion.readJsonStr(
  * @return [DataRow] from the given [text].
  */
 public fun DataRow.Companion.readJsonStr(
-    text: String,
+    @Language("json") text: String,
     header: List<String> = emptyList(),
     keyValuePaths: List<JsonPath> = emptyList(),
     typeClashTactic: TypeClashTactic = ARRAY_AND_VALUE_COLUMNS,