From bf7ba98107d74e79e4cdea90709b76c223275918 Mon Sep 17 00:00:00 2001
From: Nikita Klimenko <nikita.klimenko@jetbrains.com>
Date: Tue, 10 Sep 2024 13:27:57 +0300
Subject: [PATCH 1/2] Highlight readJsonStr argument as JSON

---
 .../main/kotlin/org/jetbrains/kotlinx/dataframe/io/json.kt   | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/io/json.kt b/core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/io/json.kt
index c9448fabdb..ccb73d6e20 100644
--- a/core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/io/json.kt
+++ b/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,

From 115e5f358df88525dd74c697eb96bab718b57e90 Mon Sep 17 00:00:00 2001
From: Nikita Klimenko <nikita.klimenko@jetbrains.com>
Date: Tue, 10 Sep 2024 13:34:28 +0300
Subject: [PATCH 2/2] Highlight split.match(regex) argument as RegExp

---
 .../main/kotlin/org/jetbrains/kotlinx/dataframe/api/split.kt | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/split.kt b/core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/split.kt
index 10291745f8..56fa8d23ab 100644
--- a/core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/split.kt
+++ b/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 {