1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
# Copyright 2013-2022 Lawrence Livermore National Security, LLC and other
# Spack Project Developers. See the top-level COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
"""Module for finding the user's preferred text editor.
Defines one function, editor(), which invokes the editor defined by the
user's VISUAL environment variable if set. We fall back to the editor
defined by the EDITOR environment variable if VISUAL is not set or the
specified editor fails (e.g. no DISPLAY for a graphical editor). If
neither variable is set, we fall back to one of several common editors,
raising an EnvironmentError if we are unable to find one.
"""
import os
import shlex
import llnl.util.tty as tty
import spack.config
from spack.util.executable import which_string
#: editors to try if VISUAL and EDITOR are not set
_default_editors = ['vim', 'vi', 'emacs', 'nano', 'notepad']
def _find_exe_from_env_var(var):
"""Find an executable from an environment variable.
Args:
var (str): environment variable name
Returns:
(str or None, list): executable string (or None if not found) and
arguments parsed from the env var
"""
# try to get the environment variable
exe = os.environ.get(var)
if not exe:
return None, []
# split env var into executable and args if needed
args = shlex.split(str(exe))
if not args:
return None, []
exe = which_string(args[0])
args = [exe] + args[1:]
return exe, args
def editor(*args, **kwargs):
"""Invoke the user's editor.
This will try to execute the following, in order:
1. $VISUAL <args> # the "visual" editor (per POSIX)
2. $EDITOR <args> # the regular editor (per POSIX)
3. some default editor (see ``_default_editors``) with <args>
If an environment variable isn't defined, it is skipped. If it
points to something that can't be executed, we'll print a
warning. And if we can't find anything that can be executed after
searching the full list above, we'll raise an error.
Arguments:
args (list): args to pass to editor
Optional Arguments:
_exec_func (function): invoke this function instead of ``os.execv()``
"""
# allow this to be customized for testing
_exec_func = kwargs.get('_exec_func', os.execv)
def try_exec(exe, args, var=None):
"""Try to execute an editor with execv, and warn if it fails.
Returns: (bool) False if the editor failed, ideally does not
return if ``execv`` succeeds, and ``True`` if the
``_exec_func`` does return successfully.
"""
try:
_exec_func(exe, args)
return True
except OSError as e:
if spack.config.get('config:debug'):
raise
# Show variable we were trying to use, if it's from one
if var:
exe = '$%s (%s)' % (var, exe)
tty.warn('Could not execute %s due to error:' % exe, str(e))
return False
def try_env_var(var):
"""Find an editor from an environment variable and try to exec it.
This will warn if the variable points to something is not
executable, or if there is an error when trying to exec it.
"""
if var not in os.environ:
return False
exe, editor_args = _find_exe_from_env_var(var)
if not exe:
tty.warn('$%s is not an executable:' % var, os.environ[var])
return False
full_args = editor_args + list(args)
return try_exec(exe, full_args, var)
# try standard environment variables
if try_env_var('VISUAL'):
return
if try_env_var('EDITOR'):
return
# nothing worked -- try the first default we can find don't bother
# trying them all -- if we get here and one fails, something is
# probably much more deeply wrong with the environment.
exe = which_string(*_default_editors)
if try_exec(exe, [exe] + list(args)):
return
# Fail if nothing could be found
raise EnvironmentError(
'No text editor found! Please set the VISUAL and/or EDITOR '
'environment variable(s) to your preferred text editor.')
|