From 0f802294689eeb0217195be0bd27e724a00722f0 Mon Sep 17 00:00:00 2001
From: phanium <91544758+phanen@users.noreply.github.com>
Date: Thu, 1 Jan 2026 18:28:07 +0800
Subject: [PATCH] Fix edit-in-kitty +lnum path

---
 kitty/launch.py                 | 12 +++---------
 tools/cmd/edit_in_kitty/main.go | 22 ++++++++++++++++++----
 2 files changed, 21 insertions(+), 13 deletions(-)

diff --git a/kitty/launch.py b/kitty/launch.py
index 233fedf54..0a8a4b4a2 100644
--- a/kitty/launch.py
+++ b/kitty/launch.py
@@ -912,6 +912,8 @@ class EditCmd:
         self.file_data = b''
         self.file_inode = -1, -1
         self.file_size = -1
+        self.file_spec = ''
+        self.line_number = 0
         self.version = 0
         self.source_window_id = self.editor_window_id = -1
         self.abort_signaled = ''
@@ -934,15 +936,7 @@ class EditCmd:
             return
         if self.version > 0:
             raise ValueError(f'Unsupported version received in edit protocol: {self.version}')
-        self.opts, extra_args = parse_opts_for_clone(['--type=overlay'] + self.args)
-        self.file_spec = extra_args.pop()
-        self.line_number = 0
-        import re
-        pat = re.compile(r'\+(-?\d+)')
-        for x in extra_args:
-            m = pat.match(x)
-            if m is not None:
-                self.line_number = int(m.group(1))
+        self.opts, _ = parse_opts_for_clone(['--type=overlay'] + self.args)
         self.file_name = os.path.basename(self.file_spec)
         self.file_localpath = os.path.normpath(os.path.join(self.cwd, self.file_spec))
         self.is_local_file = False
diff --git a/tools/cmd/edit_in_kitty/main.go b/tools/cmd/edit_in_kitty/main.go
index cef842fb5..e84370d93 100644
--- a/tools/cmd/edit_in_kitty/main.go
+++ b/tools/cmd/edit_in_kitty/main.go
@@ -165,7 +165,7 @@ func edit_loop(data_to_send string, kill_if_signaled bool, on_data OnDataCallbac
 	return
 }
 
-func edit_in_kitty(path string, opts *Options) (err error) {
+func edit_in_kitty(path string, opts *Options, line_number int) (err error) {
 	read_file, err := os.Open(path)
 	if err != nil {
 		return fmt.Errorf("Failed to open %s for reading with error: %w", path, err)
@@ -206,6 +206,8 @@ func edit_in_kitty(path string, opts *Options) (err error) {
 		return fmt.Errorf("Failed to get the current working directory with error: %w", err)
 	}
 	add_encoded("cwd", cwd)
+	add_encoded("file_spec", path)
+	add_encoded("line_number", strconv.Itoa(line_number))
 	for _, arg := range os.Args[2:] {
 		add_encoded("a", arg)
 	}
@@ -245,16 +247,28 @@ func EntryPoint(parent *cli.Command) *cli.Command {
 				fmt.Fprintln(os.Stderr, "Usage:", cmd.Usage)
 				return 1, fmt.Errorf("No file to edit specified.")
 			}
-			if len(args) != 1 {
+			lineNumber := 0
+			fileArgs := []string{}
+			for _, arg := range args {
+				if strings.HasPrefix(arg, "+") {
+					ln, err := strconv.Atoi(arg[1:])
+					if err == nil {
+						lineNumber = ln
+						continue
+					}
+				}
+				fileArgs = append(fileArgs, arg)
+			}
+			if len(fileArgs) != 1 {
 				fmt.Fprintln(os.Stderr, "Usage:", cmd.Usage)
-				return 1, fmt.Errorf("Only one file to edit must be specified")
+				return 1, fmt.Errorf("Only one file to edit must be specified, optionally with a line number")
 			}
 			var opts Options
 			err = cmd.GetOptionValues(&opts)
 			if err != nil {
 				return 1, err
 			}
-			err = edit_in_kitty(args[0], &opts)
+			err = edit_in_kitty(fileArgs[0], &opts, lineNumber)
 			return 0, err
 		},
 	})