summary refs log tree commit diff
path: root/Completion/Redhat/Command/_dnf
blob: 297c95ae9a5c181085eaa77cd41535b66c9e3199 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
#compdef dnf

# Main dispatcher
_dnf() {
  _arguments -s \
    '(- *)'{-h,--help}'[show the help message]' \
    '(-t --tolerant)'{-t,--tolerant}'[be tolerant of errors]' \
    '(-C --cacheonly)'{-C,--cacheonly}'[run entirely from cache]' \
    '(-c --config)'{-c,--config=}'[config file location]:config file:_files' \
    '(-R --randomwait)'{-R,--randomwait=}'[maximum command wait time (in minutes)]:max wait time' \
    '(-d --debuglevel)'{-d,--debuglevel=}'[debug level (0-10)]:debug level' \
    '(-e --errorlevel)'{-e,--errorlevel=}'[error level (0-10)]:error level' \
    '(-y --assumeyes)'{-y,--assumeyes}'[answer yes for all questions]' \
    '--installroot=[set install root]:install root:_files -/' \
    '*--enablerepo=[enable or or more repositories]:repos to enable:_dnf_disabled_repos_list' \
    '*--disablerepo=[disable one or more repositories]:disable repos:_dnf_enabled_repos_list' \
    {*-x,*--exclude=}'[exclude package(s) by name or glob]:exclude packages' \
    '--version[show dnf version]' \
    '--obsoletes[enable obsoletes processing during updates]' \
    '--nogpgcheck[disable gpg signature checking]' \
    '--noplugins[disable dnf plugins]' \
    '--disablepresto[disable Presto plugin and don''''t download any deltarpms]' \
    '*::dnf command:_dnf_command'
}

(( $+functions[_dnf_command] )) || _dnf_command() {
  local -a _dnf_cmds
  _dnf_cmds=(
    "install:install the latest version of a package or group of packages"
    "erase:remove an installed package (with its dependencies)"
    "remove:remove an installed package (with its dependencies)"
    "clean:clean local dnf cache"
    "check-update:check if any updates are available"
    "info:get description of available packages"
    "list:is used to list various information about packages"
    "groupinfo:get info on package groups"
    "groupinstall:install a package group or groups"
    "groupremove:remove a package group or groups"
    "grouplist:list package groups"
    "groupupdate:update a package group or groups"
    "localinstall:install packages with local rpm files"
    "localupdate:update packages with local rpm files"
    "makecache:makes a local dnf cache"
    "provides:find out which package provides some feature or file"
    "whatprovides:find out which package provides some feature or file"
    "search:find any packages matching pattern"
    "shell:enter the 'dnf shell'"
    "update:update one or more packages"
    "upgrade:upgrade one or more packages"
  )

  if (( CURRENT == 1 )); then
    _describe -t commands 'dnf command' _dnf_cmds || compadd "$@"
  else
    local curcontext="$curcontext"

    cmd="${${_dnf_cmds[(r)$words[1]:*]%%:*}}"
    # Deal with any aliases
    case $cmd in
      remove) cmd="erase";;
      whatprovides) cmd="provides";;
      upgrade) cmd="update";;
    esac

    if (( $#cmd )); then
      curcontext="${curcontext%:*:*}:dnf-${cmd}:"

      local update_policy
      zstyle -s ":completion:${curcontext}:" cache-policy update_policy
      if [[ -z "$update_policy" ]]; then
	zstyle ":completion:${curcontext}:" cache-policy _dnf_caching_policy
      fi

      _call_function ret _dnf_$cmd || _message 'no more arguments'
    else
      _message "unknown dnf command: $words[1]"
    fi
    return ret
  fi
}

# Fills the all pkg cache
_dnf_all_pkgs() {
  if ( [[ ${+_all_pkgs} -eq 0 ]] || _cache_invalid ALL ) &&
    ! _retrieve_cache ALL;
  then
    _all_pkgs=( $(dnf -C list all | sed 's/\s.*//' | grep '\.' 2>/dev/null) )
    _store_cache ALL _all_pkgs
  fi
}

# Fills the installed pkg cache
_dnf_installed_pkgs() {
  if ( [[ ${+_installed_pkgs} -eq 0 ]] || _cache_invalid INSTALLED ) &&
    ! _retrieve_cache INSTALLED;
  then
    _installed_pkgs=( $(dnf -C list installed | sed 's/\s.*//' | grep '\.' 2>/dev/null) )
    _store_cache INSTALLED _installed_pkgs
  fi
}

# Fills the available pkg cache
_dnf_available_pkgs() {
  if ( [[ ${+_available_pkgs} -eq 0 ]] || _cache_invalid AVAILABLE ) &&
    ! _retrieve_cache AVAILABLE;
  then
    _available_pkgs=( $(dnf -C list available | sed 's/\s.*//' | grep '\.' 2>/dev/null) )
    _store_cache AVAILABLE _available_pkgs
  fi
}

# Fills the upgrade pkg cache
_dnf_upgrade_pkgs()
{
  if ( [[ ${+_upgrade_pkgs} -eq 0 ]] || _cache_invalid UPGRADE ) &&
    ! _retrieve_cache UPGRADE;
  then
    _upgrade_pkgs=( $(dnf -C list upgrade | sed 's/\s.*//' | grep '\.' 2>/dev/null) )
    _store_cache UPGRADE _upgrade_pkgs
  fi
}

# Gets the list of defined repos
__dnf_repos() {
  local trepo
  local -a tarray
  tarray=( $(egrep -h '(^\[.*\]|^enabled.*=)' /etc/dnf.repos.d/*.repo /etc/dnf.conf 2>/dev/null | sed -e 's/ //g' | sed -e 's/\[//g' | sed -e 's/\].*$//g' 2>/dev/null) )
  local -i eindex=0
  local -i dindex=0
  for line in $tarray; do
    if [[ "$line" = "enabled=1" ]]; then
      enabled_dnf_repos+=($trepo)
    elif [[ "$line" = "enabled=0" ]]; then
      disabled_dnf_repos+=($trepo)
    elif [[ "$line" != "main" ]]; then
      trepo=$line
    fi
  done
}

(( $+functions[_dnf_disabled_repos_list] )) || _dnf_disabled_repos_list() {
  local -a enabled_dnf_repos disabled_dnf_repos
  __dnf_repos
  _sequence compadd "$@" - -a disabled_dnf_repos
}

(( $+functions[_dnf_enabled_repos_list] )) || _dnf_enabled_repos_list() {
  local -a enabled_dnf_repos disabled_dnf_repos
  __dnf_repos
  _sequence compadd "$@" - -a enabled_dnf_repos
}

# Completion function for erase|remove
(( $+functions[_dnf_erase] )) || _dnf_erase() {
  _dnf_installed_pkgs
  compadd "$@" -a -- _installed_pkgs
}

# Completion function for install
(( $+functions[_dnf_install] )) || _dnf_install() {
  if ! [[ $PREFIX == */* ]]; then
    _dnf_available_pkgs
  fi

  local ret=1
  _tags files packages
  while _tags; do
    if _requested files; then
      compadd "$@" -a -- _available_pkgs
    fi
    if _requested packages; then
      _call_function - _dnf_localinstall
    fi
    (( ret )) || break
  done
  return ret
}

# Completion function for localinstall
(( $+functions[_dnf_localinstall] )) || _dnf_localinstall() {
  _files -/ -g '(#i)*.rpm(-.)'
}

# Completion function for localupdate
(( $+functions[_dnf_localupdate] )) || _dnf_localupdate() {
  _files -/ -g '(#i)*.rpm(-.)'
}

# Completion function for update/upgrade
(( $+functions[_dnf_update] )) || _dnf_update() {
  _dnf_upgrade_pkgs
  compadd "$@" -a -- _upgrade_pkgs
}

_dnf_all() {
  _dnf_all_pkgs
  compadd "$@" -a -- _all_pkgs
}

_dnf_list_or_info() {
  local -a listlist
  listlist=(
    "all:all packages in repositories"
    "available:packages available in repositories"
    "updates:packages with updates available"
    "installed:installed packages"
    "extras:packages installed that are not available in any dnf repository"
    "obsoletes:packages installed that are obsoleted"
    "recent:packages recently added to repositories"
  )

  if (( CURRENT == 2 )); then
    _describe -t dnf-list-subcmds "dnf info/list sub-commands" listlist || _dnf_all
  else
    local subcmd
    subcmd="${${listlist[(r)$words[2]:*]%%:*}}"
    # offer packages selected by the subcommand
    case $subcmd in
      all) _dnf_all;;
      installed) _dnf_erase;;
      available) _dnf_install;;
      updates) _dnf_update;;
    esac
  fi
}

# Completion function for list
(( $+functions[_dnf_list] )) || _dnf_list() {
  _dnf_list_or_info
}

# Completion function for info
(( $+functions[_dnf_info] )) || _dnf_info() {
  _dnf_list_or_info
}

# Completion function for provides|whatprovides
(( $+functions[_dnf_provides] )) || _dnf_provides() {
  _files
}

# Completion function for clean
(( $+functions[_dnf_clean] )) || _dnf_clean() {
  local -a cleanlist
  cleanlist=(
    "all:all cache"
    "cache:all cache"
    "dbcache:DB cache"
    "headers:cache headers"
    "packages:cache packages"
    "metadata:cache meta-data"
  )

  if (( CURRENT == 2 )); then
    _describe -t dnf-clean-subcmds "dnf clean sub-commands" cleanlist
  fi
}

_dnf_caching_policy() {
  local _dnfrepomds
  local -a oldp

  # rebuild if cache is more than a week old
  oldp=( "$1"(mw+1) )
  (( $#oldp )) && return 0

  _dnfrepomds=( /var/cache/dnf/**/repomd.xml )

  if (( $#_dnfrepomds )); then
    for repo in $_dnfrepomds; do
      [[ "$repo" -nt "$1" ]] && return 0
    done
  fi

  return 1
}

_dnf "$@"