🐛 Fix bug when variable assigment doesn't available in caller scope

Author: uriyyo

build.gradle.kts

@@ -8,7 +8,7 @@ plugins {
 
 
 group = "com.uriyyo.evaluate_async_code"
-version = "1.1"
+version = "1.2"
 
 repositories {
     mavenCentral()

src/main/java/com/uriyyo/evaluate_async_code/Utils.kt

@@ -184,22 +184,22 @@ val String.isAsyncCode: Boolean
 
 fun String.toAsyncCode(): Pair<String, String> {
     val lines = this.prependIndent("    ").lines().toMutableList()
-    val expression = if (lines.size == 1) "    return ${lines[0].trimStart()}" else lines.joinToString("\n")
-
     val resultVar = "__async_result_var_${Random.nextInt().absoluteValue}__"
 
+    lines[lines.lastIndex] = "    $resultVar = ${lines[lines.lastIndex]}"
+    val expression = lines.joinToString("\n")
+
     return """
 $NEST_ASYNCIO_SOUCE_CODE
 
 async def __async_evaluate_expression__():
 $expression
+    return locals()
 
 from asyncio import run as __async_run_coro__
-$resultVar = __async_run_coro__(__async_evaluate_expression__())
+locals().update(__async_run_coro__(__async_evaluate_expression__()))
 
-del __async_run_coro__
-del __async_evaluate_expression__
-del __patch_asyncio__
+del __async_run_coro__, __async_evaluate_expression__, __patch_asyncio__
 """ to resultVar
 }
 

src/main/resources/META-INF/plugin.xml

@@ -1,14 +1,14 @@
 <idea-plugin>
     <id>com.uriyyo.evaluate_async_code</id>
     <name>Evaluate async code</name>
-    <version>1.1</version>
+    <version>1.2</version>
     <vendor email="[email protected]">Yurii Karabas</vendor>
 
     <description>
         Finally, you can use await in evaluate expression
     </description>
 
-    <change-notes>Fix bug when evaluate expression failed because of AttributeError</change-notes>
+    <change-notes>Fix bug when variable assigment doesn't available in caller scope</change-notes>
 
     <idea-version since-build="192"/>
     <depends>com.intellij.modules.lang</depends>