代码优化

kkevsekk1 · Sep 15, 2023 · f888975 · f888975
1 parent 48ae953
commit f888975
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/autojs/src/main/java/com/stardust/autojs/engine/RhinoJavaScriptEngine.kt b/autojs/src/main/java/com/stardust/autojs/engine/RhinoJavaScriptEngine.kt
@@ -111,7 +111,7 @@ open class RhinoJavaScriptEngine(private val mAndroidContext: android.content.Co
         }
     }
 
-    internal fun initRequireBuilder(context: Context, scope: Scriptable) {
+    private fun initRequireBuilder(context: Context, scope: Scriptable) {
         val provider = AssetAndUrlModuleSourceProvider(
             mAndroidContext,
             listOf(
@@ -126,7 +126,7 @@ open class RhinoJavaScriptEngine(private val mAndroidContext: android.content.Co
         require.install(scope)
     }
 
-    protected fun createScope(context: Context): TopLevelScope {
+    private fun createScope(context: Context): TopLevelScope {
         val topLevelScope = TopLevelScope()
         topLevelScope.initStandardObjects(context, false)
         return topLevelScope
@@ -139,7 +139,7 @@ open class RhinoJavaScriptEngine(private val mAndroidContext: android.content.Co
         return context
     }
 
-    private fun setupContext(context: Context) {
+    fun setupContext(context: Context) {
         context.wrapFactory = WrapFactory()
     }
 

diff --git a/autojs/src/main/java/com/stardust/autojs/runtime/ScriptBridges.kt b/autojs/src/main/java/com/stardust/autojs/runtime/ScriptBridges.kt
@@ -21,8 +21,12 @@ class ScriptBridges {
 
     private fun <T> useJsContext(f: (context: Context) -> T): T {
         val context = Context.getCurrentContext()
+        val cx: Context = context ?: with(Context.enter()) {
+            engine?.setupContext(this)
+            this
+        }
         try {
-            return f(context ?: engine?.enterContext() ?: Context.enter())
+            return f(cx)
         } finally {
             context ?: Context.exit()
         }