71 lines
1.8 KiB
Go
71 lines
1.8 KiB
Go
// License: GPLv3 Copyright: 2023, Kovid Goyal, <kovid at kovidgoyal.net>
|
|
|
|
package shell_integration
|
|
|
|
import (
|
|
"archive/tar"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
var _ = fmt.Print
|
|
|
|
type integration_setup_func = func(argv []string, env map[string]string) ([]string, map[string]string, error)
|
|
|
|
func extract_shell_integration_for(shell_name string, dest_dir string) (err error) {
|
|
d := Data()
|
|
for _, fname := range d.FilesMatching("shell-integration/" + shell_name + "/") {
|
|
entry := d[fname]
|
|
dest := filepath.Join(dest_dir, fname)
|
|
ddir := filepath.Dir(dest)
|
|
if err = os.MkdirAll(ddir, 0o755); err != nil {
|
|
return
|
|
}
|
|
switch entry.Metadata.Typeflag {
|
|
case tar.TypeDir:
|
|
if err = os.MkdirAll(dest, 0o755); err != nil {
|
|
return
|
|
}
|
|
case tar.TypeSymlink:
|
|
if err = os.Symlink(entry.Metadata.Linkname, dest); err != nil {
|
|
return
|
|
}
|
|
case tar.TypeReg:
|
|
if err = os.WriteFile(dest, entry.Data, 0o644); err != nil {
|
|
return
|
|
}
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func zsh_setup_func(argv []string, env map[string]string) (final_argv []string, final_env map[string]string, err error) {
|
|
return
|
|
}
|
|
|
|
func fish_setup_func(argv []string, env map[string]string) (final_argv []string, final_env map[string]string, err error) {
|
|
return
|
|
}
|
|
|
|
func bash_setup_func(argv []string, env map[string]string) (final_argv []string, final_env map[string]string, err error) {
|
|
return
|
|
}
|
|
|
|
func setup_func_for_shell(shell_name string) integration_setup_func {
|
|
switch shell_name {
|
|
case "zsh":
|
|
return zsh_setup_func
|
|
case "fish":
|
|
return fish_setup_func
|
|
case "bash":
|
|
return bash_setup_func
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func IsSupportedShell(shell_name string) bool { return setup_func_for_shell(shell_name) != nil }
|
|
|
|
func Setup(shell_name string, argv []string, env map[string]string) ([]string, map[string]string, error) {
|
|
return setup_func_for_shell(shell_name)(argv, env)
|
|
}
|