onyx-tasks/crates/bevy-tasks-cli/src/commands/task.rs

use anyhow::{Context, Result};
use bevy_tasks_core::Task;
use chrono::{DateTime, Utc};
use uuid::Uuid;
use crate::output;
use crate::commands::get_repository;

pub fn add(title: String, list_name: Option<String>, due_str: Option<String>, workspace: Option<String>) -> Result<()> {
    let (mut repo, _workspace_name) = get_repository(workspace)?;

    // Get lists
    let lists = repo.get_lists()
        .context("Failed to get lists")?;

    if lists.is_empty() {
        anyhow::bail!("No lists found. Create one with 'bevy-tasks list create <name>'");
    }

    // Find the target list
    let list = if let Some(name) = list_name {
        lists.iter()
            .find(|l| l.title == name)
            .ok_or_else(|| anyhow::anyhow!("List '{}' not found", name))?
    } else {
        // Use the first list
        &lists[0]
    };

    // Create task
    let mut task = Task::new(title.clone());

    // Parse due date if provided
    if let Some(due_str) = due_str {
        let due_date = parse_due_date(&due_str)?;
        task.due_date = Some(due_date);
    }

    // Save task
    repo.create_task(list.id, task.clone())
        .context("Failed to create task")?;

    let due_info = if let Some(due) = task.due_date {
        format!("\n  Due: {}", due.format("%Y-%m-%d"))
    } else {
        String::new()
    };

    output::success(&format!("Created task \"{}\" ({}){}", title, task.id, due_info));

    Ok(())
}

pub fn complete(task_id_str: String, workspace: Option<String>) -> Result<()> {
    let (mut repo, _workspace_name) = get_repository(workspace)?;

    let task_id = Uuid::parse_str(&task_id_str)
        .context("Invalid task ID")?;

    // Find the task across all lists
    let lists = repo.get_lists()?;
    let mut found = false;

    for list in lists {
        if let Some(mut task) = list.tasks.iter().find(|t| t.id == task_id).cloned() {
            task.complete();
            repo.update_task(list.id, task.clone())
                .context("Failed to update task")?;

            output::success(&format!("Completed task \"{}\"", task.title));
            found = true;
            break;
        }
    }

    if !found {
        anyhow::bail!("Task not found: {}", task_id_str);
    }

    Ok(())
}

pub fn delete(task_id_str: String, workspace: Option<String>) -> Result<()> {
    let (mut repo, _workspace_name) = get_repository(workspace)?;

    let task_id = Uuid::parse_str(&task_id_str)
        .context("Invalid task ID")?;

    // Find the task across all lists
    let lists = repo.get_lists()?;
    let mut found = false;

    for list in lists {
        if let Some(task) = list.tasks.iter().find(|t| t.id == task_id) {
            let title = task.title.clone();

            repo.delete_task(list.id, task_id)
                .context("Failed to delete task")?;

            output::success(&format!("Deleted task \"{}\"", title));
            found = true;
            break;
        }
    }

    if !found {
        anyhow::bail!("Task not found: {}", task_id_str);
    }

    Ok(())
}

pub fn edit(task_id_str: String, workspace: Option<String>) -> Result<()> {
    let (mut repo, _workspace_name) = get_repository(workspace)?;

    let task_id = Uuid::parse_str(&task_id_str)
        .context("Invalid task ID")?;

    // Find the task across all lists
    let lists = repo.get_lists()?;
    let mut task_list_id = None;
    let mut task_to_edit = None;

    for list in lists {
        if let Some(task) = list.tasks.iter().find(|t| t.id == task_id).cloned() {
            task_list_id = Some(list.id);
            task_to_edit = Some(task);
            break;
        }
    }

    let (list_id, task) = match (task_list_id, task_to_edit) {
        (Some(lid), Some(t)) => (lid, t),
        _ => anyhow::bail!("Task not found: {}", task_id_str),
    };

    // Create temporary file with task content
    let temp_dir = std::env::temp_dir();
    let temp_file = temp_dir.join(format!("bevy-tasks-{}.md", task.id));

    // Write current task content to temp file
    let content = format!("# {}\n\n{}", task.title, task.description);
    std::fs::write(&temp_file, content)?;

    // Get editor from environment
    let editor = std::env::var("EDITOR").unwrap_or_else(|_| {
        if cfg!(windows) {
            "notepad".to_string()
        } else {
            "nano".to_string()
        }
    });

    // Open editor
    let status = std::process::Command::new(&editor)
        .arg(&temp_file)
        .status()
        .context(format!("Failed to open editor: {}", editor))?;

    if !status.success() {
        anyhow::bail!("Editor exited with non-zero status");
    }

    // Read updated content
    let updated_content = std::fs::read_to_string(&temp_file)?;

    // Parse the content
    let lines: Vec<&str> = updated_content.lines().collect();
    let (title, description) = if !lines.is_empty() && lines[0].starts_with("# ") {
        let title = lines[0].trim_start_matches("# ").trim().to_string();
        let description = if lines.len() > 2 {
            lines[2..].join("\n").trim().to_string()
        } else {
            String::new()
        };
        (title, description)
    } else {
        (task.title.clone(), updated_content.trim().to_string())
    };

    // Update task
    let mut updated_task = task.clone();
    updated_task.title = title;
    updated_task.description = description;
    updated_task.updated_at = Utc::now();

    repo.update_task(list_id, updated_task.clone())
        .context("Failed to update task")?;

    // Clean up temp file
    std::fs::remove_file(&temp_file).ok();

    output::success(&format!("Updated task \"{}\"", updated_task.title));

    Ok(())
}

fn parse_due_date(s: &str) -> Result<DateTime<Utc>> {
    // Try parsing as date only (YYYY-MM-DD)
    if let Ok(naive_date) = chrono::NaiveDate::parse_from_str(s, "%Y-%m-%d") {
        let naive_datetime = naive_date.and_hms_opt(0, 0, 0)
            .ok_or_else(|| anyhow::anyhow!("Invalid date"))?;
        return Ok(DateTime::from_naive_utc_and_offset(naive_datetime, Utc));
    }

    // Try parsing as full datetime (ISO 8601)
    if let Ok(dt) = DateTime::parse_from_rfc3339(s) {
        return Ok(dt.with_timezone(&Utc));
    }

    anyhow::bail!("Invalid date format. Use YYYY-MM-DD or ISO 8601 format (YYYY-MM-DDTHH:MM:SS)")
}