use crate::argparse::{DescriptionMod, Modification};
use failure::Fallible;
use std::convert::TryInto;
use taskchampion::TaskMut;
use termcolor::WriteColor;

/// Apply the given modification
pub(super) fn apply_modification<W: WriteColor>(
    w: &mut W,
    task: &mut TaskMut,
    modification: &Modification,
) -> Fallible<()> {
    match modification.description {
        DescriptionMod::Set(ref description) => task.set_description(description.clone())?,
        DescriptionMod::Prepend(ref description) => {
            task.set_description(format!("{} {}", description, task.get_description()))?
        }
        DescriptionMod::Append(ref description) => {
            task.set_description(format!("{} {}", task.get_description(), description))?
        }
        DescriptionMod::None => {}
    }

    if let Some(ref status) = modification.status {
        task.set_status(status.clone())?;
    }

    if let Some(true) = modification.active {
        task.start()?;
    }

    if let Some(false) = modification.active {
        task.stop()?;
    }

    for tag in modification.add_tags.iter() {
        let tag = tag.try_into()?; // see #111
        task.add_tag(&tag)?;
    }

    for tag in modification.remove_tags.iter() {
        let tag = tag.try_into()?; // see #111
        task.remove_tag(&tag)?;
    }

    write!(w, "modified task {}\n", task.get_uuid())?;

    Ok(())
}