mirror of
https://github.com/ohmyzsh/ohmyzsh.git
synced 2026-01-23 02:35:38 +01:00
feat(taskman): add terminal task manager plugin
This commit is contained in:
parent
042605ee6b
commit
9fc3f3dcd6
5 changed files with 1120 additions and 0 deletions
342
plugins/taskman/bin/task_manager.py
Executable file
342
plugins/taskman/bin/task_manager.py
Executable file
|
|
@ -0,0 +1,342 @@
|
|||
#!/usr/bin/env python3
|
||||
"""
|
||||
Terminal Task Manager - A sidebar-style task manager for the terminal
|
||||
|
||||
Features:
|
||||
- Sidebar display in terminal
|
||||
- Keyboard shortcuts for task operations
|
||||
- Persistent task storage
|
||||
- Task prioritization and highlighting
|
||||
"""
|
||||
|
||||
import curses
|
||||
import json
|
||||
import os
|
||||
from datetime import datetime
|
||||
from typing import List, Dict, Optional
|
||||
|
||||
class Task:
|
||||
def __init__(self, id: int, text: str, completed: bool = False, priority: str = "normal", created_at: str = None):
|
||||
self.id = id
|
||||
self.text = text
|
||||
self.completed = completed
|
||||
self.priority = priority # "high", "normal", "low"
|
||||
self.created_at = created_at or datetime.now().isoformat()
|
||||
|
||||
def to_dict(self) -> Dict:
|
||||
return {
|
||||
"id": self.id,
|
||||
"text": self.text,
|
||||
"completed": self.completed,
|
||||
"priority": self.priority,
|
||||
"created_at": self.created_at
|
||||
}
|
||||
|
||||
@classmethod
|
||||
def from_dict(cls, data: Dict) -> 'Task':
|
||||
return cls(
|
||||
id=data["id"],
|
||||
text=data["text"],
|
||||
completed=data["completed"],
|
||||
priority=data.get("priority", "normal"),
|
||||
created_at=data.get("created_at")
|
||||
)
|
||||
|
||||
class TaskManager:
|
||||
def __init__(self, data_file: str = None):
|
||||
# Use environment variable or default path
|
||||
self.data_file = data_file or os.environ.get('TASKMAN_DATA_FILE', os.path.expanduser("~/.taskman/tasks.json"))
|
||||
self.tasks: List[Task] = []
|
||||
self.selected_index = 0
|
||||
self.next_id = 1
|
||||
self.load_tasks()
|
||||
|
||||
def load_tasks(self):
|
||||
"""Load tasks from JSON file"""
|
||||
if os.path.exists(self.data_file):
|
||||
try:
|
||||
with open(self.data_file, 'r') as f:
|
||||
data = json.load(f)
|
||||
self.tasks = [Task.from_dict(task_data) for task_data in data.get("tasks", [])]
|
||||
self.next_id = data.get("next_id", 1)
|
||||
except (json.JSONDecodeError, FileNotFoundError):
|
||||
self.tasks = []
|
||||
self.next_id = 1
|
||||
|
||||
def save_tasks(self):
|
||||
"""Save tasks to JSON file"""
|
||||
# Ensure directory exists
|
||||
os.makedirs(os.path.dirname(self.data_file), exist_ok=True)
|
||||
|
||||
data = {
|
||||
"tasks": [task.to_dict() for task in self.tasks],
|
||||
"next_id": self.next_id
|
||||
}
|
||||
with open(self.data_file, 'w') as f:
|
||||
json.dump(data, f, indent=2)
|
||||
|
||||
def add_task(self, text: str, priority: str = "normal") -> Task:
|
||||
"""Add a new task"""
|
||||
task = Task(self.next_id, text, priority=priority)
|
||||
self.tasks.append(task)
|
||||
self.next_id += 1
|
||||
self.save_tasks()
|
||||
return task
|
||||
|
||||
def toggle_task(self, task_id: int):
|
||||
"""Toggle task completion status"""
|
||||
for task in self.tasks:
|
||||
if task.id == task_id:
|
||||
task.completed = not task.completed
|
||||
self.save_tasks()
|
||||
break
|
||||
|
||||
def delete_task(self, task_id: int):
|
||||
"""Delete a task"""
|
||||
self.tasks = [task for task in self.tasks if task.id != task_id]
|
||||
self.save_tasks()
|
||||
if self.selected_index >= len(self.tasks) and self.tasks:
|
||||
self.selected_index = len(self.tasks) - 1
|
||||
elif not self.tasks:
|
||||
self.selected_index = 0
|
||||
|
||||
def get_selected_task(self) -> Optional[Task]:
|
||||
"""Get currently selected task"""
|
||||
if 0 <= self.selected_index < len(self.tasks):
|
||||
return self.tasks[self.selected_index]
|
||||
return None
|
||||
|
||||
def move_selection(self, direction: int):
|
||||
"""Move selection up or down"""
|
||||
if self.tasks:
|
||||
self.selected_index = max(0, min(len(self.tasks) - 1, self.selected_index + direction))
|
||||
|
||||
class TaskManagerUI:
|
||||
def __init__(self):
|
||||
self.task_manager = TaskManager()
|
||||
self.input_mode = False
|
||||
self.input_text = ""
|
||||
self.input_priority = "normal"
|
||||
self.show_help = False
|
||||
|
||||
def run(self, stdscr):
|
||||
"""Main application loop"""
|
||||
curses.curs_set(0) # Hide cursor
|
||||
stdscr.nodelay(1) # Non-blocking input
|
||||
stdscr.timeout(100) # Refresh every 100ms
|
||||
|
||||
# Color pairs
|
||||
curses.start_color()
|
||||
curses.init_pair(1, curses.COLOR_WHITE, curses.COLOR_BLUE) # Selected
|
||||
curses.init_pair(2, curses.COLOR_GREEN, curses.COLOR_BLACK) # Completed
|
||||
curses.init_pair(3, curses.COLOR_RED, curses.COLOR_BLACK) # High priority
|
||||
curses.init_pair(4, curses.COLOR_YELLOW, curses.COLOR_BLACK) # Normal priority
|
||||
curses.init_pair(5, curses.COLOR_CYAN, curses.COLOR_BLACK) # Low priority
|
||||
curses.init_pair(6, curses.COLOR_WHITE, curses.COLOR_BLACK) # Default
|
||||
curses.init_pair(7, curses.COLOR_BLACK, curses.COLOR_WHITE) # Input mode
|
||||
|
||||
while True:
|
||||
self.draw_ui(stdscr)
|
||||
|
||||
try:
|
||||
key = stdscr.getch()
|
||||
except:
|
||||
continue
|
||||
|
||||
if key == -1:
|
||||
continue
|
||||
|
||||
if self.input_mode:
|
||||
if self.handle_input_mode(key):
|
||||
break
|
||||
else:
|
||||
if self.handle_normal_mode(key):
|
||||
break
|
||||
|
||||
def draw_ui(self, stdscr):
|
||||
"""Draw the user interface"""
|
||||
stdscr.clear()
|
||||
height, width = stdscr.getmaxyx()
|
||||
|
||||
# Title
|
||||
title = "Terminal Task Manager (osh plugin)"
|
||||
stdscr.addstr(0, 2, title, curses.A_BOLD)
|
||||
stdscr.addstr(0, width - 20, f"Tasks: {len(self.task_manager.tasks)}", curses.A_DIM)
|
||||
|
||||
# Help line
|
||||
if not self.show_help:
|
||||
help_text = "Press 'h' for help, 'q' to quit"
|
||||
stdscr.addstr(1, 2, help_text, curses.A_DIM)
|
||||
|
||||
# Tasks list
|
||||
start_row = 3
|
||||
for i, task in enumerate(self.task_manager.tasks):
|
||||
if start_row + i >= height - 3:
|
||||
break
|
||||
|
||||
# Determine color based on task status and priority
|
||||
color = curses.color_pair(6) # Default
|
||||
if task.completed:
|
||||
color = curses.color_pair(2) # Green for completed
|
||||
elif task.priority == "high":
|
||||
color = curses.color_pair(3) # Red for high priority
|
||||
elif task.priority == "low":
|
||||
color = curses.color_pair(5) # Cyan for low priority
|
||||
else:
|
||||
color = curses.color_pair(4) # Yellow for normal priority
|
||||
|
||||
# Highlight selected task
|
||||
if i == self.task_manager.selected_index:
|
||||
color |= curses.A_REVERSE
|
||||
|
||||
# Task status icon
|
||||
status_icon = "✓" if task.completed else "○"
|
||||
priority_icon = {
|
||||
"high": "!",
|
||||
"normal": "-",
|
||||
"low": "·"
|
||||
}.get(task.priority, "-")
|
||||
|
||||
# Truncate task text if too long
|
||||
max_text_width = width - 15
|
||||
task_text = task.text[:max_text_width] + "..." if len(task.text) > max_text_width else task.text
|
||||
|
||||
task_line = f" {status_icon} [{priority_icon}] {task_text}"
|
||||
stdscr.addstr(start_row + i, 2, task_line, color)
|
||||
|
||||
# Input area
|
||||
if self.input_mode:
|
||||
input_y = height - 3
|
||||
stdscr.addstr(input_y, 2, "New task: ", curses.A_BOLD)
|
||||
stdscr.addstr(input_y, 12, self.input_text, curses.color_pair(7))
|
||||
stdscr.addstr(input_y + 1, 2, f"Priority: {self.input_priority} (Tab to change)", curses.A_DIM)
|
||||
curses.curs_set(1) # Show cursor in input mode
|
||||
else:
|
||||
curses.curs_set(0) # Hide cursor
|
||||
|
||||
# Help panel
|
||||
if self.show_help:
|
||||
self.draw_help(stdscr, height, width)
|
||||
|
||||
# Status line
|
||||
status_y = height - 1
|
||||
if self.input_mode:
|
||||
status_text = "Enter: Save task | Esc: Cancel | Tab: Change priority"
|
||||
else:
|
||||
status_text = "n: New | Space: Toggle | d: Delete | ↑↓: Navigate | h: Help | q: Quit"
|
||||
|
||||
stdscr.addstr(status_y, 2, status_text[:width-4], curses.A_DIM)
|
||||
|
||||
stdscr.refresh()
|
||||
|
||||
def draw_help(self, stdscr, height, width):
|
||||
"""Draw help panel"""
|
||||
help_lines = [
|
||||
"KEYBOARD SHORTCUTS:",
|
||||
"",
|
||||
"Navigation:",
|
||||
" ↑/k - Move up",
|
||||
" ↓/j - Move down",
|
||||
"",
|
||||
"Task Operations:",
|
||||
" n - New task",
|
||||
" Space - Toggle completion",
|
||||
" d - Delete task",
|
||||
"",
|
||||
"Priority Levels:",
|
||||
" ! High priority (red)",
|
||||
" - Normal priority (yellow)",
|
||||
" · Low priority (cyan)",
|
||||
"",
|
||||
"Other:",
|
||||
" h - Toggle this help",
|
||||
" q - Quit application",
|
||||
"",
|
||||
"CLI Usage:",
|
||||
" tasks add 'text' [priority]",
|
||||
" tasks list [filter]",
|
||||
" tasks done <id>",
|
||||
" tasks delete <id>",
|
||||
]
|
||||
|
||||
# Calculate help panel size
|
||||
help_width = max(len(line) for line in help_lines) + 4
|
||||
help_height = len(help_lines) + 2
|
||||
|
||||
start_x = max(2, (width - help_width) // 2)
|
||||
start_y = max(2, (height - help_height) // 2)
|
||||
|
||||
# Draw help background
|
||||
for i in range(help_height):
|
||||
stdscr.addstr(start_y + i, start_x, " " * help_width, curses.color_pair(7))
|
||||
|
||||
# Draw help content
|
||||
for i, line in enumerate(help_lines):
|
||||
stdscr.addstr(start_y + 1 + i, start_x + 2, line, curses.color_pair(7))
|
||||
|
||||
def handle_normal_mode(self, key) -> bool:
|
||||
"""Handle key presses in normal mode"""
|
||||
# Quit
|
||||
if key == ord('q'):
|
||||
return True
|
||||
|
||||
# Navigation
|
||||
elif key == curses.KEY_UP or key == ord('k'):
|
||||
self.task_manager.move_selection(-1)
|
||||
elif key == curses.KEY_DOWN or key == ord('j'):
|
||||
self.task_manager.move_selection(1)
|
||||
|
||||
# Task operations
|
||||
elif key == ord('n'):
|
||||
self.input_mode = True
|
||||
self.input_text = ""
|
||||
self.input_priority = "normal"
|
||||
elif key == ord(' '):
|
||||
selected_task = self.task_manager.get_selected_task()
|
||||
if selected_task:
|
||||
self.task_manager.toggle_task(selected_task.id)
|
||||
elif key == ord('d'):
|
||||
selected_task = self.task_manager.get_selected_task()
|
||||
if selected_task:
|
||||
self.task_manager.delete_task(selected_task.id)
|
||||
|
||||
# Help
|
||||
elif key == ord('h'):
|
||||
self.show_help = not self.show_help
|
||||
|
||||
return False
|
||||
|
||||
def handle_input_mode(self, key) -> bool:
|
||||
"""Handle key presses in input mode"""
|
||||
if key == 27: # Escape
|
||||
self.input_mode = False
|
||||
self.input_text = ""
|
||||
elif key == ord('\n') or key == 10: # Enter
|
||||
if self.input_text.strip():
|
||||
self.task_manager.add_task(self.input_text.strip(), self.input_priority)
|
||||
self.input_mode = False
|
||||
self.input_text = ""
|
||||
elif key == ord('\t'): # Tab - cycle priority
|
||||
priorities = ["normal", "high", "low"]
|
||||
current_index = priorities.index(self.input_priority)
|
||||
self.input_priority = priorities[(current_index + 1) % len(priorities)]
|
||||
elif key == curses.KEY_BACKSPACE or key == 127:
|
||||
self.input_text = self.input_text[:-1]
|
||||
elif 32 <= key <= 126: # Printable characters
|
||||
self.input_text += chr(key)
|
||||
|
||||
return False
|
||||
|
||||
def main():
|
||||
"""Main entry point"""
|
||||
ui = TaskManagerUI()
|
||||
try:
|
||||
curses.wrapper(ui.run)
|
||||
except KeyboardInterrupt:
|
||||
print("\nGoodbye!")
|
||||
except Exception as e:
|
||||
print(f"Error: {e}")
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue