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
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
|
'''This module contains helper functions for Bash.'''
from . import helpers
_COMPGEN_W_REPLACEMENT = helpers.ShellFunction('compgen_w_replacement', r'''
local cur word append=0
[[ "$1" == "-a" ]] && { shift; append=1; }
[[ "$1" == "--" ]] && { shift; }
cur="$1"; shift
(( append )) || COMPREPLY=()
for word; do
if [[ "$word" == "$cur"* ]]; then
COMPREPLY+=("$(printf '%q' "$word")")
fi
done
''')
_EXEC = helpers.ShellFunction('exec', r'''
local item desc
while IFS=$'\t' read -r item desc; do
if [[ "$item" == "$cur"* ]]; then
COMPREPLY+=("$(printf '%q' "$item")")
fi
done < <(eval "$1")
''')
_EXEC_FAST = helpers.ShellFunction('exec_fast', r'''
local item desc
while IFS=$'\t' read -r item desc; do
if [[ "$item" == "$cur"* ]]; then
COMPREPLY+=("$item")
fi
done < <(eval "$1")
''')
_VALUE_LIST = helpers.ShellFunction('value_list', r'''
local separator="$1"; shift
if [[ -z "$cur" ]]; then
COMPREPLY=("$@")
return
fi
local value having_value found_value
local -a having_values remaining_values
IFS="$separator" read -r -a having_values <<< "$cur"
for value; do
found_value=0
for having_value in "${having_values[@]}"; do
if [[ "$value" == "$having_value" ]]; then
found_value=1
break
fi
done
if (( ! found_value )); then
remaining_values+=("$value")
fi
done
COMPREPLY=()
if [[ "${cur: -1}" == "$separator" ]]; then
for value in "${remaining_values[@]}"; do
COMPREPLY+=("$cur$value")
done
elif (( ${#having_values[@]} )); then
local cur_last_value=${having_values[-1]}
for value in "${remaining_values[@]}"; do
if [[ "$value" == "$cur_last_value"* ]]; then
COMPREPLY+=("${cur%"$cur_last_value"}$value")
fi
done
fi
''')
_PREFIX_COMPREPLY = helpers.ShellFunction('prefix_compreply', r'''
local i prefix="$1"
for ((i=0; i < ${#COMPREPLY[@]}; ++i)); do
COMPREPLY[i]="$prefix${COMPREPLY[i]}"
done
''')
_HISTORY = helpers.ShellFunction('history', r'''
[[ -f "$HISTFILE" ]] || return
local match
while read -r match; do
if [[ "$match" == "$cur"* ]]; then
COMPREPLY+=("$(printf '%q' "$match")")
fi
done < <(command grep -E -o -- "$1" "$HISTFILE")
''')
# =============================================================================
# Bonus
# =============================================================================
_NET_INTERFACES_LIST = helpers.ShellFunction('net_interfaces_list', r'''
if [[ -d /sys/class/net ]]; then
command ls /sys/class/net
elif command ifconfig -l &>/dev/null; then
command ifconfig -l # BSD / macOS
else
command ifconfig 2>/dev/null | command awk '/^[a-z0-9]/ {print $1}' | command sed 's/://'
fi''')
_TIMEZONE_LIST = helpers.ShellFunction('timezone_list', r'''
if ! command timedatectl list-timezones 2>/dev/null; then
command find /usr/share/zoneinfo -type f |\
command sed 's|/usr/share/zoneinfo/||g' |\
command grep -E -v '^(posix|right)'
fi''')
_ALSA_LIST_CARDS = helpers.ShellFunction('alsa_list_cards', r'''
local card
command aplay -l \
| command grep -Eo '^card [0-9]+: [^,]+' \
| command uniq \
| while builtin read card; do
card="${card#card }"
local id="${card%%: *}"
builtin echo "$id"
done''')
_ALSA_LIST_DEVICES = helpers.ShellFunction('alsa_list_devices', r'''
local card
command aplay -l \
| command grep -Eo '^card [0-9]+: [^,]+' \
| command uniq \
| while builtin read card; do
card="${card#card }"
local id="${card%%: *}"
builtin echo "hw:$id"
done''')
class BashHelpers(helpers.GeneralHelpers):
'''Class holding helper functions for Bash.'''
def __init__(self, function_prefix):
super().__init__(function_prefix, helpers.ShellFunction)
self.add_function(_COMPGEN_W_REPLACEMENT)
self.add_function(_EXEC)
self.add_function(_EXEC_FAST)
self.add_function(_VALUE_LIST)
self.add_function(_PREFIX_COMPREPLY)
self.add_function(_HISTORY)
self.add_function(_NET_INTERFACES_LIST)
self.add_function(_TIMEZONE_LIST)
self.add_function(_ALSA_LIST_CARDS)
self.add_function(_ALSA_LIST_DEVICES)
|
