chore(vscode): simplify types (#4456)

Author: benfdking

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

@@ -8,6 +8,7 @@ import { Position, type NodeProps } from 'reactflow'
 import { type Column } from '@/api/client'
 import ModelNodeHeaderHandles from './ModelNodeHeaderHandles'
 import ModelColumns from './ModelColumns'
+import type { Model } from '@/domain/model'
 
 export const EnumLineageNodeModelType = {
   ...ModelType,

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

@@ -9,6 +9,9 @@ import {
 import { getNodeMap, hasActiveEdge, hasActiveEdgeConnector } from './help'
 import { EnumSide } from './types'
 import { type Node } from 'reactflow'
+import type { Lineage } from '@/domain/lineage'
+import type { ModelSQLMeshModel } from '@/domain/sqlmesh-model'
+import type { Model } from '@/domain/model'
 export interface Connections {
   left: string[]
   right: string[]
@@ -19,10 +22,6 @@ export type ActiveNodes = Set<string>
 export type SelectedNodes = Set<string>
 export type HighlightedNodes = Record<string, string[]>
 
-type Lineage = any
-type ModelSQLMeshModel = any
-type Model = any
-
 interface LineageFlow {
   lineage: Record<string, Lineage>
   lineageCache?: Record<string, Lineage>
@@ -33,7 +32,7 @@ interface LineageFlow {
   selectedNodes: SelectedNodes
   selectedEdges: any[]
   models: Record<string, Model>
-  unknownModels: Record<string, Model>
+  unknownModels: Set<string>
   connections: Map<string, Connections>
   withConnected: boolean
   withColumns: boolean
@@ -84,7 +83,7 @@ export const LineageFlowContext = createContext<LineageFlow>({
   mainNode: undefined,
   activeEdges: new Map(),
   activeNodes: new Set(),
-  models: new Map(),
+  models: {},
   unknownModels: new Set(),
   manuallySelectedColumn: undefined,
   connections: new Map(),

vscode/react/src/components/graph/help.ts

@@ -17,8 +17,7 @@ import {
 } from './ModelNode'
 import type { Lineage } from '@/domain/lineage'
 import type { ConnectedNode } from '@/workers/lineage'
-
-type ModelSQLMeshModel = any
+import type { ModelSQLMeshModel } from '@/domain/sqlmesh-model'
 
 export interface GraphNodeData {
   label: string

vscode/react/src/domain/model.ts

@@ -1,11 +1,14 @@
-interface Model {
+import type { Column } from '@/api/client'
+
+export interface Model {
+  displayName: string
   name: string
   fqn: string
   path: string
   dialect: string
   full_path: string
   // type: ModelType
-  // columns: Column[]
+  columns: Column[]
   // default_catalog?: ModelDefaultCatalog
   // description?: ModelDescription
   // sql?: ModelSql

vscode/react/src/pages/lineage.tsx

@@ -14,6 +14,7 @@ import { ModelSQLMeshModel } from '@/domain/sqlmesh-model'
 import { useEventBus } from '@/hooks/eventBus'
 import type { VSCodeEvent } from '@bus/callbacks'
 import { URI } from 'vscode-uri'
+import type { Model } from '@/domain/model'
 
 export function LineagePage() {
   const { emit } = useEventBus()