fix(web_common): adjust types in lineage component (#5476)

Author: mykalmax

web/common/src/components/Lineage/help.ts

@@ -23,7 +23,9 @@ export function getOnlySelectedNodes<
   TNodeData extends LineageNodeData = LineageNodeData,
   TNodeID extends string = NodeId,
 >(nodeMaps: LineageNodesMap<TNodeData, TNodeID>, selectedNodes: Set<TNodeID>) {
-  return (Object.values(nodeMaps) as LineageNode<TNodeData, TNodeID>[]).reduce(
+  return (
+    Object.values(nodeMaps) satisfies LineageNode<TNodeData, TNodeID>[]
+  ).reduce(
     (acc, node) =>
       selectedNodes.has(node.id) ? { ...acc, [node.id]: node } : acc,
     {} as LineageNodesMap<TNodeData, TNodeID>,

web/common/src/components/Lineage/index.ts

@@ -21,6 +21,7 @@ export * from './node/useNodeMetadata'
 export * from './edge/EdgeWithGradient'
 export * from './edge/FactoryEdgeWithGradient'
 export * from './layout/dagreLayout'
+export * from './layout/help'
 export * from './LineageColumnLevel/ColumnLevelLineageContext'
 export * from './LineageColumnLevel/FactoryColumn'
 export * from './LineageColumnLevel/useColumns'

web/common/src/components/Lineage/stories/ModelNode.tsx

@@ -254,7 +254,7 @@ export const ModelNode = React.memo(function ModelNode({
                     className="p-1 first:border-t-0 h-6"
                     columnLineageData={
                       (
-                        column as Column & {
+                        column satisfies Column & {
                           columnLineageData?: ColumnLevelLineageAdjacencyList<
                             ModelNameType,
                             ColumnName
@@ -287,7 +287,7 @@ export const ModelNode = React.memo(function ModelNode({
                     className="p-1 border-t border-lineage-divider first:border-t-0 h-6"
                     columnLineageData={
                       (
-                        column as Column & {
+                        column satisfies Column & {
                           columnLineageData?: ColumnLevelLineageAdjacencyList<
                             ModelNameType,
                             ColumnName

web/common/src/components/Lineage/stories/dagreLayout.worker.ts

@@ -2,23 +2,37 @@ import {
   type LayoutedGraph,
   type LineageEdgeData,
   type LineageNodeData,
+  type EdgeId,
+  type NodeId,
+  type PortId,
 } from '../utils'
 import { buildLayout } from '../layout/dagreLayout'
 
 self.onmessage = <
   TNodeData extends LineageNodeData = LineageNodeData,
   TEdgeData extends LineageEdgeData = LineageEdgeData,
+  TNodeID extends string = NodeId,
+  TEdgeID extends string = EdgeId,
+  TPortID extends string = PortId,
 >(
-  event: MessageEvent<LayoutedGraph<TNodeData, TEdgeData>>,
+  event: MessageEvent<
+    LayoutedGraph<TNodeData, TEdgeData, TNodeID, TEdgeID, TPortID>
+  >,
 ) => {
   try {
-    const { edges, nodesMap } = buildLayout(event.data)
+    const { edges, nodesMap } = buildLayout<
+      TNodeData,
+      TEdgeData,
+      TNodeID,
+      TEdgeID,
+      TPortID
+    >(event.data)
 
     self.postMessage({
       edges,
       nodesMap,
-    } as LayoutedGraph<TNodeData, TEdgeData>)
-  } catch (outerError) {
-    self.postMessage({ error: outerError } as { error: ErrorEvent })
+    } satisfies LayoutedGraph<TNodeData, TEdgeData, TNodeID, TEdgeID, TPortID>)
+  } catch (error) {
+    self.postMessage({ error })
   }
 }