service-forge 0.1.18__py3-none-any.whl → 0.1.39__py3-none-any.whl

service_forge/api/routers/service/service_router.py

@@ -1,10 +1,12 @@
+import json
 import os
 import uuid
 import tempfile
 from fastapi import APIRouter, HTTPException, UploadFile, File, Form
+from fastapi.params import Body, Path
 from fastapi.responses import JSONResponse
 from loguru import logger
-from typing import Optional, TYPE_CHECKING
+from typing import Optional, TYPE_CHECKING, Dict, Any
 from pydantic import BaseModel
 from omegaconf import OmegaConf
 from service_forge.current_service import get_service
@@ -18,18 +20,32 @@ class WorkflowStatusResponse(BaseModel):
     workflows: list[dict]
 
 class WorkflowActionResponse(BaseModel):
-    workflow_id: str
+    workflow_id: Optional[str] = None
     success: bool
     message: str
+    task_id: Optional[str] = None
 
 @service_router.get("/status", response_model=WorkflowStatusResponse)
 async def get_service_status():
     service = get_service()
     if service is None:
         raise HTTPException(status_code=503, detail="Service not initialized")
-    
+    # 排除调试版本
+    try:
+        status = service.get_service_status(exclude_debug=True)
+        return status
+    except Exception as e:
+        logger.error(f"Error getting service status: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
+
+@service_router.get("/workflow/{workflow_id}/status", response_model=dict)
+def get_workflow_data(workflow_id: str):
+    service = get_service()
+    if service is None:
+        raise HTTPException(status_code=503, detail="Service not initialized")
+
     try:
-        status = service.get_service_status()
+        status = service.get_workflow_status(workflow_id)
         return status
     except Exception as e:
         logger.error(f"Error getting service status: {e}")
@@ -42,7 +58,7 @@ async def start_workflow(workflow_id: str):
         raise HTTPException(status_code=503, detail="Service not initialized")
     
     try:
-        success = service.start_workflow_by_id(uuid.UUID(workflow_id))
+        success = service.start_workflow_by_id(workflow_id)
         if success:
             return WorkflowActionResponse(success=True, message=f"Workflow {workflow_id} started successfully")
         else:
@@ -58,7 +74,7 @@ async def stop_workflow(workflow_id: str):
         raise HTTPException(status_code=503, detail="Service not initialized")
     
     try:
-        success = await service.stop_workflow_by_id(uuid.UUID(workflow_id))
+        success = await service.stop_workflow_by_id(workflow_id)
         if success:
             return WorkflowActionResponse(success=True, message=f"Workflow {workflow_id} stopped successfully")
         else:
@@ -67,6 +83,26 @@ async def stop_workflow(workflow_id: str):
         logger.error(f"Error stopping workflow {workflow_id}: {e}")
         raise HTTPException(status_code=500, detail=str(e))
 
+class TriggerWorkflowRequest(BaseModel):
+    kwargs: Optional[Dict[str, Any]] = {}
+
+@service_router.post("/workflow/{workflow_id}/trigger", response_model=WorkflowActionResponse)
+async def trigger_workflow(workflow_id: str = Path(...), request_body: TriggerWorkflowRequest = Body(...)):
+    service = get_service()
+    if service is None:
+        raise HTTPException(status_code=503, detail="Service not initialized")
+
+    try:
+        task_id = service.trigger_workflow_by_id(workflow_id, "", None, **request_body.kwargs)   # Trigger Name实际上没有使用
+        if task_id is not None:
+            return WorkflowActionResponse(workflow_id=workflow_id, task_id=str(task_id), success=True, message=f"Workflow {workflow_id} triggered successfully with task_id {task_id}")
+        else:
+            return WorkflowActionResponse(workflow_id=workflow_id, success=False, message=f"Failed to trigger workflow {workflow_id}")
+    except Exception as e:
+        logger.error(f"Error triggering workflow {workflow_id}: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
+
+
 @service_router.post("/workflow/upload", response_model=WorkflowActionResponse)
 async def upload_workflow_config(
     file: Optional[UploadFile] = File(None),

service_forge/api/routers/trace/trace_router.py

@@ -0,0 +1,326 @@
+from datetime import datetime, timezone
+
+from fastapi import APIRouter, HTTPException
+from httpx import AsyncClient, HTTPStatusError
+from loguru import logger
+
+from service_forge.current_service import get_service
+from service_forge.model.trace import (
+    GetTraceListParams, GetTraceListResponse,
+    GetTraceDetailParams, GetTraceDetailResponse, TraceListItem, TraceDetail, Span, StatusCode, SpanKind
+)
+
+trace_router = APIRouter(prefix="/sdk/trace", tags=["trace"])
+
+
+def _get_signoz_api_base():
+    _service = get_service()
+    if not _service:
+        return None
+    if not _service.config.signoz:
+        return None
+    return _service.config.signoz.api_url
+
+def _get_signoz_api_key():
+    _service = get_service()
+    if not _service:
+        return None
+    if not _service.config.signoz:
+        return None
+    return _service.config.signoz.api_key
+
+@trace_router.post("/list", response_model=GetTraceListResponse)
+async def get_trace_list(params: GetTraceListParams):
+    """
+    获取跟踪列表，调用外部SigNoz API
+    """
+    _service = get_service()
+    internal_service_name = _service.name if _service else ""
+
+    # 构建SQL查询语句
+    where_clauses = []
+
+    # 添加服务名过滤
+    if params.service_name:
+        where_clauses.append(f"resource_string_service$$name = '{params.service_name}'")
+    elif internal_service_name:
+        where_clauses.append(f"resource_string_service$$name = '{internal_service_name}'")
+
+    # 添加工作流名过滤
+    if params.workflow_name:
+        where_clauses.append(f"attributes_string['workflow.name'] = '{params.workflow_name}'")
+
+    # 添加工作流任务ID过滤
+    if params.workflow_task_id:
+        where_clauses.append(f"attributes_string['workflow.task_id'] = '{params.workflow_task_id}'")
+
+    # 添加错误状态过滤
+    if params.has_error is not None:
+        where_clauses.append(f"has_error = {1 if params.has_error else 0}")
+
+    # 组合WHERE子句
+    where_part = ""
+    if where_clauses:
+        where_part = "WHERE " + " AND ".join(where_clauses)
+
+    # 构建完整的查询
+    # 注意：我们需要按trace_id分组，并计算每个trace的span_count
+    # 同时获取最新的span信息（最大timestamp）
+    query = f"""
+    SELECT 
+        trace_id,
+        anyIf(resource_string_service$$name, resource_string_service$$name != '' AND resource_string_service$$name IS NOT NULL) as service_name,
+        anyIf(attributes_string['workflow.name'], attributes_string['workflow.name'] != '' AND attributes_string['workflow.name'] IS NOT NULL) as workflow_name,
+        anyIf(attributes_string['workflow.task_id'], attributes_string['workflow.task_id'] != '' AND attributes_string['workflow.task_id'] IS NOT NULL) as workflow_task_id,
+        max(timestamp) as timestamp,
+        sum(duration_nano) as duration_nano,
+        count() as span_count,
+        max(has_error) as has_error,
+        max(status_code) as status_code
+    FROM signoz_traces.distributed_signoz_index_v3
+    {where_part}
+    GROUP BY trace_id
+    ORDER BY timestamp DESC
+    LIMIT {params.limit or 100}
+    OFFSET {params.offset or 0}
+    """
+
+    # 同时计算总数
+    count_query = f"""
+    SELECT COUNT(DISTINCT trace_id) as total
+    FROM signoz_traces.distributed_signoz_index_v3
+    {where_part}
+    """
+
+    async with AsyncClient() as client:
+        try:
+            current_time = datetime.now(timezone.utc)
+            # 调用外部API
+            response = await client.post(
+                f"{_get_signoz_api_base()}/api/v5/query_range",
+                headers={
+                    "SIGNOZ-API-KEY": _get_signoz_api_key()
+                },
+                json={
+                    "start": int(params.start_time or 0),
+                    "end": int(params.end_time or (current_time.timestamp() * 1000)),
+                    "requestType": "raw",
+                    "compositeQuery": {
+                        "queries": [
+                            {
+                                "type": "clickhouse_sql",
+                                "spec": {
+                                    "name": "query_1",
+                                    "query": query,
+                                    "disabled": False
+                                }
+                            },
+                            {
+                                "type": "clickhouse_sql",
+                                "spec": {
+                                    "name": "query_2",
+                                    "query": count_query,
+                                    "disabled": False
+                                }
+                            }
+                        ]
+                    }
+
+                }
+            )
+
+            response.raise_for_status()  # 检查HTTP状态码
+
+            # 假设外部API返回的数据格式与我们的响应模型一致
+            # 如果不一致，需要在这里进行数据转换
+            data = response.json()
+
+            query_results = data["data"]["data"]["results"]
+            query_1_result = None
+            query_2_result = None
+            for result in query_results:
+                if result["queryName"] == "query_1":
+                    query_1_result = result
+                elif result["queryName"] == "query_2":
+                    query_2_result = result
+
+            traces = [
+                TraceListItem(
+                    trace_id=item["data"].get("trace_id", ""),
+                    service_name=item["data"].get("service_name", params.service_name or internal_service_name),
+                    workflow_name=item["data"].get("workflow_name", params.workflow_name or ""),
+                    workflow_task_id=item["data"].get("workflow_task_id", params.workflow_task_id or ""),
+                    timestamp=item["data"].get("timestamp", current_time.isoformat().replace("+00:00", "Z")),
+                    duration_nano=item["data"].get("duration_nano", 0),
+                    span_count=item["data"].get("span_count", 0),
+                    has_error=item["data"].get("has_error", False),
+                    status_code=item["data"].get("status_code", StatusCode.OK),
+                )
+                for item in query_1_result["rows"] or []
+            ]
+
+            # 转换为内部响应模型并返回
+            return GetTraceListResponse(
+                traces=traces,
+                total=query_2_result["rows"][0]["data"].get("total", len(traces)),
+                limit=params.limit or 100,
+                offset=params.offset or 0,
+            )
+
+        except HTTPStatusError as e:
+            # 处理HTTP错误
+            raise HTTPException(status_code=e.response.status_code, detail=str(e))
+        except Exception as e:
+            # 处理其他错误
+            logger.error(e)
+            raise HTTPException(status_code=500, detail=f"API处理失败: {str(e)}")
+
+
+@trace_router.post("/detail", response_model=GetTraceDetailResponse)
+async def get_trace_detail(params: GetTraceDetailParams):
+    # 构建SQL查询语句
+    where_clauses = [f"trace_id = '{params.trace_id}'"]
+
+    # 添加服务名过滤
+    if params.service_name:
+        where_clauses.append(f"resource_string_service$$name = '{params.service_name}'")
+
+    # 组合WHERE子句
+    where_part = "WHERE " + " AND ".join(where_clauses)
+
+    # 构建完整的查询获取所有span信息
+    span_query = f"""
+    SELECT 
+        trace_id,
+        span_id,
+        parent_span_id,
+        name,
+        kind,
+        timestamp,
+        duration_nano,
+        status_code,
+        has_error,
+        resource_string_service$$name as service_name,
+        attributes_string as attributes_string
+    FROM signoz_traces.distributed_signoz_index_v3
+    {where_part}
+    ORDER BY timestamp ASC
+    """
+
+    count_query = f"""
+    SELECT 
+        count(DISTINCT span_id) as total
+    FROM signoz_traces.distributed_signoz_index_v3
+    {where_part}
+    """
+
+    async with AsyncClient() as client:
+        try:
+            current_time = datetime.now(timezone.utc)
+            # 调用外部API获取span数据
+            response = await client.post(
+                f"{_get_signoz_api_base()}/api/v5/query_range",
+                headers={
+                    "SIGNOZ-API-KEY": _get_signoz_api_key()
+                },
+                json={
+                    "start": 0,  # 获取完整trace，不限制时间范围
+                    "end": int(current_time.timestamp() * 1000),
+                    "requestType": "raw",
+                    "compositeQuery": {
+                        "queries": [
+                            {
+                                "type": "clickhouse_sql",
+                                "spec": {
+                                    "name": "query_1",
+                                    "query": span_query,
+                                    "disabled": False
+                                }
+                            },
+                            {
+                                "type": "clickhouse_sql",
+                                "spec": {
+                                    "name": "query_2",
+                                    "query": count_query,
+                                    "disabled": False
+                                }
+                            },
+                        ]
+                    }
+                }
+            )
+            response.raise_for_status()  # 检查HTTP状态码
+
+            # 解析响应数据
+            data = response.json()
+
+            query_results = data["data"]["data"]["results"]
+            query_1_result = None
+            query_2_result = None
+            for result in query_results:
+                if result["queryName"] == "query_1":
+                    query_1_result = result
+                elif result["queryName"] == "query_2":
+                    query_2_result = result
+
+            _service = get_service()
+            internal_service_name = _service.name if _service else ""
+
+            spans = [
+                Span(
+                    span_id=item["data"].get("span_id", ""),
+                    parent_span_id=item["data"].get("parent_span_id", ""),
+                    name=item["data"].get("name", ""),
+                    kind=item["data"].get("kind", SpanKind.UNKNOWN),
+                    timestamp=item["data"].get("timestamp", current_time.isoformat().replace("+00:00", "Z")),
+                    duration_nano=item["data"].get("duration_nano", 0),
+                    status_code=item["data"].get("status_code", StatusCode.OK),
+                    service_name=item["data"].get("attributes_string", {}).get("service.name", params.service_name or internal_service_name),
+                    workflow_name=item["data"].get("attributes_string", {}).get("workflow.name", ""),
+                    workflow_task_id=item["data"].get("attributes_string", {}).get("workflow.task_id", ""),
+                    node_name=item["data"].get("attributes_string", {}).get("node.name", ""),
+                    attributes=item["data"].get("attributes_string", {}),
+                )
+                for item in query_1_result["rows"] or []
+            ]
+
+            has_error = False
+            for row in (query_1_result["rows"] or []):
+                if row["data"].get("has_error", False):
+                    has_error = True
+                    break
+
+            workflow_name = ""
+            for span in spans:
+                if span.workflow_name:
+                    workflow_name = span.workflow_name
+                    break
+
+            workflow_task_id = ""
+            for span in spans:
+                if span.workflow_task_id:
+                    workflow_task_id = span.workflow_task_id
+                    break
+
+            return GetTraceDetailResponse(
+                trace=TraceDetail(
+                    trace_id=params.trace_id,
+                    service_name=params.service_name or "", # 注入当前服务名
+                    spans=spans,
+                    start_time=spans[0].timestamp if spans else None,
+                    end_time=spans[-1].timestamp if spans else None,
+                    span_count=query_2_result["rows"][0]["data"].get("total", len(spans)),
+                    has_error=has_error,
+                    workflow_name=workflow_name,
+                    workflow_task_id=workflow_task_id,
+                )
+            )
+
+
+
+        except HTTPStatusError as e:
+            # 处理HTTP错误
+            raise HTTPException(status_code=e.response.status_code, detail=str(e))
+        except Exception as e:
+            # 处理其他错误
+            raise HTTPException(status_code=500, detail=f"API处理失败: {str(e)}")

service_forge/api/routers/websocket/websocket_router.py

@@ -1,14 +1,33 @@
-from fastapi import WebSocket, WebSocketDisconnect
+from http.client import HTTPException
+
+from fastapi import WebSocket, WebSocketDisconnect, Form
 from fastapi.routing import APIRouter
 from loguru import logger
 import json
 import uuid
+
+from omegaconf import OmegaConf
+
+from service_forge.current_service import get_service
 from .websocket_manager import websocket_manager
 
 websocket_router = APIRouter()
 
 @websocket_router.websocket("/sdk/ws")
 async def sdk_websocket_endpoint(websocket: WebSocket):
+    # Authenticate WebSocket connection before accepting
+    # Get trusted_domain from app.state if available
+    # trusted_domain = getattr(websocket.app.state, "trusted_domain", "ring.shiweinan.com")
+    # enable_auth = getattr(websocket.app.state, "enable_auth_middleware", True)
+    
+    # if enable_auth:
+    #     from service_forge.api.http_api import authenticate_websocket
+    #     await authenticate_websocket(websocket, trusted_domain)
+    # else:
+    #     # If auth is disabled, set default values
+    #     websocket.state.user_id = websocket.headers.get("X-User-ID", "0")
+    #     websocket.state.auth_token = websocket.headers.get("X-User-Token")
+    
     await websocket.accept()
     try:
         while True:
@@ -61,6 +80,10 @@ async def sdk_websocket_endpoint(websocket: WebSocket):
                             await websocket.send_text(
                                 json.dumps({"error": "Invalid task_id format"})
                             )
+                    # TODO: 支持调试模式下取消订阅后删除debug版本
+                # 调式模式下订阅（客户端无需实现提供task_id，而是提供工作流配置，并由服务端创建debug版本、触发debug版本、登记订阅并返回task_id和workflow_id）
+                elif message_type == "subscribe_debug":
+                    await handle_subscribe_debug_message(websocket, message)
                 else:
                     await websocket.send_text(
                         json.dumps({"error": f"Unknown message type: {message_type}"})
@@ -76,3 +99,48 @@ async def sdk_websocket_endpoint(websocket: WebSocket):
         logger.error(f"SDK WebSocket连接处理异常: {e}")
         await websocket_manager.disconnect(websocket)
 
+
+async def handle_subscribe_debug_message(websocket: WebSocket, message: dict):
+    # 解析debug版本的工作流配置
+    try:
+        workflow_config_str = message.get("workflow_config")
+        config = OmegaConf.to_object(OmegaConf.create(workflow_config_str))
+    except Exception as e:
+        await websocket.send_text(
+            json.dumps({"error": f"Failed to parse workflow config: {str(e)}"})
+        )
+        return
+
+    # 加载debug版本的工作流
+    try:
+        service = get_service()
+        workflow_id = await service.load_workflow_from_config(config=config, debug_version=True)
+    except Exception as e:
+        await websocket.send_text(
+            json.dumps({"error": f"Failed to load workflow of the debug version: {str(e)}"})
+        )
+        return
+
+    # 先指定task_id并订阅
+    task_id = uuid.uuid4()
+    try:
+        success = await websocket_manager.subscribe(websocket, task_id)
+        response = {"success": success, "type": "subscribe_response", "task_id": str(task_id), "workflow_id": str(workflow_id)}
+        await websocket.send_text(json.dumps(response))
+    except Exception as e:
+        await websocket.send_text(
+            json.dumps({"error": f"Failed to subscribe the debug task: {str(e)}"})
+        )
+        return
+
+    # 触发debug版本的工作流
+    try:
+        trigger_args = message.get("trigger_args", {})
+        task_id = service.trigger_workflow_by_id(str(workflow_id), "", assigned_task_id=task_id, **trigger_args)
+        response = {"success": True, "type": "trigger_response", "task_id": str(task_id), "workflow_id": str(workflow_id)}
+        await websocket.send_text(json.dumps(response))
+    except Exception as e:
+        await websocket.send_text(
+            json.dumps({"error": f"Failed to trigger workflow of the debug version: {str(e)}"})
+        )
+        return

service_forge/api/service_studio.py

@@ -0,0 +1,9 @@
+from pathlib import Path
+
+from starlette.staticfiles import StaticFiles
+
+def get_studio_static_files() -> StaticFiles:
+    frontend_dist_path = Path(__file__).parent.parent / "frontend" / "static"
+    return StaticFiles(directory=str(frontend_dist_path), html=True)
+
+studio_static_files = get_studio_static_files()

service_forge/db/database.py

@@ -7,6 +7,7 @@ from typing import AsyncGenerator
 from loguru import logger
 from sqlalchemy.ext.asyncio import AsyncEngine, AsyncSession, async_sessionmaker, create_async_engine
 from service_forge.service_config import ServiceConfig
+from pymongo import AsyncMongoClient
 
 class PostgresDatabase:
     def __init__(
@@ -114,6 +115,7 @@ class MongoDatabase:
         self.mongo_password = mongo_password
         self.mongo_db = mongo_db or ""
         self.client = pymongo.MongoClient(self.database_url)
+        self.async_client = AsyncMongoClient(self.database_url)
         self.test_connection()
 
     @property
@@ -129,6 +131,21 @@ class MongoDatabase:
             logger.error(f"MongoDB connection test failed for database '{self.name}': {e}")
             return False
 
+    async def test_async_connection(self) -> bool:
+        try:
+            await self.async_client.admin.command('ping')
+            logger.info(f"Async MongoDB connection test successful for database '{self.name}'")
+            return True
+        except Exception as e:
+            logger.error(f"Async MongoDB connection test failed for database '{self.name}': {e}")
+            return False
+
+    def get_sync_collection(self, collection_name: str):
+        return self.client[self.mongo_db][collection_name]
+
+    def get_async_collection(self, collection_name: str):
+        return self.async_client[self.mongo_db][collection_name]
+
 class RedisDatabase:
     def __init__(
         self,

service_forge/execution_context.py

@@ -0,0 +1,106 @@
+from __future__ import annotations
+
+import json
+from contextlib import contextmanager
+from contextvars import ContextVar, Token
+from dataclasses import dataclass, field, replace
+from typing import Any, Iterator, Mapping
+
+from loguru import logger
+from opentelemetry.context import Context as OtelContext
+from opentelemetry.trace import Span
+
+"""
+ExecutionContext helper for tracing/logging/state propagation.
+
+Usage constraints (async-aware):
+- Entrypoints must call set_current_context/use_execution_context so spawned coroutines inherit the context; create tasks after setting it.
+- ContextVar does not flow into run_in_executor/thread pools; set_current_context manually in threads if needed.
+- Use with_state to copy + mutate state; instances are immutable to avoid concurrent writes.
+- Logger/span are shared references; set a new ExecutionContext when switching spans to prevent cross-task leakage.
+- State payload is masked for sensitive keys and capped at ~4KB after JSON serialization; oversize payload raises ValueError.
+"""
+
+STATE_SIZE_LIMIT_BYTES = 4096
+SENSITIVE_KEYS = (
+    "password",
+    "passwd",
+    "secret",
+    "token",
+    "api_key",
+    "apikey",
+    "authorization",
+    "auth",
+    "credential",
+    "cookie",
+)
+
+
+def _mask_sensitive_value(key: str, value: Any) -> Any:
+    lower_key = key.lower()
+    if any(sensitive in lower_key for sensitive in SENSITIVE_KEYS):
+        return "***"
+    return value
+
+
+def _validate_state_payload(state: Mapping[str, Any]) -> None:
+    try:
+        payload = json.dumps(state, default=str)
+    except Exception:
+        payload = str(state)
+    if len(payload.encode("utf-8")) > STATE_SIZE_LIMIT_BYTES:
+        raise ValueError(
+            f"ExecutionContext state exceeds {STATE_SIZE_LIMIT_BYTES} bytes after serialization; "
+            "store a compact summary instead of raw payloads."
+        )
+
+
+@dataclass(frozen=True)
+class ExecutionContext:
+    """Shared execution-scoped data passed across API, workflow, and node layers."""
+
+    trace_context: OtelContext | None = None
+    span: Span | None = None
+    logger = logger
+    state: Mapping[str, Any] = field(default_factory=dict)
+    metadata: Mapping[str, Any] = field(default_factory=dict)
+
+    def with_state(self, **updates: Any) -> ExecutionContext:
+        merged_state = {
+            key: _mask_sensitive_value(key, value)
+            for key, value in {**self.state, **updates}.items()
+        }
+        _validate_state_payload(merged_state)
+        return replace(self, state=merged_state)
+
+
+current_context: ContextVar[ExecutionContext | None] = ContextVar(
+    "current_execution_context",
+    default=None,
+)
+
+
+def get_current_context(
+    default: ExecutionContext | None = None,
+) -> ExecutionContext | None:
+    context = current_context.get()
+    if context is None:
+        return default
+    return context
+
+
+def set_current_context(context: ExecutionContext | None) -> Token:
+    return current_context.set(context)
+
+
+def reset_current_context(token: Token) -> None:
+    current_context.reset(token)
+
+
+@contextmanager
+def use_execution_context(context: ExecutionContext) -> Iterator[ExecutionContext]:
+    token = set_current_context(context)
+    try:
+        yield context
+    finally:
+        reset_current_context(token)

service_forge/frontend/static/assets/CreateNewNodeDialog-DkrEMxSH.js

@@ -0,0 +1 @@
+import"./api-DyiqpKJK.js";import{u as S,W as x}from"./WorkflowStudio-FCyhGD4y.js";/* empty css                  *//* empty css               */import{u as B}from"./vue-flow-Dn7R8GPr.js";import{u as O}from"./WorkflowEditor-CcaGGbko.js";import{e as T,C as D,D as F,F as j,G as U,t as W}from"./element-ui-D3x2y3TA.js";import{v as h,i as G,c as f,r as K,w as M,J as v,y as r,K as s,U as a,a2 as _,B as p,x as R,H as A,a9 as H,k as I,u as J}from"./vue-core-DL-LgTX0.js";import{_ as L}from"./index-BMvodlwc.js";import"./ui-components-DQ7-U3pr.js";import"./js-yaml-yTPt38rv.js";import"./elkjs-Dm5QV7uy.js";import"./code-editor-DBSql_sc.js";const P={class:"type-info"},q=h({__name:"CreateNewNodeDialog",props:{pos:{}},setup(b){const y=b,l=G(x.CREATE_NEW_NODE_OPEN),d=S().nodeSchemaMap,N=B(),{addNewNode:E}=O(N),n=f(()=>[...d.values()]),t=K(n.value[0]?.name??"");M(n,()=>{n.value[0]&&(t.value=t.value||n.value[0]?.name)});const i=f(()=>d.get(t.value)),w=()=>{l&&(l.value=!1)},V=()=>{t.value&&(E(t.value,y.pos),l&&(l.value=!1))};return(z,e)=>{const C=F,g=D,u=U,m=j,c=T,k=W;return r(),v(k,{modelValue:J(l),"onUpdate:modelValue":e[1]||(e[1]=o=>I(l)?l.value=o:null),title:"新建节点 · 选择节点类型","align-center":""},{default:s(()=>[a(g,{modelValue:t.value,"onUpdate:modelValue":e[0]||(e[0]=o=>t.value=o),placeholder:n.value[0]?.name},{default:s(()=>[(r(!0),R(A,null,H(n.value,o=>(r(),v(C,{key:o.name,label:o.name,value:o.name},null,8,["label","value"]))),128))]),_:1},8,["modelValue","placeholder"]),p("div",P,[p("div",null,[e[2]||(e[2]=p("p",{class:"type-info-entry-title"},"输入",-1)),a(m,{data:Object.values(i.value?.inputs||{})},{default:s(()=>[a(u,{prop:"name",label:"端口名",width:"140"}),a(u,{prop:"type",label:"数据类型"}),a(u,{prop:"default_value",label:"默认值"})]),_:1},8,["data"])]),p("div",null,[e[3]||(e[3]=p("p",{class:"type-info-entry-title"},"输出",-1)),a(m,{data:Object.values(i.value?.outputs||{})},{default:s(()=>[a(u,{prop:"name",label:"端口名",width:"140"}),a(u,{prop:"type",label:"数据类型"})]),_:1},8,["data"])])])]),footer:s(()=>[a(c,{onClick:w},{default:s(()=>[...e[4]||(e[4]=[_("取消",-1)])]),_:1}),a(c,{type:"primary",onClick:V,disabled:!t.value},{default:s(()=>[...e[5]||(e[5]=[_(" 新建 ",-1)])]),_:1},8,["disabled"])]),_:1},8,["modelValue"])}}}),pe=L(q,[["__scopeId","data-v-ce5c662a"]]);export{pe as default};