#[derive(Clone, Debug, PartialEq, Eq, Default)]
#[cfg_attr(
feature = "serde-support",
derive(serde_derive::Serialize, serde_derive::Deserialize)
)]
pub struct Extended {
#[cfg_attr(feature = "serde-support", serde(flatten))]
pub inner: super::Task,
#[cfg_attr(feature = "serde-support", serde(default))]
pub note: super::Note,
pub recurrence: Option<super::Recurrence>,
#[cfg_attr(feature = "serde-support", serde(default))]
pub flagged: bool,
#[cfg_attr(feature = "serde-support", serde(default))]
pub hidden: bool,
}
impl Extended {
pub fn has_note(&self) -> bool {
self.note != super::Note::None
}
fn note(task: &super::Task) -> super::Note {
if let Some(file) = task.tags.get(&Self::tag_name()) {
super::Note::from_file(file)
} else {
super::Note::None
}
}
fn tag_name() -> String {
match std::env::var("TODO_NOTE_TAG") {
Ok(tag) => tag,
Err(_) => "note".to_owned(),
}
}
}
impl std::convert::From<super::Task> for Extended {
fn from(task: super::Task) -> Self {
use std::str::FromStr;
let mut inner = task;
let note = Self::note(&inner);
inner.tags.remove(&Self::tag_name());
let mut recurrence = None;
if let Some(rec) = inner.tags.get(&"rec".to_owned()) {
recurrence = match super::Recurrence::from_str(rec) {
Ok(rec) => Some(rec),
Err(_) => None,
};
}
inner.tags.remove(&"rec".to_owned());
let flagged = inner.tags.contains_key(&"f".to_owned());
inner.tags.remove(&"f".to_owned());
let hidden = inner.tags.contains_key(&"h".to_owned());
inner.tags.remove(&"h".to_owned());
Self {
inner,
note,
recurrence,
flagged,
hidden,
}
}
}
impl std::str::FromStr for Extended {
type Err = crate::Error;
fn from_str(s: &str) -> Result<Self, Self::Err> {
let task = super::Task::from_str(s)?;
Ok(task.into())
}
}
impl std::ops::Deref for Extended {
type Target = super::Task;
fn deref(&self) -> &Self::Target {
&self.inner
}
}
impl std::ops::DerefMut for Extended {
fn deref_mut(&mut self) -> &mut Self::Target {
&mut self.inner
}
}
impl std::fmt::Display for Extended {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
use std::ops::Deref;
f.write_str(format!("{}", self.deref()).as_str())?;
if self.note != super::Note::None {
f.write_str(format!(" {}", self.note).as_str())?;
}
if let Some(ref recurrence) = self.recurrence {
f.write_str(format!(" rec:{}", recurrence).as_str())?;
}
if self.flagged {
f.write_str(" f:1")?;
}
if self.hidden {
f.write_str(" h:1")?;
}
Ok(())
}
}
impl std::cmp::PartialOrd for Extended {
fn partial_cmp(&self, other: &Self) -> Option<std::cmp::Ordering> {
Some(self.cmp(other))
}
}
impl std::cmp::Ord for Extended {
fn cmp(&self, other: &Self) -> std::cmp::Ordering {
self.inner.cmp(&other.inner)
}
}