From 924b87a16aaba2821bc20f77f960ed2add8509d6 Mon Sep 17 00:00:00 2001
From: Kovid Goyal <kovid@kovidgoyal.net>
Date: Thu, 21 Mar 2024 10:58:46 +0530
Subject: [PATCH] Convenience function to get terminfo data in kittens

---
 tools/tui/shell_integration/api.go      | 6 ++++++
 tools/tui/shell_integration/api_test.go | 3 ++-
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/tools/tui/shell_integration/api.go b/tools/tui/shell_integration/api.go
index beb004b60..bbf07f136 100644
--- a/tools/tui/shell_integration/api.go
+++ b/tools/tui/shell_integration/api.go
@@ -23,6 +23,12 @@ var _ = fmt.Print
 
 type integration_setup_func = func(shell_integration_dir string, argv []string, env map[string]string) ([]string, map[string]string, error)
 
+func TerminfoData() string {
+	d := Data()
+	entry := d["terminfo/x/xterm-kitty"]
+	return utils.UnsafeBytesToString(entry.Data)
+}
+
 func extract_files(match, dest_dir string) (err error) {
 	d := Data()
 	for _, fname := range d.FilesMatching(match) {
diff --git a/tools/tui/shell_integration/api_test.go b/tools/tui/shell_integration/api_test.go
index 868bdaf04..b36a4ef57 100644
--- a/tools/tui/shell_integration/api_test.go
+++ b/tools/tui/shell_integration/api_test.go
@@ -29,7 +29,7 @@ func TestExtractShellIntegration(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	os.WriteFile(kzsh, []byte("changed"), 0o644)
+	_ = os.WriteFile(kzsh, []byte("changed"), 0o644)
 	if err := extract_shell_integration_for("zsh", tdir); err != nil {
 		t.Fatal(err)
 	}
@@ -47,4 +47,5 @@ func TestExtractShellIntegration(t *testing.T) {
 	if _, err := os.Stat(filepath.Join(tdir, "terminfo", "78", kitty.DefaultTermName)); err != nil {
 		t.Fatal(err)
 	}
+	TerminfoData()
 }