task_cli.py

#!/usr/bin/env python3

import json
import os
import sys
from datetime import datetime, timezone


TASK_DIR = "task"
TASK_FILE = os.path.join(TASK_DIR, "task.json")
VALID_STATUSES = {"todo", "in-progress", "done"}


def now_iso():
    return datetime.now(timezone.utc).isoformat()


def ensure_task_file():
    os.makedirs(TASK_DIR, exist_ok=True)
    if not os.path.exists(TASK_FILE):
        with open(TASK_FILE, "w", encoding="utf-8") as f:
            json.dump([], f, indent=2)


def load_tasks():
    ensure_task_file()

    try:
        with open(TASK_FILE, "r", encoding="utf-8") as f:
            data = json.load(f)
            if isinstance(data, list):
                return data
    except json.JSONDecodeError:
        pass

    return []


def save_tasks(tasks):
    with open(TASK_FILE, "w", encoding="utf-8") as f:
        json.dump(tasks, f, indent=2)


def next_id(tasks):
    return max((task.get("id", 0) for task in tasks), default=0) + 1


def find_task(tasks, task_id):
    for task in tasks:
        if task.get("id") == task_id:
            return task
    return None


def add_task(description):
    tasks = load_tasks()
    task = {
        "id": next_id(tasks),
        "description": description,
        "status": "todo",
        "createdAt": now_iso(),
        "updatedAt": now_iso(),
    }
    tasks.append(task)
    save_tasks(tasks)
    print(f"Task added successfully (ID: {task['id']})")


def update_task(task_id, description):
    tasks = load_tasks()
    task = find_task(tasks, task_id)

    if not task:
        print(f"Task with ID {task_id} not found.")
        return

    task["description"] = description
    task["updatedAt"] = now_iso()
    save_tasks(tasks)
    print(f"Task updated successfully (ID: {task_id})")


def delete_task(task_id):
    tasks = load_tasks()
    updated_tasks = [task for task in tasks if task.get("id") != task_id]

    if len(updated_tasks) == len(tasks):
        print(f"Task with ID {task_id} not found.")
        return

    save_tasks(updated_tasks)
    print(f"Task deleted successfully (ID: {task_id})")


def mark_task(task_id, status):
    tasks = load_tasks()
    task = find_task(tasks, task_id)

    if not task:
        print(f"Task with ID {task_id} not found.")
        return

    task["status"] = status
    task["updatedAt"] = now_iso()
    save_tasks(tasks)

    if status == "done":
        print(f"Task marked as done (ID: {task_id})")
    else:
        print(f"Task marked as in progress (ID: {task_id})")


def list_tasks(status_filter=None):
    tasks = load_tasks()

    if status_filter:
        tasks = [task for task in tasks if task.get("status") == status_filter]

    if not tasks:
        print("No tasks found.")
        return

    for task in tasks:
        print(
            f"[{task.get('id')}] {task.get('description')} "
            f"({task.get('status')})"
        )


def parse_task_id(value):
    try:
        return int(value)
    except ValueError:
        print("Task ID must be a number.")
        return None


def main():
    if len(sys.argv) < 2:
        print(
            "Usage: task-cli add|update|delete|mark-in-progress|mark-done|list ..."
        )
        return

    command = sys.argv[1]

    if command == "add":
        if len(sys.argv) < 3:
            print("Usage: task-cli add \"description\"")
            return
        add_task(" ".join(sys.argv[2:]))
        return

    if command == "update":
        if len(sys.argv) < 4:
            print('Usage: task-cli update <id> "new description"')
            return
        task_id = parse_task_id(sys.argv[2])
        if task_id is None:
            return
        update_task(task_id, " ".join(sys.argv[3:]))
        return

    if command == "delete":
        if len(sys.argv) < 3:
            print("Usage: task-cli delete <id>")
            return
        task_id = parse_task_id(sys.argv[2])
        if task_id is None:
            return
        delete_task(task_id)
        return

    if command == "mark-in-progress":
        if len(sys.argv) < 3:
            print("Usage: task-cli mark-in-progress <id>")
            return
        task_id = parse_task_id(sys.argv[2])
        if task_id is None:
            return
        mark_task(task_id, "in-progress")
        return

    if command == "mark-done":
        if len(sys.argv) < 3:
            print("Usage: task-cli mark-done <id>")
            return
        task_id = parse_task_id(sys.argv[2])
        if task_id is None:
            return
        mark_task(task_id, "done")
        return

    if command == "list":
        if len(sys.argv) == 2:
            list_tasks()
            return

        status = sys.argv[2]
        if status not in VALID_STATUSES:
            print("Usage: task-cli list [todo|in-progress|done]")
            return

        list_tasks(status)
        return

    print(f"Unknown command: {command}")


if __name__ == "__main__":
    main()