#compdef task
#
# zsh completion for taskwarrior
#
# Copyright 2010 - 2011 Johannes Schlatow
# Copyright 2009 P.C. Shyamshankar
# All rights reserved.
#
# This script is part of the taskwarrior project.
#
# This program is free software; you can redistribute it and/or modify it under
# the terms of the GNU General Public License as published by the Free Software
# Foundation; either version 2 of the License, or (at your option) any later
# version.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
# details.
#
# You should have received a copy of the GNU General Public License along with
# this program; if not, write to the
#
#     Free Software Foundation, Inc.,
#     51 Franklin Street, Fifth Floor,
#     Boston, MA
#     02110-1301
#     USA
#
typeset -g _task_cmds _task_projects _task_tags _task_config _task_modifiers
_task_projects=($(task _projects))
_task_tags=($(task _tags))
_task_ids=($(task _ids))
_task_config=($(task _config))
_task_modifiers=(
	'before' \
	'after' \
	'none' \
	'any' \
	'is' \
	'isnt' \
	'has' \
	'hasnt' \
	'startswith' \
	'endswith' \
	'word' \
	'noword'
)
_task_cmds=($(task _commands))
_task_zshcmds=( ${(f)"$(task _zshcommands)"} )


_task_idCmds=(
	'append' \
	'prepend' \
	'annotate' \
	'denotate' \
	'edit' \
	'duplicate' \
	'info' \
	'start' \
	'stop' \
	'done'
)

_task_idCmdsDesc=(
	'append:Appends more description to an existing task.' \
	'prepend:Prepends more description to an existing task.' \
	'annotate:Adds an annotation to an existing task.' \
	'denotate:Deletes an annotation of an existing task.' \
	'edit:Launches an editor to let you modify a task directly.' \
	'duplicate:Duplicates the specified task, and allows modifications.' \
	'info:Shows all data, metadata for specified task.' \
	'start:Marks specified task as started.' \
	'stop:Removes the start time from a task.' \
	'done:Marks the specified task as completed.'
)

_task() {
    _arguments -s -S \
        "*::task command:_task_commands"
    return 0
}

local -a reply args word
word=$'[^\0]#\0'

# priorities
local -a task_priorities
_regex_words values 'task priorities' \
	'H:High' \
	'M:Middle' \
	'L:Low'
task_priorities=("$reply[@]")

# projects
local -a task_projects
task_projects=(
	/"$word"/
	":values:task projects:compadd -a _task_projects"
)

local -a _task_dates
_regex_words values 'task dates' \
	'tod*ay:Today' \
	'yes*terday:Yesterday' \
	'tom*orrow:Tomorrow' \
	'sow:Start of week' \
	'soww:Start of work week' \
	'socw:Start of calendar week' \
	'som:Start of month' \
	'soy:Start of year' \
	'eow:End of week' \
	'eoww:End of work week' \
	'eocw:End of calendar week' \
	'eom:End of month' \
	'eoy:End of year' \
	'mon:Monday' \
	'tue:Tuesday'\
	'wed:Wednesday' \
	'thu:Thursday' \
	'fri:Friday' \
	'sat:Saturday' \
	'sun:Sunday'
_task_dates=("$reply[@]")

local -a _task_reldates
_regex_words values 'task reldates' \
	'hrs:n hours' \
	'day:n days' \
	'1st:first' \
	'2nd:second' \
	'3rd:third' \
	'th:4th, 5th, etc.' \
	'wks:weeks'
_task_reldates=("$reply[@]")

task_dates=(
	\( "$_task_dates[@]" \|
    \( /$'[0-9][0-9]#'/- \( "$_task_reldates[@]" \) \)
	\)
)

_regex_words values 'task frequencies' \
	'daily:Every day' \
	'day:Every day' \
	'weekdays:Every day skipping weekend days' \
	'weekly:Every week' \
	'biweekly:Every two weeks' \
	'fortnight:Every two weeks' \
	'quarterly:Every three months' \
	'semiannual:Every six months' \
	'annual:Every year' \
	'yearly:Every year' \
	'biannual:Every two years' \
	'biyearly:Every two years'
_task_freqs=("$reply[@]")

local -a _task_frequencies
_regex_words values 'task frequencies' \
	'd:days' \
	'w:weeks' \
	'q:quarters' \
	'y:years'
_task_frequencies=("$reply[@]")

task_freqs=(
	\( "$_task_freqs[@]" \|
     \( /$'[0-9][0-9]#'/- \( "$_task_frequencies[@]" \) \)
	\)
)

# attributes
local -a task_attributes
_regex_words -t ':' default 'task attributes' \
	'pro*ject:Project name:$task_projects' \
	'du*e:Due date:$task_dates' \
	'wa*it:Date until task becomes pending:$task_dates' \
	're*cur:Recurrence frequency:$task_freqs' \
	'pri*ority:priority:$task_priorities' \
	'un*til:Recurrence end date:$task_dates' \
	'fg:Foreground color' \
	'bg:Background color' \
	'li*mit:Desired number of rows in report'
task_attributes=("$reply[@]")

args=(
	\( "$task_attributes[@]" \|
	\( /'(project|due|wait|recur|priority|until|fg|bg|limit).'/- \( /$'[^:]#:'/ ":default:modifiers:compadd -S ':' -a _task_modifiers" \) \) \|
	\( /'(rc).'/- \( /$'[^:]#:'/ ":arguments:config:compadd -S ':' -a _task_config" \) \) \|
	\( /'(+|-)'/- \( /"$word"/ ":values:remove tag:compadd -a _task_tags" \) \) \|
	\( /"$word"/ \)
	\) \#
)
_regex_arguments _task_attributes "${args[@]}"

## task commands

# default completion
(( $+functions[_task_default] )) ||
_task_default() {
	_task_attributes "$@"
}

# commands expecting an ID
(( $+functions[_task_id] )) ||
_task_id() {
	if (( CURRENT < 3 )); then
		# update IDs
		_task_zshids=( ${(f)"$(task _zshids)"} )
		_describe -t values 'task IDs' _task_zshids
	else
		_task_attributes "$@"
	fi
}

# merge completion
(( $+functions[_task_merge] )) ||
_task_merge() {
	# TODO match URIs in .taskrc
	_files
}

# push completion
(( $+functions[_task_push] )) ||
_task_push() {
	# TODO match URIs in .taskrc
	_files
}

# pull completion
(( $+functions[_task_pull] )) ||
_task_pull() {
	# TODO match URIs in .taskrc
	_files
}


# modify (task [0-9]* ...) completion
(( $+functions[_task_modify] )) ||
_task_modify() {
	_describe -t commands 'task command' _task_idCmdsDesc
	_task_attributes "$@"
}

## first level completion => task sub-command completion
(( $+functions[_task_commands] )) ||
_task_commands() {
    local cmd ret=1
	if (( CURRENT == 1 )); then
		# update IDs
		_task_zshids=( ${(f)"$(task _zshids)"} )

		_describe -t commands 'task command' _task_zshcmds
		_describe -t values 'task IDs' _task_zshids
		# TODO match more than one ID
	elif [[ $words[1] =~ ^[0-9]*$ ]] then
		 _call_function ret _task_modify
		 return ret
	else
#        local curcontext="${curcontext}"
#        cmd="${_task_cmds[(r)$words[1]:*]%%:*}"
		  cmd="${_task_cmds[(r)$words[1]]}"
		  idCmd="${(M)_task_idCmds[@]:#$words[1]}"
        if (( $#cmd )); then
#            curcontext="${curcontext%:*:*}:task-${cmd}"

				if (( $#idCmd )); then
					_call_function ret _task_id
				else
					_call_function ret _task_${cmd} ||
						_call_function ret _task_default ||
							_message "No command remaining."
				fi
        else
            _message "Unknown subcommand ${cmd}"
        fi
        return ret
    fi
}