fix cll sytyling for source/expressions

Author: mykalmax

web/common/src/components/Lineage/LineageColumnLevel/FactoryColumn.tsx

@@ -65,7 +65,10 @@ export function FactoryColumn<
     error?: Error | null
     renderError?: (error: Error) => React.ReactNode
     renderExpression?: (expression: string) => React.ReactNode
-    renderSource?: (source: string) => React.ReactNode
+    renderSource?: (
+      source: string,
+      expression?: string | null,
+    ) => React.ReactNode
     onClick?: () => void
     onCancel?: () => void
   }) {
@@ -155,7 +158,7 @@ export function FactoryColumn<
               className="p-0 min-w-[30rem] max-w-xl bg-lineage-model-column-source-background"
               delayDuration={0}
             >
-              {renderSource(column.source)}
+              {renderSource(column.source, column.expression)}
             </Tooltip>
           )}
           {column?.expression && renderExpression && (

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

@@ -38,12 +38,7 @@ export function getTransformedNodes<
 >(
   adjacencyListKeys: TAdjacencyListKey[],
   lineageDetails: LineageDetails<TAdjacencyListKey, TDetailsNode>,
-  transformNode: TransformNodeFn<
-    TDetailsNode,
-    TNodeData,
-    TAdjacencyListKey,
-    TNodeID
-  >,
+  transformNode: TransformNodeFn<TDetailsNode, TNodeData, TNodeID>,
 ): LineageNodesMap<TNodeData, TNodeID> {
   const nodesCount = adjacencyListKeys.length
   const nodesMap: LineageNodesMap<TNodeData, TNodeID> = Object.create(null)
@@ -53,7 +48,6 @@ export function getTransformedNodes<
     const encodedNodeId = toNodeID<TNodeID>(adjacencyListKey)
     nodesMap[encodedNodeId] = transformNode(
       encodedNodeId,
-      adjacencyListKey,
       lineageDetails[adjacencyListKey],
     )
   }

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

@@ -120,7 +120,7 @@ export const ModelLineage = ({
   }, [adjacencyListKeysColumnLevel, adjacencyList])
 
   const transformNode = React.useCallback(
-    (nodeId: ModelNodeId, _: ModelName, detail: ModelLineageNodeDetails) => {
+    (nodeId: ModelNodeId, detail: ModelLineageNodeDetails) => {
       const columns = detail.columns
 
       const node = createNode('node', nodeId, {

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

@@ -55,13 +55,8 @@ export type PathType = 'bezier' | 'smoothstep' | 'step' | 'straight'
 export type TransformNodeFn<
   TData,
   TNodeData extends LineageNodeData = LineageNodeData,
-  TAdjacencyListKey extends string = string,
   TNodeID extends string = NodeId,
-> = (
-  nodeId: TNodeID,
-  adjacencyListKey: TAdjacencyListKey,
-  data: TData,
-) => LineageNode<TNodeData, TNodeID>
+> = (nodeId: TNodeID, data: TData) => LineageNode<TNodeData, TNodeID>
 
 export type TransformEdgeFn<
   TEdgeData extends LineageEdgeData = LineageEdgeData,