use anyhow::Result;
use chrono::Utc;
use rusqlite::Connection;
use serde_json::{Value, json};
use crate::infrastructure::config::Config;
use crate::infrastructure::db;
use crate::infrastructure::model::Status;
pub fn done_value(conn: &Connection, cfg: &Config, id_or_uuid: &str, force: bool) -> Result<Value> {
let mut task = db::resolve_task(conn, id_or_uuid)?;
let blockers = db::get_blockers(conn, &task.uuid)?;
if !blockers.is_empty() && !force {
let blocker_ids: Vec<String> = blockers
.iter()
.map(|u| {
db::get_task_by_uuid_prefix(conn, &u.to_string()[..8])
.ok()
.flatten()
.and_then(|t| t.id)
.map(|i| i.to_string())
.unwrap_or_else(|| u.to_string()[..8].to_string())
})
.collect();
anyhow::bail!(
"Task {} is blocked by tasks: {}. Use --force to complete anyway.",
task.id.unwrap_or(0),
blocker_ids.join(", ")
);
}
if let Some(started) = task.started_at {
task.time_spent += (Utc::now() - started).num_seconds().max(0);
task.started_at = None;
}
task.status = Status::Completed;
task.end = Some(Utc::now());
task.modified = Utc::now();
db::update_task(conn, &task)?;
db::repack_ids(conn)?;
let was_blocking = db::get_blocking(conn, &task.uuid)?;
for dep_uuid in was_blocking {
let _ = db::refresh_urgency(conn, &cfg.urgency, &dep_uuid);
}
let mut recurrence = Value::Null;
if let Some(ref interval) = task.recur.clone() {
let base = task.due.unwrap_or_else(Utc::now);
let next_due = crate::infrastructure::model::advance_by_interval(base, interval);
let mut next =
crate::infrastructure::model::Task::new(task.description.clone(), task.project.clone());
next.priority = task.priority.clone();
next.tags = task.tags.clone();
next.due = Some(next_due);
next.recur = Some(interval.clone());
next.estimate_mins = task.estimate_mins;
next.urgency = db::compute_urgency(&next, &cfg.urgency, false, 0);
db::insert_task(conn, &mut next)?;
recurrence = json!({
"id": next.id,
"due": next_due.with_timezone(&chrono::Local).format("%Y-%m-%d").to_string(),
});
}
Ok(json!({
"task": task.id,
"uuid": task.uuid.to_string(),
"project": task.project,
"description": task.description,
"status": "completed",
"recurrence": recurrence,
}))
}
pub fn run(conn: &Connection, cfg: &Config, id_or_uuid: &str, force: bool) -> Result<()> {
let v = done_value(conn, cfg, id_or_uuid, force)?;
println!(
"Done: [{}] {}",
v["project"].as_str().unwrap_or_default(),
v["description"].as_str().unwrap_or_default()
);
if let Some(rec) = v.get("recurrence").filter(|r| !r.is_null()) {
println!(
"♺ Next recurrence: #{} due {}",
rec.get("id").and_then(|i| i.as_i64()).unwrap_or(0),
rec.get("due").and_then(|d| d.as_str()).unwrap_or_default()
);
}
Ok(())
}