aven-core 0.1.28

use anyhow::{Result, bail};
use chrono::{DateTime, Utc};

use crate::choices::TaskStatus;
use crate::db::Database;
use crate::ids::{TaskId, WorkspaceId};
use crate::recurrence::RecurrenceSeriesId;
use crate::refs::DisplayRefContext;
use crate::types::Task;

mod database_stats;
mod dependencies;
mod details;
mod doctor;
pub mod fragments;
mod hydration;
mod projects;
mod recent_actions;
mod recurrence;
mod related;
mod search;
mod sidebar;
mod sorting;
mod sync_history;
mod tasks;
#[cfg(test)]
mod test_support;
mod types;

pub use database_stats::{DatabaseStats, DatabaseStatsPriorityCounts, DatabaseStatsStatusCounts};
pub use dependencies::{TaskDependencyItem, TaskDependencySummary};
pub(crate) use dependencies::{task_dependency_summary, task_dependency_summary_with_display_refs};
pub use details::{TaskDetail, TaskDetailConflict};
pub(crate) use details::{conflict_display_value, task_detail, task_detail_with_display_refs};
pub use doctor::WorkspaceTaskCounts;
pub(crate) use doctor::{unresolved_conflict_count, workspace_task_counts};
pub(crate) use projects::list_project_items_in_workspace;
pub(crate) use recent_actions::{
    list_recent_actions_in_workspace, task_activity_for_tasks_in_workspace,
};
pub(crate) use recurrence::task_recurrence_summaries;
pub use related::TaskRelatedLink;
pub(crate) use related::{related_links_for_tasks, task_related_links};
pub use search::{
    SearchMatchedField, TaskSearchPreviewResultSet, TaskSearchQuery, TaskSearchResult,
};
pub(crate) use search::{
    search_task_items_in_workspace, search_task_occurrence_items_in_workspace,
    search_task_preview_set_in_workspace,
};
pub(crate) use sidebar::sidebar_counts_for_scope_in_workspace;
pub use sync_history::SyncHistoryStats;
pub(crate) use sync_history::sync_history_stats;
pub(crate) use tasks::{
    list_bulk_update_task_items_in_workspace, list_task_items_in_workspace,
    list_task_items_with_display_refs, list_task_items_without_activity_in_workspace,
    list_task_summary_items_in_workspace,
};
pub use types::RecentActionTarget;
pub use types::{
    AttachmentMetadata, EpicRollup, MetadataFilter, ProjectListItem, RecentActionItem,
    RecurrenceCounts, RecurrenceHistoryEntry, RecurrenceHistoryKind, RecurrenceHistoryPage,
    RecurrenceOccurrenceLink, RecurrenceReconciliation, RecurrenceSeriesConflict,
    RecurrenceSeriesDetail, RecurrenceSeriesLifecycleFilter, RecurrenceSeriesListItem,
    RecurrenceSeriesListQuery, RecurrenceSeriesSummary, RecurrenceTaskGroup, SidebarCounts,
    SortDirection, TaskAvailabilityFilter, TaskDependencyLink, TaskFilters, TaskIdFilter,
    TaskItemHydration, TaskListItem, TaskNote, TaskQueryMode, TaskRecurrenceSummary, TaskSort,
};

pub const MAX_RECURRENCE_HISTORY_LIMIT: usize = 500;

pub(crate) fn status_order(status: TaskStatus) -> u8 {
    match status {
        TaskStatus::Active => 0,
        TaskStatus::Todo => 1,
        TaskStatus::Inbox => 2,
        TaskStatus::Backlog => 3,
        TaskStatus::Done => 4,
        TaskStatus::Canceled => 5,
    }
}

fn validate_search_limit(limit: usize) -> Result<()> {
    if limit == 0 {
        bail!("error search-limit-invalid limit=0 min=1 hint=\"pass a limit of 1 or greater\"");
    }
    Ok(())
}

fn validate_recurrence_history_limit(limit: usize) -> Result<()> {
    if !(1..=MAX_RECURRENCE_HISTORY_LIMIT).contains(&limit) {
        bail!(
            "error recurrence-history-limit-invalid limit={} min=1 max={} hint=\"pass a limit between 1 and {}\"",
            limit,
            MAX_RECURRENCE_HISTORY_LIMIT,
            MAX_RECURRENCE_HISTORY_LIMIT
        );
    }
    Ok(())
}

impl Database {
    /// Reconciles recurrence projections before a report reads current state.
    ///
    /// The candidate count is bounded. Each changed series writes one atomic projection
    /// transaction that can archive one superseded task and materialize one successor.
    pub async fn reconcile_recurrence_reports_at(
        &self,
        workspace_id: &WorkspaceId,
        at: DateTime<Utc>,
    ) -> Result<RecurrenceReconciliation> {
        let (workspace, candidates, incomplete) = {
            let mut conn = self.acquire_reader().await?;
            let workspace = crate::workspaces::workspace_for_id(&mut conn, workspace_id).await?;
            let (candidates, incomplete) =
                recurrence::recurrence_reconciliation_candidates(&mut conn, workspace_id, at)
                    .await?;
            (workspace, candidates, incomplete)
        };
        let mut report = RecurrenceReconciliation {
            workspace_id: Some(workspace_id.clone()),
            examined: candidates.len(),
            incomplete,
            ..RecurrenceReconciliation::default()
        };
        for series_id in candidates {
            let result = self
                .reconcile_recurrence_series(&workspace, &series_id, at)
                .await?;
            report.changed += usize::from(result.changed);
            report.lifecycle_blocked += usize::from(result.lifecycle_blocked);
        }
        recurrence::validate_reconciliation(&report)?;
        Ok(report)
    }

    pub async fn reconcile_recurrence_reports(
        &self,
        workspace_id: &WorkspaceId,
    ) -> Result<RecurrenceReconciliation> {
        let at = crate::ids::now_utc();
        self.reconcile_recurrence_reports_at(workspace_id, at).await
    }

    pub async fn list_recurrence_series_view_at(
        &self,
        workspace_id: &WorkspaceId,
        at: DateTime<Utc>,
        query: RecurrenceSeriesListQuery,
    ) -> Result<Vec<RecurrenceSeriesListItem>> {
        self.reconcile_recurrence_reports_at(workspace_id, at)
            .await?;
        self.list_recurrence_series_view_from_current_projection(workspace_id, query)
            .await
    }

    /// Reads the persisted recurrence projection as-is.
    ///
    /// Callers establish projection currency by reconciling the workspace for the report snapshot.
    #[doc(hidden)]
    pub async fn list_recurrence_series_view_from_current_projection(
        &self,
        workspace_id: &WorkspaceId,
        query: RecurrenceSeriesListQuery,
    ) -> Result<Vec<RecurrenceSeriesListItem>> {
        let mut conn = self.acquire_reader().await?;
        recurrence::list_recurrence_series_view(&mut conn, workspace_id, &query).await
    }

    pub async fn list_recurrence_series_view(
        &self,
        workspace_id: &WorkspaceId,
        query: RecurrenceSeriesListQuery,
    ) -> Result<Vec<RecurrenceSeriesListItem>> {
        let at = crate::ids::now_utc();
        self.list_recurrence_series_view_at(workspace_id, at, query)
            .await
    }

    pub async fn list_recurrence_series_at(
        &self,
        workspace_id: &WorkspaceId,
        at: DateTime<Utc>,
    ) -> Result<Vec<RecurrenceSeriesSummary>> {
        self.reconcile_recurrence_reports_at(workspace_id, at)
            .await?;
        let mut conn = self.acquire_reader().await?;
        recurrence::list_recurrence_series(&mut conn, workspace_id, at).await
    }

    pub async fn recurrence_series_detail_at(
        &self,
        workspace_id: &WorkspaceId,
        series_id: &RecurrenceSeriesId,
        at: DateTime<Utc>,
    ) -> Result<RecurrenceSeriesDetail> {
        self.reconcile_recurrence_reports_at(workspace_id, at)
            .await?;
        self.recurrence_series_detail_at_from_current_projection(workspace_id, series_id, at)
            .await
    }

    /// Reads the persisted recurrence projection as-is.
    ///
    /// Callers establish projection currency by reconciling the workspace for the report snapshot.
    #[doc(hidden)]
    pub async fn recurrence_series_detail_at_from_current_projection(
        &self,
        workspace_id: &WorkspaceId,
        series_id: &RecurrenceSeriesId,
        at: DateTime<Utc>,
    ) -> Result<RecurrenceSeriesDetail> {
        let mut conn = self.acquire_reader().await?;
        recurrence::recurrence_series_detail(&mut conn, workspace_id, series_id, at).await
    }

    pub async fn recurrence_history_at(
        &self,
        workspace_id: &WorkspaceId,
        series_id: &RecurrenceSeriesId,
        at: DateTime<Utc>,
        offset: usize,
        limit: usize,
    ) -> Result<RecurrenceHistoryPage> {
        validate_recurrence_history_limit(limit)?;
        self.reconcile_recurrence_reports_at(workspace_id, at)
            .await?;
        let mut conn = self.acquire_reader().await?;
        recurrence::recurrence_history(&mut conn, workspace_id, series_id, at, offset, limit).await
    }

    pub async fn list_recurrence_series(
        &self,
        workspace_id: &WorkspaceId,
    ) -> Result<Vec<RecurrenceSeriesSummary>> {
        let at = crate::ids::now_utc();
        self.list_recurrence_series_at(workspace_id, at).await
    }

    pub async fn recurrence_series_detail(
        &self,
        workspace_id: &WorkspaceId,
        series_id: &RecurrenceSeriesId,
    ) -> Result<RecurrenceSeriesDetail> {
        let at = crate::ids::now_utc();
        self.recurrence_series_detail_at(workspace_id, series_id, at)
            .await
    }

    /// Reads the persisted recurrence projection as-is.
    ///
    /// Callers establish projection currency by reconciling the workspace for the report snapshot.
    #[doc(hidden)]
    pub async fn recurrence_series_detail_from_current_projection(
        &self,
        workspace_id: &WorkspaceId,
        series_id: &RecurrenceSeriesId,
    ) -> Result<RecurrenceSeriesDetail> {
        let at = crate::ids::now_utc();
        self.recurrence_series_detail_at_from_current_projection(workspace_id, series_id, at)
            .await
    }

    pub async fn recurrence_history(
        &self,
        workspace_id: &WorkspaceId,
        series_id: &RecurrenceSeriesId,
        offset: usize,
        limit: usize,
    ) -> Result<RecurrenceHistoryPage> {
        let at = crate::ids::now_utc();
        self.recurrence_history_at(workspace_id, series_id, at, offset, limit)
            .await
    }

    pub async fn list_project_items(
        &self,
        workspace_id: &WorkspaceId,
    ) -> Result<Vec<ProjectListItem>> {
        self.reconcile_recurrence_reports(workspace_id).await?;
        self.list_project_items_from_current_projection(workspace_id)
            .await
    }

    /// Reads the persisted recurrence projection as-is.
    ///
    /// Callers establish projection currency by reconciling the workspace for the report snapshot.
    #[doc(hidden)]
    pub async fn list_project_items_from_current_projection(
        &self,
        workspace_id: &WorkspaceId,
    ) -> Result<Vec<ProjectListItem>> {
        let mut conn = self.acquire_reader().await?;
        list_project_items_in_workspace(&mut conn, workspace_id).await
    }

    pub async fn list_task_items(
        &self,
        workspace_id: &WorkspaceId,
        filters: TaskFilters,
        mode: TaskQueryMode,
        sort: TaskSort,
        direction: SortDirection,
    ) -> Result<Vec<TaskListItem>> {
        self.reconcile_recurrence_reports(workspace_id).await?;
        self.list_task_items_from_current_projection(workspace_id, filters, mode, sort, direction)
            .await
    }

    pub async fn list_bulk_update_task_items(
        &self,
        workspace_id: &WorkspaceId,
        filters: TaskFilters,
        mode: TaskQueryMode,
        sort: TaskSort,
        direction: SortDirection,
    ) -> Result<Vec<TaskListItem>> {
        self.reconcile_recurrence_reports(workspace_id).await?;
        let mut conn = self.acquire_reader().await?;
        list_bulk_update_task_items_in_workspace(
            &mut conn,
            workspace_id,
            filters,
            mode,
            sort,
            direction,
        )
        .await
    }

    /// Reads the persisted recurrence projection as-is.
    ///
    /// Callers establish projection currency by reconciling the workspace for the report snapshot.
    #[doc(hidden)]
    pub async fn list_task_items_from_current_projection(
        &self,
        workspace_id: &WorkspaceId,
        filters: TaskFilters,
        mode: TaskQueryMode,
        sort: TaskSort,
        direction: SortDirection,
    ) -> Result<Vec<TaskListItem>> {
        let mut conn = self.acquire_reader().await?;
        list_task_items_in_workspace(&mut conn, workspace_id, filters, mode, sort, direction).await
    }

    pub async fn list_task_items_without_activity_from_current_projection(
        &self,
        workspace_id: &WorkspaceId,
        filters: TaskFilters,
        mode: TaskQueryMode,
        sort: TaskSort,
        direction: SortDirection,
    ) -> Result<Vec<TaskListItem>> {
        let mut conn = self.acquire_reader().await?;
        list_task_items_without_activity_in_workspace(
            &mut conn,
            workspace_id,
            filters,
            mode,
            sort,
            direction,
        )
        .await
    }

    pub async fn list_task_summary_items(
        &self,
        workspace_id: &WorkspaceId,
        filters: TaskFilters,
        mode: TaskQueryMode,
        sort: TaskSort,
        direction: SortDirection,
        limit: Option<usize>,
    ) -> Result<Vec<TaskListItem>> {
        self.reconcile_recurrence_reports(workspace_id).await?;
        self.list_task_summary_items_from_current_projection(
            workspace_id,
            filters,
            mode,
            sort,
            direction,
            limit,
        )
        .await
    }

    /// Reads the persisted task projection as-is.
    ///
    /// Callers establish projection currency by reconciling the workspace for the report snapshot.
    #[doc(hidden)]
    pub async fn list_task_summary_items_from_current_projection(
        &self,
        workspace_id: &WorkspaceId,
        filters: TaskFilters,
        mode: TaskQueryMode,
        sort: TaskSort,
        direction: SortDirection,
        limit: Option<usize>,
    ) -> Result<Vec<TaskListItem>> {
        let mut conn = self.acquire_reader().await?;
        list_task_summary_items_in_workspace(
            &mut conn,
            workspace_id,
            filters,
            mode,
            sort,
            direction,
            limit,
        )
        .await
    }

    pub async fn sidebar_counts_for_scope(
        &self,
        workspace_id: &WorkspaceId,
        project_key: Option<&str>,
    ) -> Result<SidebarCounts> {
        self.reconcile_recurrence_reports(workspace_id).await?;
        self.sidebar_counts_for_scope_from_current_projection(workspace_id, project_key)
            .await
    }

    /// Reads the persisted recurrence projection as-is.
    ///
    /// Callers establish projection currency by reconciling the workspace for the report snapshot.
    #[doc(hidden)]
    pub async fn sidebar_counts_for_scope_from_current_projection(
        &self,
        workspace_id: &WorkspaceId,
        project_key: Option<&str>,
    ) -> Result<SidebarCounts> {
        let mut conn = self.acquire_reader().await?;
        sidebar_counts_for_scope_in_workspace(&mut conn, workspace_id, project_key).await
    }

    pub async fn list_recent_actions(
        &self,
        workspace_id: &WorkspaceId,
        project_scope: Option<&str>,
    ) -> Result<Vec<RecentActionItem>> {
        self.reconcile_recurrence_reports(workspace_id).await?;
        self.list_recent_actions_from_current_projection(workspace_id, project_scope)
            .await
    }

    /// Reads the persisted recurrence projection as-is.
    ///
    /// Callers establish projection currency by reconciling the workspace for the report snapshot.
    #[doc(hidden)]
    pub async fn list_recent_actions_from_current_projection(
        &self,
        workspace_id: &WorkspaceId,
        project_scope: Option<&str>,
    ) -> Result<Vec<RecentActionItem>> {
        let mut conn = self.acquire_reader().await?;
        list_recent_actions_in_workspace(&mut conn, workspace_id, project_scope).await
    }

    pub async fn task_activity_for_task(
        &self,
        workspace_id: &WorkspaceId,
        task_id: &TaskId,
    ) -> Result<Vec<RecentActionItem>> {
        let mut conn = self.acquire_reader().await?;
        Ok(task_activity_for_tasks_in_workspace(
            &mut conn,
            workspace_id,
            std::slice::from_ref(task_id),
        )
        .await?
        .remove(task_id)
        .unwrap_or_default())
    }

    pub async fn search_task_items(
        &self,
        workspace_id: &WorkspaceId,
        query: TaskSearchQuery,
    ) -> Result<Vec<TaskSearchResult>> {
        validate_search_limit(query.limit)?;
        self.reconcile_recurrence_reports(workspace_id).await?;
        let mut conn = self.acquire_reader().await?;
        search_task_items_in_workspace(&mut conn, workspace_id, query).await
    }

    pub async fn search_task_occurrence_items(
        &self,
        workspace_id: &WorkspaceId,
        query: TaskSearchQuery,
    ) -> Result<Vec<TaskSearchResult>> {
        validate_search_limit(query.limit)?;
        self.reconcile_recurrence_reports(workspace_id).await?;
        let mut conn = self.acquire_reader().await?;
        search_task_occurrence_items_in_workspace(&mut conn, workspace_id, query).await
    }

    pub async fn search_task_preview_set(
        &self,
        workspace_id: &WorkspaceId,
        query: TaskSearchQuery,
    ) -> Result<TaskSearchPreviewResultSet> {
        validate_search_limit(query.limit)?;
        self.reconcile_recurrence_reports(workspace_id).await?;
        self.search_task_preview_set_from_current_projection(workspace_id, query)
            .await
    }

    /// Reads the persisted recurrence projection as-is.
    ///
    /// Callers establish projection currency by reconciling the workspace for the report snapshot.
    #[doc(hidden)]
    pub async fn search_task_preview_set_from_current_projection(
        &self,
        workspace_id: &WorkspaceId,
        query: TaskSearchQuery,
    ) -> Result<TaskSearchPreviewResultSet> {
        validate_search_limit(query.limit)?;
        let mut conn = self.acquire_reader().await?;
        search_task_preview_set_in_workspace(&mut conn, workspace_id, query).await
    }

    pub async fn task_detail(&self, task: &Task) -> Result<TaskDetail> {
        self.reconcile_recurrence_reports(&task.workspace_id)
            .await?;
        let mut conn = self.acquire_reader().await?;
        task_detail(&mut conn, task).await
    }

    pub async fn task_detail_with_display_refs(
        &self,
        task: &Task,
        display_refs: &DisplayRefContext,
    ) -> Result<TaskDetail> {
        self.reconcile_recurrence_reports(&task.workspace_id)
            .await?;
        let mut conn = self.acquire_reader().await?;
        task_detail_with_display_refs(&mut conn, task, display_refs).await
    }

    pub async fn conflict_display_value(
        &self,
        workspace_id: &WorkspaceId,
        field: &str,
        value: &str,
    ) -> Result<String> {
        let mut conn = self.acquire_reader().await?;
        conflict_display_value(&mut conn, workspace_id, field, value).await
    }

    pub async fn task_dependency_summary(
        &self,
        workspace_id: &WorkspaceId,
        task_id: &TaskId,
    ) -> Result<TaskDependencySummary> {
        let mut conn = self.acquire_reader().await?;
        task_dependency_summary(&mut conn, workspace_id, task_id).await
    }

    pub async fn task_related_links(
        &self,
        workspace_id: &WorkspaceId,
        task_id: &TaskId,
    ) -> Result<Vec<TaskRelatedLink>> {
        let mut conn = self.acquire_reader().await?;
        task_related_links(&mut conn, workspace_id, task_id, None).await
    }

    pub async fn workspace_task_counts(
        &self,
        workspace_id: &WorkspaceId,
    ) -> Result<WorkspaceTaskCounts> {
        let mut conn = self.acquire_reader().await?;
        workspace_task_counts(&mut conn, workspace_id).await
    }

    pub async fn unresolved_conflict_count(&self) -> Result<i64> {
        let mut conn = self.acquire_reader().await?;
        unresolved_conflict_count(&mut conn).await
    }

    pub async fn sync_history_stats(&self) -> Result<SyncHistoryStats> {
        let mut conn = self.acquire_reader().await?;
        sync_history_stats(&mut conn).await
    }

    pub async fn database_stats(
        &self,
        workspace: &crate::workspaces::Workspace,
    ) -> Result<DatabaseStats> {
        let mut conn = self.acquire_reader().await?;
        database_stats::database_stats(&mut conn, workspace).await
    }
}