fix: pass category argument through to add_task properly

The add_task() function hardcoded category='General' regardless of user
input, and _handle_todo_command() concatenated both arguments into a
single string, discarding the category.

Fix:
- Update add_task() to parse quoted arguments via shlex.split()
- Update _handle_todo_command() to pass arguments with quotes preserved

Closes TruFoundation#46

Author: Ziang Zhang

trushell/cli.py

@@ -26,11 +26,11 @@
 
 def app_with_lower() -> None:
     """Entry point that normalizes the first argument to lowercase for case-insensitive invocation."""
-    if len(sys.argv) > 1:
-        # Create a local copy to avoid mutating the global sys.argv
-        argv_copy = sys.argv.copy()
-        if argv_copy[1].lower() not in {"--help", "-h", "version"}:
-            raw = " ".join(argv_copy[1:])
+    # Create a local copy to avoid mutating the global sys.argv
+    argv = sys.argv.copy()
+    if len(argv) > 1:
+        if argv[1].lower() not in {"--help", "-h", "version"}:
+            raw = " ".join(argv[1:]).lower()
             get_kernel().execute_command(raw)
             return
 
@@ -193,7 +193,7 @@ def _handle_todo_command(command: str) -> bool:
     if add_match:
         from trushell.commands.tasks import add_task
 
-        add_task(f"{add_match.group(1)} {add_match.group(2)}")
+        add_task(f'"{add_match.group(1)}" "{add_match.group(2)}"')
         return True
 
     update_match = re.match(r'updatetask\s+(\d+)\s+"([^"]+)"\s+"([^"]+)"', command)

trushell/commands/tasks.py

@@ -7,13 +7,31 @@
 
 
 def add_task(args: str) -> None:
-    """Add a new task to the todo list. The full remainder is treated as the task."""
+    """Add a new task to the todo list.
+
+    Supports two formats:
+        addtask "task text" "category"
+        addtask task text here
+    """
     if not args.strip():
-        print("Usage: task add <task>")
+        print("Usage: task add <task> [category]")
         return
 
-    task_text = args.strip()
-    todo = Todo(task=task_text, category="General")
+    # Try to parse quoted arguments: "task text" "category"
+    import shlex
+    try:
+        parts = shlex.split(args)
+    except ValueError:
+        parts = args.strip().split(maxsplit=1)
+
+    if len(parts) >= 2:
+        task_text = parts[0]
+        category = parts[1]
+    else:
+        task_text = parts[0]
+        category = "General"
+
+    todo = Todo(task=task_text, category=category)
     insert_todo(todo)
     print("Task added.")