Skip to content

Commit 8094417

Browse files
reubenmillerreubenmiller
authored
feat: add linux package completions (#28)
1 parent f2fd5c6 commit 8094417

File tree

3 files changed

+574
-0
lines changed

3 files changed

+574
-0
lines changed

‎.goreleaser.yaml‎

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,14 @@ nfpms:
7979
dst: /etc/tedge/plugins/tedge-oscar.toml
8080
file_info:
8181
mode: 0644
82+
- src: packaging/completions/_zsh
83+
dst: /usr/share/zsh/vendor-completions/_tedge-oscar
84+
file_info:
85+
mode: 0644
86+
- src: packaging/completions/_bash
87+
dst: /usr/share/bash-completion/completions/tedge-oscar
88+
file_info:
89+
mode: 0644
8290
- id: rpm
8391
ids: [tedge-oscar]
8492
formats:
@@ -107,6 +115,14 @@ nfpms:
107115
dst: /etc/tedge/plugins/tedge-oscar.toml
108116
file_info:
109117
mode: 0644
118+
- src: packaging/completions/_zsh
119+
dst: /usr/share/zsh/vendor-completions/_tedge-oscar
120+
file_info:
121+
mode: 0644
122+
- src: packaging/completions/_bash
123+
dst: /usr/share/bash-completion/completions/tedge-oscar
124+
file_info:
125+
mode: 0644
110126
- id: apk
111127
ids: [tedge-oscar]
112128
formats:
@@ -135,3 +151,11 @@ nfpms:
135151
dst: /etc/tedge/plugins/tedge-oscar.toml
136152
file_info:
137153
mode: 0644
154+
- src: packaging/completions/_zsh
155+
dst: /usr/share/zsh/vendor-completions/_tedge-oscar
156+
file_info:
157+
mode: 0644
158+
- src: packaging/completions/_bash
159+
dst: /usr/share/bash-completion/completions/tedge-oscar
160+
file_info:
161+
mode: 0644

‎packaging/completions/_bash‎

Lines changed: 338 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,338 @@
1+
# bash completion V2 for tedge-oscar -*- shell-script -*-
2+
3+
__tedge-oscar_debug()
4+
{
5+
if [[ -n ${BASH_COMP_DEBUG_FILE-} ]]; then
6+
echo "$*" >> "${BASH_COMP_DEBUG_FILE}"
7+
fi
8+
}
9+
10+
# Macs have bash3 for which the bash-completion package doesn't include
11+
# _init_completion. This is a minimal version of that function.
12+
__tedge-oscar_init_completion()
13+
{
14+
COMPREPLY=()
15+
_get_comp_words_by_ref "$@" cur prev words cword
16+
}
17+
18+
# This function calls the tedge-oscar program to obtain the completion
19+
# results and the directive. It fills the 'out' and 'directive' vars.
20+
__tedge-oscar_get_completion_results() {
21+
local requestComp lastParam lastChar args
22+
23+
# Prepare the command to request completions for the program.
24+
# Calling ${words[0]} instead of directly tedge-oscar allows handling aliases
25+
args=("${words[@]:1}")
26+
requestComp="${words[0]} __complete ${args[*]}"
27+
28+
lastParam=${words[$((${#words[@]}-1))]}
29+
lastChar=${lastParam:$((${#lastParam}-1)):1}
30+
__tedge-oscar_debug "lastParam ${lastParam}, lastChar ${lastChar}"
31+
32+
if [[ -z ${cur} && ${lastChar} != = ]]; then
33+
# If the last parameter is complete (there is a space following it)
34+
# We add an extra empty parameter so we can indicate this to the go method.
35+
__tedge-oscar_debug "Adding extra empty parameter"
36+
requestComp="${requestComp} ''"
37+
fi
38+
39+
# When completing a flag with an = (e.g., tedge-oscar -n=<TAB>)
40+
# bash focuses on the part after the =, so we need to remove
41+
# the flag part from $cur
42+
if [[ ${cur} == -*=* ]]; then
43+
cur="${cur#*=}"
44+
fi
45+
46+
__tedge-oscar_debug "Calling ${requestComp}"
47+
# Use eval to handle any environment variables and such
48+
out=$(eval "${requestComp}" 2>/dev/null)
49+
50+
# Extract the directive integer at the very end of the output following a colon (:)
51+
directive=${out##*:}
52+
# Remove the directive
53+
out=${out%:*}
54+
if [[ ${directive} == "${out}" ]]; then
55+
# There is not directive specified
56+
directive=0
57+
fi
58+
__tedge-oscar_debug "The completion directive is: ${directive}"
59+
__tedge-oscar_debug "The completions are: ${out}"
60+
}
61+
62+
__tedge-oscar_process_completion_results() {
63+
local shellCompDirectiveError=1
64+
local shellCompDirectiveNoSpace=2
65+
local shellCompDirectiveNoFileComp=4
66+
local shellCompDirectiveFilterFileExt=8
67+
local shellCompDirectiveFilterDirs=16
68+
local shellCompDirectiveKeepOrder=32
69+
70+
if (((directive & shellCompDirectiveError) != 0)); then
71+
# Error code. No completion.
72+
__tedge-oscar_debug "Received error from custom completion go code"
73+
return
74+
else
75+
if (((directive & shellCompDirectiveNoSpace) != 0)); then
76+
if [[ $(type -t compopt) == builtin ]]; then
77+
__tedge-oscar_debug "Activating no space"
78+
compopt -o nospace
79+
else
80+
__tedge-oscar_debug "No space directive not supported in this version of bash"
81+
fi
82+
fi
83+
if (((directive & shellCompDirectiveKeepOrder) != 0)); then
84+
if [[ $(type -t compopt) == builtin ]]; then
85+
# no sort isn't supported for bash less than < 4.4
86+
if [[ ${BASH_VERSINFO[0]} -lt 4 || ( ${BASH_VERSINFO[0]} -eq 4 && ${BASH_VERSINFO[1]} -lt 4 ) ]]; then
87+
__tedge-oscar_debug "No sort directive not supported in this version of bash"
88+
else
89+
__tedge-oscar_debug "Activating keep order"
90+
compopt -o nosort
91+
fi
92+
else
93+
__tedge-oscar_debug "No sort directive not supported in this version of bash"
94+
fi
95+
fi
96+
if (((directive & shellCompDirectiveNoFileComp) != 0)); then
97+
if [[ $(type -t compopt) == builtin ]]; then
98+
__tedge-oscar_debug "Activating no file completion"
99+
compopt +o default
100+
else
101+
__tedge-oscar_debug "No file completion directive not supported in this version of bash"
102+
fi
103+
fi
104+
fi
105+
106+
# Separate activeHelp from normal completions
107+
local completions=()
108+
local activeHelp=()
109+
__tedge-oscar_extract_activeHelp
110+
111+
if (((directive & shellCompDirectiveFilterFileExt) != 0)); then
112+
# File extension filtering
113+
local fullFilter filter filteringCmd
114+
115+
# Do not use quotes around the $completions variable or else newline
116+
# characters will be kept.
117+
for filter in ${completions[*]}; do
118+
fullFilter+="$filter|"
119+
done
120+
121+
filteringCmd="_filedir $fullFilter"
122+
__tedge-oscar_debug "File filtering command: $filteringCmd"
123+
$filteringCmd
124+
elif (((directive & shellCompDirectiveFilterDirs) != 0)); then
125+
# File completion for directories only
126+
127+
local subdir
128+
subdir=${completions[0]}
129+
if [[ -n $subdir ]]; then
130+
__tedge-oscar_debug "Listing directories in $subdir"
131+
pushd "$subdir" >/dev/null 2>&1 && _filedir -d && popd >/dev/null 2>&1 || return
132+
else
133+
__tedge-oscar_debug "Listing directories in ."
134+
_filedir -d
135+
fi
136+
else
137+
__tedge-oscar_handle_completion_types
138+
fi
139+
140+
__tedge-oscar_handle_special_char "$cur" :
141+
__tedge-oscar_handle_special_char "$cur" =
142+
143+
# Print the activeHelp statements before we finish
144+
if ((${#activeHelp[*]} != 0)); then
145+
printf "\n";
146+
printf "%s\n" "${activeHelp[@]}"
147+
printf "\n"
148+
149+
# The prompt format is only available from bash 4.4.
150+
# We test if it is available before using it.
151+
if (x=${PS1@P}) 2> /dev/null; then
152+
printf "%s" "${PS1@P}${COMP_LINE[@]}"
153+
else
154+
# Can't print the prompt. Just print the
155+
# text the user had typed, it is workable enough.
156+
printf "%s" "${COMP_LINE[@]}"
157+
fi
158+
fi
159+
}
160+
161+
# Separate activeHelp lines from real completions.
162+
# Fills the $activeHelp and $completions arrays.
163+
__tedge-oscar_extract_activeHelp() {
164+
local activeHelpMarker="_activeHelp_ "
165+
local endIndex=${#activeHelpMarker}
166+
167+
while IFS='' read -r comp; do
168+
if [[ ${comp:0:endIndex} == $activeHelpMarker ]]; then
169+
comp=${comp:endIndex}
170+
__tedge-oscar_debug "ActiveHelp found: $comp"
171+
if [[ -n $comp ]]; then
172+
activeHelp+=("$comp")
173+
fi
174+
else
175+
# Not an activeHelp line but a normal completion
176+
completions+=("$comp")
177+
fi
178+
done <<<"${out}"
179+
}
180+
181+
__tedge-oscar_handle_completion_types() {
182+
__tedge-oscar_debug "__tedge-oscar_handle_completion_types: COMP_TYPE is $COMP_TYPE"
183+
184+
case $COMP_TYPE in
185+
37|42)
186+
# Type: menu-complete/menu-complete-backward and insert-completions
187+
# If the user requested inserting one completion at a time, or all
188+
# completions at once on the command-line we must remove the descriptions.
189+
# https://github.com/spf13/cobra/issues/1508
190+
local tab=$'\t' comp
191+
while IFS='' read -r comp; do
192+
[[ -z $comp ]] && continue
193+
# Strip any description
194+
comp=${comp%%$tab*}
195+
# Only consider the completions that match
196+
if [[ $comp == "$cur"* ]]; then
197+
COMPREPLY+=("$comp")
198+
fi
199+
done < <(printf "%s\n" "${completions[@]}")
200+
;;
201+
202+
*)
203+
# Type: complete (normal completion)
204+
__tedge-oscar_handle_standard_completion_case
205+
;;
206+
esac
207+
}
208+
209+
__tedge-oscar_handle_standard_completion_case() {
210+
local tab=$'\t' comp
211+
212+
# Short circuit to optimize if we don't have descriptions
213+
if [[ "${completions[*]}" != *$tab* ]]; then
214+
IFS=$'\n' read -ra COMPREPLY -d '' < <(compgen -W "${completions[*]}" -- "$cur")
215+
return 0
216+
fi
217+
218+
local longest=0
219+
local compline
220+
# Look for the longest completion so that we can format things nicely
221+
while IFS='' read -r compline; do
222+
[[ -z $compline ]] && continue
223+
# Strip any description before checking the length
224+
comp=${compline%%$tab*}
225+
# Only consider the completions that match
226+
[[ $comp == "$cur"* ]] || continue
227+
COMPREPLY+=("$compline")
228+
if ((${#comp}>longest)); then
229+
longest=${#comp}
230+
fi
231+
done < <(printf "%s\n" "${completions[@]}")
232+
233+
# If there is a single completion left, remove the description text
234+
if ((${#COMPREPLY[*]} == 1)); then
235+
__tedge-oscar_debug "COMPREPLY[0]: ${COMPREPLY[0]}"
236+
comp="${COMPREPLY[0]%%$tab*}"
237+
__tedge-oscar_debug "Removed description from single completion, which is now: ${comp}"
238+
COMPREPLY[0]=$comp
239+
else # Format the descriptions
240+
__tedge-oscar_format_comp_descriptions $longest
241+
fi
242+
}
243+
244+
__tedge-oscar_handle_special_char()
245+
{
246+
local comp="$1"
247+
local char=$2
248+
if [[ "$comp" == *${char}* && "$COMP_WORDBREAKS" == *${char}* ]]; then
249+
local word=${comp%"${comp##*${char}}"}
250+
local idx=${#COMPREPLY[*]}
251+
while ((--idx >= 0)); do
252+
COMPREPLY[idx]=${COMPREPLY[idx]#"$word"}
253+
done
254+
fi
255+
}
256+
257+
__tedge-oscar_format_comp_descriptions()
258+
{
259+
local tab=$'\t'
260+
local comp desc maxdesclength
261+
local longest=$1
262+
263+
local i ci
264+
for ci in ${!COMPREPLY[*]}; do
265+
comp=${COMPREPLY[ci]}
266+
# Properly format the description string which follows a tab character if there is one
267+
if [[ "$comp" == *$tab* ]]; then
268+
__tedge-oscar_debug "Original comp: $comp"
269+
desc=${comp#*$tab}
270+
comp=${comp%%$tab*}
271+
272+
# $COLUMNS stores the current shell width.
273+
# Remove an extra 4 because we add 2 spaces and 2 parentheses.
274+
maxdesclength=$(( COLUMNS - longest - 4 ))
275+
276+
# Make sure we can fit a description of at least 8 characters
277+
# if we are to align the descriptions.
278+
if ((maxdesclength > 8)); then
279+
# Add the proper number of spaces to align the descriptions
280+
for ((i = ${#comp} ; i < longest ; i++)); do
281+
comp+=" "
282+
done
283+
else
284+
# Don't pad the descriptions so we can fit more text after the completion
285+
maxdesclength=$(( COLUMNS - ${#comp} - 4 ))
286+
fi
287+
288+
# If there is enough space for any description text,
289+
# truncate the descriptions that are too long for the shell width
290+
if ((maxdesclength > 0)); then
291+
if ((${#desc} > maxdesclength)); then
292+
desc=${desc:0:$(( maxdesclength - 1 ))}
293+
desc+="…"
294+
fi
295+
comp+=" ($desc)"
296+
fi
297+
COMPREPLY[ci]=$comp
298+
__tedge-oscar_debug "Final comp: $comp"
299+
fi
300+
done
301+
}
302+
303+
__start_tedge-oscar()
304+
{
305+
local cur prev words cword split
306+
307+
COMPREPLY=()
308+
309+
# Call _init_completion from the bash-completion package
310+
# to prepare the arguments properly
311+
if declare -F _init_completion >/dev/null 2>&1; then
312+
_init_completion -n =: || return
313+
else
314+
__tedge-oscar_init_completion -n =: || return
315+
fi
316+
317+
__tedge-oscar_debug
318+
__tedge-oscar_debug "========= starting completion logic =========="
319+
__tedge-oscar_debug "cur is ${cur}, words[*] is ${words[*]}, #words[@] is ${#words[@]}, cword is $cword"
320+
321+
# The user could have moved the cursor backwards on the command-line.
322+
# We need to trigger completion from the $cword location, so we need
323+
# to truncate the command-line ($words) up to the $cword location.
324+
words=("${words[@]:0:$cword+1}")
325+
__tedge-oscar_debug "Truncated words[*]: ${words[*]},"
326+
327+
local out directive
328+
__tedge-oscar_get_completion_results
329+
__tedge-oscar_process_completion_results
330+
}
331+
332+
if [[ $(type -t compopt) = "builtin" ]]; then
333+
complete -o default -F __start_tedge-oscar tedge-oscar
334+
else
335+
complete -o default -o nospace -F __start_tedge-oscar tedge-oscar
336+
fi
337+
338+
# ex: ts=4 sw=4 et filetype=sh

0 commit comments

Comments
 (0)