feat: 完成异步任务跟踪架构基础

pkg/api/http/controller/groups/plugins.py

@@ -5,7 +5,7 @@ import traceback
 
 import quart
 
-from .....core import app
+from .....core import app, taskmgr
 from .. import group
 
 
@@ -23,12 +23,26 @@ class PluginsRouterGroup(group.RouterGroup):
                 'plugins': plugins_data
             })
         
-        @self.route('/toggle/<author>/<plugin_name>', methods=['PUT'])
+        @self.route('/<author>/<plugin_name>/toggle', methods=['PUT'])
         async def _(author: str, plugin_name: str) -> str:
             data = await quart.request.json
             target_enabled = data.get('target_enabled')
             await self.ap.plugin_mgr.update_plugin_status(plugin_name, target_enabled)
             return self.success()
+        
+        @self.route('/<author>/<plugin_name>/update', methods=['POST'])
+        async def _(author: str, plugin_name: str) -> str:
+            ctx = taskmgr.TaskContext.new()
+            wrapper = self.ap.task_mgr.create_user_task(
+                self.ap.plugin_mgr.update_plugin(plugin_name, task_context=ctx),
+                kind="plugin-operation",
+                name=f"plugin-update-{plugin_name}",
+                label=f"更新插件 {plugin_name}",
+                context=ctx
+            )
+            return self.success(data={
+                'task_id': wrapper.id
+            })
 
         @self.route('/reorder', methods=['PUT'])
         async def _() -> str:

pkg/api/http/controller/groups/system.py

@@ -1,6 +1,7 @@
 import quart
+import asyncio
 
-from .....core import app
+from .....core import app, taskmgr
 from .. import group
 from .....utils import constants
 
@@ -17,3 +18,23 @@ class SystemRouterGroup(group.RouterGroup):
                     "debug": constants.debug_mode
                 }
             )
+
+        @self.route('/tasks', methods=['GET'])
+        async def _() -> str:
+            task_type = quart.request.args.get("type")
+
+            if task_type == '':
+                task_type = None
+
+            return self.success(
+                data=self.ap.task_mgr.get_tasks_dict(task_type)
+            )
+        
+        @self.route('/tasks/<task_id>', methods=['GET'])
+        async def _(task_id: str) -> str:
+            task = self.ap.task_mgr.get_task_by_id(int(task_id))
+
+            if task is None:
+                return self.http_status(404, 404, "Task not found")
+            
+            return self.success(data=task.to_dict())

pkg/api/http/controller/main.py

@@ -19,39 +19,33 @@ class HTTPController:
     def __init__(self, ap: app.Application) -> None:
         self.ap = ap
         self.quart_app = quart.Quart(__name__)
-        quart_cors.cors(self.quart_app, allow_origin='*')
+        quart_cors.cors(self.quart_app, allow_origin="*")
 
     async def initialize(self) -> None:
         await self.register_routes()
 
     async def run(self) -> None:
-        if self.ap.system_cfg.data['http-api']['enable']:
+        if self.ap.system_cfg.data["http-api"]["enable"]:
+
             async def shutdown_trigger_placeholder():
                 while True:
                     await asyncio.sleep(1)
 
-            # task = asyncio.create_task(self.quart_app.run_task(
-            #     host=self.ap.system_cfg.data['http-api']['host'],
-            #     port=self.ap.system_cfg.data['http-api']['port'],
-            #     shutdown_trigger=shutdown_trigger_placeholder
-            # ))
-            # self.ap.asyncio_tasks.append(task)
-            self.ap.task_mgr.create_task(self.quart_app.run_task(
-                host=self.ap.system_cfg.data['http-api']['host'],
-                port=self.ap.system_cfg.data['http-api']['port'],
-                shutdown_trigger=shutdown_trigger_placeholder
-            ))
-
+            self.ap.task_mgr.create_task(
+                self.quart_app.run_task(
+                    host=self.ap.system_cfg.data["http-api"]["host"],
+                    port=self.ap.system_cfg.data["http-api"]["port"],
+                    shutdown_trigger=shutdown_trigger_placeholder,
+                ),
+                name="http-api-quart",
+            )
 
     async def register_routes(self) -> None:
-        
-        @self.quart_app.route('/healthz')
+
+        @self.quart_app.route("/healthz")
         async def healthz():
-            return {
-                "code": 0,
-                "msg": "ok"
-            }
-        
+            return {"code": 0, "msg": "ok"}
+
         for g in group.preregistered_groups:
             ginst = g(self.ap, self.quart_app)
             await ginst.initialize()