diff options
| -rw-r--r-- | packages/ui/src/components/session-review.tsx | 14 |
1 files changed, 10 insertions, 4 deletions
diff --git a/packages/ui/src/components/session-review.tsx b/packages/ui/src/components/session-review.tsx index 1ae0b1a13..42c0bc2aa 100644 --- a/packages/ui/src/components/session-review.tsx +++ b/packages/ui/src/components/session-review.tsx @@ -123,11 +123,17 @@ type SessionReviewSelection = { range: SelectedLineRange } -function findSide(element: HTMLElement): "additions" | "deletions" { +function findSide(element: HTMLElement): "additions" | "deletions" | undefined { + const typed = element.closest("[data-line-type]") + if (typed instanceof HTMLElement) { + const type = typed.dataset.lineType + if (type === "change-deletion") return "deletions" + if (type === "change-addition" || type === "change-additions") return "additions" + } + const code = element.closest("[data-code]") - if (!(code instanceof HTMLElement)) return "additions" - if (code.hasAttribute("data-deletions")) return "deletions" - return "additions" + if (!(code instanceof HTMLElement)) return + return code.hasAttribute("data-deletions") ? "deletions" : "additions" } function findMarker(root: ShadowRoot, range: SelectedLineRange) { |