mirror of
https://github.com/YunoHost/yunohost.git
synced 2024-09-03 20:06:10 +02:00
3550be63f2
Added the output of the script in data/bash-completion.d/yunohost_completion This is probably not the correct place for the script and the generation should be done at some other time and place also.
84 lines
3.7 KiB
Python
84 lines
3.7 KiB
Python
"""
|
|
Simple automated generation of a bash_completion file
|
|
for yunohost command from the actionsmap.
|
|
|
|
Generates a bash completion file assuming the structure
|
|
`yunohost domain action`
|
|
adds `--help` at the end if one presses [tab] again.
|
|
|
|
author: Christophe Vuillot
|
|
"""
|
|
import yaml
|
|
|
|
ACTIONSMAP_FILE = 'yunohost.yml'
|
|
BASH_COMPLETION_FILE = '../bash-completion.d/yunohost_completion'
|
|
|
|
with open(ACTIONSMAP_FILE, 'r') as stream:
|
|
|
|
# Getting the dictionary containning what actions are possible per domain
|
|
OPTION_TREE = yaml.load(stream)
|
|
DOMAINS = [str for str in OPTION_TREE.keys() if not str.startswith('_')]
|
|
DOMAINS_STR = '"{}"'.format(' '.join(DOMAINS))
|
|
ACTIONS_DICT = {}
|
|
for domain in DOMAINS:
|
|
ACTIONS = [str for str in OPTION_TREE[domain]['actions'].keys()
|
|
if not str.startswith('_')]
|
|
ACTIONS_STR = '"{}"'.format(' '.join(ACTIONS))
|
|
ACTIONS_DICT[domain] = ACTIONS_STR
|
|
|
|
with open(BASH_COMPLETION_FILE, 'w') as generated_file:
|
|
|
|
# header of the file
|
|
generated_file.write('#\n')
|
|
generated_file.write('# completion for yunohost\n')
|
|
generated_file.write('# automatically generated from the actionsmap\n')
|
|
generated_file.write('#\n\n')
|
|
|
|
# Start of the completion function
|
|
generated_file.write('_yunohost_completion()\n')
|
|
generated_file.write('{\n')
|
|
|
|
# Defining local variable for previously and currently typed words
|
|
generated_file.write('\tlocal cur prev opts narg\n')
|
|
generated_file.write('\tCOMPREPLY=()\n\n')
|
|
generated_file.write('\t# the number of words already typed\n')
|
|
generated_file.write('\tnarg=${#COMP_WORDS[@]}\n\n')
|
|
generated_file.write('\t# the current word being typed\n')
|
|
generated_file.write('\tcur="${COMP_WORDS[COMP_CWORD]}"\n\n')
|
|
generated_file.write('\t# the last typed word\n')
|
|
generated_file.write('\tprev="${COMP_WORDS[COMP_CWORD-1]}"\n\n')
|
|
|
|
# If one is currently typing a domain then match with the domain list
|
|
generated_file.write('\t# If one is currently typing a domain,\n')
|
|
generated_file.write('\t# match with domains\n')
|
|
generated_file.write('\tif [[ $narg == 2 ]]; then\n')
|
|
generated_file.write('\t\topts={}\n'.format(DOMAINS_STR))
|
|
generated_file.write('\tfi\n\n')
|
|
|
|
# If one is currently typing an action then match with the action list
|
|
# of the previously typed domain
|
|
generated_file.write('\t# If one already typed a domain,\n')
|
|
generated_file.write('\t# match the actions of that domain\n')
|
|
generated_file.write('\tif [[ $narg == 3 ]]; then\n')
|
|
for domain in DOMAINS:
|
|
generated_file.write('\t\tif [[ $prev == "{}" ]]; then\n'.format(domain))
|
|
generated_file.write('\t\t\topts={}\n'.format(ACTIONS_DICT[domain]))
|
|
generated_file.write('\t\tfi\n')
|
|
generated_file.write('\tfi\n\n')
|
|
|
|
# If both domain and action have been typed or the domain
|
|
# was not recognized propose --help (only once)
|
|
generated_file.write('\t# If no options were found propose --help\n')
|
|
generated_file.write('\tif [ -z "$opts" ]; then\n')
|
|
generated_file.write('\t\tif [[ $prev != "--help" ]]; then\n')
|
|
generated_file.write('\t\t\topts=( --help )\n')
|
|
generated_file.write('\t\tfi\n')
|
|
generated_file.write('\tfi\n')
|
|
|
|
# generate the completion list from the possible options
|
|
generated_file.write('\tCOMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )\n')
|
|
generated_file.write('\treturn 0\n')
|
|
generated_file.write('}\n\n')
|
|
|
|
# Add the function to bash completion
|
|
generated_file.write('complete -F _yunohost_completion yunohost')
|