#compdef pkg

_pkg_installed() {
	local expl
	_wanted packages expl package \
	compadd "$@" - $(/usr/local/sbin/pkg query "%n-%v")
}

_pkg_available_name() {
	local expl
	_wanted packages expl package \
	compadd "$@" - $(/usr/local/sbin/pkg rquery "%n")
}

_pkg_aliases() {
	local expl
	_wanted aliases expl alias \
	compadd "$@" - $(/usr/local/sbin/pkg alias -ql)
}

_pkg_available() {
	local ret=1
	_tags files packages
	while _tags; do
		if _requested files; then
			_files "$@" -g "*.t?z" && ret=0
		fi
		if _requested packages; then
			compadd "$@" - $(/usr/local/sbin/pkg rquery "%n-%v") && ret=0
		fi
		(( ret )) || break
	done
	return ret
}

_pkg_config_opts() {
	local -a _pkg_opts
	_pkg_opts=(
	'ABI[ABI of the package you want to install]:string:'
	'ALIAS[Define local aliases for various pkg(8) standard command lines]:key/value list:'
	'DEFAULT_ALWAYS_YES[Default to "yes" for all questions requiring user confirmation]:boolean:(yes no)'
	'ASSUME_ALWAYS_YES[Assumes "yes" to all questions requiring user confirmation]:boolean:(yes no)'
	'CONSERVATIVE_UPGRADE[Give priority to the repo where a package was first installed from]:boolean:(yes no)'
	'CUDF_SOLVER[Experimental\: use an external CUDF solver]:CUDF solver:_files'
	'CASE_SENSITIVE_MATCH[Case sensitive package matching]:boolean:(yes no)'
	'DEBUG_LEVEL[Debugging level]:debug level:( {0..4} )'
	'DEVELOPER_MODE[Makes certain errors immediately fatal, adds warnings and suggestions]:boolean:(yes no)'
	'EVENT_PIPE[Send all event messages to specified FIFO or UNIX socket]:event pipe:_files'
	'FETCH_RETRY[Number of times to retry a failed fetch of a file]:number of retries:'
	'FETCH_TIMEOUT[Number of seconds to wait for a file to download]:timeout (seconds):'
	'HANDLE_RC_SCRIPTS[Automatically perform start/stop of service on install/removal]:boolean:(yes no)'
	'INDEXDIR[The directory to search for INDEXFILE in]: : _files -/'
	'INDEXFILE[The ports index file]: : _files'
	'IP_VERSION[Restrict network access to specified IP version]:IP version:((0\:"system default" 4\:IPv4 6\:IPv6 ))'
	'LOCK_RETRIES[Number of retries to obtain a lock]:retries:'
	'LOCK_WAIT[Wait time in seconds to regain a lock]:wait time (seconds):'
	'NAMESERVER[Hostname or IPv\[46\] address of a nameserver for DNS resolution]: : _hosts'
	'PERMISSIVE[Ignore conflicts while registering a package]:boolean:(yes no)'
	'PKG_CACHEDIR[Specifies the cache directory for packages]: : _files -/'
	'PKG_CREATE_VERBOSE[Make pkg_create(8) use verbose mode]:boolean:(yes no)'
	'PKG_DBDIR[Specifies the directory to use for storing the package database files]: : _files -/'
	'PKG_ENABLE_PLUGINS[Activate plugin support]:boolean:(yes no)'
	'PKG_ENV[key/value pair of environment variables]:key/value list:'
	'PKG_PLUGINS_DIR[Specific directory for plugins]: : _files -/'
	'PKG_SSH_ARGS[Extra arguments for ssh(1)]:ssh(1) arguments:'
	'PLIST_KEYWORDS_DIR[Directory containing definitions of plist keywords]: : _files -/'
	'PLIST_ACCEPT_DIRECTORIES[Accept directories listed like plain files in plist]:boolean:(yes no)'
	'PLUGINS[List of plugins pkg(8) should load]:plugins:'
	'PLUGINS_CONF_DIR[Directory containing per-plugin configurations files]: : _files -/'
	'PORTSDIR[Specifies the location to the Ports directory]: : _files -/'
	'REPOS_DIR[List of directories to search for repository configuration files]: : _files -/' # completing one directory is better than nothing.
	'REPO_AUTOUPDATE[Automatically check for repo.sqlsite updates]:boolean:(yes no)'
	'RUN_SCRIPTS[Run pre-/post-installation action scripts]:boolean:(yes no)'
	'SAT_SOLVER[Experimental\: use an external SAT solver]:SAT solver:_files'
	'SQLITE_PROFILE[Profile SQLite queries]:boolean:(yes no)'
	'SSH_RESTRICT_DIR[Directory which the ssh subsystem will be restricted to]:chroot:_files -/'
	'SYSLOG[Log install/deinstall/upgrade operations to syslog(3)]:boolean:(yes no)'
	'UNSET_TIMESTAMP[Do not include timestamps in the package tar(1) archive]:boolean:(yes no)'
	'VERSION_SOURCE[Default database for comparing version numbers in pkg-version(8)]:database:(( I\:index P\:ports R\:remote ))'
	'VULNXML_SITE[Specifies the URL to fetch the vuln.xml vulnerability database from]: : _urls -F "( gopher:* file:* ftp:* )"'
	'WORKERS_COUNT[How many workers are used for pkg-repo]:workers:( {0..$(sysctl -n hw.ncpu)} )'
	)
	_values 'configuration option' "$_pkg_opts[@]"
}

_pkg() {
	local curcontext="$curcontext" state state_descr line ret=1
	local earlyargs subcmd
	local subcmd_list cmd
	earlyargs=(
	'(-d --debug)'{-d,--debug}'[increment debug level]'
	'(-j --jail)'{-j,--jail}'[execute pkg(8) inside a jail(8)]:jail:_jails'
	'(-c --chroot)'{-c,--chroot}'[execute pkg(8) inside a chroot(8)]:chroot:_files -/'
	\*{-o,--option}'[set configuration option]:option:_pkg_config_opts'
	'(-C --config)'{-C,--config}'[use the specified configuration file]:configuration file:_files'
	'(-l --list)'{-l,--list}'[list available command and exit]'
	'(-v --version)'{-v,--version}'[display pkg(8) version]'
	'-N[test if pkg(8) is activated and avoid auto-activation]'
	'(-R --repo-conf-dir)'{-R,--repo-conf-dir}'[specify directory for per-repository configuration files]: : _files'
	'-4[use IPv4 for fetching repository and packages]'
	'-6[use IPv6 for fetching repository and packages]'
	)

	subcmd=(
		'add[compatibility interface to install a package]'
		'alias[list the command line aliases]'
		'annotate[add, modify or delete tag-value style annotations on packages]'
		'audit[report vulnerable packages]'
		'autoremove[remove orphaned packages]'
		'backup[back-up and restore the local package database]'
		'check[check for missing dependencies and database consistency]'
		'clean[clean old packages from the cache]'
		'config[display value of a configuration option]'
		'convert[convert database from/to pkgng]'
		'create[create software package distributions]'
		'delete[delete packages from the database and the system]'
		'fetch[fetch packages from a remote repository]'
		'help[display help information]'
		'info[display information about installed packages]'
		'install[install packages from remote package repositories]'
		'lock[lock package against modifications or deletion]'
		'query[query information about installed packages]'
		'register[register a package with the local database]'
		'remove[delete packages from the database and the system]'
		'repo[create a package repository catalogue]'
		'rquery[query information in repository catalogues]'
		'search[perform a search of package repository catalogues]'
		'set[modify information about packages in the local database]'
		'shell[open a debug shell]'
		'shlib[list packages that link against a specific shared library]'
		'stats[display package database statistics]'
		'unlock[unlock a package, allowing modification or deletion]'
		'update[update package repository catalogues]'
		'updating[display UPDATING information for a package]'
		'upgrade[perform upgrades of packaged software distributions]'
		'version[display versions of installed packages]'
		'which[display which package installed a specific file]'
	)

	if [[ $service = "pkg" ]]; then

		_arguments -C -A "-*" \
			$earlyargs \
			'*::command:->subcmd' && return

		if (( CURRENT == 1 )); then
			_values 'pkg subcommands' \
				$subcmd
			return
		fi
		service="$words[1]"
		curcontext="${curcontext%:*}-$service:"
	fi

	case $service in
		(add)
			_arguments -C -A 'package' \
				'(-A --automatic)'{-A,--automatic}'[Mark the installed packages as automatic]' \
				'(-f --force)'{-f,--force}'[Force the reinstallation of package]' \
				'(-I --no-scripts)'{-I,--no-scripts}'[Ignore pre-/post-install scripts]' \
				'(-M --accept-missing)'{-M,--accept-missing}'[Ignore missing dependencies when installing]' \
				'(-q --quiet)'{-q,--quiet}'[Force quiet output]' \
				'*:package:->packages'

			if [[ $state[1] == packages ]]; then
				local ret=1
				_tags packages urls
				while _tags; do
					if _requested packages; then
						_files -g "*.t?z" && ret=0
					fi
					if _requested urls; then
						_urls -F "( gopher:* file:* )" && ret=0
					fi
					(( ret )) || break
				done
				return
			fi
			return
			;;
		(alias)
			_arguments -A '-*' -s \
				'(-l --list)'{-l,--list}'[list aliases]' \
				'(-q --quiet)'{-q,--quiet}'[quiet]' \
				'*:aliases:_pkg_aliases'
			return
			;;
		(audit)
			_arguments -A '-*' -s \
				'(-F --fetch)'{-F,--fetch}'[fetch database before checking]' \
				'(-q --quiet)'{-q,--quiet}'[quiet]' \
				'(-r --recursive)'{-r,--recursive}'[print packages that depend on vulnerable packages]' \
				'(-f --file)'{-f,--file}'[local copy of the vulnerability database]:audit file:_files' \
				'*:package:_pkg_installed'
			return
			;;
		(autoremove)
			_arguments -s \
				'(-q --quiet)'{-q,--quiet}'[force quiet output]' \
				- '(yes)' \
				{-y,--yes}'[assume yes when asked for confirmation]' \
				- '(dryrun)' \
				{-n,--dry-run}'[assume no (dry run) when asked for confirmation]'
			return
			;;
		(backup)
			_arguments -s \
				'(-q --quiet)'{-q,--quiet}'[force quiet output]' \
				- '(dump)' \
				{-d,--dump}'[dumps the local package database]:destination:_files' \
				- '(restore)' \
				{-r,--restore}'[restore the local package database]:destination:_files'
			return
			;;
		(check)
			_arguments -A '-*' -s \
				'(-B --shlibs)'{-B,--shlibs}'[reanalyze the shared libraries]' \
				'(-d --dependencies)'{-d,--dependencies}'[check for and install missing dependencies]' \
				'(-r --recompute)'{-r,--recompute}'[recompute sizes and checksums of installed]' \
				'(-s --checksums)'{-s,--checksums}'[find invalid checksums]' \
				'(-v --verbose)'{-v,--verbose}'[be verbose]' \
				'(-n --dry-run)'{-n,--dry-run}'[check for missing dependencies but do not install them]' \
				'(-y --yes)'{-y,--yes}'[assume yes for confirmations]' \
				'(-q --quiet)'{-q,--quiet}'[force quiet output]' \
				'(-a --all -i --case-insensitive -C --case-sensitive)'{-i,--case-insensitive}'[case insensitive pattern matching]' \
				'(-a --all -i --case-insensitive -C --case-sensitive)'{-C,--case-sensitive}'[case sensitive pattern matching]' \
				${words[(r)(-*a*|--all)]+!}'*:available package:_pkg_installed' \
				- '(all)' \
				'(-i --case-insensitive -C --case-sensitive)'{-a,--all}'[process all packages]' \
				- '(patterns)' \
				{-x,--regex}'[process packages that match the regex pattern]' \
				{-g,--glob}'[process packages that match the glob pattern]'
			return
			;;
		(clean)
			_arguments -s \
				'(-a --all)'{-a,--all}'[delete all packages]' \
				'(-n --dry-run)'{-n,--dry-run}'[show what packages would be removed]' \
				'(-q --quiet)'{-q,--quiet}'[force quiet output]' \
				'(-y --yes)'{-y,--yes}'[assume yes for confirmations]'
			return
			;;
		(convert)
			_arguments -s \
				'(-d --pkg-dbdir)'{-d,--pkg-dbdir}'[specify the location of the pkg_add(1) dbdir]: : _files -/' \
				'(-n --dry-run)'{-n,--dry-run}'[dry run mode]'
			return
			;;
		(create)
			_arguments -A '-*' -s \
				'(-q --quiet)'{-q,--quiet}'[force quiet output]' \
				'(-v --verbose)'{-v,--verbose}'[be verbose]' \
				'(-n --no-clobber)'{-n,--no-clobber}'[no not overwrite existing packages]' \
				'(-f --format)'{-f,--format}'[format]:format:((tar tgz tbz txz))' \
				'(-o --out-dir)'{-o,--out-dir}'[output directory]:outdir:_files -/' \
				'(-r --root-dir)'{-r,--root-dir}'[specify root directory]:rootdir:_files -/' \
				- '(manifest)' \
				{-M,--manifest}'[specify manifest file]:manifest file:_files' \
				- metadata \
				'(-m --metadata)'{-m,--metadata}'[specify manifest directory]:manifestdir:_files -/' \
				'(-p --plist)'{-p,--plist}'[specify package metadata using the legacy plist format]:plist' \
				- '(all)' \
				{-a,--all}'[process all packages]' \
				- patterns \
				'(-g --glob -x --regex)'{-g,--glob}'[process packages that match a glob pattern]' \
				'(-g --glob -x --regex)'{-x,--regex}'[process packages that match a regex pattern]' \
				'*:package:_pkg_installed'
			return
			;;
		(delete|remove)
			_arguments -A '-*' -s \
				'(-q --quiet)'{-q,--quiet}'[force quiet output]' \
				'(-n --dry-run -y --yes)'{-y,--yes}'[assume yes when asked for confirmation]' \
				'(-y --yes -n --dry-run)'{-n,--dry-run}'[assume no (dry run) when asked for confirmation]' \
				'(-f --force)'{-f,--force}'[force the package(s) to be removed]' \
				'(-a --all -i --case-insensitive -C --case-sensitive)'{-i,--case-insensitive}'[case insensitive pattern matching]' \
				'(-a --all -i --case-insensitive -C --case-sensitive)'{-C,--case-sensitive}'[case sensitive pattern matching]' \
				'(-D --no-deinstall-script)'{-D,--no-deinstall-script}'[do not execute deinstallation scripts]' \
				'(-R --recursive)'{-R,--recursive}'[delete all packages that require the list packages as well]' \
				- '(all)' \
				'(-i --case-insensitive -C --case-sensitive)'{-a,--all}'[process all packages]' \
				- patterns \
				'(-g --glob -x --regex)'{-g,--glob}'[process packages that match a glob pattern]' \
				'(-g --glob -x --regex)'{-x,--regex}'[process packages that match a regex pattern]' \
				'*:package:_pkg_installed'
			return
			;;
		(fetch)
			_arguments -A '-*' -s \
				'(-y --yes)'{-y,--yes}'[assume yes when asked for confirmation]' \
				'(-r --repository)'{-r,--repository}'[specify the repository to fetch from]:reponame' \
				'(-q --quiet)'{-q,--quiet}'[force quiet output]' \
				'(-o --output)'{-o,--output}'[place files in a subdirectory named `All'\'' of the specified directory]: : _files -/' \
				'(-a --all -i --case-insensitive -C --case-sensitive)'{-i,--case-insensitive}'[case insensitive pattern matching]' \
				'(-d --dependencies)'{-d,--dependencies}'[Fetch dependencies as well]' \
				'(-a --all -i --case-insensitive -C --case-sensitive)'{-C,--case-sensitive}'[case sensitive pattern matching]' \
				'(-U --no-repo-update)'{-U,--no-repo-update}'[suppress the automatic update of the repo catalogue]' \
				- '(all)' \
				'(-i --case-insensitive -C --case-sensitive)'{-a,--all}'[process all packages]' \
				- patterns \
				'(-g --glob -x --regex)'{-g,--glob}'[process packages that match a glob pattern]' \
				'(-g --glob -x --regex)'{-x,--regex}'[process packages that match a regex pattern]' \
				'*:available package:_pkg_available' \
				- '(updates)' \
				{-u,--available-update}'[fetch all available updates for currently installed packages]'
			return
			;;
		(help)
			_arguments -s \
				':command:_values -S " " -w "pkg subcommand" ${subcmd/\[*/}'
			return 0
			;;
		(info)
			local mutexopts='(-e --exists -d --dependencies -r --required-by -l --list-files -I --comment'
			mutexopts+=' -R --raw -o --origin -O --by-origin -p --prefix -D --pkg-message -f --full)'
			_arguments -A '-*' -s \
				'(-q --quiet)'{-q,--quiet}'[force quiet output]' \
				'(-a --all -i --case-insensitive -C --case-sensitive)'{-i,--case-insensitive}'[case insensitive pattern matching]' \
				'(-a --all -i --case-insensitive -C --case-sensitive)'{-C,--case-sensitive}'[case sensitive pattern matching]' \
				'--raw-format[output format for raw output]:format:(json json-compact yaml)' \
				'(-f --full)'{-f,--full}'[display full information]' \
				${mutexopts}{-e,--exists}'[return 0 if specified package is installed]' \
				${mutexopts}{-R,--raw}'[display the full manifest]' \
				'(-d --dependencies)'{-d,--dependencies}'[display the dependencies]' \
				'(-r --required-by)'{-r,--required-by}'[display the reverse dependencies]' \
				'(-l --list-files)'{-l,--list-files}'[display all files]' \
				'(-o --origin -O --by-origin)'{-o,--origin,-O,--by-origin}'[display origin]' \
				'(-p --prefix)'{-p,--prefix}'[display prefix]' \
				'(-D --pkg-message)'{-D,--pkg-message}'[display message]' \
				'(-k --locked)'{-k,--locked}'[show the locking status]' \
				'(-I --comment)'{-I,--comment}'[display comments]' \
				'(-b --provided-shlibs)'{-b,--provided-shlibs}'[display shared libraries provided by the package]' \
				'(-B --required-shlibs)'{-B,--required-shlibs}'[display all shared libraries used by the package]' \
				'(-s --size)'{-s,--size}'[display the total size of the files installed by the package]' \
				- '(all)' \
				'(-i --case-insensitive -C --case-sensitive)'{-a,--all}'[process all packages]' \
				- '(pkg-file)' \
				{-F,--file}'[display information from an package archive]:package:_files' \
				- patterns \
				'(-g --glob -x --regex)'{-g,--glob}'[process packages that match a glob pattern]' \
				'(-g --glob -x --regex)'{-x,--regex}'[process packages that match a regex pattern]' \
				'*:available package:_pkg_installed'
			return
			;;
		(install)
			_arguments -A '-*' -s \
				'(-A --automatic)'{-A,--automatic}'[Mark the installed packages as automatic]' \
				'(-a --all -i --case-insensitive -C --case-sensitive)'{-i,--case-insensitive}'[case insensitive pattern matching]' \
				'(-a --all -i --case-insensitive -C --case-sensitive)'{-C,--case-sensitive}'[case sensitive pattern matching]' \
				'(-r --repository)'{-r,--repository}'[specify the repository to install packages from]:reponame' \
				'(-U --no-repo-update)'{-U,--no-repo-update}'[suppress the automatic update of the repo catalogue]' \
				'(-I --no-install-scripts)'{-I,--no-install-scripts}'[do not execute any pre/post-install scripts]' \
				'(-M --ignore-missing)'{-M,--ignore-missing}'[ignore missing dependencies]' \
				'(-n --dry-run -y --yes)'{-y,--yes}'[assume yes when asked for confirmation]' \
				'(-y --yes -n --dry-run)'{-n,--dry-run}'[assume no (dry run) when asked for confirmation]' \
				'(-f --force)'{-f,--force}'[force reinstallation if needed]' \
				'(-R --from-root)'{-R,--from-root}'[reinstall every package depending on matching expressions]' \
				${words[(r)(-*a*|--all)]+!}'*:available package:_pkg_available' \
				- '(all)' \
				'(-i --case-insensitive -C --case-sensitive)'{-a,--all}'[process all packages]' \
				- patterns \
				'(-g --glob -x --regex)'{-g,--glob}'[process packages that match a glob pattern]' \
				'(-g --glob -x --regex)'{-x,--regex}'[process packages that match a regex pattern]'
			return
			;;
		(query)
			_arguments -C -s \
				'(-a --all -i --case-insensitive -C --case-sensitive)'{-C,--case-sensitive}'[case sensitive pattern matching]' \
				'(-a --all -i --case-insensitive -C --case-sensitive)'{-i,--case-insensitive}'[case insensitive pattern matching]' \
				':output format:->query' \
				- '(all)' \
				{-a,--all}'[process all packages]' \
				- '(patterns)' \
				{-g,--glob}'[process packages that match a glob pattern]' \
				{-x,--regex}'[process packages that match a regex pattern]' \
				- '(eval)' \
				{-e,--evaluate}'[process packages that match an evaluation]:evaluation:->evaluation' \
				- '(files)' \
				{-F,--file}'[process the specified package]:package:_files -g "*.t?z"' \
				&& ret=0
			;;
		(register)
			_arguments -A '-*' -s \
				'(-l --legacy)'{-l,--legacy}'[register as a legacy format]' \
				'(-d --debug)'{-d,--debug}'[mark the package as an automatic dependency]' \
				'(-t --test)'{-t,--test}'[enable testing mode]' \
				- metadatadir \
				'(-f --plist)'{-f,--plist}'[packing list file]:packing list file:_files' \
				'(-m --metadata)'{-m,--metadata}'[metadata directory]:metadatadir:_files -/' \
				'(-i --root)'{-i,--root}'[input path (aka root directory)]:input path:_files -/' \
				- '(metadatafile)' \
				{-M,--manifest}'[specify package manifest file]:manifest file:_files'
			return
			;;
		(repo)
			_arguments -A '-*' -s \
				'(-l --list-files)'{-l,--list-files}'[generate list of all files in repo as filesite.txz archive]' \
				'(-q --quiet)'{-q,--quiet}'[force quiet output]' \
				'(-L --legacy)'{-L,--legacy}'[create an pkg 1.2 compatiable repository]' \
				'(-o --output-dir)'{-o,--output-dir}'[specify the location of the new repo]:repo location:_files -/' \
				'(-m --meta-file)'{-m,--meta-file}'[use specified file as repository meta file instead of the defaults]:meta file:_files' \
				':repository path:_files -/' \
				':RSA key: _alternative "files\:RSA key\:_files" "commands\:commands\:_cmdstring"'
			return
			;;
		(rquery)
			_arguments -A '-*' -C -s \
				'(-U --no-repo-update)'{-U,--no-repo-update}'[suppress the automatic update of the repo catalogue]' \
				'(-i --case-insensitive -C --case-sensitive)'{-C,--case-sensitive}'[case sensitive pattern matching]' \
				'(-i --case-insensitive -C --case-sensitive)'{-i,--case-insensitive}'[case insensitive pattern matching]' \
				'(-r --repository)'{-r,--repository}'[specify the repository to search]:reponame' \
				'(-I --index-line)'{-I,--index-line}'[print corresponding line from ports index file]' \
				${words[(r)(-I|--index-line)]+!}':output format:->query' \
				'*:available package:_pkg_available_name' \
				- '(all)' \
				'(-i --case-insensitive -C --case-sensitive)'{-a,--all}'[process all packages]' \
				- '(patterns)' \
				{-g,--glob}'[process packages that match the glob pattern]' \
				{-x,--regex}'[process packages that match a regex pattern]' \
				- '(eval)' \
				{-e,--evalute}'[process packages that match an evaluation]:evaluation:->evaluation' \
				&& ret=0
			;;
		(search)
			local -a label_opts modifier_opts
			label_opts=(comment description name origin pkg-name)
			modifier_opts=(
			annotations arch categories comment depends-on description full
			licenses maintainer name options pkg-size prefix repository
			required-by shared-libs-{required,provided} size url version www
			)
			_arguments -A '-*' -s \
				'(-x --regex -g --glob -e --exact)'{-g,--glob}'[process packages that match the glob pattern]' \
				'(-x --regex -g --glob -e --exact)'{-x,--regex}'[process packages that match the regex pattern]' \
				'(-x --regex -g --glob -e --exact)'{-e,--exact}'[process packages that match the string exactly]' \
				'(-i --case-insensitive -C --case-sensitive)'{-C,--case-sensitive}'[case sensitive pattern matching]' \
				'(-i --case-insensitive -C --case-sensitive)'{-i,--case-insensitive}'[case insensitive pattern matching]' \
				'(-U --no-repo-update)'{-U,--no-repo-update}'[suppress the automatic update of the repo catalogue]' \
				'(-r --repository)'{-r,--repository}'[specify the repository to search]:reponame' \
				'1:available package:_pkg_available_name' \
				- set1 \
				'(-S --search)'{-S,--search}'[specify the field to search the repository catalogue on]:field:( $label_opts )' \
				\*{-Q,--query-modifier}'[add an additional field to the result]:modifier:( $modifier_opts )' \
				- set2 \
				'(-D --description -c --comment)'{-c,--comment}'[use pattern matching on comments text]' \
				'(-D --description -c --comment)'{-D,--description}'[use pattern matching on description text]' \
				'(-d --depends-on)'{-d,--depends-on}'[list dependencies of the matched packages]' \
				'(-f --full)'{-f,--full}"[show \`\`full'' information about the package]" \
				'(-o --origin)'{-o,--origins}'[list packages by origins]' \
				'(-p --prefix)'{-p,--prefix}'[display package installation prefix]' \
				'(-q --quiet)'{-q,--quiet}'[be quiet]' \
				'(-R --raw)'{-R,--raw}'[display full manifest]' \
				'(-s --size)'{-s,--size}'[display the installed size]' \
				'--raw-format[output format for raw output]:format:(json json-compact yaml)'
			return
			;;
		(set)
			local mutexopts='(-o --change-origin -A --automatic -n --change-name)'
			_arguments -A '-*' -s \
				'(-a --all -i --case-insensitive -C --case-sensitive)'{-C,--case-sensitive}'[case sensitive pattern matching]' \
				'(-a --all -i --case-insensitive -C --case-sensitive)'{-i,--case-insensitive}'[case insensitive pattern matching]' \
				${mutexopts}{-A,--automatic}'[mark as automatic or not]:flag:((1\:automatic 0\:not\ automatic))' \
				${mutexopts}{-o,--change-origin}'[change the origin]:oldorigin\:neworigin' \
				${mutexopts}{-n,--change-name}'[change the name]:oldname\:newname' \
				'(-y --yes)'{-y,--yes}'[assume yes when asked for confirmation]' \
				'*:package:_pkg_installed' \
				- '(all)' \
				{-a,--all}'[process all packages]' \
				- '(patterns)' \
				{-g,--glob}'[process packages that match the glob pattern]' \
				{-x,--regex}'[process packages that match the regex pattern]'
			return
			;;
		(shell)
			_arguments -s \
				':database:_files'
			return
			;;
		(shlib)
			_arguments -A '-*' -s \
				'(-q --quiet)'{-q,--quiet}'[force quiet output]' \
				'(-P --provides)'{-P,--provides}'[show the package which provides the library]' \
				'(-R --requires)'{-R,--requires}'[show installed packages which requires the library]' \
				':library:'
			return
			;;
		(stats)
			_arguments -A '-*' -s \
				'(-q --quiet)'{-q,--quiet}'[be quiet]' \
				- '(remote)' \
				{-r,--remote}'[display stats only for the remote package database]' \
				- '(local)' \
				{-l,--local}'[display stats only for the local package database(s)]' \
				- '(size)' \
				{-b,--bytes}'[display space usage, in bytes]'
			return
			;;
		(update)
			_arguments -A '-*' -s \
				'(-f --force)'{-f,--force}'[force update]' \
				'(-q --quiet)'{-q,--quiet}'[be quiet]' \
				'(-r --repository)'{-r,--repository}'[specify the repository to update from]:reponame'
			return
			;;
		(updating)
			# recall dates starting the from the beginning of this month.
			# zstyle ':completion:*:*:pkg-updating:option-(d|-date)-1:*' fake ${(%):-%D{%Y%m}}{01..${(%):-%D{%d}}}
			_arguments -A '-*' -s \
				'(-d --date)'{-d,--date}'[only entries newer than date are shown]:date' \
				'(-f --file)'{-f,--file}'[specify alternative location of the UPDATING file]:UPDATING file:_files' \
				'(-i --case-insensitive)'{-i,--case-insensitive}'[case insensitive pattern matching]' \
				'*:package:_pkg_installed'
			return
			;;
		(upgrade)
			_arguments -A '-*' -s \
				"(-F --fetch-only)"{-F,--fetch-only}"[don't install packages, merely fetch them]" \
				'(-y --yes -n --dry-run)'{-n,--dry-run}'[assume no (dry run) when asked for confirmation]' \
				'(-n --dry-run -y --yes)'{-y,--yes}'[assume yes when asked for confirmation]' \
				'(-f --force)'{-f,--force}'[upgrade/reinstall everything]' \
				'(-i --case-insensitive -C --case-sensitive)'{-C,--case-sensitive}'[case sensitive pattern matching]' \
				'(-i --case-insensitive -C --case-sensitive)'{-i,--case-insensitive}'[case insensitive pattern matching]' \
				'(-U --no-repo-update)'{-U,--no-repo-update}'[suppress the automatic update of the repo catalogue]' \
				'(-I --no-install-scripts)'{-I,--no-install-scripts}'[do not execute any pre/post-install scripts]' \
				'(-r --repository)'{-r,--repository}'[specify the repository to upgrade from]:reponame' \
				'*:available package:_pkg_installed' \
				- '(glob)' \
				{-g,--glob}'[process packages that match a glob pattern]' \
				- '(regex)' \
				{-x,--regex}'[process packages that match a regex pattern]'
			return
			;;
		(version)
			_arguments -A '-*' -s \
				- '(testversion)' \
				{-t,--test-version}'[test a pair of version number strings]: : _message -e values "version string": : _message -e values "version string"' \
				- '(testpattern)' \
				{-T,--test-pattern}'[compare pkgname againist a shell glob pattern]:package:_pkg_installed: : _message -e values glob' \
				- other \
				'(-r --repository -P --ports -R --remote -I --index)'{-I,--index}'[use INDEX file]' \
				'(-r --repository -P --ports -R --remote -I --index)'{-P,--ports}'[force checking against the ports tree]' \
				'(-P --ports -R --remote -I --index)'{-R,--remote}'[use remote repository]' \
				'(-o --origin)'{-o,--origin}'[display package origin, instead of package name]' \
				'(-q --quiet)'{-q,--quiet}'[be quiet]' \
				'(-v --verbose)'{-v,--verbose}'[be verbose]' \
				${words[(r)(-*[PI]*|--(index|ports))]+!}'(-r --repository)'{-r,--repository}'[specify the repository to ]:reponame' \
				'(-U --no-repo-update)'{-U,--no-repo-update}'[suppress the automatic update of the repo catalogue]' \
				'(-L --not-like -l --like)'{-l,--like}'[display only the packages with a given status flag]:flag:((\< = \>))' \
				'(-l --like -L --not-like)'{-L,--not-like}'[display only the packages without a given status flag]:flag:((\< = \>))' \
				'(-i --case-insensitive -C --case-sensitive)'{-C,--case-sensitive}'[case sensitive pattern matching]' \
				'(-i --case-insensitive -C --case-sensitive)'{-i,--case-insensitive}'[case insensitive pattern matching]' \
				'(-x --regex -g --glob -e --exact)'{-g,--glob}'[display packages that match the glob pattern]:glob pattern' \
				'(-x --regex -g --glob -e --exact)'{-x,--regex}'[display packages that match the regex pattern]:regex pattern' \
				'(-x --regex -g --glob -e --exact)'{-e,--exact}'[display the packages that exactly match the string]:string'
			return
			;;
		(which)
			_arguments -A '-*' -s \
				'(-q --quiet)'{-q,--quiet}'[be quiet]' \
				'(-o --origin)'{-o,--origin}'[display origin]' \
				':file:_files' \
				- '(glob)' \
				{-g,--glob}'[treat <file> as a glob pattern]' \
				- '(PATH)' \
				{-p,--path-search}'[search PATH for <file>]'
			return
			;;
	esac

	[[ -z $state ]] && return ret
	local -a specs elements
	elements=(
		'd:dependencies' 'r:reverse dependencies' 'C:categories'
		'O:options' 'L:licenses' 'A:annotations'
		'B:required share libraries' 'b:provided shared libraries'
	)
	specs=(
		'n:package name' 'o:origin' 'p:prefix' 'm:maintainer'
		'c:comment' 'e:description' 'w:home page' 's:size'
		'q:architecture' 'M:message'
		'?:if info exists' '#:no of elements'
	)
	if [[ $service = query ]]; then
		specs+=( 'a:automatic' 'k:locked' 't:timestamp' )
		elements+=( 'F:files' 'D:directories'  'U:users' 'G:groups' )
  fi
	if [[ $state = query ]]; then
    specs+=( 'v:version' 'l:license' 'R:repository' )
		compset -P '(%[^sOFdrA?#]|%?[^%]|[^%])#'
  else
		compset -P '(%[^OFdrA?#]|%?[^%]|[^%])#'
  fi
	if compset -P '%s'; then
		_describe -t units unit '( b:bytes h:human\ readable )' -S '' && ret=0
	elif compset -P '%F'; then
		_describe -t properties property '( p:path s:sum )' -S '' && ret=0
	elif compset -P '%O'; then
		_describe -t properties property '( k:key v:value d:default D:description )' -S '' && ret=0
	elif compset -P '%[dr]'; then
		_describe -t properties 'dependency property' '( n:name o:origin v:version )' -S '' && ret=0
	elif compset -P '%A'; then
		_describe -t properties property '( t:tag s:value )' -S '' && ret=0
	elif compset -P '%(|\\)[?#]'; then
		_describe -t lists list elements -S '' && ret=0
	else
		[[ $IPREFIX = *%(#b)([${(j..)elements%%:*}])* ]] &&
			elements=( ${(M)elements:#$match[1]:*} )
		if [[ $state = evaluation ]]; then
		  _tags variables operators
		else
		  _tags patterns elements
		fi
		while _tags; do
			_requested elements && _describe -t elements 'list element' elements -p % -S '' && ret=0
			_requested variables && _describe -t variables 'variable' specs -p % -S '' && ret=0
			_requested patterns && _describe -t patterns 'query specifier' specs -p % -S '' && ret=0
			_requested operators expl operator compadd -S '' \~ {,\!,\>,\<}= \> \< \&\& \|\| && ret=0
		  (( ret )) || break
		done
	fi
	return ret
}

_pkg "$@"
