issue-bot/src/cron.rs

93 lines
3.4 KiB
Rust
Raw Normal View History

2019-09-20 12:59:13 +03:00
use super::*;
pub fn check(conn: Connection, conf: Config) {
let mut stmt = conn.prepare("SELECT * FROM issue").unwrap();
let mut results = stmt
.query_map(NO_PARAMS, |row| {
let submitter: String = row.get(1)?;
let password: Vec<u8> = row.get(2)?;
let last_update: Option<String> = row.get(7)?;
Ok(Issue {
id: row.get(0)?,
submitter: new_address(submitter.as_str()),
password: Password::from_slice(password.as_slice()).unwrap(),
time_created: row.get(3)?,
anonymous: row.get(4)?,
subscribed: row.get(5)?,
title: row.get(6)?,
last_update: last_update.unwrap_or(String::new()),
})
})
.unwrap()
.map(|r| r.unwrap())
.collect::<Vec<Issue>>();
for issue in &mut results {
let mut update = false;
let mut comments = api::comments(issue.id, &issue.last_update, &conf);
let mut new_value = issue.last_update.clone();
comments.retain(|c| {
let created_at = c["created_at"].to_string();
if created_at > issue.last_update {
if created_at > new_value {
new_value = created_at;
update = true;
}
true
} else {
false
}
});
if update {
let mut stmt = conn
.prepare("UPDATE issue SET last_update = (:last_update) WHERE id = (:id)")
.unwrap();
assert_eq!(
stmt.execute_named(&[(":last_update", &new_value), (":id", &issue.id),])
.unwrap(),
1
);
if issue.subscribed {
let comments = comments
.into_iter()
.map(|c| {
if c["user"]["login"].as_str().unwrap() == &conf.bot_username {
c["body"].as_str().unwrap().to_string()
} else {
format!(
"User {} replied:\n\n{}",
c["user"]["login"],
c["body"].as_str().unwrap()
)
}
})
.collect::<Vec<String>>();
let mut notice = melib::Draft::default();
notice.headers_mut().insert(
"From".to_string(),
new_address(&format!(
"{local_part}@{domain}",
local_part = &conf.local_part,
domain = &conf.domain
))
.to_string(),
);
notice.headers_mut().insert(
"Subject".to_string(),
format!(
"[{tag}] new replies in issue `{title}`",
tag = &conf.tag,
title = &issue.title
)
.to_string(),
);
notice
.headers_mut()
.insert("To".to_string(), issue.submitter.to_string());
notice.set_body(templates::reply_update(&issue, &conf, comments));
send_mail(notice, &conf);
}
}
}
}