Claude Code 업그레이드 후 Caps Lock 한영전환 오류 해결 (Kitty Keyboard Protocol)

❯ [57358u[57358uasdfasdf[57358u[57358uㄴㅁㅇㄹ

# v1.0.x
KITTY_PUSH = b'\x1b[>1u'
stdout_buf = stdout_buf.replace(KITTY_PUSH, b'', 1)

# v1.1.0
CAPSLOCK_PATTERN = re.compile(rb'\x1b\[57358(;\d+)*u')
stdout_buf = CAPSLOCK_PATTERN.sub(b'', stdout_buf)

# v1.0.x: kitty protocol 전체 제거
사용자 ← stdout ← [ESC[>1u 제거] ← [claude]
                   (legacy 모드 → Ctrl+V 이미지 붙여넣기 불가)

# v1.1.0: CapsLock 키코드만 제거
사용자 ← stdout ← [57358u만 제거] ← [claude]
                   (kitty protocol 유지 → Ctrl+V 이미지 붙여넣기 정상)

#!/usr/bin/env python3
"""Claude Code wrapper - CapsLock 한/영 전환 키코드 필터.

Changelog:
  v1.0.0 (2026-03-26): kitty keyboard protocol 전체 비활성화 (ESC[>1u 제거)
  v1.0.1 (2026-03-26): PTY 크기 전달 버그 수정 — ioctl TIOCGWINSZ 기반으로 교체
  v1.0.2 (2026-03-28): SIGWINCH 핸들러 안정화, set_winsize() ioctl 통합
  v1.1.0 (2026-03-30): kitty protocol 유지, CapsLock(57358u) 키코드만 필터링
                        → Ctrl+V 이미지 붙여넣기 정상화
"""
__version__ = "1.1.0"

import os
import re
import sys
import pty
import signal
import select
import struct
import fcntl
import termios

CAPSLOCK_PATTERN = re.compile(rb'\x1b\[57358(;\d+)*u')


def get_real_claude():
    """실제 claude 바이너리 경로 반환."""
    for path in [
        '/opt/homebrew/bin/claude',
        '/usr/local/bin/claude',
    ]:
        if os.path.exists(path):
            return path
    versions_dir = os.path.expanduser('~/.local/share/claude/versions')
    if os.path.isdir(versions_dir):
        versions = sorted(os.listdir(versions_dir))
        if versions:
            return os.path.join(versions_dir, versions[-1])
    return None


def set_winsize(fd):
    """터미널 윈도우 크기를 pty에 전달."""
    try:
        s = struct.pack('HHHH', 0, 0, 0, 0)
        result = fcntl.ioctl(sys.stdout.fileno(), termios.TIOCGWINSZ, s)
        fcntl.ioctl(fd, termios.TIOCSWINSZ, result)
    except (OSError, AttributeError):
        pass


def main():
    claude_bin = get_real_claude()
    if not claude_bin:
        print("Error: Claude Code binary not found", file=sys.stderr)
        sys.exit(1)

    pid, master_fd = pty.fork()

    if pid == 0:
        os.execvp(claude_bin, [claude_bin] + sys.argv[1:])
        sys.exit(1)

    set_winsize(master_fd)

    def handle_winch(signum, frame):
        set_winsize(master_fd)
        os.kill(pid, signal.SIGWINCH)

    signal.signal(signal.SIGWINCH, handle_winch)

    old_settings = None
    try:
        old_settings = termios.tcgetattr(sys.stdin.fileno())
    except termios.error:
        pass

    try:
        if old_settings:
            import tty
            tty.setraw(sys.stdin.fileno())

        stdout_buf = b''

        while True:
            try:
                rlist, _, _ = select.select(
                    [sys.stdin.fileno(), master_fd], [], [], 0.1
                )
            except (select.error, ValueError):
                break

            if sys.stdin.fileno() in rlist:
                try:
                    data = os.read(sys.stdin.fileno(), 4096)
                    if not data:
                        break
                    os.write(master_fd, data)
                except OSError:
                    break

            if master_fd in rlist:
                try:
                    data = os.read(master_fd, 4096)
                    if not data:
                        break

                    stdout_buf += data

                    if stdout_buf.endswith(b'\x1b'):
                        continue

                    stdout_buf = CAPSLOCK_PATTERN.sub(b'', stdout_buf)

                    os.write(sys.stdout.fileno(), stdout_buf)
                    stdout_buf = b''
                except OSError:
                    break

        if stdout_buf:
            stdout_buf = CAPSLOCK_PATTERN.sub(b'', stdout_buf)
            os.write(sys.stdout.fileno(), stdout_buf)

    finally:
        if old_settings:
            termios.tcsetattr(
                sys.stdin.fileno(), termios.TCSADRAIN, old_settings
            )

        try:
            _, status = os.waitpid(pid, 0)
            sys.exit(os.WEXITSTATUS(status) if os.WIFEXITED(status) else 1)
        except ChildProcessError:
            pass


if __name__ == '__main__':
    main()

# ~/.zshrc에 alias 추가
alias claude='python3 ~/.local/bin/claude-wrapper'

_disable_kitty_protocol() {
  printf '\e[<u\e[<u\e[<u\e[<u\e[<u'
}
autoload -Uz add-zsh-hook
add-zsh-hook precmd _disable_kitty_protocol