From 3fb38cdbf37e299b76afe2f446cdd6c69fefc4c7 Mon Sep 17 00:00:00 2001
From: LeonOstrez <leon.ostrez@gmail.com>
Date: Tue, 25 Jun 2024 16:59:13 +0100
Subject: [PATCH] add more info to appFinished event for extension

---
 core/agents/task_completer.py | 12 ++++++++++--
 core/ui/base.py               | 22 ++++++++++++++++++++--
 core/ui/console.py            | 16 +++++++++++++---
 core/ui/ipc_client.py         | 32 ++++++++++++++++++++++++++++----
 core/ui/virtual.py            | 16 +++++++++++++---
 5 files changed, 84 insertions(+), 14 deletions(-)

diff --git a/core/agents/task_completer.py b/core/agents/task_completer.py
index bda0c9ab..98094968 100644
--- a/core/agents/task_completer.py
+++ b/core/agents/task_completer.py
@@ -38,8 +38,16 @@ class TaskCompleter(BaseAgent):
 
         if current_task_index1 == len(tasks):
             if source == "app":
-                await self.ui.send_app_finished()
+                await self.ui.send_app_finished(
+                    app_id=str(self.state_manager.project.id),
+                    app_name=self.state_manager.project.name,
+                    folder_name=self.state_manager.project.folder_name,
+                )
             elif source == "feature":
-                await self.ui.send_feature_finished()
+                await self.ui.send_feature_finished(
+                    app_id=str(self.state_manager.project.id),
+                    app_name=self.state_manager.project.name,
+                    folder_name=self.state_manager.project.folder_name,
+                )
 
         return AgentResponse.done(self)
diff --git a/core/ui/base.py b/core/ui/base.py
index 26d6df72..0cb8dbdd 100644
--- a/core/ui/base.py
+++ b/core/ui/base.py
@@ -121,15 +121,33 @@ class UIBase:
         """
         raise NotImplementedError()
 
-    async def send_app_finished(self):
+    async def send_app_finished(
+        self,
+        app_id: Optional[str] = None,
+        app_name: Optional[str] = None,
+        folder_name: Optional[str] = None,
+    ):
         """
         Send the app finished message.
+
+        :param app_id: App ID.
+        :param app_name: App name.
+        :param folder_name: Folder name.
         """
         raise NotImplementedError()
 
-    async def send_feature_finished(self):
+    async def send_feature_finished(
+        self,
+        app_id: Optional[str] = None,
+        app_name: Optional[str] = None,
+        folder_name: Optional[str] = None,
+    ):
         """
         Send the feature finished message.
+
+        :param app_id: App ID.
+        :param app_name: App name.
+        :param folder_name: Folder name.
         """
         raise NotImplementedError()
 
diff --git a/core/ui/console.py b/core/ui/console.py
index 813f0025..ed312812 100644
--- a/core/ui/console.py
+++ b/core/ui/console.py
@@ -33,14 +33,24 @@ class PlainConsoleUI(UIBase):
         else:
             print(message)
 
-    async def send_key_expired(self, message: Optional[str]):
+    async def send_key_expired(self, message: Optional[str] = None):
         if message:
             await self.send_message(message)
 
-    async def send_app_finished(self):
+    async def send_app_finished(
+        self,
+        app_id: Optional[str] = None,
+        app_name: Optional[str] = None,
+        folder_name: Optional[str] = None,
+    ):
         pass
 
-    async def send_feature_finished(self):
+    async def send_feature_finished(
+        self,
+        app_id: Optional[str] = None,
+        app_name: Optional[str] = None,
+        folder_name: Optional[str] = None,
+    ):
         pass
 
     async def ask_question(
diff --git a/core/ui/ipc_client.py b/core/ui/ipc_client.py
index 53eff7ab..89157454 100644
--- a/core/ui/ipc_client.py
+++ b/core/ui/ipc_client.py
@@ -199,11 +199,35 @@ class IPCClientUI(UIBase):
     async def send_key_expired(self, message: Optional[str] = None):
         await self._send(MessageType.KEY_EXPIRED)
 
-    async def send_app_finished(self):
-        await self._send(MessageType.APP_FINISHED)
+    async def send_app_finished(
+        self,
+        app_id: Optional[str] = None,
+        app_name: Optional[str] = None,
+        folder_name: Optional[str] = None,
+    ):
+        await self._send(
+            MessageType.APP_FINISHED,
+            content={
+                "app_id": app_id,
+                "app_name": app_name,
+                "folder_name": folder_name,
+            },
+        )
 
-    async def send_feature_finished(self):
-        await self._send(MessageType.FEATURE_FINISHED)
+    async def send_feature_finished(
+        self,
+        app_id: Optional[str] = None,
+        app_name: Optional[str] = None,
+        folder_name: Optional[str] = None,
+    ):
+        await self._send(
+            MessageType.FEATURE_FINISHED,
+            content={
+                "app_id": app_id,
+                "app_name": app_name,
+                "folder_name": folder_name,
+            },
+        )
 
     async def ask_question(
         self,
diff --git a/core/ui/virtual.py b/core/ui/virtual.py
index 1d3f51b1..0d07a58f 100644
--- a/core/ui/virtual.py
+++ b/core/ui/virtual.py
@@ -34,13 +34,23 @@ class VirtualUI(UIBase):
         else:
             print(message)
 
-    async def send_key_expired(self, message: Optional[str]):
+    async def send_key_expired(self, message: Optional[str] = None):
         pass
 
-    async def send_app_finished(self):
+    async def send_app_finished(
+        self,
+        app_id: Optional[str] = None,
+        app_name: Optional[str] = None,
+        folder_name: Optional[str] = None,
+    ):
         pass
 
-    async def send_feature_finished(self):
+    async def send_feature_finished(
+        self,
+        app_id: Optional[str] = None,
+        app_name: Optional[str] = None,
+        folder_name: Optional[str] = None,
+    ):
         pass
 
     async def ask_question(