Improve typing of PipelineWrapper

jvalue · Jan 15, 2024 · 815a580 · 815a580
1 parent a224ee9
commit 815a580
Showing 1 changed file with 5 additions and 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
@@ -13,16 +13,15 @@ import {
 import { AstNodeWrapper } from './ast-node-wrapper';
 import { PipeWrapper, createWrappersFromPipeChain } from './pipe-wrapper';
 
-export class PipelineWrapper
-  implements AstNodeWrapper<PipelineDefinition | CompositeBlocktypeDefinition>
+export class PipelineWrapper<
+  T extends PipelineDefinition | CompositeBlocktypeDefinition,
+> implements AstNodeWrapper<T>
 {
-  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) =>