Support vim screen line motions

Author: SqlRush

docs/claude-code-go-rewrite-plan.md

@@ -238,6 +238,7 @@ test/parity/                 # golden tests against TS/official behavior
 - 本轮补充：tail、byte-tail、window 和 metadata-only transcript loader 也改用 canonical message type 处理 progress bridge 与 compact-boundary，`progress_update` 和 `system-event` 等别名不再只在 full loader 路径生效。
 - 本轮补充：tail、byte-tail、window 和 streaming transcript search 现在复用 full/index loader 的 wrapped record 展开逻辑，可从 JSON:API/resource、GraphQL edge/node 和 collection/list wrapper 中恢复 transcript 批次，并保留 progress bridge。
 - 本轮补充：TUI Vim prompt editing 增加基础 visual/visual-line 模式，支持 `v`/`V` 进入选择、motion 扩展 selection、visual `o` 切换 active end、visual `<`/`>` 行缩进/反缩进、visual `~` 大小写切换、visual `u`/`U` 小写/大写转换、`y`/`d`/`c` 以及常用 visual `x`/`s` aliases 作用于选择范围、Esc 回到 normal，并让 interaction script 可用 `visual`/`visualLine` 断言当前 mode。
+- 本轮补充：TUI Vim prompt editing 增加 prompt-local `H`/`M`/`L` screen-line motions，normal/visual/operator 路径都可用，`H`/`L` 支持 count 定位首/末附近行，operator line-motion 可 dot-repeat。
 - 本轮补充：TUI Vim prompt editing 支持 normal-mode `gv` 重新进入上一次 characterwise/linewise visual selection，后续 visual operator 会复用恢复出的选择范围。
 - 本轮补充：TUI Vim prompt editing 增加 `gu`/`gU`/`g~` case-conversion operator，复用 motion、linewise、find/till、text-object 和 dot-repeat operator 管线，并保持大小写转换不写入 yank register。
 - 本轮补充：TUI Vim prompt editing 增加 normal-mode `gJ` raw line join，不插入/规范化空格，并接入 dot-repeat。

docs/first-second-parity-audit.md

@@ -245,6 +245,7 @@ M7 progress now includes:
 - `internal/tui`: Vim prompt editing now includes normal-mode `gJ` raw line joins that do not insert or normalize whitespace and can be dot-repeated.
 - `internal/tui`: Vim prompt editing now supports normal, operator, and visual `+`/`-`/`_` first-nonblank line motions, including linewise operator ranges and dot-repeat replay.
 - `internal/tui`: Vim prompt editing now supports normal, operator, and visual `g_` last-nonblank line motions, preserving charwise operator ranges for commands such as `dg_`.
+- `internal/tui`: Vim prompt editing now supports prompt-local `H`/`M`/`L` screen-line motions across normal, visual, and linewise operator paths, including counted top/bottom targeting and dot-repeat replay for operators.
 - `internal/tui`: Vim prompt editing now includes visual and visual-line `J`/`gJ` joins across selected line ranges, sharing the normal whitespace-normalized and raw join semantics plus undo, `gv`, and dot-repeat recording.
 - `internal/tui`: Vim prompt editing now includes visual and visual-line `p`/`P` paste-over-selection behavior for characterwise and linewise registers, updating the unnamed register with the replaced text and avoiding extra trailing blank lines for end-of-buffer line replacements.
 - `internal/tui`: Vim prompt editing now includes visual and visual-line `r{char}` selection replacement, replacing non-newline characters while preserving line structure, supporting undo, and remembering the previous visual range for `gv`.

internal/tui/tui_test.go

@@ -3573,6 +3573,59 @@ func TestREPLScreenVimFirstNonBlankLineMotions(t *testing.T) {
 	}
 }
 
+func TestREPLScreenVimScreenLineMotions(t *testing.T) {
+	screen := NewREPLScreen(40, 8, nil)
+	screen.SetVimEnabled(true)
+	typePromptText(&screen, "one\ntwo\nthree\nfour\nfive")
+	screen.ApplyKey(ParseKey("\x1b"))
+	screen.ApplyKey(ParseKey("H"))
+	if screen.Prompt.Cursor != 0 {
+		t.Fatalf("cursor after H = %d", screen.Prompt.Cursor)
+	}
+	screen.ApplyKey(ParseKey("L"))
+	if screen.Prompt.Cursor != len([]rune("one\ntwo\nthree\nfour\n")) {
+		t.Fatalf("cursor after L = %d", screen.Prompt.Cursor)
+	}
+	screen.ApplyKey(ParseKey("M"))
+	if screen.Prompt.Cursor != len([]rune("one\ntwo\n")) {
+		t.Fatalf("cursor after M = %d", screen.Prompt.Cursor)
+	}
+	for _, seq := range []string{"2", "H"} {
+		screen.ApplyKey(ParseKey(seq))
+	}
+	if screen.Prompt.Cursor != len([]rune("one\n")) {
+		t.Fatalf("cursor after 2H = %d", screen.Prompt.Cursor)
+	}
+	for _, seq := range []string{"2", "L"} {
+		screen.ApplyKey(ParseKey(seq))
+	}
+	if screen.Prompt.Cursor != len([]rune("one\ntwo\nthree\n")) {
+		t.Fatalf("cursor after 2L = %d", screen.Prompt.Cursor)
+	}
+
+	screen = NewREPLScreen(40, 8, nil)
+	screen.SetVimEnabled(true)
+	typePromptText(&screen, "one\ntwo\nthree\nfour\nfive")
+	screen.ApplyKey(ParseKey("\x1b"))
+	for _, seq := range []string{"g", "g", "v", "L"} {
+		screen.ApplyKey(ParseKey(seq))
+	}
+	if screen.VimMode != VimVisual || screen.Prompt.Cursor != len([]rune("one\ntwo\nthree\nfour\n")) {
+		t.Fatalf("after visual L screen = %#v", screen)
+	}
+
+	screen = NewREPLScreen(40, 8, nil)
+	screen.SetVimEnabled(true)
+	typePromptText(&screen, "one\ntwo\nthree\nfour\nfive")
+	screen.ApplyKey(ParseKey("\x1b"))
+	for _, seq := range []string{"g", "g", "j", "d", "L"} {
+		screen.ApplyKey(ParseKey(seq))
+	}
+	if screen.Prompt.Text != "one" || screen.Prompt.Cursor != len([]rune("one")) || screen.VimRegister != "two\nthree\nfour\nfive\n" || !screen.VimRegisterLinewise {
+		t.Fatalf("after dL screen = %#v", screen)
+	}
+}
+
 func TestREPLScreenVimLastNonBlankGMotion(t *testing.T) {
 	screen := NewREPLScreen(40, 8, nil)
 	screen.SetVimEnabled(true)

internal/tui/vim.go

@@ -235,6 +235,8 @@ func (s *REPLScreen) applyVimNormalRune(r rune) ScreenEvent {
 		s.VimPendingCount = count
 	case 'G':
 		s.Prompt.goToLine(vimGTargetLine(s.Prompt.lineCount(), count))
+	case 'H', 'M', 'L':
+		s.Prompt.goToLine(vimScreenTargetLine(s.Prompt.lineCount(), r, count))
 	case '+', '-', '_':
 		s.Prompt.moveVimFirstNonBlankMotion(r, count)
 	case '/', '?':
@@ -408,7 +410,7 @@ func (s *REPLScreen) applyVimVisualRune(r rune) ScreenEvent {
 		s.applyVimVisualOperator('d')
 	case 's':
 		s.applyVimVisualOperator('c')
-	case 'h', 'l', 'j', 'k', 'w', 'W', 'b', 'B', 'e', 'E', '$', '^', '|', '%', 'G', '+', '-', '_':
+	case 'h', 'l', 'j', 'k', 'w', 'W', 'b', 'B', 'e', 'E', '$', '^', '|', '%', 'G', 'H', 'M', 'L', '+', '-', '_':
 		s.applyVimVisualMotion(r, count)
 	}
 	return ScreenEvent{}
@@ -623,6 +625,8 @@ func (s *REPLScreen) applyVimVisualMotion(motion rune, count int) {
 		s.Prompt.moveMatchingPair()
 	case 'G':
 		s.Prompt.goToLine(vimGTargetLine(s.Prompt.lineCount(), count))
+	case 'H', 'M', 'L':
+		s.Prompt.goToLine(vimScreenTargetLine(s.Prompt.lineCount(), motion, count))
 	case '+', '-', '_':
 		s.Prompt.moveVimFirstNonBlankMotion(motion, count)
 	}
@@ -788,6 +792,8 @@ func (s *REPLScreen) applyVimOperator(r rune) ScreenEvent {
 		s.enterVimSearchOperator(r == '?', operator, count)
 	case 'G':
 		s.applyVimLineMotionOperator(operator, vimGTargetLine(s.Prompt.lineCount(), count), 'G', count)
+	case 'H', 'M', 'L':
+		s.applyVimLineMotionOperator(operator, vimScreenTargetLine(s.Prompt.lineCount(), r, count), r, count)
 	case '+', '-', '_':
 		s.applyVimLineMotionOperator(operator, s.Prompt.vimFirstNonBlankMotionTargetLine(r, count), r, count)
 	case 'g':
@@ -1492,6 +1498,8 @@ func (s *REPLScreen) replayVimLastChange() {
 		switch change.Motion {
 		case 'G':
 			s.applyVimLineMotionOperator(change.Operator, vimGTargetLine(s.Prompt.lineCount(), change.Count), change.Motion, change.Count)
+		case 'H', 'M', 'L':
+			s.applyVimLineMotionOperator(change.Operator, vimScreenTargetLine(s.Prompt.lineCount(), change.Motion, change.Count), change.Motion, change.Count)
 		case 'g':
 			targetLine := 1
 			if change.Count > 1 {
@@ -1666,6 +1674,25 @@ func vimGTargetLine(lineCount int, count int) int {
 	return count
 }
 
+func vimScreenTargetLine(lineCount int, motion rune, count int) int {
+	if lineCount < 1 {
+		lineCount = 1
+	}
+	if count <= 0 {
+		count = 1
+	}
+	switch motion {
+	case 'H':
+		return count
+	case 'M':
+		return (lineCount + 1) / 2
+	case 'L':
+		return lineCount - count + 1
+	default:
+		return 1
+	}
+}
+
 func (p *PromptState) vimFirstNonBlankMotionTargetLine(motion rune, count int) int {
 	if count <= 0 {
 		count = 1