jZdZddlmZddlZddlmZmZmZmZm Z dd Z ddZ ddZ gdZ dS)u$Diff-aware line-shift map for cross-edit LSP delta filtering. When an edit deletes or inserts lines in the middle of a file, every diagnostic below the edit point shifts to a new line number. The LSPService delta filter subtracts the pre-edit baseline from the post-edit diagnostics keyed on ``(severity, code, source, message, range)`` — without an adjustment, the shifted-but-otherwise-identical diagnostics look brand-new and the agent gets flooded with noise. The fix used here is the same trick git's blame and unified diff use: build a piecewise-linear map from pre-edit line numbers to post-edit line numbers, then apply that map to baseline diagnostics before the set-difference. Diagnostics whose pre-edit line is in a region the edit deleted return ``None`` and are dropped from the baseline (they genuinely no longer apply). Trade-off vs. dropping range from the key entirely (the previous fix): preserves the "new instance of an identical error at a different line" signal — if the model introduces a second instance of the same error class at a different location, that one will be surfaced as new instead of swallowed by content-only dedup. The map is derived from ``difflib.SequenceMatcher.get_opcodes()`` and exposed as a single callable so callers don't have to reason about diff regions. ) annotationsN)AnyCallableDictListOptionalpre_textstr post_textreturnCallable[[int], Optional[int]]c|r|ng}|r|ng|krdStj|d}|d fd }|S) a|Build a function mapping pre-edit line numbers to post-edit line numbers. Lines are 0-indexed to match the LSP wire format (``range.start.line`` is 0-indexed). The returned callable takes a pre-edit 0-indexed line number and returns the corresponding post-edit 0-indexed line number, or ``None`` if that line was deleted by the edit (no post-edit counterpart exists). Cost: one ``SequenceMatcher.get_opcodes()`` call up front; the returned closure is O(log n) per call (binary search over opcode regions). Cheap enough to call once per write/patch and apply to every baseline diagnostic. c|S)N)lines 4/usr/local/lib/hermes-agent/agent/lsp/range_shift.pyz"build_line_shift..6sDF)abautojunkrintr Optional[int]cD]B\}}}}}||cxkr|kr%nn"|dkr ||z |zcS|dkrdS|dkrdS||krnCr tdtdz ndS)Nequaldeletereplacer)maxlen)rtagi1i2j1j2opcodes post_liness rshiftzbuild_line_shift..shift@s$+   CRRTB'>>"9r>)))(??44)## 44byy /9Bs1c*oo)***dBr)rrr r) splitlinesdifflibSequenceMatcher get_opcodes)r r pre_linessmr(r&r's @@rbuild_line_shiftr/!s *29##%%%rI+4<%%'''"JJ     9 U K K KBnnGCCCCCCC0 LrdiagDict[str, Any]r(Optional[Dict[str, Any]]c|dpi}|dpi}|dpi}t|dd}t|d|}||}|dS||}||}t|} |t|ddd|t|dddd | d<| S) uHReturn a copy of ``diag`` with its line range remapped through ``shift``. Returns ``None`` if the diagnostic's start line maps to ``None`` (the line was deleted by the edit) — caller drops it from the baseline since the diagnostic no longer applies. Both ``start.line`` and ``end.line`` are remapped independently; when only the end maps to ``None`` (rare, multi-line diagnostic straddling the edit boundary) we collapse to a single-line range at the shifted start to keep the diagnostic in the baseline. The original ``diag`` is not mutated. rangestartendrrN character)rr7)r5r6)getrdict) r0r(rngr5r6pre_start_line pre_end_linenew_start_line new_end_lineshifteds rshift_diagnostic_ranger@[s! ((7   !rC GGG   "E ''%.. BC61--..Nswwv~6677LU>**Nt5&&L% 4jjG#UYY{A6677  !SWW[!4455   GG NrbaselineList[Dict[str, Any]]cg}|D]?}t|tst||}|||@|S)zNApply ``shift`` to every diagnostic in ``baseline``, dropping deleted entries.) isinstancer9r@append)rAr(outdr?s rshift_baselinerHsZ!#C   !T""  (E22   JJw    Jr)r/r@rH)r r r r r r )r0r1r(r r r2)rArBr(r r rB)__doc__ __future__rr*typingrrrrrr/r@rH__all__rrrrMs4#"""""666666666666667777t****Z     K J Jr