ohmyzsh/plugins/taskman/taskman.plugin.zsh
2025-06-13 15:37:10 +08:00

243 lines
7.4 KiB
Bash

#!/usr/bin/env zsh
#
# Terminal Task Manager Plugin for Oh-My-Zsh
# A powerful sidebar-style task manager that runs entirely in your terminal
#
# Author: @oiahoon
# Version: 1.0.0
# Repository: https://github.com/ohmyzsh/ohmyzsh/tree/master/plugins/taskman
#
# Features:
# - Interactive terminal UI with keyboard shortcuts
# - Command line interface for quick operations
# - Priority system with color coding
# - Persistent JSON storage
# - Vim-like keybindings
# - Zero external dependencies (Python 3 only)
#
# Plugin directory detection (Oh-My-Zsh compatible)
TASKMAN_PLUGIN_DIR="${0:h}"
# Data directory (store tasks in home directory)
TASKMAN_DATA_DIR="$HOME/.taskman"
# Ensure data directory exists
[[ ! -d "$TASKMAN_DATA_DIR" ]] && mkdir -p "$TASKMAN_DATA_DIR"
# Color definitions for output
TASKMAN_COLOR_RED="\033[31m"
TASKMAN_COLOR_GREEN="\033[32m"
TASKMAN_COLOR_YELLOW="\033[33m"
TASKMAN_COLOR_BLUE="\033[34m"
TASKMAN_COLOR_CYAN="\033[36m"
TASKMAN_COLOR_RESET="\033[0m"
# Main task manager function
tasks() {
local action="$1"
shift
case "$action" in
"" | "ui" | "show")
# Launch the full UI
_taskman_launch_ui
;;
"add" | "new" | "create")
# Quick add task from command line
_taskman_add_task "$@"
;;
"list" | "ls")
# List tasks in terminal
_taskman_list_tasks "$@"
;;
"done" | "complete")
# Mark task as complete
_taskman_complete_task "$@"
;;
"delete" | "del" | "rm")
# Delete a task
_taskman_delete_task "$@"
;;
"help" | "-h" | "--help")
_taskman_show_help
;;
*)
echo "${TASKMAN_COLOR_RED}Unknown action: $action${TASKMAN_COLOR_RESET}"
_taskman_show_help
return 1
;;
esac
}
# Launch the full UI
_taskman_launch_ui() {
if ! command -v python3 >/dev/null 2>&1; then
echo "${TASKMAN_COLOR_RED}Error: Python 3 is required but not installed.${TASKMAN_COLOR_RESET}"
echo "${TASKMAN_COLOR_YELLOW}Please install Python 3 to use the interactive UI.${TASKMAN_COLOR_RESET}"
return 1
fi
# Set the data file path
export TASKMAN_DATA_FILE="$TASKMAN_DATA_DIR/tasks.json"
# Run the Python task manager
python3 "$TASKMAN_PLUGIN_DIR/bin/task_manager.py"
}
# Quick add task from command line
_taskman_add_task() {
if [[ $# -eq 0 ]]; then
echo "${TASKMAN_COLOR_RED}Error: Please provide task description${TASKMAN_COLOR_RESET}"
echo "Usage: tasks add <task description> [priority]"
return 1
fi
local task_text="$1"
local priority="${2:-normal}"
# Validate priority
case "$priority" in
"high"|"normal"|"low")
;;
*)
echo "${TASKMAN_COLOR_YELLOW}Warning: Invalid priority '$priority', using 'normal'${TASKMAN_COLOR_RESET}"
priority="normal"
;;
esac
if command -v python3 >/dev/null 2>&1; then
python3 "$TASKMAN_PLUGIN_DIR/bin/task_cli.py" add "$task_text" "$priority"
# Show color-coded confirmation
local priority_color
case "$priority" in
"high") priority_color="$TASKMAN_COLOR_RED" ;;
"low") priority_color="$TASKMAN_COLOR_CYAN" ;;
*) priority_color="$TASKMAN_COLOR_YELLOW" ;;
esac
echo "${TASKMAN_COLOR_GREEN}✓ Added task:${TASKMAN_COLOR_RESET} ${priority_color}[$priority]${TASKMAN_COLOR_RESET} $task_text"
else
echo "${TASKMAN_COLOR_RED}Error: Python 3 is required for task management.${TASKMAN_COLOR_RESET}"
return 1
fi
}
# List tasks in terminal
_taskman_list_tasks() {
local filter="$1"
if command -v python3 >/dev/null 2>&1; then
python3 "$TASKMAN_PLUGIN_DIR/bin/task_cli.py" list "$filter"
else
echo "${TASKMAN_COLOR_RED}Error: Python 3 is required for task management.${TASKMAN_COLOR_RESET}"
return 1
fi
}
# Complete a task
_taskman_complete_task() {
if [[ $# -eq 0 ]]; then
echo "${TASKMAN_COLOR_RED}Error: Please provide task ID${TASKMAN_COLOR_RESET}"
echo "Usage: tasks done <task_id>"
return 1
fi
local task_id="$1"
if command -v python3 >/dev/null 2>&1; then
python3 "$TASKMAN_PLUGIN_DIR/bin/task_cli.py" complete "$task_id"
else
echo "${TASKMAN_COLOR_RED}Error: Python 3 is required for task management.${TASKMAN_COLOR_RESET}"
return 1
fi
}
# Delete a task
_taskman_delete_task() {
if [[ $# -eq 0 ]]; then
echo "${TASKMAN_COLOR_RED}Error: Please provide task ID${TASKMAN_COLOR_RESET}"
echo "Usage: tasks delete <task_id>"
return 1
fi
local task_id="$1"
if command -v python3 >/dev/null 2>&1; then
python3 "$TASKMAN_PLUGIN_DIR/bin/task_cli.py" delete "$task_id"
else
echo "${TASKMAN_COLOR_RED}Error: Python 3 is required for task management.${TASKMAN_COLOR_RESET}"
return 1
fi
}
# Show help
_taskman_show_help() {
cat << 'EOF'
Terminal Task Manager - Oh-My-Zsh Plugin
Usage:
tasks [action] [arguments]
Actions:
(no action) Launch full interactive UI
ui, show Launch full interactive UI
add <text> [priority] Add new task (priority: high, normal, low)
list [filter] List tasks (filter: all, pending, completed)
done <id> Mark task as completed
delete <id> Delete a task
help Show this help
Examples:
tasks # Launch interactive UI
tasks add "Fix bug in login" # Add normal priority task
tasks add "Deploy to prod" high # Add high priority task
tasks list # List all tasks
tasks list pending # List only pending tasks
tasks done 3 # Mark task ID 3 as completed
tasks delete 5 # Delete task ID 5
Interactive UI Keys:
↑/k Move up n New task
↓/j Move down Space Toggle completion
h Help d Delete task
q Quit
Aliases:
tm, task, todo - All point to 'tasks' command
Data Storage:
Tasks are stored in: ~/.taskman/tasks.json
Requirements:
Python 3.6+ (for interactive UI and CLI operations)
EOF
}
# Convenient aliases
alias tm='tasks'
alias task='tasks'
alias todo='tasks'
# Quick access function for sidebar workflow
task-sidebar() {
echo "${TASKMAN_COLOR_BLUE}Starting task manager in sidebar mode...${TASKMAN_COLOR_RESET}"
echo "${TASKMAN_COLOR_YELLOW}Tip: Use terminal split to run this alongside your work${TASKMAN_COLOR_RESET}"
tasks ui
}
# Optional: Show task summary on shell startup
# Uncomment the next line to enable startup summary
# _taskman_startup_summary
_taskman_startup_summary() {
if [[ -f "$TASKMAN_DATA_DIR/tasks.json" ]] && command -v python3 >/dev/null 2>&1; then
local pending_count=$(python3 "$TASKMAN_PLUGIN_DIR/bin/task_cli.py" count pending 2>/dev/null || echo "0")
local completed_count=$(python3 "$TASKMAN_PLUGIN_DIR/bin/task_cli.py" count completed 2>/dev/null || echo "0")
if [[ "$pending_count" -gt 0 ]]; then
echo "${TASKMAN_COLOR_CYAN}📋 Task Summary: ${pending_count} pending, ${completed_count} completed${TASKMAN_COLOR_RESET}"
echo "${TASKMAN_COLOR_YELLOW} Type 'tasks' to manage your tasks${TASKMAN_COLOR_RESET}"
fi
fi
}