`gjhdZddlmZddlZddlmZmZmZmZm Z d dZ d dZ d dZ gdZ y) 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)AnyCallableDictListOptionalc|r|jng}|r|jng|k(rdStj|d}|jdfd }|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 H/root/.hermes/venv/lib/python3.12/site-packages/agent/lsp/range_shift.pyz"build_line_shift..6sDF)abautojunkcD]:\}}}}}||cxkr|kr nn|dk(r ||z |zcS|dk(ry|dk(ry||ks:nrtdtdz SdS)Nequaldeletereplacer)maxlen)r tagi1i2j1j2opcodes post_liness r shiftzbuild_line_shift..shift@s$+  CRRTB'>"9r>)(?)# by $/9s1c*o)*BdBr)r intreturnz Optional[int]) splitlinesdifflibSequenceMatcher get_opcodes)pre_text post_text pre_linessmr!rr s @@r build_line_shiftr,!sh *2##%rI+4%%'"JJ    9 U KBnnGC0 Lrc|jdxsi}|jdxsi}|jdxsi}t|jdd}t|jd|}||}|y||}||}t|} |t|jddd|t|jdddd | 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. rangestartendr rN character)r r1)r/r0)getr"dict) diagr!rngr/r0pre_start_line pre_end_linenew_start_line new_end_lineshifteds r shift_diagnostic_ranger;[s ((7  !rC GGG  "E ''%. BC61-.Nswwv~67L>*N&L% 4jG#UYY{A67 !SWW[!45 GG Nrczg}|D]3}t|tst||}|#|j|5|S)zNApply ``shift`` to every diagnostic in ``baseline``, dropping deleted entries.) isinstancer3r;append)baseliner!outdr:s r shift_baselinerBsJ!#C  !T" (E2   JJw   Jr)r,r;rB)r(strr)rCr#Callable[[int], Optional[int]])r4zDict[str, Any]r!rDr#zOptional[Dict[str, Any]])r?List[Dict[str, Any]]r!rDr#rE)__doc__ __future__rr%typingrrrrrr,r;rB__all__r rr rJsQ4#667t*"@*E]*Z 8 =Q  Kr