From ff5ac898da86d73c0bba18f60327933934cd3dbf Mon Sep 17 00:00:00 2001
From: lideming <me@yuuza.net>
Date: Sun, 26 Nov 2023 06:22:45 +0800
Subject: [PATCH] fix: actions

---
 neetbox/daemon/server/_server.py                      |  2 +-
 .../src/components/dashboard/project/actions.tsx      | 11 +++++++----
 neetbox/frontend/src/services/projects.ts             |  2 +-
 tests/client/test.py                                  |  2 ++
 4 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/neetbox/daemon/server/_server.py b/neetbox/daemon/server/_server.py
index 4da0747f..02c87fd0 100644
--- a/neetbox/daemon/server/_server.py
+++ b/neetbox/daemon/server/_server.py
@@ -196,7 +196,7 @@ def handle_ws_message(client, server: WebsocketServer, message):
                     _target_bridge = Bridge(name=_project_name)  # create new bridge for this name
                     __BRIDGES[_project_name] = _target_bridge
                 __BRIDGES[_project_name].cli_ws = client  # assign cli to bridge
-                connected_clients[client["id"]] = (_project_name, "web")
+                connected_clients[client["id"]] = (_project_name, "cli")
                 server.send_message(
                     client=client,
                     msg=json.dumps(
diff --git a/neetbox/frontend/src/components/dashboard/project/actions.tsx b/neetbox/frontend/src/components/dashboard/project/actions.tsx
index af63ec69..bda09a4e 100644
--- a/neetbox/frontend/src/components/dashboard/project/actions.tsx
+++ b/neetbox/frontend/src/components/dashboard/project/actions.tsx
@@ -64,18 +64,21 @@ export function ActionItem({
             {options.description}
           </div>
         )}
-        {options.args.map((argName) => (
-          <Row align="middle" style={{ alignSelf: "stretch" }}>
-            <Col span={6}>
+        {Object.entries(options.args).map(([argName, argType]) => (
+          <Row align="middle" type="flex" justify="space-between" style={{ alignSelf: "stretch" }}>
+            <Col span={4}>
               <Typography.Text ellipsis>{argName}</Typography.Text>
             </Col>
-            <Col span={18}>
+            <Col span={16}>
               <Input
                 size="small"
                 value={args[argName]}
                 onChange={(val) => setArgs({ ...args, [argName]: val })}
               />
             </Col>
+            <Col span={3}>
+              <Typography.Text ellipsis>({argType})</Typography.Text>
+            </Col>
           </Row>
         ))}
         <Button
diff --git a/neetbox/frontend/src/services/projects.ts b/neetbox/frontend/src/services/projects.ts
index f560208a..1101c024 100644
--- a/neetbox/frontend/src/services/projects.ts
+++ b/neetbox/frontend/src/services/projects.ts
@@ -38,7 +38,7 @@ export interface ProjectStatus {
     Record<
       string,
       {
-        args: string[];
+        args: Record<string, string>;
         blocking: boolean;
         description: string;
       }
diff --git a/tests/client/test.py b/tests/client/test.py
index 7d7aa8c8..a0403c73 100644
--- a/tests/client/test.py
+++ b/tests/client/test.py
@@ -1,4 +1,5 @@
 import os
+import time
 from random import random
 from time import sleep
 
@@ -72,6 +73,7 @@ def action_2(text1, text2):
 def action_2(sec):
     sec = int(sec)
     logger.log(f"wait for {sec} sec.")
+    time.sleep(sec)
 
 
 @action(name="shutdown", description="shutdown your process", blocking=True)