From a4b9bbbe08cc7c5e52f6321ed837da9258a126b8 Mon Sep 17 00:00:00 2001
From: haesol1013 <shinhaesol11@gmail.com>
Date: Thu, 6 Mar 2025 01:18:09 +0900
Subject: [PATCH] Feat: Enable switching between UI Thread and Multi Process
 modes

---
 app/src/androidMain/python/main.py            | 32 ++++++++++++++++---
 .../python/repl/kernel/kernel_android.py      |  4 +--
 app/src/commonMain/python/repl/__init__.py    |  2 +-
 app/src/commonMain/python/repl/server.py      |  4 +++
 4 files changed, 34 insertions(+), 8 deletions(-)

diff --git a/app/src/androidMain/python/main.py b/app/src/androidMain/python/main.py
index 3e9cae4..ccfe8f0 100644
--- a/app/src/androidMain/python/main.py
+++ b/app/src/androidMain/python/main.py
@@ -11,7 +11,7 @@
 
 import time
 from model.config import ChatHistory
-from repl import REPLConfig, send_server_launch_intent, kernel
+from repl import REPLConfig, send_server_launch_intent, send_server_stop_intent, kernel, UIThreadKernelService
 
 
 config = REPLConfig(manager_class=kernel.UIThreadKernelManager)
@@ -49,12 +49,34 @@ def compose(cls):
 
         cls.user_prompt = remember_saveable("안녕하세요!")
 
+        cls.target_manager =  remember_saveable(config.manager.split(".")[-1])
+
+        def get_target_manager_class():
+            if cls.target_manager.getValue() == "UIThreadKernelManager":
+                return kernel.UIThreadKernelManager
+            else:
+                return kernel.InAppKernelManager
+
+        def change_mode():
+            current_manager = cls.target_manager.getValue()
+            new_value = "InAppKernelManager" if current_manager == "UIThreadKernelManager" else "UIThreadKernelManager"
+            cls.target_manager.setValue(new_value)
+
         def run_jupyter():
-            browser_intent = Intent(Intent.ACTION_VIEW, Uri.parse(config.uri))
+            new_config = REPLConfig(manager_class=get_target_manager_class())
+            browser_intent = Intent(Intent.ACTION_VIEW, Uri.parse(new_config.uri))
 
             def runner():
-                send_server_launch_intent(context, config)
+                kernel_service_class = UIThreadKernelService.getClass()
+                intent = Intent(context, kernel_service_class)
+                context.stopService(intent)
+
+                send_server_stop_intent(context)
                 time.sleep(1)
+
+                send_server_launch_intent(context, new_config)
+                time.sleep(1)
+
                 context.startActivity(browser_intent)
 
             cls.scope.launch(runner)
@@ -72,10 +94,10 @@ def View():
             Text("")
 
             Button(
-                onclick=lambda: None,
+                onclick=change_mode,
                 color=0xFF000000,
                 content=lambda: {
-                    Text(f"실행 모드 변경 (현재 옵션: UI 스레드 동기화 모드)", color=0xFFFFFFFF)
+                    Text(f"실행 모드 변경 (현재 옵션: {cls.target_manager.getValue()[:-7]} 모드)", color=0xFFFFFFFF)
                 }
             )
             Button(
diff --git a/app/src/androidMain/python/repl/kernel/kernel_android.py b/app/src/androidMain/python/repl/kernel/kernel_android.py
index 1b903c9..5593c68 100644
--- a/app/src/androidMain/python/repl/kernel/kernel_android.py
+++ b/app/src/androidMain/python/repl/kernel/kernel_android.py
@@ -183,7 +183,7 @@ def process_info(self):
             :return: The process info if the kernel is running, None otherwise
             """
             activity_manager = app.getSystemService(app.ACTIVITY_SERVICE)
-            for p_info in activity_manager.getRunningAppProcesses(Integer.MAX_VALUE).toArray():
+            for p_info in activity_manager.getRunningAppProcesses().toArray():
                 if ":".join(["", *p_info.processName.split(":")[1:]]) == f":kernel{self.process_name}":
                     return p_info
             return None
@@ -252,4 +252,4 @@ async def launch_kernel(self, cmd: List[str], **kwargs: Any) -> KernelConnection
         return self.connection_info
 
 
-#provisioning.LocalProvisioner = InAppLocalPrivateProvisioner
+provisioning.LocalProvisioner = InAppLocalPrivateProvisioner
diff --git a/app/src/commonMain/python/repl/__init__.py b/app/src/commonMain/python/repl/__init__.py
index 249b6a6..aee6150 100644
--- a/app/src/commonMain/python/repl/__init__.py
+++ b/app/src/commonMain/python/repl/__init__.py
@@ -1,5 +1,5 @@
 from .config import REPLConfig
-from .server import run_lab_server, send_server_launch_intent
+from .server import run_lab_server, send_server_launch_intent, send_server_stop_intent
 
 import signal
 
diff --git a/app/src/commonMain/python/repl/server.py b/app/src/commonMain/python/repl/server.py
index ca79593..8b6641e 100644
--- a/app/src/commonMain/python/repl/server.py
+++ b/app/src/commonMain/python/repl/server.py
@@ -22,3 +22,7 @@ def send_server_launch_intent(context, config: REPLConfig):
     for key, value in config.dict.items():
         intent.putExtra(key, value)
     context.startService(intent)
+
+def send_server_stop_intent(context):
+    intent = Intent(context, InAppLabServerService.getClass())
+    context.stopService(intent)