[java-shell] override set operator for GraalVM Value to use 'v[...] = ...' notation

Author: kornilova203

packages/java-shell/src/main/kotlin/com/mongodb/mongosh/ConsoleLogSupport.kt

@@ -10,12 +10,12 @@ internal class ConsoleLogSupport(context: MongoShellContext, converter: MongoShe
         val print = context.jsFun { args ->
             printedValues?.add(args.map { converter.toJava(it).value })
         }
-        context.bindings.putMember("print", print)
+        context.bindings["print"] = print
         @Suppress("JSPrimitiveTypeWrapperUsage")
         val console = context.eval("new Object()")
-        console.putMember("log", print)
-        console.putMember("error", print)
-        context.bindings.putMember("console", console)
+        console["log"] = print
+        console["error"] = print
+        context.bindings["console"] = console
     }
 
     fun <T> withConsoleLogEnabled(printedValues: MutableList<List<Any?>>, func: () -> T): T {

packages/java-shell/src/main/kotlin/com/mongodb/mongosh/MongoShellConverter.kt

@@ -34,7 +34,7 @@ internal class MongoShellConverter(private val context: MongoShellContext, priva
         @Suppress("JSPrimitiveTypeWrapperUsage")
         val jsMap = context.eval("new Object()")
         for ((key, value) in map.entries) {
-            jsMap.putMember(key as String, toJs(value))
+            jsMap[key as String] = toJs(value)
         }
         return jsMap
     }

packages/java-shell/src/main/kotlin/com/mongodb/mongosh/MongoShellEvaluator.kt

@@ -49,15 +49,14 @@ internal class MongoShellEvaluator(client: MongoClient, private val context: Mon
     }
 
     private fun initContext(bindings: Value, jsSymbol: Value) {
-        bindings.putMember("BSONSymbol", bindings["Symbol"])
-        bindings.putMember("Symbol", jsSymbol)
+        bindings["BSONSymbol"] = bindings["Symbol"]
+        bindings["Symbol"] = jsSymbol
         val date = context.eval("(dateHelper) => function inner() { return dateHelper(new.target !== undefined, ...arguments) }", "dateHelper_script")
                 .execute(ProxyExecutable { args -> dateHelper(args[0].asBoolean(), args.drop(1)) })
-        date.putMember("now", ProxyExecutable { System.currentTimeMillis() })
-        bindings.putMember("Date", date)
-        val isoDate = context.jsFun { args -> dateHelper(true, args.toList()) }
-        bindings.putMember("ISODate", isoDate)
-        bindings.putMember("UUID", context.jsFun { args -> if (args.isEmpty()) UUID.randomUUID() else UUID.fromString(args[0].asString()) })
+        date["now"] = ProxyExecutable { System.currentTimeMillis() }
+        bindings["Date"] = date
+        bindings["ISODate"] = context.jsFun { args -> dateHelper(true, args.toList()) }
+        bindings["UUID"] = context.jsFun { args -> if (args.isEmpty()) UUID.randomUUID() else UUID.fromString(args[0].asString()) }
     }
 
     private fun shellResult(printable: Value, type: String): Value {

packages/java-shell/src/main/kotlin/com/mongodb/mongosh/util.kt

@@ -6,9 +6,8 @@ import org.graalvm.polyglot.Value
 import org.intellij.lang.annotations.Language
 
 
-internal inline operator fun Value.get(identifier: String): Value? {
-    return getMember(identifier)
-}
+internal inline operator fun Value.get(identifier: String): Value? = getMember(identifier)
+internal inline operator fun Value.set(identifier: String, value: Any?) = putMember(identifier, value)
 
 internal inline fun Value.instanceOf(context: MongoShellContext, @Language("js") clazz: String): Boolean {
     return context.eval("(x) => x instanceof $clazz", "instance_of_script").execute(this).asBoolean()