temp

Author: benfdking

vscode/extension/src/webviews/lineagePanel.ts

@@ -65,7 +65,6 @@ export class LineagePanel implements WebviewViewProvider, Disposable {
     // Set up message listener for events from the iframe
     webviewView.webview.onDidReceiveMessage(
       async request => {
-        console.log('request', request)
         if (!request) {
           return
         }
@@ -85,7 +84,6 @@ export class LineagePanel implements WebviewViewProvider, Disposable {
             break
           }
           case 'rpcRequest': {
-            console.log('rpcRequest', message.payload)
             const payload: RPCRequest = message.payload
             const requestId = payload.requestId
             switch (payload.method) {

vscode/react/src/api/index.ts

@@ -70,7 +70,14 @@ export function useApiModelLineage(
   return useQuery({
     queryKey: ['/api/lineage', modelName],
     queryFn: async ({ signal }) => {
-      return await modelLineageApiLineageModelNameGet(modelName, { signal })
+      try {
+      const response = await modelLineageApiLineageModelNameGet(modelName, { signal })
+        console.log('lineage response', response)
+        return response
+      } catch (error) {
+        console.error('error fetching lineage', error)
+        throw error
+      }
     },
   })
 }

vscode/react/src/components/graph/ModelLineage.tsx

@@ -74,8 +74,9 @@ export function ModelLineage({
     setWithImpacted(true)
   }, [setWithSecondary])
 
-  const { refetch: getModelLineage, isFetching: isFetchingModelLineage } =
+  const { refetch: getModelLineage, isFetching: isFetchingModelLineage, data, error } =
     useApiModelLineage(model.name)
+  console.log('useApiModelLineage', isFetchingModelLineage, data, error)
   const { isFetching: isFetchingModels } = useApiModels()
 
   const [isMergingModels, setIsMergingModels] = useState(false)
@@ -163,12 +164,13 @@ export function ModelLineage({
     }
   }
 
+  console.log('isFetching', isFetchingModelLineage, isFetchingModels, isMergingModels)
   const isFetching =
     isFetchingModelLineage || isFetchingModels || isMergingModels
 
   return (
     <div className="relative h-full w-full overflow-hidden">
-      {isFetching && (
+      {/* {isFetching && (
         <div className="absolute top-0 left-0 z-10 flex justify-center items-center w-full h-full">
           <span className="absolute w-full h-full z-10 bg-transparent-20 backdrop-blur-lg"></span>
           <Loading className="inline-block z-10">
@@ -178,7 +180,7 @@ export function ModelLineage({
             </h3>
           </Loading>
         </div>
-      )}
+      )} */}
       <ReactFlowProvider>
         <ModelColumnLineage />
       </ReactFlowProvider>

vscode/react/src/components/graph/context.tsx

@@ -104,7 +104,7 @@ export const LineageFlowContext = createContext<LineageFlow>({
   setActiveEdges: () => {},
   handleClickModel: () => {},
   setManuallySelectedColumn: () => {},
-  handleError: () => {},
+  handleError: (error) => console.error(error),
   setLineage: () => {},
   setLineageCache: () => {},
   isActiveColumn: () => false,

vscode/react/src/pages/lineage.tsx

@@ -16,7 +16,6 @@ import type { VSCodeEvent } from '@bus/callbacks'
 import { URI } from 'vscode-uri'
 import type { Model } from '@/api/client'
 import { useRpc } from '@/utils/rpc'
-import { isErr } from '@bus/result'
 
 export function LineagePage() {
   const { emit } = useEventBus()
@@ -76,9 +75,8 @@ function Lineage() {
 
   const { data: models, isLoading: isLoadingModels } = useApiModels()
   const rpc = useRpc()
-
   React.useEffect(() => {
-    const fetchFirstModelIfNotSet = async (
+    const fetchFirstTimeModelIfNotSet = async (
       models: Model[],
     ): Promise<string | undefined> => {
       console.log('fetchFirstModelIfNotSet', models)
@@ -101,8 +99,9 @@ function Lineage() {
       return undefined
     }
     if (selectedModel === undefined && Array.isArray(models)) {
-      fetchFirstModelIfNotSet(models).then(modelName => {
+      fetchFirstTimeModelIfNotSet(models).then(modelName => {
         if (modelName && selectedModel === undefined) {
+          console.log('setting selected model', modelName, selectedModel, models)
           setSelectedModel(modelName)
         }
       })
@@ -186,7 +185,7 @@ export function LineageComponentFromWeb({
   }
 
   function handleError(error: any): void {
-    console.log(error)
+    console.error(error)
   }
 
   const model = models[selectedModel]

vscode/react/src/utils/rpc.ts

@@ -17,7 +17,6 @@ export const callRpc = async <T extends keyof RPCMethods>(
   return new Promise((resolve, reject) => {
     const requestId = `query_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`
     const messageHandler = (event: MessageEvent) => {
-      console.log('messageHandler', event)
       if (event.data) {
         const eventData = event.data as CallbackEvent
         if (eventData.key !== 'rpcResponse') {
@@ -28,9 +27,8 @@ export const callRpc = async <T extends keyof RPCMethods>(
         }
 
         // Remove the listener once we get our response
-        window.removeEventListener('message', messageHandler)
+        // window.removeEventListener('message', messageHandler)
         const payload = eventData.payload.result
-        console.log('resolved payload', payload)
         return resolve(payload)
       }
     }
@@ -47,7 +45,7 @@ export const callRpc = async <T extends keyof RPCMethods>(
 
     // Set a timeout to prevent hanging promises
     setTimeout(() => {
-      window.removeEventListener('message', messageHandler)
+    //   window.removeEventListener('message', messageHandler)
       reject(new Error('Query request timed out'))
     }, 30000) // 30 second timeout
   })