Fixes <2370>: made required changes to accommodate tempLineageTable in the Lineage grapgh Tab (#27709)

* added necessary changes to accomodate tempLineageTables nodes

* updated code as per comments from Gitar

---------

Co-authored-by: Satender <sommy@Satenders-MacBook-Pro.local>

Author: satender-kumar-collate

openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityLineage/LineageNodeLabelV1.tsx

@@ -84,9 +84,11 @@ const EntityLabel = ({ node }: Pick<LineageNodeLabelProps, 'node'>) => {
         childrenCount > 0 ? 'with-footer' : ''
       )}>
       <Col className="d-flex items-center" flex="auto">
-        <div className="d-flex entity-service-icon m-r-xs">
-          {getServiceIcon(node)}
-        </div>
+        {!node.isTempTable && (
+          <div className="d-flex entity-service-icon m-r-xs">
+            {getServiceIcon(node)}
+          </div>
+        )}
         <Space align="start" className="flex-1" direction="vertical" size={0}>
           <Typography.Text
             className="m-b-0 d-block text-left entity-header-display-name text-md font-medium w-54"

openmetadata-ui/src/main/resources/ui/src/components/Lineage/Lineage.interface.ts

@@ -21,7 +21,10 @@ import { MessageSchemaObject } from '../../generated/entity/data/topic';
 import { EntityReference } from '../../generated/entity/type';
 import { TagLabel } from '../../generated/tests/testCase';
 import { APISchema } from '../../generated/type/apiSchema';
-import { ColumnLineage } from '../../generated/type/entityLineage';
+import {
+  ColumnLineage,
+  TempLineageTable,
+} from '../../generated/type/entityLineage';
 import {
   SearchSourceAlias,
   TableSearchSource,
@@ -65,6 +68,7 @@ export interface EdgeDetails {
   pipelineEntityType?: EntityType.PIPELINE | EntityType.STORED_PROCEDURE;
   docId?: string;
   extraInfo?: EdgeDetails;
+  tempLineageTables?: TempLineageTable[];
 }
 
 export interface ColumnLevelLineageNode
@@ -168,4 +172,5 @@ export interface LineageNodeType
   responseSchema?: APISchema;
   requestSchema?: APISchema;
   fields?: SearchIndexField[];
+  isTempTable?: boolean;
 }

openmetadata-ui/src/main/resources/ui/src/context/LineageProvider/LineageProvider.tsx

@@ -1231,6 +1231,10 @@ const LineageProvider = ({ children }: LineageProviderProps) => {
         return;
       }
 
+      if (node.data?.node?.isTempTable) {
+        return;
+      }
+
       if (node.type === EntityLineageNodeType.LOAD_MORE) {
         selectLoadMoreNode(node);
       } else {

openmetadata-ui/src/main/resources/ui/src/utils/CanvasUtils.test.ts

@@ -11,6 +11,7 @@
  *  limitations under the License.
  */
 import { Edge, Node, Viewport } from 'reactflow';
+import { EntityType } from '../enums/entity.enum';
 import {
   BoundingBox,
   boundsIntersect,
@@ -218,11 +219,12 @@ describe('CanvasUtils', () => {
 
         const result = getEdgeCoordinates(edge, sourceNode, targetNode);
 
+        // Y uses node.height (100) from createMockNode, so midpoint = 50
         expect(result).toEqual({
           sourceX: 400,
-          sourceY: 33,
+          sourceY: 50,
           targetX: 490,
-          targetY: 33,
+          targetY: 50,
         });
       });
 
@@ -237,9 +239,9 @@ describe('CanvasUtils', () => {
 
         expect(result).not.toBeNull();
         expect(result?.sourceX).toBe(500);
-        expect(result?.sourceY).toBe(233);
+        expect(result?.sourceY).toBe(250); // 200 + 100/2
         expect(result?.targetX).toBe(590);
-        expect(result?.targetY).toBe(333);
+        expect(result?.targetY).toBe(350); // 300 + 100/2
       });
     });
 
@@ -391,6 +393,54 @@ describe('CanvasUtils', () => {
         expect(result?.targetX).toBe(490);
       });
     });
+
+    describe('temp lineage nodes', () => {
+      const createTempNode = (id: string, measuredHeight: number): Node => ({
+        id,
+        position: { x: 0, y: 0 },
+        data: {
+          node: {
+            id,
+            name: id,
+            entityType: EntityType.TABLE,
+            isTempTable: true,
+            columns: [],
+          },
+          isRootNode: false,
+        },
+        width: 400,
+        height: measuredHeight,
+      });
+
+      it('uses measured node.height for temp node entity-level edge', () => {
+        const edge = createMockEdge('edge1', 'temp_staging', 'node2', false);
+        const tempNode = createTempNode('temp_staging', 80);
+        const targetNode = createMockNode('node2');
+        targetNode.position = { x: 500, y: 0 };
+
+        const result = getEdgeCoordinates(edge, tempNode, targetNode);
+
+        expect(result).not.toBeNull();
+        expect(result?.sourceY).toBe(40); // 0 + 80/2 (measured height), not 33 (getNodeHeight formula)
+        expect(result?.targetY).toBe(50); // 0 + 100/2
+      });
+
+      it('centers edge at actual midpoint when measured height differs from computed height', () => {
+        const edge = createMockEdge('edge1', 'node1', 'node2', false);
+        const sourceNode = createMockNode('node1');
+        sourceNode.height = 150;
+        sourceNode.position = { x: 0, y: 100 };
+        const targetNode = createMockNode('node2');
+        targetNode.height = 200;
+        targetNode.position = { x: 500, y: 100 };
+
+        const result = getEdgeCoordinates(edge, sourceNode, targetNode);
+
+        expect(result).not.toBeNull();
+        expect(result?.sourceY).toBe(175); // 100 + 150/2
+        expect(result?.targetY).toBe(200); // 100 + 200/2
+      });
+    });
   });
 
   describe('getEdgeBounds', () => {
@@ -414,7 +464,8 @@ describe('CanvasUtils', () => {
       expect(result).not.toBeNull();
       expect(result!.minX).toBeLessThan(351);
       expect(result!.maxX).toBeGreaterThan(500);
-      expect(result!.minY).toBeLessThan(0);
+      // sourceY = 50 (node.height=100 / 2), padding=50 → minY = 0
+      expect(result!.minY).toBeLessThanOrEqual(0);
       expect(result!.maxY).toBeGreaterThan(100);
     });
   });

openmetadata-ui/src/main/resources/ui/src/utils/CanvasUtils.ts

@@ -93,11 +93,19 @@ const getBaseNodeHeightFromType = (
   return isRootNode ? baseHeight + 10 : baseHeight;
 };
 
+const getNodeYPadding = (node: Node): number => {
+  const { children } = getEntityChildrenAndLabel(node.data.node);
+
+  const sourceYPadding = children.length > 0 ? 48 : 0;
+
+  return sourceYPadding;
+};
+
 export function getNodeHeight(
   node: Node,
-  isColumnLineage: boolean,
+  isColumnLineage?: boolean,
   columnCount?: number
-) {
+): number {
   const isRootNode = node.data?.isRootNode ?? false;
 
   const visibleColumnCount = isColumnLineage
@@ -124,15 +132,6 @@ export function getNodeHeight(
   return height;
 }
 
-const getNodeYPadding = (node: Node): number => {
-  const { children } = getEntityChildrenAndLabel(node.data.node);
-
-  const sourceYPadding = children.length > 0 ? 48 : 0;
-
-  // Add padding for the node's border
-  return sourceYPadding;
-};
-
 interface ColumnLineageData {
   columnIds: string[];
   columnIndex: number;
@@ -252,10 +251,10 @@ function getColumnLineageCoordinates(
 function getEntityLineageCoordinates(
   sourceNode: Node,
   targetNode: Node,
-  isColumnLineage: boolean
+  _isColumnLineage: boolean
 ): EdgeCoordinates {
-  const sourceHeight = getNodeHeight(sourceNode, isColumnLineage, 0);
-  const targetHeight = getNodeHeight(targetNode, isColumnLineage, 0);
+  const sourceHeight = sourceNode.height ?? 0;
+  const targetHeight = targetNode.height ?? 0;
 
   return {
     sourceX: sourceNode.position.x + (sourceNode.width ?? 0),