Move the kittens Go code into the kittens folder

kittens/diff/main.go

@@ -0,0 +1,174 @@
+// License: GPLv3 Copyright: 2023, Kovid Goyal, <kovid at kovidgoyal.net>
+
+package diff
+
+import (
+	"archive/tar"
+	"bytes"
+	"fmt"
+	"io/fs"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"strings"
+
+	"kitty/kittens/ssh"
+	"kitty/tools/cli"
+	"kitty/tools/config"
+	"kitty/tools/tui/loop"
+	"kitty/tools/utils"
+)
+
+var _ = fmt.Print
+
+func load_config(opts *Options) (ans *Config, err error) {
+	ans = NewConfig()
+	p := config.ConfigParser{LineHandler: ans.Parse}
+	err = p.LoadConfig("diff.conf", opts.Config, opts.Override)
+	if err != nil {
+		return nil, err
+	}
+	return ans, nil
+}
+
+var conf *Config
+var opts *Options
+var lp *loop.Loop
+
+func isdir(path string) bool {
+	if s, err := os.Stat(path); err == nil {
+		return s.IsDir()
+	}
+	return false
+}
+
+func exists(path string) bool {
+	_, err := os.Stat(path)
+	return err == nil
+}
+
+func get_ssh_file(hostname, rpath string) (string, error) {
+	tdir, err := os.MkdirTemp("", "*-"+hostname)
+	if err != nil {
+		return "", err
+	}
+	add_remote_dir(tdir)
+	is_abs := strings.HasPrefix(rpath, "/")
+	for strings.HasPrefix(rpath, "/") {
+		rpath = rpath[1:]
+	}
+	cmd := []string{ssh.SSHExe(), hostname, "tar", "-c", "-f", "-"}
+	if is_abs {
+		cmd = append(cmd, "-C", "/")
+	}
+	cmd = append(cmd, rpath)
+	c := exec.Command(cmd[0], cmd[1:]...)
+	c.Stdin, c.Stderr = os.Stdin, os.Stderr
+	stdout, err := c.Output()
+	if err != nil {
+		return "", fmt.Errorf("Failed to ssh into remote host %s to get file %s with error: %w", hostname, rpath, err)
+	}
+	tf := tar.NewReader(bytes.NewReader(stdout))
+	count, err := utils.ExtractAllFromTar(tf, tdir)
+	if err != nil {
+		return "", fmt.Errorf("Failed to untar data from remote host %s to get file %s with error: %w", hostname, rpath, err)
+	}
+	ans := filepath.Join(tdir, rpath)
+	if count == 1 {
+		filepath.WalkDir(tdir, func(path string, d fs.DirEntry, err error) error {
+			if !d.IsDir() {
+				ans = path
+				return fs.SkipAll
+			}
+			return nil
+		})
+	}
+	return ans, nil
+}
+
+func get_remote_file(path string) (string, error) {
+	if strings.HasPrefix(path, "ssh:") {
+		parts := strings.SplitN(path, ":", 3)
+		if len(parts) == 3 {
+			return get_ssh_file(parts[1], parts[2])
+		}
+	}
+	return path, nil
+}
+
+func main(_ *cli.Command, opts_ *Options, args []string) (rc int, err error) {
+	opts = opts_
+	conf, err = load_config(opts)
+	if err != nil {
+		return 1, err
+	}
+	if len(args) != 2 {
+		return 1, fmt.Errorf("You must specify exactly two files/directories to compare")
+	}
+	if err = set_diff_command(conf.Diff_cmd); err != nil {
+		return 1, err
+	}
+	init_caches()
+	create_formatters()
+	defer func() {
+		for tdir := range remote_dirs {
+			os.RemoveAll(tdir)
+		}
+	}()
+	left, err := get_remote_file(args[0])
+	if err != nil {
+		return 1, err
+	}
+	right, err := get_remote_file(args[1])
+	if err != nil {
+		return 1, err
+	}
+	if isdir(left) != isdir(right) {
+		return 1, fmt.Errorf("The items to be diffed should both be either directories or files. Comparing a directory to a file is not valid.'")
+	}
+	if !exists(left) {
+		return 1, fmt.Errorf("%s does not exist", left)
+	}
+	if !exists(right) {
+		return 1, fmt.Errorf("%s does not exist", right)
+	}
+	lp, err = loop.New()
+	if err != nil {
+		return 1, err
+	}
+	h := Handler{left: left, right: right, lp: lp}
+	lp.OnInitialize = func() (string, error) {
+		lp.SetCursorVisible(false)
+		lp.SetCursorShape(loop.BAR_CURSOR, true)
+		lp.AllowLineWrapping(false)
+		lp.SetWindowTitle(fmt.Sprintf("%s vs. %s", left, right))
+		h.initialize()
+		return "", nil
+	}
+	lp.OnWakeup = h.on_wakeup
+	lp.OnFinalize = func() string {
+		lp.SetCursorVisible(true)
+		lp.SetCursorShape(loop.BLOCK_CURSOR, true)
+		h.finalize()
+		return ""
+	}
+	lp.OnResize = h.on_resize
+	lp.OnKeyEvent = h.on_key_event
+	lp.OnText = h.on_text
+	err = lp.Run()
+	if err != nil {
+		return 1, err
+	}
+	ds := lp.DeathSignalName()
+	if ds != "" {
+		fmt.Println("Killed by signal: ", ds)
+		lp.KillIfSignalled()
+		return 1, nil
+	}
+
+	return
+}
+
+func EntryPoint(parent *cli.Command) {
+	create_cmd(parent, main)
+}