You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
241 lines
7.4 KiB
241 lines
7.4 KiB
#compdef knife |
|
|
|
# You can override the path to knife.rb and your cookbooks by setting |
|
# KNIFE_CONF_PATH=/path/to/my/.chef/knife.rb |
|
# KNIFE_COOKBOOK_PATH=/path/to/my/chef/cookbooks |
|
# If you want your local cookbooks path to be calculated relative to where you are then |
|
# set the below option |
|
# KNIFE_RELATIVE_PATH=true |
|
# Read around where these are used for more detail. |
|
|
|
# These flags should be available everywhere according to man knife |
|
knife_general_flags=( --help --server-url --key --config --editor --format --log_level --logfile --no-editor --user --print-after --version --yes ) |
|
|
|
# knife has a very special syntax, some example calls are: |
|
# knife status |
|
# knife cookbook list |
|
# knife role show ROLENAME |
|
# knife data bag show DATABAGNAME |
|
# knife role show ROLENAME --attribute ATTRIBUTENAME |
|
# knife cookbook show COOKBOOKNAME COOKBOOKVERSION recipes |
|
|
|
# The -Q switch in compadd allow for completions of things like "data bag" without having to go through two rounds of completion and avoids zsh inserting a \ for escaping spaces |
|
_knife() { |
|
local curcontext="$curcontext" state line |
|
typeset -A opt_args |
|
cloudproviders=(bluebox ec2 rackspace slicehost terremark) |
|
_arguments \ |
|
'1: :->knifecmd'\ |
|
'2: :->knifesubcmd'\ |
|
'3: :->knifesubcmd2' \ |
|
'4: :->knifesubcmd3' \ |
|
'5: :->knifesubcmd4' \ |
|
'6: :->knifesubcmd5' |
|
|
|
case $state in |
|
knifecmd) |
|
compadd -Q "$@" bootstrap client configure cookbook "cookbook site" "data bag" diff exec environment index node recipe role search ssh status upload vault windows $cloudproviders |
|
;; |
|
knifesubcmd) |
|
case $words[2] in |
|
(bluebox|ec2|rackspace|slicehost|terremark) |
|
compadd "$@" server images |
|
;; |
|
client) |
|
compadd -Q "$@" "bulk delete" list create show delete edit reregister |
|
;; |
|
configure) |
|
compadd "$@" client |
|
;; |
|
cookbook) |
|
compadd -Q "$@" test list create download delete "metadata from" show "bulk delete" metadata upload |
|
;; |
|
diff) |
|
_arguments '*:file or directory:_files -g "*"' |
|
;; |
|
environment) |
|
compadd -Q "$@" list create delete edit show "from file" |
|
;; |
|
node) |
|
compadd -Q "$@" "from file" create show edit delete list run_list "bulk delete" |
|
;; |
|
recipe) |
|
compadd "$@" list |
|
;; |
|
role) |
|
compadd -Q "$@" "bulk delete" create delete edit "from file" list show |
|
;; |
|
upload) |
|
_arguments '*:file or directory:_files -g "*"' |
|
;; |
|
vault) |
|
compadd -Q "$@" create decrypt delete edit remove "rotate all keys" "rotate keys" show update |
|
;; |
|
windows) |
|
compadd "$@" bootstrap |
|
;; |
|
*) |
|
_arguments '2:Subsubcommands:($(_knife_options1))' |
|
esac |
|
;; |
|
knifesubcmd2) |
|
case $words[3] in |
|
server) |
|
compadd "$@" list create delete |
|
;; |
|
images) |
|
compadd "$@" list |
|
;; |
|
site) |
|
compadd "$@" vendor show share search download list unshare |
|
;; |
|
(show|delete|edit) |
|
_arguments '3:Subsubcommands:($(_chef_$words[2]s_remote))' |
|
;; |
|
(upload|test) |
|
_arguments '3:Subsubcommands:($(_chef_$words[2]s_local) --all)' |
|
;; |
|
list) |
|
compadd -a "$@" knife_general_flags |
|
;; |
|
bag) |
|
compadd -Q "$@" show edit list "from file" create delete |
|
;; |
|
*) |
|
_arguments '3:Subsubcommands:($(_knife_options2))' |
|
esac |
|
;; |
|
knifesubcmd3) |
|
case $words[3] in |
|
show) |
|
case $words[2] in |
|
cookbook) |
|
versioncomp=1 |
|
_arguments '4:Cookbookversions:($(_cookbook_versions) latest)' |
|
;; |
|
(node|client|role) |
|
compadd "$@" --attribute |
|
esac |
|
esac |
|
case $words[4] in |
|
(show|edit) |
|
_arguments '4:Subsubsubcommands:($(_chef_$words[2]_$words[3]s_remote))' |
|
;; |
|
file) |
|
case $words[2] in |
|
environment) |
|
_arguments '*:files:_path_files -g "*.(rb|json)" -W "$(_chef_root)/environments"' |
|
;; |
|
node) |
|
_arguments '*:files:_path_files -g "*.(rb|json)" -W "$(_chef_root)/nodes"' |
|
;; |
|
role) |
|
_arguments '*:files:_path_files -g "*.(rb|json)" -W "$(_chef_root)/roles"' |
|
;; |
|
*) |
|
_arguments '*:Subsubcommands:($(_knife_options3))' |
|
esac |
|
;; |
|
list) |
|
compadd -a "$@" knife_general_flags |
|
;; |
|
*) |
|
_arguments '*:Subsubcommands:($(_knife_options3))' |
|
esac |
|
;; |
|
knifesubcmd4) |
|
if (( versioncomp > 0 )); then |
|
compadd "$@" attributes definitions files libraries providers recipes resources templates |
|
else |
|
case $words[5] in |
|
file) |
|
_arguments '*:directory:_path_files -/ -W "$(_chef_root)/data_bags" -qS \ ' |
|
;; |
|
*) _arguments '*:Subsubcommands:($(_knife_options2))' |
|
esac |
|
fi |
|
;; |
|
knifesubcmd5) |
|
case $words[5] in |
|
file) |
|
_arguments '*:files:_path_files -g "*.json" -W "$(_chef_root)/data_bags/$words[6]"' |
|
;; |
|
*) |
|
_arguments '*:Subsubcommands:($(_knife_options3))' |
|
esac |
|
esac |
|
} |
|
|
|
# Helper functions to provide the argument completion for several depths of commands |
|
_knife_options1() { |
|
( for line in $( knife $words[2] --help | grep -v "^knife" ); do echo $line | grep "\-\-"; done ) |
|
} |
|
|
|
_knife_options2() { |
|
( for line in $( knife $words[2] $words[3] --help | grep -v "^knife" ); do echo $line | grep "\-\-"; done ) |
|
} |
|
|
|
_knife_options3() { |
|
( for line in $( knife $words[2] $words[3] $words[4] --help | grep -v "^knife" ); do echo $line | grep "\-\-"; done ) |
|
} |
|
|
|
# The chef_x_remote functions use knife to get a list of objects of type x on the server |
|
_chef_roles_remote() { |
|
(knife role list --format json | grep \" | awk '{print $1}' | awk -F"," '{print $1}' | awk -F"\"" '{print $2}') |
|
} |
|
|
|
_chef_clients_remote() { |
|
(knife client list --format json | grep \" | awk '{print $1}' | awk -F"," '{print $1}' | awk -F"\"" '{print $2}') |
|
} |
|
|
|
_chef_nodes_remote() { |
|
(knife node list --format json | grep \" | awk '{print $1}' | awk -F"," '{print $1}' | awk -F"\"" '{print $2}') |
|
} |
|
|
|
_chef_cookbooks_remote() { |
|
(knife cookbook list --format json | grep \" | awk '{print $1}' | awk -F"," '{print $1}' | awk -F"\"" '{print $2}') |
|
} |
|
|
|
_chef_sitecookbooks_remote() { |
|
(knife cookbook site list --format json | grep \" | awk '{print $1}' | awk -F"," '{print $1}' | awk -F"\"" '{print $2}') |
|
} |
|
|
|
_chef_data_bags_remote() { |
|
(knife data bag list --format json | grep \" | awk '{print $1}' | awk -F"," '{print $1}' | awk -F"\"" '{print $2}') |
|
} |
|
|
|
_chef_environments_remote() { |
|
(knife environment list | awk '{print $1}') |
|
} |
|
|
|
# The chef_x_local functions use the knife config to find the paths of relevant objects x to be uploaded to the server |
|
_chef_cookbooks_local() { |
|
if [ $KNIFE_RELATIVE_PATH ]; then |
|
local cookbook_path="$(_chef_root)/cookbooks" |
|
else |
|
local knife_rb=${KNIFE_CONF_PATH:-${HOME}/.chef/knife.rb} |
|
if [ -f ./.chef/knife.rb ]; then |
|
knife_rb="./.chef/knife.rb" |
|
fi |
|
local cookbook_path=${KNIFE_COOKBOOK_PATH:-$(grep cookbook_path $knife_rb | awk 'BEGIN {FS = "[" }; {print $2}' | sed 's/\,//g' | sed "s/'//g" | sed 's/\(.*\)]/\1/' )} |
|
fi |
|
(for i in $cookbook_path; do ls $i; done) |
|
} |
|
|
|
# This function extracts the available cookbook versions on the chef server |
|
_cookbook_versions() { |
|
(knife cookbook show $words[4] | grep -v $words[4] | grep -v -E '\]|\[|\{|\}' | sed 's/ //g' | sed 's/"//g') |
|
} |
|
|
|
# Searches up from current directory to find the closest folder that has a .chef folder |
|
# Useful for the knife upload/from file commands |
|
_chef_root () { |
|
directory="$PWD" |
|
while [ $directory != '/' ] |
|
do |
|
test -e "$directory/.chef" && echo "$directory" && return |
|
directory="${directory:h}" |
|
done |
|
} |
|
|
|
_knife "$@"
|
|
|