Ensure pipeline wrapper can be created in validations

jvalue · Jan 12, 2024 · cda9e9a · cda9e9a
1 parent 32c87d9
commit cda9e9a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/libs/language-server/src/lib/validation/checks/block-definition.ts b/libs/language-server/src/lib/validation/checks/block-definition.ts
@@ -31,7 +31,10 @@ function checkPipesOfBlock(
   whatToCheck: 'input' | 'output',
   context: ValidationContext,
 ): void {
-  if (!BlockTypeWrapper.canBeWrapped(block?.type)) {
+  if (
+    !BlockTypeWrapper.canBeWrapped(block?.type) ||
+    !PipelineWrapper.canBeWrapped(block.$container)
+  ) {
     return;
   }
   const blockType = new BlockTypeWrapper(block?.type);
@@ -98,6 +101,7 @@ function collectPipes(
   whatToCheck: 'input' | 'output',
 ): PipeWrapper[] {
   const pipelineWrapper = new PipelineWrapper(block.$container);
+
   let pipes: PipeWrapper[];
   switch (whatToCheck) {
     case 'input': {

diff --git a/libs/language-server/src/lib/validation/checks/pipeline-definition.ts b/libs/language-server/src/lib/validation/checks/pipeline-definition.ts
@@ -22,7 +22,11 @@ function checkStartingBlocks(
   pipeline: PipelineDefinition,
   context: ValidationContext,
 ): void {
+  if (PipelineWrapper.canBeWrapped(pipeline)) {
+    return;
+  }
   const pipelineWrapper = new PipelineWrapper(pipeline);
+
   const startingBlocks = pipelineWrapper.getStartingBlocks();
   if (startingBlocks.length === 0) {
     context.accept(