validate agent connection

Author: chriskapp

src/Service/Agent/Validator.php

@@ -20,14 +20,17 @@
 
 namespace Fusio\Impl\Service\Agent;
 
+use Fusio\Engine\ConnectorInterface;
 use Fusio\Engine\Exception\ActionNotFoundException;
+use Fusio\Engine\Exception\ConnectionNotFoundException;
 use Fusio\Engine\Exception\FactoryResolveException;
 use Fusio\Engine\ProcessorInterface;
 use Fusio\Impl\Action\Scheme as ActionScheme;
 use Fusio\Impl\Service\Tenant\UsageLimiter;
 use Fusio\Impl\Table;
 use Fusio\Model\Backend\Agent;
 use PSX\Http\Exception as StatusCode;
+use Symfony\AI\Agent\AgentInterface;
 
 /**
  * Validator
@@ -41,6 +44,7 @@
     public function __construct(
         private Table\Agent $agentTable,
         private Table\Action $actionTable,
+        private ConnectorInterface $connector,
         private ProcessorInterface $processor,
         private UsageLimiter $usageLimiter
     ) {
@@ -64,6 +68,15 @@ public function assert(Agent $agent, int $categoryId, ?string $tenantId, ?Table\
             $this->assertType($type);
         }
 
+        $connection = $agent->getConnection();
+        if ($connection !== null) {
+            $this->assertConnection($connection);
+        } else {
+            if ($existing === null) {
+                throw new StatusCode\BadRequestException('Connection must not be empty');
+            }
+        }
+
         $action = $agent->getAction();
         if ($action !== null) {
             $this->assertAction($action, $categoryId, $tenantId);
@@ -92,6 +105,19 @@ private function assertType(int $type): void
         }
     }
 
+    private function assertConnection(int $connectionId): void
+    {
+        try {
+            $connection = $this->connector->getConnection($connectionId);
+        } catch (ConnectionNotFoundException) {
+            throw new StatusCode\BadRequestException('Provided connection is not available');
+        }
+
+        if (!$connection instanceof AgentInterface) {
+            throw new StatusCode\BadRequestException('Provided an invalid connection type, the connection must be an agent connection');
+        }
+    }
+
     private function assertAction(string $action, int $categoryId, ?string $tenantId): void
     {
         $scheme = ActionScheme::wrap($action);