tftio-asana-cli 3.1.0

//! High level task operations built on the core API client.

use crate::{
    api::{ApiClient, ApiError},
    api::{SingleResponse, api_delete, api_post, api_put},
    models::{
        Task, TaskCreateRequest, TaskListParams, TaskReference, TaskSearchParams, TaskSort,
        TaskUpdateRequest,
    },
};
use futures_util::{StreamExt, pin_mut};
use serde::{Deserialize, Serialize};
use std::collections::BTreeSet;
use tracing::debug;

/// Retrieve tasks according to the supplied parameters.
///
/// # Errors
///
/// Returns an error if the API request fails, if deserialization fails, or if the response is invalid.
pub async fn list_tasks(
    client: &ApiClient,
    mut params: TaskListParams,
) -> Result<Vec<Task>, ApiError> {
    ensure_default_fields(&mut params);

    let query = params.to_query();
    let max_items = params.limit;
    let stream = client.paginate_with_limit::<Task>("/tasks", query, max_items);
    pin_mut!(stream);

    let mut tasks = Vec::new();
    while let Some(page) = stream.next().await {
        let mut page = page?;
        tasks.append(&mut page);
    }

    // Fetch subtasks if requested. This makes separate API calls for each task
    // to get subtasks with complete field data. The deprecated opt_expand=subtasks
    // no longer returns full field information, and num_subtasks is not reliably
    // populated by the API, so we attempt to fetch subtasks for all tasks.
    if params.include_subtasks {
        debug!("Fetching subtasks for {} parent tasks", tasks.len());
        let mut all_subtasks = Vec::new();
        for task in &tasks {
            // Continue on error - tasks without subtasks may return empty results or errors
            match list_subtasks(client, &task.gid, params.fields.iter().cloned().collect()).await {
                Ok(subtasks) => {
                    if !subtasks.is_empty() {
                        debug!("Found {} subtasks for task {}", subtasks.len(), task.gid);
                    }
                    all_subtasks.extend(subtasks);
                }
                Err(e) => {
                    debug!("Failed to fetch subtasks for task {}: {}", task.gid, e);
                    // Task has no subtasks or fetch failed, skip it
                }
            }
        }
        debug!("Total subtasks fetched: {}", all_subtasks.len());
        tasks.extend(all_subtasks);
    }

    params.apply_post_filters(&mut tasks);

    if let Some(sort) = params.sort {
        sort_tasks(&mut tasks, sort);
    }

    Ok(tasks)
}

/// Search tasks in a workspace.
///
/// # Errors
///
/// Returns an error if the API request fails, if deserialization fails, or if the response is invalid.
pub async fn search_tasks(
    client: &ApiClient,
    mut params: TaskSearchParams,
) -> Result<Vec<Task>, ApiError> {
    ensure_default_fields_for_search(&mut params);

    let query = params.to_query();
    let max_items = params.limit;
    let endpoint = format!("/workspaces/{}/tasks/search", params.workspace);
    let stream = client.paginate_with_limit::<Task>(&endpoint, query, max_items);
    pin_mut!(stream);

    let mut tasks = Vec::new();
    while let Some(page) = stream.next().await {
        let mut page = page?;
        tasks.append(&mut page);
    }

    Ok(tasks)
}

fn ensure_default_fields_for_search(params: &mut TaskSearchParams) {
    // Add same defaults as list_tasks
    let defaults = [
        "gid",
        "name",
        "completed",
        "completed_at",
        "due_on",
        "due_at",
        "start_on",
        "start_at",
        "assignee.name",
        "assignee.gid",
        "assignee.email",
        "resource_type",
        "resource_subtype",
        "modified_at",
        "workspace.name",
        "workspace.gid",
        "projects.name",
        "projects.gid",
        "tags.name",
        "tags.gid",
        "memberships.project.name",
        "memberships.project.gid",
        "memberships.section.name",
        "memberships.section.gid",
        "permalink_url",
        "num_subtasks",
    ];
    for field in defaults {
        params.fields.insert(field.to_string());
    }
}

/// Retrieve a single task by gid.
///
/// # Errors
///
/// Returns an error if the API request fails, if deserialization fails, or if the response is invalid.
pub async fn get_task(
    client: &ApiClient,
    gid: &str,
    fields: Vec<String>,
) -> Result<Task, ApiError> {
    let mut field_set: BTreeSet<String> = fields.into_iter().collect();
    for field in detail_defaults() {
        field_set.insert((*field).to_string());
    }

    let mut query = Vec::new();
    if !field_set.is_empty() {
        query.push((
            "opt_fields".into(),
            field_set.into_iter().collect::<Vec<_>>().join(","),
        ));
    }

    let response: SingleResponse<Task> = client
        .get_json_with_pairs(&format!("/tasks/{gid}"), query)
        .await?;
    Ok(response.data)
}

/// Create a task using the provided payload.
///
/// # Errors
///
/// Returns an error if the API request fails, if deserialization fails, or if the response is invalid.
pub async fn create_task(client: &ApiClient, request: TaskCreateRequest) -> Result<Task, ApiError> {
    api_post(client, "/tasks", &request).await
}

/// Update a task using the given payload.
///
/// # Errors
///
/// Returns an error if the API request fails, if deserialization fails, or if the response is invalid.
pub async fn update_task(
    client: &ApiClient,
    gid: &str,
    request: TaskUpdateRequest,
) -> Result<Task, ApiError> {
    api_put(client, &format!("/tasks/{gid}"), &request).await
}

/// Delete a task permanently.
///
/// # Errors
///
/// Returns an error if the API request fails or if the response is invalid.
pub async fn delete_task(client: &ApiClient, gid: &str) -> Result<(), ApiError> {
    api_delete(client, &format!("/tasks/{gid}")).await
}

/// List subtasks for a parent task.
///
/// # Errors
///
/// Returns an error if the API request fails or deserialization fails.
pub async fn list_subtasks(
    client: &ApiClient,
    gid: &str,
    fields: Vec<String>,
) -> Result<Vec<Task>, ApiError> {
    let mut field_set: BTreeSet<String> = fields.into_iter().collect();
    ensure_subtask_fields(&mut field_set);

    let mut query = Vec::new();
    if !field_set.is_empty() {
        let list = field_set.into_iter().collect::<Vec<_>>().join(",");
        query.push(("opt_fields".into(), list));
    }

    let stream = client.paginate_with_limit::<Task>(&format!("/tasks/{gid}/subtasks"), query, None);
    pin_mut!(stream);

    let mut tasks = Vec::new();
    while let Some(page) = stream.next().await {
        let mut page = page?;
        tasks.append(&mut page);
    }
    Ok(tasks)
}

/// Retrieve the tasks that this task depends on.
///
/// # Errors
///
/// Returns an error if the API request fails or deserialization fails.
pub async fn list_dependencies(
    client: &ApiClient,
    gid: &str,
) -> Result<Vec<TaskReference>, ApiError> {
    let response: TaskReferenceResponse = client
        .get_json_with_pairs(&format!("/tasks/{gid}/dependencies"), vec![])
        .await?;
    Ok(response.data)
}

/// Retrieve the tasks blocked by this task.
///
/// # Errors
///
/// Returns an error if the API request fails or deserialization fails.
pub async fn list_dependents(
    client: &ApiClient,
    gid: &str,
) -> Result<Vec<TaskReference>, ApiError> {
    let response: TaskReferenceResponse = client
        .get_json_with_pairs(&format!("/tasks/{gid}/dependents"), vec![])
        .await?;
    Ok(response.data)
}

/// Add dependencies to a task.
///
/// # Errors
///
/// Returns an error if the API request fails.
pub async fn add_dependencies(
    client: &ApiClient,
    gid: &str,
    dependencies: Vec<String>,
) -> Result<(), ApiError> {
    if dependencies.is_empty() {
        return Ok(());
    }

    let payload = DependencyModifyRequest {
        data: DependencyList { dependencies },
    };
    client
        .post_void(&format!("/tasks/{gid}/addDependencies"), &payload)
        .await
}

/// Remove dependencies from a task.
///
/// # Errors
///
/// Returns an error if the API request fails.
pub async fn remove_dependencies(
    client: &ApiClient,
    gid: &str,
    dependencies: Vec<String>,
) -> Result<(), ApiError> {
    if dependencies.is_empty() {
        return Ok(());
    }

    let payload = DependencyModifyRequest {
        data: DependencyList { dependencies },
    };
    client
        .post_void(&format!("/tasks/{gid}/removeDependencies"), &payload)
        .await
}

/// Add dependents to a task (tasks blocked by this task).
///
/// # Errors
///
/// Returns an error if the API request fails.
pub async fn add_dependents(
    client: &ApiClient,
    gid: &str,
    dependents: Vec<String>,
) -> Result<(), ApiError> {
    if dependents.is_empty() {
        return Ok(());
    }

    let payload = DependentModifyRequest {
        data: DependentList { dependents },
    };
    client
        .post_void(&format!("/tasks/{gid}/addDependents"), &payload)
        .await
}

/// Remove dependents from a task.
///
/// # Errors
///
/// Returns an error if the API request fails.
pub async fn remove_dependents(
    client: &ApiClient,
    gid: &str,
    dependents: Vec<String>,
) -> Result<(), ApiError> {
    if dependents.is_empty() {
        return Ok(());
    }

    let payload = DependentModifyRequest {
        data: DependentList { dependents },
    };
    client
        .post_void(&format!("/tasks/{gid}/removeDependents"), &payload)
        .await
}

/// Add the task to a project, optionally targeting a section.
///
/// # Errors
///
/// Returns an error if the API request fails.
pub async fn add_project(
    client: &ApiClient,
    gid: &str,
    project: String,
    section: Option<String>,
) -> Result<(), ApiError> {
    let payload = ProjectModifyRequest {
        data: ProjectModifyData { project, section },
    };
    client
        .post_void(&format!("/tasks/{gid}/addProject"), &payload)
        .await
}

/// Remove the task from a project.
///
/// # Errors
///
/// Returns an error if the API request fails.
pub async fn remove_project(
    client: &ApiClient,
    gid: &str,
    project: String,
) -> Result<(), ApiError> {
    let payload = ProjectModifyRequest {
        data: ProjectModifyData {
            project,
            section: None,
        },
    };
    client
        .post_void(&format!("/tasks/{gid}/removeProject"), &payload)
        .await
}

/// Add followers to a task.
///
/// # Errors
///
/// Returns an error if the API request fails.
pub async fn add_followers(
    client: &ApiClient,
    gid: &str,
    followers: Vec<String>,
) -> Result<(), ApiError> {
    if followers.is_empty() {
        return Ok(());
    }

    let payload = FollowersModifyRequest {
        data: FollowersList { followers },
    };
    client
        .post_void(&format!("/tasks/{gid}/addFollowers"), &payload)
        .await
}

/// Remove followers from a task.
///
/// # Errors
///
/// Returns an error if the API request fails.
pub async fn remove_followers(
    client: &ApiClient,
    gid: &str,
    followers: Vec<String>,
) -> Result<(), ApiError> {
    if followers.is_empty() {
        return Ok(());
    }

    let payload = FollowersModifyRequest {
        data: FollowersList { followers },
    };
    client
        .post_void(&format!("/tasks/{gid}/removeFollowers"), &payload)
        .await
}

/// Add a tag to a task.
///
/// # Errors
///
/// Returns an error if the API request fails.
pub async fn add_tag(client: &ApiClient, gid: &str, tag: String) -> Result<(), ApiError> {
    let payload = TagModifyRequest {
        data: TagModifyData { tag },
    };
    client
        .post_void(&format!("/tasks/{gid}/addTag"), &payload)
        .await
}

/// Remove a tag from a task.
///
/// # Errors
///
/// Returns an error if the API request fails.
pub async fn remove_tag(client: &ApiClient, gid: &str, tag: String) -> Result<(), ApiError> {
    let payload = TagModifyRequest {
        data: TagModifyData { tag },
    };
    client
        .post_void(&format!("/tasks/{gid}/removeTag"), &payload)
        .await
}

fn ensure_default_fields(params: &mut TaskListParams) {
    let defaults = [
        "gid",
        "name",
        "completed",
        "completed_at",
        "due_on",
        "due_at",
        "start_on",
        "start_at",
        "assignee.name",
        "assignee.gid",
        "assignee.email",
        "resource_type",
        "resource_subtype",
        "modified_at",
        "workspace.name",
        "workspace.gid",
        "projects.name",
        "projects.gid",
        "tags.name",
        "tags.gid",
        "memberships.project.name",
        "memberships.project.gid",
        "memberships.section.name",
        "memberships.section.gid",
        "permalink_url",
        "num_subtasks",
    ];
    for field in defaults {
        params.fields.insert(field.to_string());
    }
}

const fn detail_defaults() -> &'static [&'static str] {
    &[
        "gid",
        "name",
        "completed",
        "completed_at",
        "due_on",
        "due_at",
        "start_on",
        "start_at",
        "notes",
        "html_notes",
        "assignee",
        "assignee_status",
        "assignee.name",
        "assignee.email",
        "completed_by",
        "created_at",
        "modified_at",
        "workspace",
        "workspace.name",
        "workspace.gid",
        "parent",
        "projects",
        "projects.name",
        "projects.gid",
        "memberships",
        "tags",
        "followers",
        "followers.name",
        "followers.email",
        "dependencies",
        "dependents",
        "custom_fields",
        "attachments",
        "permalink_url",
        "resource_subtype",
        "num_subtasks",
    ]
}

fn ensure_subtask_fields(fields: &mut BTreeSet<String>) {
    // Always ensure parent field is included so we can identify subtasks
    fields.insert("parent".to_string());
    fields.insert("parent.gid".to_string());
    fields.insert("parent.name".to_string());

    if fields.len() == 3 {
        // Only parent fields were added, add minimal defaults
        let defaults = ["gid", "name", "completed", "assignee.name"];
        for field in defaults {
            fields.insert(field.to_string());
        }
    }
}

fn sort_tasks(tasks: &mut [Task], sort: TaskSort) {
    match sort {
        TaskSort::Name => tasks.sort_by(|a, b| {
            a.name
                .to_ascii_lowercase()
                .cmp(&b.name.to_ascii_lowercase())
        }),
        TaskSort::DueOn => tasks.sort_by(|a, b| {
            a.due_on
                .cmp(&b.due_on)
                .then_with(|| a.due_at.cmp(&b.due_at))
        }),
        TaskSort::CreatedAt => tasks.sort_by(|a, b| a.created_at.cmp(&b.created_at)),
        TaskSort::ModifiedAt => tasks.sort_by(|a, b| a.modified_at.cmp(&b.modified_at)),
        TaskSort::Assignee => tasks.sort_by_key(assignee_label),
    }
}

fn assignee_label(task: &Task) -> Option<String> {
    task.assignee
        .as_ref()
        .map(crate::models::user::UserReference::label)
}

#[derive(Debug, Deserialize)]
struct TaskReferenceResponse {
    data: Vec<TaskReference>,
}

#[derive(Debug, Serialize)]
struct DependencyModifyRequest {
    data: DependencyList,
}

#[derive(Debug, Serialize)]
struct DependencyList {
    #[serde(skip_serializing_if = "Vec::is_empty")]
    dependencies: Vec<String>,
}

#[derive(Debug, Serialize)]
struct DependentModifyRequest {
    data: DependentList,
}

#[derive(Debug, Serialize)]
struct DependentList {
    #[serde(skip_serializing_if = "Vec::is_empty")]
    dependents: Vec<String>,
}

#[derive(Debug, Serialize)]
struct ProjectModifyRequest {
    data: ProjectModifyData,
}

#[derive(Debug, Serialize)]
struct ProjectModifyData {
    project: String,
    #[serde(skip_serializing_if = "Option::is_none")]
    section: Option<String>,
}

#[derive(Debug, Serialize)]
struct FollowersModifyRequest {
    data: FollowersList,
}

#[derive(Debug, Serialize)]
struct FollowersList {
    #[serde(skip_serializing_if = "Vec::is_empty")]
    followers: Vec<String>,
}

#[derive(Debug, Serialize)]
struct TagModifyRequest {
    data: TagModifyData,
}

#[derive(Debug, Serialize)]
struct TagModifyData {
    tag: String,
}