cleaning up lineage

benfdking · benfdking · commit 51a28c69899d · 2025-04-09T19:05:54.000+01:00
[ci skip]
diff --git a/vscode/extension/src/webviews/lineagePanel.ts b/vscode/extension/src/webviews/lineagePanel.ts
@@ -76,8 +76,6 @@ import {
   <meta http-equiv="Content-Security-Policy" content="default-src 'none'; frame-src ${externalAuthority}; connect-src ${externalAuthority}; img-src ${externalAuthority} data:; script-src 'unsafe-inline' ${externalAuthority}; style-src 'unsafe-inline' ${externalAuthority};">
 </head>
 <body>
-  <div>${isProd ? "Production" : "Development"}</div>
-  <div>${externalUrl}</div>
   <iframe src="${externalUrl}" style="width:100%; height:100vh;" frameborder="0" allow="clipboard-read; clipboard-write"></iframe>
 </body>
 </html>        `
diff --git a/vscode/react/src/components/graph/ModelLineage.tsx b/vscode/react/src/components/graph/ModelLineage.tsx
@@ -19,7 +19,7 @@ import ReactFlow, {
 } from 'reactflow'
 import Loading from '@/components/loading/Loading'
 import Spinner from '@/components/logo/Spinner'
-import { createLineageWorker } from '@/workers'
+import { createLineageWorker } from '@/workers/index'
 import { isArrayEmpty, isFalse, isNil, isNotNil } from '@/utils/index'
 import ListboxShow from '@/components/listbox/ListboxShow'
 import clsx from 'clsx'
@@ -62,8 +62,18 @@ export function ModelLineage({
     setUnknownModels,
     models,
     unknownModels,
+    setWithSecondary,
+    setWithConnected,
+    setWithImpacted,
   } = useLineageFlow()
 
+  useEffect(() => {
+    setWithColumns(true)
+    setWithSecondary(true)
+    setWithConnected(true)
+    setWithImpacted(true)
+  }, [setWithSecondary])
+
   const {
     refetch: getModelLineage,
     isFetching: isFetchingModelLineage,
@@ -415,6 +425,13 @@ function GraphControls({ nodes = [] }: { nodes: Node[] }): JSX.Element {
     setHasBackground,
   } = useLineageFlow()
 
+  useEffect(() => {
+    setWithColumns(true)
+    setWithSecondary(true)
+    setWithConnected(true)
+    setWithImpacted(true)
+  }, [setWithSecondary])
+
   const lineageInfoTrigger = useRef<HTMLButtonElement>(null)
 
   const highlightedNodeModels = useMemo(
diff --git a/vscode/react/src/routes/lineage.tsx b/vscode/react/src/routes/lineage.tsx
@@ -92,6 +92,7 @@ export function LineageComponentFromWeb({
         handleClickModel={handleClickModel}
         handleError={handleError}
         models={models}
+        showControls={false}
       >
         <ModelLineage model={model} />
       </LineageFlowProvider>

Original file line number	Diff line number	Diff line change
`@@ -92,6 +92,7 @@ export function LineageComponentFromWeb({`
`92`	`92`	`handleClickModel={handleClickModel}`
`93`	`93`	`handleError={handleError}`
`94`	`94`	`models={models}`
	`95`	`+ showControls={false}`
`95`	`96`	`>`
`96`	`97`	`<ModelLineage model={model} />`
`97`	`98`	`</LineageFlowProvider>`