mirror of
https://github.com/kopia/kopia.git
synced 2026-01-25 14:58:00 -05:00
133 lines
2.8 KiB
Go
133 lines
2.8 KiB
Go
// Package editor encapsulates working with external text editor.
|
|
package editor
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
|
|
"github.com/kopia/kopia/internal/kopialogging"
|
|
)
|
|
|
|
var log = kopialogging.Logger("editor")
|
|
|
|
// EditLoop launches OS-specific editor (VI, notepad.exe or anoter editor configured through environment variables)
|
|
// It creates a temporary file with 'initial' contents and repeatedly invokes the editor until the provided 'parse' function
|
|
// returns nil result indicating success. The 'parse' function is passed the contents of edited files without # line commments.
|
|
func EditLoop(fname string, initial string, parse func(updated string) error) error {
|
|
tmpDir, err := ioutil.TempDir("", "kopia")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
tmpFile := filepath.Join(tmpDir, fname)
|
|
defer os.RemoveAll(tmpDir) //nolint:errcheck
|
|
|
|
if err := ioutil.WriteFile(tmpFile, []byte(initial), 0600); err != nil {
|
|
return err
|
|
}
|
|
|
|
for {
|
|
if err := editFile(tmpFile); err != nil {
|
|
return err
|
|
}
|
|
|
|
txt, err := readAndStripComments(tmpFile)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = parse(txt)
|
|
if err == nil {
|
|
return nil
|
|
}
|
|
|
|
log.Errorf("%v", err)
|
|
fmt.Print("Reopen editor to fix? (Y/n) ")
|
|
|
|
var shouldReopen string
|
|
fmt.Scanf("%s", &shouldReopen) //nolint:errcheck
|
|
if strings.HasPrefix(strings.ToLower(shouldReopen), "n") {
|
|
return fmt.Errorf("aborted")
|
|
}
|
|
}
|
|
}
|
|
|
|
func readAndStripComments(fname string) (string, error) {
|
|
f, err := os.Open(fname)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer f.Close() //nolint:errcheck
|
|
|
|
var result []string
|
|
|
|
s := bufio.NewScanner(f)
|
|
for s.Scan() {
|
|
l := s.Text()
|
|
l = strings.TrimSpace(strings.Split(l, "#")[0])
|
|
if l != "" {
|
|
result = append(result, l)
|
|
}
|
|
}
|
|
return strings.Join(result, "\n"), nil
|
|
}
|
|
|
|
func editFile(file string) error {
|
|
editor, editorArgs := getEditorCommand()
|
|
|
|
var args []string
|
|
args = append(args, editorArgs...)
|
|
args = append(args, file)
|
|
|
|
cmd := exec.Command(editor, args...)
|
|
cmd.Stderr = os.Stderr
|
|
cmd.Stdin = os.Stdin
|
|
cmd.Stdout = os.Stdout
|
|
|
|
log.Debugf("launching editor %q on file %q", editor, file)
|
|
err := cmd.Run()
|
|
if err != nil {
|
|
log.Errorf("unable to launch editor: %v", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func getEditorCommand() (string, []string) {
|
|
editor := os.Getenv("VISUAL")
|
|
if editor == "" {
|
|
editor = os.Getenv("EDITOR")
|
|
}
|
|
|
|
if editor != "" {
|
|
return parseEditor(editor)
|
|
}
|
|
|
|
if runtime.GOOS == "windows" {
|
|
return "notepad.exe", nil
|
|
}
|
|
|
|
return "vi", nil
|
|
}
|
|
|
|
func parseEditor(s string) (string, []string) {
|
|
// quoted editor path
|
|
if s[0] == '"' {
|
|
p := strings.Index(s[1:], "\"")
|
|
if p == -1 {
|
|
// invalid
|
|
return s, nil
|
|
}
|
|
|
|
return s[1 : p+1], strings.Split(strings.TrimSpace(s[p+1:]), " ")
|
|
}
|
|
|
|
parts := strings.Split(s, " ")
|
|
return parts[0], parts[1:]
|
|
}
|