@@ -28,23 +28,7 @@ export interface GraphNodeData {
2828 [ key : string ] : any
2929}
3030
31- export {
32- getNodeMap ,
33- getEdges ,
34- createGraphLayout ,
35- mergeLineageWithColumns ,
36- mergeConnections ,
37- getLineageIndex ,
38- getActiveNodes ,
39- getUpdatedNodes ,
40- getUpdatedEdges ,
41- hasActiveEdge ,
42- hasActiveEdgeConnector ,
43- getModelNodeTypeTitle ,
44- getModelAncestors ,
45- }
46-
47- function createGraphLayout ( {
31+ export function createGraphLayout ( {
4832 nodesMap,
4933 nodes = [ ] ,
5034 edges = [ ] ,
@@ -97,7 +81,7 @@ function createGraphLayout({
9781 }
9882}
9983
100- function getEdges ( lineage : Record < string , Lineage > = { } ) : Edge [ ] {
84+ export function getEdges ( lineage : Record < string , Lineage > = { } ) : Edge [ ] {
10185 const modelNames = Object . keys ( lineage )
10286 const outputEdges : Edge [ ] = [ ]
10387
@@ -151,7 +135,7 @@ function getEdges(lineage: Record<string, Lineage> = {}): Edge[] {
151135 return outputEdges
152136}
153137
154- function getNodeMap ( {
138+ export function getNodeMap ( {
155139 lineage,
156140 models,
157141 unknownModels,
@@ -321,7 +305,7 @@ function createGraphEdge<TData = any>(
321305 return output
322306}
323307
324- function mergeLineageWithColumns (
308+ export function mergeLineageWithColumns (
325309 currentLineage : Record < string , Lineage > = { } ,
326310 newLineage : Record < string , Record < string , LineageColumn > > = { } ,
327311) : Record < string , Lineage > {
@@ -380,7 +364,7 @@ function mergeLineageWithColumns(
380364 return currentLineage
381365}
382366
383- function mergeConnections (
367+ export function mergeConnections (
384368 connections : Map < string , Connections > ,
385369 lineage : Record < string , Record < string , LineageColumn > > = { } ,
386370) : {
@@ -472,7 +456,7 @@ function mergeConnections(
472456 }
473457}
474458
475- function getLineageIndex ( lineage : Record < string , Lineage > = { } ) : string {
459+ export function getLineageIndex ( lineage : Record < string , Lineage > = { } ) : string {
476460 return Object . keys ( lineage )
477461 . reduce ( ( acc : string [ ] , key ) => {
478462 const { models = [ ] , columns = { } } = lineage [ key ] !
@@ -496,7 +480,7 @@ function getLineageIndex(lineage: Record<string, Lineage> = {}): string {
496480 . join ( '' )
497481}
498482
499- function getModelAncestors (
483+ export function getModelAncestors (
500484 lineage : Record < string , Lineage > = { } ,
501485 name : string ,
502486 output = new Set < string > ( ) ,
@@ -513,7 +497,7 @@ function getModelAncestors(
513497 return output
514498}
515499
516- function getActiveNodes (
500+ export function getActiveNodes (
517501 edges : Edge [ ] = [ ] ,
518502 activeEdges : ActiveEdges ,
519503 selectedEdges : ConnectedNode [ ] ,
@@ -563,7 +547,7 @@ function getActiveNodes(
563547 )
564548}
565549
566- function getUpdatedEdges (
550+ export function getUpdatedEdges (
567551 edges : Edge [ ] = [ ] ,
568552 connections : Map < string , Connections > ,
569553 activeEdges : ActiveEdges ,
@@ -655,7 +639,7 @@ function getUpdatedEdges(
655639 return tempEdges
656640}
657641
658- function getUpdatedNodes (
642+ export function getUpdatedNodes (
659643 nodes : Node [ ] = [ ] ,
660644 activeNodes : Set < string > ,
661645 mainNode : string ,
@@ -696,7 +680,7 @@ function getUpdatedNodes(
696680 } )
697681}
698682
699- function hasActiveEdge (
683+ export function hasActiveEdge (
700684 activeEdges : ActiveEdges = new Map ( ) ,
701685 [ leftConnect , rightConnect ] : [
702686 string | undefined | null ,
@@ -720,14 +704,14 @@ function hasActiveEdge(
720704 return inLeft || inRight
721705}
722706
723- function hasActiveEdgeConnector (
707+ export function hasActiveEdgeConnector (
724708 activeEdges : ActiveEdges = new Map ( ) ,
725709 connector : string ,
726710) : boolean {
727711 return ( activeEdges . get ( connector ) ?? [ ] ) . length > 0
728712}
729713
730- function getModelNodeTypeTitle ( type : LineageNodeModelType ) : string {
714+ export function getModelNodeTypeTitle ( type : LineageNodeModelType ) : string {
731715 if ( type === EnumLineageNodeModelType . python ) return 'PYTHON'
732716 if ( type === EnumLineageNodeModelType . sql ) return 'SQL'
733717 if ( type === EnumLineageNodeModelType . seed ) return 'SEED'
0 commit comments