From 815a580f4f4112fae01c7ebe64903fa2b58223cf Mon Sep 17 00:00:00 2001 From: Georg Schwarz Date: Mon, 15 Jan 2024 15:02:38 +0100 Subject: [PATCH] Improve typing of PipelineWrapper --- .../src/lib/ast/wrappers/pipeline-wrapper.ts | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/libs/language-server/src/lib/ast/wrappers/pipeline-wrapper.ts b/libs/language-server/src/lib/ast/wrappers/pipeline-wrapper.ts index acbc577a3..f5d40c123 100644 --- a/libs/language-server/src/lib/ast/wrappers/pipeline-wrapper.ts +++ b/libs/language-server/src/lib/ast/wrappers/pipeline-wrapper.ts @@ -13,16 +13,15 @@ import { import { AstNodeWrapper } from './ast-node-wrapper'; import { PipeWrapper, createWrappersFromPipeChain } from './pipe-wrapper'; -export class PipelineWrapper - implements AstNodeWrapper +export class PipelineWrapper< + T extends PipelineDefinition | CompositeBlocktypeDefinition, +> implements AstNodeWrapper { - public readonly astNode: PipelineDefinition | CompositeBlocktypeDefinition; + public readonly astNode: T; allPipes: PipeWrapper[] = []; - constructor( - pipesContainer: PipelineDefinition | CompositeBlocktypeDefinition, - ) { + constructor(pipesContainer: T) { this.astNode = pipesContainer; this.allPipes = pipesContainer.pipes.flatMap((pipe) =>