Merge pull request #22 from daemonp/visual_editor_env

Use $VISUAL and $EDITOR env vars for editor selection
adzimzf · Jun 24, 2024 · c5a57fa · c5a57fa
2 parents 868c0ab + d82bea8
commit c5a57fa
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/editor/edit.go b/editor/edit.go
@@ -8,9 +8,8 @@ import (
 
 const DefaultEditor = "nano"
 
-// Edit edit the text using editor
+// Edit the text using editor
 func Edit(text string, tmpPattern string) (string, error) {
-
 	if tmpPattern == "" {
 		tmpPattern = "tpot_*.txt"
 	}
@@ -25,7 +24,15 @@ func Edit(text string, tmpPattern string) (string, error) {
 		return "", err
 	}
 
-	cmd := exec.Command(DefaultEditor, f.Name())
+	editor := os.Getenv("VISUAL")
+	if editor == "" {
+		editor = os.Getenv("EDITOR")
+		if editor == "" {
+			editor = DefaultEditor
+		}
+	}
+
+	cmd := exec.Command(editor, f.Name())
 	cmd.Stdin = os.Stdin
 	cmd.Stdout = os.Stdout
 	cmd.Stderr = os.Stderr