From 766e69bbf1a7502752bcc4989fb3da0e805a0925 Mon Sep 17 00:00:00 2001
From: filtered <176114999+webfiltered@users.noreply.github.com>
Date: Sat, 22 Mar 2025 06:10:26 +1100
Subject: [PATCH] Fix regression: shift-drag new link from link line (#814)

- Resolves #375
- Resolves https://github.com/Comfy-Org/ComfyUI_frontend/issues/2941
---
 src/canvas/LinkConnector.ts | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/canvas/LinkConnector.ts b/src/canvas/LinkConnector.ts
index d750ccd683..496d2b42ea 100644
--- a/src/canvas/LinkConnector.ts
+++ b/src/canvas/LinkConnector.ts
@@ -11,7 +11,6 @@ import { LiteGraph } from "@/litegraph"
 import { LLink } from "@/LLink"
 import { LinkDirection } from "@/types/globalEnums"
 
-import { getNodeOutputOnPos } from "./measureSlots"
 import { MovingRenderLink } from "./MovingRenderLink"
 import { ToInputRenderLink } from "./ToInputRenderLink"
 import { ToOutputRenderLink } from "./ToOutputRenderLink"
@@ -246,10 +245,10 @@ export class LinkConnector {
     const node = network.getNodeById(linkSegment.origin_id)
     if (!node) return
 
-    const slot = getNodeOutputOnPos(node, linkSegment._pos[0], linkSegment._pos[1])?.output
+    const slot = node.outputs.at(linkSegment.origin_slot)
     if (!slot) return
 
-    const reroute = linkSegment.parentId ? network.reroutes.get(linkSegment.parentId) : undefined
+    const reroute = network.getReroute(linkSegment.parentId)
     if (!reroute) return
 
     const renderLink = new ToInputRenderLink(network, node, slot, reroute)
@@ -257,6 +256,8 @@ export class LinkConnector {
     this.renderLinks.push(renderLink)
 
     state.connectingTo = "input"
+
+    this.#setLegacyLinks(false)
   }
 
   /**