Commit 592f1460b0300ec9287843109c1a057fb24f7eb0
1 parent
de4644c0f5
Load completions after running an instalation.
Showing 1 changed file with 3 additions and 0 deletions Inline Diff
antigen.zsh
| 1 | #!/bin/zsh | 1 | #!/bin/zsh |
| 2 | 2 | ||
| 3 | # Each line in this string has the following entries separated by a space | 3 | # Each line in this string has the following entries separated by a space |
| 4 | # character. | 4 | # character. |
| 5 | # <bundle-name>, <repo-url>, <plugin-location>, <repo-local-clone-dir> | 5 | # <bundle-name>, <repo-url>, <plugin-location>, <repo-local-clone-dir> |
| 6 | # FIXME: Is not kept local by zsh! | 6 | # FIXME: Is not kept local by zsh! |
| 7 | local bundles="" | 7 | local bundles="" |
| 8 | 8 | ||
| 9 | # Syntaxes | 9 | # Syntaxes |
| 10 | # bundle <url> [<loc>=/] [<name>] | 10 | # bundle <url> [<loc>=/] [<name>] |
| 11 | bundle () { | 11 | bundle () { |
| 12 | 12 | ||
| 13 | # Bundle spec arguments' default values. | 13 | # Bundle spec arguments' default values. |
| 14 | local url="$ANTIGEN_DEFAULT_REPO_URL" | 14 | local url="$ANTIGEN_DEFAULT_REPO_URL" |
| 15 | local loc=/ | 15 | local loc=/ |
| 16 | local name= | 16 | local name= |
| 17 | local load=true | 17 | local load=true |
| 18 | 18 | ||
| 19 | # Set spec values based on the positional arguments. | 19 | # Set spec values based on the positional arguments. |
| 20 | local position_args='url loc name' | 20 | local position_args='url loc name' |
| 21 | local i=1 | 21 | local i=1 |
| 22 | while ! [[ -z $1 || $1 == --*=* ]]; do | 22 | while ! [[ -z $1 || $1 == --*=* ]]; do |
| 23 | arg_name="$(echo "$position_args" | cut -d\ -f$i)" | 23 | arg_name="$(echo "$position_args" | cut -d\ -f$i)" |
| 24 | arg_value="$1" | 24 | arg_value="$1" |
| 25 | eval "local $arg_name='$arg_value'" | 25 | eval "local $arg_name='$arg_value'" |
| 26 | shift | 26 | shift |
| 27 | i=$(($i + 1)) | 27 | i=$(($i + 1)) |
| 28 | done | 28 | done |
| 29 | 29 | ||
| 30 | # Check if url is just the plugin name. Super short syntax. | 30 | # Check if url is just the plugin name. Super short syntax. |
| 31 | if [[ "$url" != */* ]]; then | 31 | if [[ "$url" != */* ]]; then |
| 32 | loc="plugins/$url" | 32 | loc="plugins/$url" |
| 33 | url="$ANTIGEN_DEFAULT_REPO_URL" | 33 | url="$ANTIGEN_DEFAULT_REPO_URL" |
| 34 | fi | 34 | fi |
| 35 | 35 | ||
| 36 | # Set spec values from keyword arguments, if any. The remaining arguments | 36 | # Set spec values from keyword arguments, if any. The remaining arguments |
| 37 | # are all assumed to be keyword arguments. | 37 | # are all assumed to be keyword arguments. |
| 38 | while [[ $1 == --*=* ]]; do | 38 | while [[ $1 == --*=* ]]; do |
| 39 | arg_name="$(echo "$1" | cut -d= -f1 | sed 's/^--//')" | 39 | arg_name="$(echo "$1" | cut -d= -f1 | sed 's/^--//')" |
| 40 | arg_value="$(echo "$1" | cut -d= -f2)" | 40 | arg_value="$(echo "$1" | cut -d= -f2)" |
| 41 | eval "local $arg_name='$arg_value'" | 41 | eval "local $arg_name='$arg_value'" |
| 42 | shift | 42 | shift |
| 43 | done | 43 | done |
| 44 | 44 | ||
| 45 | # Resolve the url. | 45 | # Resolve the url. |
| 46 | if [[ $url != git://* && $url != https://* ]]; then | 46 | if [[ $url != git://* && $url != https://* ]]; then |
| 47 | url="${url%.git}" | 47 | url="${url%.git}" |
| 48 | name="$(basename "$url")" | 48 | name="$(basename "$url")" |
| 49 | url="https://github.com/$url.git" | 49 | url="https://github.com/$url.git" |
| 50 | fi | 50 | fi |
| 51 | 51 | ||
| 52 | # Plugin's repo will be cloned here. | 52 | # Plugin's repo will be cloned here. |
| 53 | local clone_dir="$ANTIGEN_REPO_CACHE/$(echo "$url" \ | 53 | local clone_dir="$ANTIGEN_REPO_CACHE/$(echo "$url" \ |
| 54 | | sed -e 's/\.git$//' -e 's./.-SLASH-.g' -e 's.:.-COLON-.g')" | 54 | | sed -e 's/\.git$//' -e 's./.-SLASH-.g' -e 's.:.-COLON-.g')" |
| 55 | 55 | ||
| 56 | # Make an intelligent guess about the name of the plugin, if not already | 56 | # Make an intelligent guess about the name of the plugin, if not already |
| 57 | # done or is explicitly specified. | 57 | # done or is explicitly specified. |
| 58 | if [[ -z $name ]]; then | 58 | if [[ -z $name ]]; then |
| 59 | name="$(basename $url/$loc)" | 59 | name="$(basename $url/$loc)" |
| 60 | fi | 60 | fi |
| 61 | 61 | ||
| 62 | # Add it to the record. | 62 | # Add it to the record. |
| 63 | bundles="$bundles\n$name $url $loc $clone_dir" | 63 | bundles="$bundles\n$name $url $loc $clone_dir" |
| 64 | 64 | ||
| 65 | # Load it, unless specified otherwise. | 65 | # Load it, unless specified otherwise. |
| 66 | if $load; then | 66 | if $load; then |
| 67 | bundle-load "$name" | 67 | bundle-load "$name" |
| 68 | fi | 68 | fi |
| 69 | } | 69 | } |
| 70 | 70 | ||
| 71 | bundle-install () { | 71 | bundle-install () { |
| 72 | 72 | ||
| 73 | if [[ $1 == --update ]]; then | 73 | if [[ $1 == --update ]]; then |
| 74 | local update=true | 74 | local update=true |
| 75 | shift | 75 | shift |
| 76 | else | 76 | else |
| 77 | local update=false | 77 | local update=false |
| 78 | fi | 78 | fi |
| 79 | 79 | ||
| 80 | mkdir -p "$ANTIGEN_BUNDLE_DIR" | 80 | mkdir -p "$ANTIGEN_BUNDLE_DIR" |
| 81 | 81 | ||
| 82 | local handled_repos="" | 82 | local handled_repos="" |
| 83 | 83 | ||
| 84 | if [[ $# != 0 ]]; then | 84 | if [[ $# != 0 ]]; then |
| 85 | # Record and install just the given plugin here and now. | 85 | # Record and install just the given plugin here and now. |
| 86 | bundle "$@" | 86 | bundle "$@" |
| 87 | echo "$bundles" | tail -1 | 87 | echo "$bundles" | tail -1 |
| 88 | 88 | ||
| 89 | else | 89 | else |
| 90 | # Install all the plugins, previously recorded. | 90 | # Install all the plugins, previously recorded. |
| 91 | echo-non-empty "$bundles" | 91 | echo-non-empty "$bundles" |
| 92 | 92 | ||
| 93 | fi | while read spec; do | 93 | fi | while read spec; do |
| 94 | echo "-> $spec" | 94 | echo "-> $spec" |
| 95 | 95 | ||
| 96 | local name="$(echo "$spec" | awk '{print $1}')" | 96 | local name="$(echo "$spec" | awk '{print $1}')" |
| 97 | local url="$(echo "$spec" | awk '{print $2}')" | 97 | local url="$(echo "$spec" | awk '{print $2}')" |
| 98 | local loc="$(echo "$spec" | awk '{print $3}')" | 98 | local loc="$(echo "$spec" | awk '{print $3}')" |
| 99 | local clone_dir="$(echo "$spec" | awk '{print $4}')" | 99 | local clone_dir="$(echo "$spec" | awk '{print $4}')" |
| 100 | 100 | ||
| 101 | if [[ -z "$(echo "$handled_repos" | grep -Fm1 "$url")" ]]; then | 101 | if [[ -z "$(echo "$handled_repos" | grep -Fm1 "$url")" ]]; then |
| 102 | if [[ ! -d $clone_dir ]]; then | 102 | if [[ ! -d $clone_dir ]]; then |
| 103 | git clone "$url" "$clone_dir" | 103 | git clone "$url" "$clone_dir" |
| 104 | elif $update; then | 104 | elif $update; then |
| 105 | git --git-dir "$clone_dir/.git" pull | 105 | git --git-dir "$clone_dir/.git" pull |
| 106 | fi | 106 | fi |
| 107 | 107 | ||
| 108 | handled_repos="$handled_repos\n$url" | 108 | handled_repos="$handled_repos\n$url" |
| 109 | fi | 109 | fi |
| 110 | 110 | ||
| 111 | if [[ $name != *.theme ]]; then | 111 | if [[ $name != *.theme ]]; then |
| 112 | echo Installing $name | 112 | echo Installing $name |
| 113 | local bundle_dest="$ANTIGEN_BUNDLE_DIR/$name" | 113 | local bundle_dest="$ANTIGEN_BUNDLE_DIR/$name" |
| 114 | test -e "$bundle_dest" && rm -rf "$bundle_dest" | 114 | test -e "$bundle_dest" && rm -rf "$bundle_dest" |
| 115 | ln -s "$clone_dir/$loc" "$bundle_dest" | 115 | ln -s "$clone_dir/$loc" "$bundle_dest" |
| 116 | else | 116 | else |
| 117 | mkdir -p "$ANTIGEN_BUNDLE_DIR/$name" | 117 | mkdir -p "$ANTIGEN_BUNDLE_DIR/$name" |
| 118 | cp "$clone_dir/$loc" "$ANTIGEN_BUNDLE_DIR/$name" | 118 | cp "$clone_dir/$loc" "$ANTIGEN_BUNDLE_DIR/$name" |
| 119 | fi | 119 | fi |
| 120 | 120 | ||
| 121 | bundle-load "$name" | 121 | bundle-load "$name" |
| 122 | 122 | ||
| 123 | done | 123 | done |
| 124 | 124 | ||
| 125 | # Initialize completions after installing | ||
| 126 | bundle-apply | ||
| 127 | |||
| 125 | } | 128 | } |
| 126 | 129 | ||
| 127 | bundle-install! () { | 130 | bundle-install! () { |
| 128 | bundle-install --update | 131 | bundle-install --update |
| 129 | } | 132 | } |
| 130 | 133 | ||
| 131 | bundle-cleanup () { | 134 | bundle-cleanup () { |
| 132 | 135 | ||
| 133 | # Find directores in ANTIGEN_BUNDLE_DIR, that are not in the bundles record. | 136 | # Find directores in ANTIGEN_BUNDLE_DIR, that are not in the bundles record. |
| 134 | local unidentified_bundles="$(comm -13 \ | 137 | local unidentified_bundles="$(comm -13 \ |
| 135 | <(echo-non-empty "$bundles" | awk '{print $1}' | sort) \ | 138 | <(echo-non-empty "$bundles" | awk '{print $1}' | sort) \ |
| 136 | <(ls -1 "$ANTIGEN_BUNDLE_DIR"))" | 139 | <(ls -1 "$ANTIGEN_BUNDLE_DIR"))" |
| 137 | 140 | ||
| 138 | if [[ -z $unidentified_bundles ]]; then | 141 | if [[ -z $unidentified_bundles ]]; then |
| 139 | echo "You don't have any unidentified bundles." | 142 | echo "You don't have any unidentified bundles." |
| 140 | return 0 | 143 | return 0 |
| 141 | fi | 144 | fi |
| 142 | 145 | ||
| 143 | echo The following bundles are not recorded: | 146 | echo The following bundles are not recorded: |
| 144 | echo "$unidentified_bundles" | sed 's/^/ /' | 147 | echo "$unidentified_bundles" | sed 's/^/ /' |
| 145 | 148 | ||
| 146 | echo -n '\nDelete them all? [y/N] ' | 149 | echo -n '\nDelete them all? [y/N] ' |
| 147 | if read -q; then | 150 | if read -q; then |
| 148 | echo | 151 | echo |
| 149 | echo | 152 | echo |
| 150 | echo "$unidentified_bundles" | while read name; do | 153 | echo "$unidentified_bundles" | while read name; do |
| 151 | echo -n Deleting $name... | 154 | echo -n Deleting $name... |
| 152 | rm -rf "$ANTIGEN_BUNDLE_DIR/$name" | 155 | rm -rf "$ANTIGEN_BUNDLE_DIR/$name" |
| 153 | echo ' done.' | 156 | echo ' done.' |
| 154 | done | 157 | done |
| 155 | else | 158 | else |
| 156 | echo | 159 | echo |
| 157 | echo Nothing deleted. | 160 | echo Nothing deleted. |
| 158 | fi | 161 | fi |
| 159 | } | 162 | } |
| 160 | 163 | ||
| 161 | bundle-load () { | 164 | bundle-load () { |
| 162 | if [[ $1 == --init ]]; then | 165 | if [[ $1 == --init ]]; then |
| 163 | do_init=true | 166 | do_init=true |
| 164 | shift | 167 | shift |
| 165 | else | 168 | else |
| 166 | do_init=false | 169 | do_init=false |
| 167 | fi | 170 | fi |
| 168 | 171 | ||
| 169 | name="$1" | 172 | name="$1" |
| 170 | bundle_dir="$ANTIGEN_BUNDLE_DIR/$name" | 173 | bundle_dir="$ANTIGEN_BUNDLE_DIR/$name" |
| 171 | 174 | ||
| 172 | # Source the plugin script | 175 | # Source the plugin script |
| 173 | script_loc="$bundle_dir/$name.plugin.zsh" | 176 | script_loc="$bundle_dir/$name.plugin.zsh" |
| 174 | if [[ -f $script_loc ]]; then | 177 | if [[ -f $script_loc ]]; then |
| 175 | source "$script_loc" | 178 | source "$script_loc" |
| 176 | fi | 179 | fi |
| 177 | 180 | ||
| 178 | # If the name of the plugin ends with `.lib`, all the *.zsh files in it are | 181 | # If the name of the plugin ends with `.lib`, all the *.zsh files in it are |
| 179 | # sourced. This is kind of a hack to source the libraries of oh-my-zsh. | 182 | # sourced. This is kind of a hack to source the libraries of oh-my-zsh. |
| 180 | if [[ $name == *.lib ]]; then | 183 | if [[ $name == *.lib ]]; then |
| 181 | # FIXME: This throws an error if no files match the given glob pattern. | 184 | # FIXME: This throws an error if no files match the given glob pattern. |
| 182 | for lib ($bundle_dir/*.zsh) source $lib | 185 | for lib ($bundle_dir/*.zsh) source $lib |
| 183 | fi | 186 | fi |
| 184 | 187 | ||
| 185 | # If the name ends with `.theme`, it is handled as if it were a zsh-theme | 188 | # If the name ends with `.theme`, it is handled as if it were a zsh-theme |
| 186 | # plugin. | 189 | # plugin. |
| 187 | if [[ $name == *.theme ]]; then | 190 | if [[ $name == *.theme ]]; then |
| 188 | source "$bundle_dir/${name%.theme}.zsh-theme" | 191 | source "$bundle_dir/${name%.theme}.zsh-theme" |
| 189 | fi | 192 | fi |
| 190 | 193 | ||
| 191 | # Add to $fpath, for completion(s) | 194 | # Add to $fpath, for completion(s) |
| 192 | fpath=($bundle_dir $fpath) | 195 | fpath=($bundle_dir $fpath) |
| 193 | 196 | ||
| 194 | if $do_init; then | 197 | if $do_init; then |
| 195 | bundle-init | 198 | bundle-init |
| 196 | fi | 199 | fi |
| 197 | } | 200 | } |
| 198 | 201 | ||
| 199 | bundle-lib () { | 202 | bundle-lib () { |
| 200 | bundle --name=oh-my-zsh.lib --loc=lib | 203 | bundle --name=oh-my-zsh.lib --loc=lib |
| 201 | } | 204 | } |
| 202 | 205 | ||
| 203 | bundle-theme () { | 206 | bundle-theme () { |
| 204 | local url="$ANTIGEN_DEFAULT_REPO_URL" | 207 | local url="$ANTIGEN_DEFAULT_REPO_URL" |
| 205 | local name="${1:-robbyrussell}" | 208 | local name="${1:-robbyrussell}" |
| 206 | bundle "$url" --name=$name.theme --loc=themes/$name.zsh-theme | 209 | bundle "$url" --name=$name.theme --loc=themes/$name.zsh-theme |
| 207 | } | 210 | } |
| 208 | 211 | ||
| 209 | bundle-apply () { | 212 | bundle-apply () { |
| 210 | # Initialize completion. | 213 | # Initialize completion. |
| 211 | compinit -i | 214 | compinit -i |
| 212 | } | 215 | } |
| 213 | 216 | ||
| 214 | # Does what it says. | 217 | # Does what it says. |
| 215 | echo-non-empty () { | 218 | echo-non-empty () { |
| 216 | echo "$@" | while read line; do | 219 | echo "$@" | while read line; do |
| 217 | [[ $line != "" ]] && echo $line | 220 | [[ $line != "" ]] && echo $line |
| 218 | done | 221 | done |
| 219 | } | 222 | } |
| 220 | 223 | ||
| 221 | -bundle-env-setup () { | 224 | -bundle-env-setup () { |
| 222 | # Pre-startup initializations | 225 | # Pre-startup initializations |
| 223 | -set-default ANTIGEN_DEFAULT_REPO_URL \ | 226 | -set-default ANTIGEN_DEFAULT_REPO_URL \ |
| 224 | https://github.com/robbyrussell/oh-my-zsh.git | 227 | https://github.com/robbyrussell/oh-my-zsh.git |
| 225 | -set-default ANTIGEN_REPO_CACHE $HOME/.antigen/cache | 228 | -set-default ANTIGEN_REPO_CACHE $HOME/.antigen/cache |
| 226 | -set-default ANTIGEN_BUNDLE_DIR $HOME/.antigen/bundles | 229 | -set-default ANTIGEN_BUNDLE_DIR $HOME/.antigen/bundles |
| 227 | 230 | ||
| 228 | # Load the compinit module | 231 | # Load the compinit module |
| 229 | autoload -U compinit | 232 | autoload -U compinit |
| 230 | 233 | ||
| 231 | # Without the following, `compdef` function is not defined. | 234 | # Without the following, `compdef` function is not defined. |
| 232 | compinit -i | 235 | compinit -i |
| 233 | } | 236 | } |
| 234 | 237 | ||
| 235 | # Same as `export $1=$2`, but will only happen if the name specified by `$1` is | 238 | # Same as `export $1=$2`, but will only happen if the name specified by `$1` is |
| 236 | # not already set. | 239 | # not already set. |
| 237 | -set-default () { | 240 | -set-default () { |
| 238 | arg_name="$1" | 241 | arg_name="$1" |
| 239 | arg_value="$2" | 242 | arg_value="$2" |
| 240 | eval "test -z \"\$$arg_name\" && export $arg_name='$arg_value'" | 243 | eval "test -z \"\$$arg_name\" && export $arg_name='$arg_value'" |
| 241 | } | 244 | } |
| 242 | 245 | ||
| 243 | -bundle-env-setup | 246 | -bundle-env-setup |
| 244 | 247 |