Blame view

antigen.zsh 6.39 KB
e5dfac7ea   Shrikant Sharat   Initial commit. W...
1
2
3
4
  #!/bin/zsh
  
  # Each line in this string has the following entries separated by a space
  # character.
a12d3140a   Shrikant Sharat   Removed bundle-in...
5
  # <repo-url>, <plugin-location>, <bundle-type>
e5dfac7ea   Shrikant Sharat   Initial commit. W...
6
  # FIXME: Is not kept local by zsh!
9dc631003   Shrikant Sharat   Rename bundle rec...
7
  local _ANTIGEN_BUNDLE_RECORD=""
e5dfac7ea   Shrikant Sharat   Initial commit. W...
8
9
  
  # Syntaxes
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
10
11
  #   antigen-bundle <url> [<loc>=/]
  antigen-bundle () {
e5dfac7ea   Shrikant Sharat   Initial commit. W...
12
13
14
15
  
      # Bundle spec arguments' default values.
      local url="$ANTIGEN_DEFAULT_REPO_URL"
      local loc=/
d535ba468   Shrikant Sharat   Introduced a new ...
16
      local btype=plugin
e5dfac7ea   Shrikant Sharat   Initial commit. W...
17
18
  
      # Set spec values based on the positional arguments.
280ec4481   Shrikant Sharat   name argument for...
19
      local position_args='url loc'
e5dfac7ea   Shrikant Sharat   Initial commit. W...
20
21
      local i=1
      while ! [[ -z $1 || $1 == --*=* ]]; do
43bb2cef1   Shrikant Sharat   Variable declarat...
22
23
          local arg_name="$(echo "$position_args" | cut -d\  -f$i)"
          local arg_value="$1"
e5dfac7ea   Shrikant Sharat   Initial commit. W...
24
25
26
27
          eval "local $arg_name='$arg_value'"
          shift
          i=$(($i + 1))
      done
1abf1faf6   Shrikant Sharat   Plugins can be sp...
28
29
30
31
32
      # Check if url is just the plugin name. Super short syntax.
      if [[ "$url" != */* ]]; then
          loc="plugins/$url"
          url="$ANTIGEN_DEFAULT_REPO_URL"
      fi
e5dfac7ea   Shrikant Sharat   Initial commit. W...
33
34
35
      # Set spec values from keyword arguments, if any. The remaining arguments
      # are all assumed to be keyword arguments.
      while [[ $1 == --*=* ]]; do
43bb2cef1   Shrikant Sharat   Variable declarat...
36
37
          local arg_name="$(echo "$1" | cut -d= -f1 | sed 's/^--//')"
          local arg_value="$(echo "$1" | cut -d= -f2)"
e5dfac7ea   Shrikant Sharat   Initial commit. W...
38
39
40
41
42
          eval "local $arg_name='$arg_value'"
          shift
      done
  
      # Resolve the url.
8110fb2f2   Shrikant Sharat   Add support for a...
43
      if [[ $url != git://* && $url != https://* && $url != /* ]]; then
bb140864c   Shrikant Sharat   The plugin name s...
44
          url="${url%.git}"
bb140864c   Shrikant Sharat   The plugin name s...
45
          url="https://github.com/$url.git"
e5dfac7ea   Shrikant Sharat   Initial commit. W...
46
      fi
e5dfac7ea   Shrikant Sharat   Initial commit. W...
47
      # Add it to the record.
a12d3140a   Shrikant Sharat   Removed bundle-in...
48
49
      _ANTIGEN_BUNDLE_RECORD="$_ANTIGEN_BUNDLE_RECORD
  $url $loc $btype"
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
50
      -antigen-ensure-repo "$url"
e5dfac7ea   Shrikant Sharat   Initial commit. W...
51

a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
52
      -antigen-load "$url" "$loc" "$btype"
2ac7c0cb0   Shrikant Sharat   Bundles are insta...
53

a12d3140a   Shrikant Sharat   Removed bundle-in...
54
  }
2ac7c0cb0   Shrikant Sharat   Bundles are insta...
55

3c5ff1c1e   Shrikant Sharat   Add antigen-bundl...
56
57
58
59
60
61
62
63
64
65
66
67
68
  antigen-bundles () {
      # Bulk add many bundles at one go. Empty lines and lines starting with a `#`
      # are ignored. Everything else is given to `antigen-bundle` as is, no
      # quoting rules applied.
  
      local line
  
      grep -v '^$\|^#' | while read line; do
          # Using `eval` so that we can use the shell-style quoting in each line
          # piped to `antigen-bundles`.
          eval "antigen-bundle $line"
      done
  }
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
69
  antigen-bundle-update () {
158407151   Shrikant Sharat   Refactoring of pr...
70
      # Update your bundles, i.e., `git pull` in all the plugin repos.
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
71
72
      -antigen-echo-record | awk '{print $1}' | sort -u | while read url; do
          -antigen-ensure-repo --update "$url"
158407151   Shrikant Sharat   Refactoring of pr...
73
74
      done
  }
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
75
  -antigen-get-clone-dir () {
a12d3140a   Shrikant Sharat   Removed bundle-in...
76
77
78
79
80
81
82
83
      # Takes a repo url and gives out the path that this url needs to be cloned
      # to. Doesn't actually clone anything.
      # TODO: Memoize?
      echo -n $ADOTDIR/repos/
      echo "$1" | sed \
          -e 's/\.git$//' \
          -e 's./.-SLASH-.g' \
          -e 's.:.-COLON-.g'
2ac7c0cb0   Shrikant Sharat   Bundles are insta...
84
  }
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
85
  -antigen-get-clone-url () {
0b32bec27   Shrikant Sharat   bundle-cleanup co...
86
87
88
89
90
91
92
93
94
      # Takes a repo's clone dir and gives out the repo's original url that was
      # used to create the given directory path.
      # TODO: Memoize?
      echo "$1" | sed \
          -e "s:^$ADOTDIR/repos/::" \
          -e 's/$/.git/' \
          -e 's.-SLASH-./.g' \
          -e 's.-COLON-.:.g'
  }
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
95
  -antigen-ensure-repo () {
2ac7c0cb0   Shrikant Sharat   Bundles are insta...
96
97
98
99
100
101
  
      local update=false
      if [[ $1 == --update ]]; then
          update=true
          shift
      fi
2ac7c0cb0   Shrikant Sharat   Bundles are insta...
102
      local url="$1"
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
103
      local clone_dir="$(-antigen-get-clone-dir $url)"
2ac7c0cb0   Shrikant Sharat   Bundles are insta...
104

aa510273d   Shrikant Sharat   Removed useless l...
105
106
107
108
      if [[ ! -d $clone_dir ]]; then
          git clone "$url" "$clone_dir"
      elif $update; then
          git --git-dir "$clone_dir/.git" pull
2ac7c0cb0   Shrikant Sharat   Bundles are insta...
109
      fi
e5dfac7ea   Shrikant Sharat   Initial commit. W...
110
  }
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
111
  -antigen-load () {
e5dfac7ea   Shrikant Sharat   Initial commit. W...
112

a12d3140a   Shrikant Sharat   Removed bundle-in...
113
      local url="$1"
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
114
      local location="$(-antigen-get-clone-dir "$url")/$2"
a12d3140a   Shrikant Sharat   Removed bundle-in...
115
      local btype="$3"
e5dfac7ea   Shrikant Sharat   Initial commit. W...
116

d535ba468   Shrikant Sharat   Introduced a new ...
117
      if [[ $btype == theme ]]; then
e5dfac7ea   Shrikant Sharat   Initial commit. W...
118

d535ba468   Shrikant Sharat   Introduced a new ...
119
120
121
          # Of course, if its a theme, the location would point to the script
          # file.
          source "$location"
e5dfac7ea   Shrikant Sharat   Initial commit. W...
122

d535ba468   Shrikant Sharat   Introduced a new ...
123
      else
e5dfac7ea   Shrikant Sharat   Initial commit. W...
124

d535ba468   Shrikant Sharat   Introduced a new ...
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
          # Source the plugin script
          # FIXME: I don't know. Looks very very ugly. Needs a better
          # implementation once tests are ready.
          local script_loc="$(ls "$location" | grep -m1 '.plugin.zsh$')"
          if [[ -f $script_loc ]]; then
              # If we have a `*.plugin.zsh`, source it.
              source "$script_loc"
          elif [[ ! -z "$(ls "$location" | grep -m1 '.zsh$')" ]]; then
              # If there is no `*.plugin.zsh` file, source *all* the `*.zsh`
              # files.
              for script ($location/*.zsh) source "$script"
          fi
  
          # Add to $fpath, for completion(s)
          fpath=($location $fpath)
  
      fi
e5dfac7ea   Shrikant Sharat   Initial commit. W...
142

e5dfac7ea   Shrikant Sharat   Initial commit. W...
143
  }
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
144
  antigen-cleanup () {
0b32bec27   Shrikant Sharat   bundle-cleanup co...
145
146
147
148
149
150
151
152
  
      if [[ ! -d "$ADOTDIR/repos" || -z "$(ls "$ADOTDIR/repos/")" ]]; then
          echo "You don't have any bundles."
          return 0
      fi
  
      # Find directores in ADOTDIR/repos, that are not in the bundles record.
      local unused_clones="$(comm -13 \
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
153
          <(-antigen-echo-record | awk '{print $1}' | sort -u) \
0b32bec27   Shrikant Sharat   bundle-cleanup co...
154
          <(ls "$ADOTDIR/repos" | while read line; do
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
155
                  -antigen-get-clone-url "$line"
0b32bec27   Shrikant Sharat   bundle-cleanup co...
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
              done))"
  
      if [[ -z $unused_clones ]]; then
          echo "You don't have any unidentified bundles."
          return 0
      fi
  
      echo 'You have clones for the following repos, but are not used.'
      echo "$unused_clones" | sed 's/^/  /'
  
      echo -n '
  Delete them all? [y/N] '
      if read -q; then
          echo
          echo
          echo "$unused_clones" | while read url; do
              echo -n "Deleting clone for $url..."
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
173
              rm -rf "$(-antigen-get-clone-dir $url)"
0b32bec27   Shrikant Sharat   bundle-cleanup co...
174
175
176
177
178
179
180
              echo ' done.'
          done
      else
          echo
          echo Nothing deleted.
      fi
  }
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
181
182
  antigen-lib () {
      antigen-bundle --loc=lib
e5dfac7ea   Shrikant Sharat   Initial commit. W...
183
  }
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
184
  antigen-theme () {
e5dfac7ea   Shrikant Sharat   Initial commit. W...
185
      local name="${1:-robbyrussell}"
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
186
      antigen-bundle --loc=themes/$name.zsh-theme --btype=theme
e5dfac7ea   Shrikant Sharat   Initial commit. W...
187
  }
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
188
  antigen-apply () {
e5dfac7ea   Shrikant Sharat   Initial commit. W...
189
      # Initialize completion.
a91386310   Shrikant Sharat   Add a todo note o...
190
191
      # TODO: Only load completions if there are any changes to the bundle
      # repositories.
e5dfac7ea   Shrikant Sharat   Initial commit. W...
192
193
      compinit -i
  }
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
194
  antigen-list () {
df942673b   Shrikant Sharat   Added bundle-list...
195
      # List all currently installed bundles
9dc631003   Shrikant Sharat   Rename bundle rec...
196
      if [[ -z "$_ANTIGEN_BUNDLE_RECORD" ]]; then
df942673b   Shrikant Sharat   Added bundle-list...
197
198
199
          echo "You don't have any bundles." >&2
          return 1
      else
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
200
          -antigen-echo-record
df942673b   Shrikant Sharat   Added bundle-list...
201
202
      fi
  }
bc9d20c7a   Shrikant Sharat   Add comments.
203
204
  # Echo the bundle specs as in the record. The first line is not echoed since it
  # is a blank line.
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
205
  -antigen-echo-record () {
9dc631003   Shrikant Sharat   Rename bundle rec...
206
      echo "$_ANTIGEN_BUNDLE_RECORD" | sed -n '1!p'
e5dfac7ea   Shrikant Sharat   Initial commit. W...
207
  }
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
208
  -antigen-env-setup () {
3047bfeb7   Shrikant Sharat   Load compinit at ...
209
      # Pre-startup initializations
e5dfac7ea   Shrikant Sharat   Initial commit. W...
210
211
      -set-default ANTIGEN_DEFAULT_REPO_URL \
          https://github.com/robbyrussell/oh-my-zsh.git
911cc8cb5   Shrikant Sharat   Use ADOTDIR as th...
212
      -set-default ADOTDIR $HOME/.antigen
3047bfeb7   Shrikant Sharat   Load compinit at ...
213
214
215
216
217
218
  
      # Load the compinit module
      autoload -U compinit
  
      # Without the following, `compdef` function is not defined.
      compinit -i
e5dfac7ea   Shrikant Sharat   Initial commit. W...
219
220
221
222
223
  }
  
  # Same as `export $1=$2`, but will only happen if the name specified by `$1` is
  # not already set.
  -set-default () {
43bb2cef1   Shrikant Sharat   Variable declarat...
224
225
      local arg_name="$1"
      local arg_value="$2"
e5dfac7ea   Shrikant Sharat   Initial commit. W...
226
227
      eval "test -z \"\$$arg_name\" && export $arg_name='$arg_value'"
  }
a4f6e2fc8   Shrikant Sharat   Fix #2. Add `anti...
228
  -antigen-env-setup