temp [ci skip]

benfdking · benfdking · commit c072cbc49458 · 2025-04-10T15:14:29.000+01:00
diff --git a/vscode/extension/src/webviews/lineagePanel.ts b/vscode/extension/src/webviews/lineagePanel.ts
@@ -139,6 +139,7 @@ export class LineagePanel implements WebviewViewProvider, Disposable {
   <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>
+  ${isProduction() ? '' : `<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/ModelNode.tsx b/vscode/react/src/components/graph/ModelNode.tsx
@@ -46,10 +46,15 @@ export default function ModelNode({
   } = useLineageFlow()
 
   const columns: Column[] = useMemo(() => {
-    const model = models[id]
+    const modelsArray = Object.values(models)
+    const decodedId = decodeURIComponent(id)
+    const model = modelsArray.find((m: Model) => m.fqn === decodedId)
+    if (!model) {
+      throw new Error(`Model not found: ${id}`)
+  }
     const modelColumns = model?.columns ?? []
 
-    Object.keys(lineage[id]?.columns ?? {}).forEach((column: string) => {
+    Object.keys(lineage[decodedId]?.columns ?? {}).forEach((column: string) => {
       const found = modelColumns.find(({ name }) => {
         try {
           return name === decodeURI(column)
@@ -131,11 +136,6 @@ export default function ModelNode({
   const isModelSeed = nodeData.type === EnumLineageNodeModelType.seed
   const isModelUnknown = nodeData.type === EnumLineageNodeModelType.unknown
   const showColumns =
-    (hasSelectedColumns ||
-      nodeData.withColumns ||
-      isMouseOver ||
-      isSelected ||
-      isMainNode) &&
     isArrayNotEmpty(columns) &&
     isFalse(hasHighlightedNodes)
   const isActiveNode =
@@ -151,6 +151,10 @@ export default function ModelNode({
     // isFalse(isModelUnknown)
   const shouldDisableColumns = isFalse(isModelSQL)
 
+  console.log(
+    'showcolumns', showColumns,
+  )
+
   return (
     <div
       onMouseEnter={() => setIsMouseOver(true)}
diff --git a/vscode/react/src/components/graph/context.tsx b/vscode/react/src/components/graph/context.tsx
@@ -121,7 +121,7 @@ export default function LineageFlowProvider({
   handleError,
   handleClickModel,
   children,
-  showColumns = false,
+  showColumns = true,
   showConnected = false,
   showControls = true,
   models,
@@ -157,6 +157,7 @@ export default function LineageFlowProvider({
   const [hasBackground, setHasBackground] = useState(true)
   const [withImpacted, setWithImpacted] = useState(true)
   const [withSecondary, setWithSecondary] = useState(false)
+  console.log('withColumns in provider', withColumns)
 
   const nodesMap = useMemo(
     () =>
