mirror of
https://git.meli.delivery/meli/issue-bot
synced 2024-05-02 19:43:11 +02:00
Update to 0.2.0
This commit is contained in:
parent
ca4580790b
commit
6aa2f12e2c
636
Cargo.lock
generated
636
Cargo.lock
generated
File diff suppressed because it is too large
Load diff
11
Cargo.toml
11
Cargo.toml
|
@ -1,6 +1,6 @@
|
|||
[package]
|
||||
name = "issue-bot"
|
||||
version = "0.1.0"
|
||||
version = "0.2.0"
|
||||
authors = ["Manos Pitsidianakis <el13635@mail.ntua.gr>"]
|
||||
edition = "2018"
|
||||
|
||||
|
@ -11,12 +11,15 @@ rusqlite = { version="0.20.0", features=["uuid",]}
|
|||
uuid = "*"
|
||||
time = "*"
|
||||
serde_json = "1.0.40"
|
||||
serde= { version = "1.0.101", features = ["derive"]}
|
||||
serde = { version = "1.0.101", features = ["derive"]}
|
||||
reqwest = "0.9.20"
|
||||
toml = "0.5.3"
|
||||
log = "0.4.11"
|
||||
simplelog = "^0.8.0"
|
||||
error-chain = "0.12.4"
|
||||
|
||||
[dependencies.melib]
|
||||
git = "https://git.meli.delivery/meli/meli"
|
||||
version = "0.3.2"
|
||||
default-features = false
|
||||
version = "0.6.1"
|
||||
default-features = false
|
||||
features = []
|
||||
|
|
10
sample-config.toml
Normal file
10
sample-config.toml
Normal file
|
@ -0,0 +1,10 @@
|
|||
tag = "repo-issues"
|
||||
auth_token= "Basic xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
|
||||
local_part= "issues"
|
||||
domain= "example.tld"
|
||||
base_url = "https://git.example.tld"
|
||||
repo = "username/repo"
|
||||
bot_name = "IssueBot"
|
||||
bot_username = "username"
|
||||
mailer = "cat"
|
||||
log_file = "issue-bot.log"
|
42
src/api.rs
42
src/api.rs
|
@ -1,3 +1,19 @@
|
|||
/* This file is part of issue-bot.
|
||||
*
|
||||
* issue-bot is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* issue-bot is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with issue-bot. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
use super::*;
|
||||
|
||||
static ISSUES_BASE_URL: &'static str = "{base_url}/api/v1/repos/{repo}/issues";
|
||||
|
@ -20,7 +36,7 @@ pub fn new_issue(
|
|||
body: String,
|
||||
anonymous: bool,
|
||||
submitter: Address,
|
||||
conf: &Config,
|
||||
conf: &Configuration,
|
||||
) -> Result<(Password, i64)> {
|
||||
let issue = CreateIssueOption {
|
||||
title,
|
||||
|
@ -88,7 +104,7 @@ pub fn new_reply(
|
|||
body: String,
|
||||
password: Password,
|
||||
submitter: Address,
|
||||
conf: &Config,
|
||||
conf: &Configuration,
|
||||
) -> Result<(String, i64, bool)> {
|
||||
let mut stmt =
|
||||
conn.prepare("SELECT id, title, subscribed, anonymous FROM issue WHERE password = ?")?;
|
||||
|
@ -99,7 +115,7 @@ pub fn new_reply(
|
|||
.map(|r| r.unwrap())
|
||||
.collect::<Vec<(i64, String, bool, bool)>>();
|
||||
if results.is_empty() {
|
||||
return Err(IssueError::new("Not found".to_string()));
|
||||
return Err(Error::new("Not found".to_string()));
|
||||
}
|
||||
let client = reqwest::Client::new();
|
||||
let response = client
|
||||
|
@ -133,7 +149,7 @@ pub fn new_reply(
|
|||
submitter.to_string(),
|
||||
body
|
||||
);
|
||||
Err(IssueError::new(
|
||||
Err(Error::new(
|
||||
"You can not reply to this issue due to an internal error.",
|
||||
))
|
||||
}
|
||||
|
@ -144,7 +160,11 @@ struct EditIssueOption {
|
|||
state: String,
|
||||
}
|
||||
|
||||
pub fn close(conn: &Connection, password: Password, conf: &Config) -> Result<(String, i64, bool)> {
|
||||
pub fn close(
|
||||
conn: &Connection,
|
||||
password: Password,
|
||||
conf: &Configuration,
|
||||
) -> Result<(String, i64, bool)> {
|
||||
let mut stmt = conn.prepare("SELECT id, title, subscribed FROM issue WHERE password = ?")?;
|
||||
let mut results = stmt
|
||||
.query_map(&[password.as_bytes().to_vec()], |row| {
|
||||
|
@ -153,7 +173,7 @@ pub fn close(conn: &Connection, password: Password, conf: &Config) -> Result<(St
|
|||
.map(|r| r.unwrap())
|
||||
.collect::<Vec<(i64, String, bool)>>();
|
||||
if results.is_empty() {
|
||||
return Err(IssueError::new("Not found".to_string()));
|
||||
return Err(Error::new("Not found".to_string()));
|
||||
}
|
||||
let client = reqwest::Client::new();
|
||||
let res = client
|
||||
|
@ -177,7 +197,7 @@ pub fn close(conn: &Connection, password: Password, conf: &Config) -> Result<(St
|
|||
Ok((title, issue_id, is_subscribed))
|
||||
} else {
|
||||
eprintln!("Issue could not be closed: {:#?}", map);
|
||||
Err(IssueError::new(
|
||||
Err(Error::new(
|
||||
"Issue cannot be closed due to an internal error.",
|
||||
))
|
||||
}
|
||||
|
@ -196,16 +216,16 @@ pub fn change_subscription(
|
|||
.map(|r| r.unwrap())
|
||||
.collect::<Vec<(i64, String, bool)>>();
|
||||
if results.is_empty() {
|
||||
return Err(IssueError::new("Issue not found".to_string()));
|
||||
return Err(Error::new("Issue not found".to_string()));
|
||||
}
|
||||
let (issue_id, title, is_subscribed) = results.remove(0);
|
||||
if !is_subscribed && !new_val {
|
||||
return Err(IssueError::new(format!(
|
||||
return Err(Error::new(format!(
|
||||
"You are not subscribed to issue `{}`",
|
||||
&title
|
||||
)));
|
||||
} else if is_subscribed && new_val {
|
||||
return Err(IssueError::new(format!(
|
||||
return Err(Error::new(format!(
|
||||
"You are already subscribed to issue `{}`",
|
||||
&title
|
||||
)));
|
||||
|
@ -226,7 +246,7 @@ pub fn change_subscription(
|
|||
pub fn comments(
|
||||
id: i64,
|
||||
since: &str,
|
||||
conf: &Config,
|
||||
conf: &Configuration,
|
||||
) -> Vec<serde_json::map::Map<String, serde_json::Value>> {
|
||||
let client = reqwest::Client::new();
|
||||
let result = client
|
||||
|
|
20
src/conf.rs
20
src/conf.rs
|
@ -1,7 +1,23 @@
|
|||
/* This file is part of issue-bot.
|
||||
*
|
||||
* issue-bot is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* issue-bot is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with issue-bot. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
use serde::Deserialize;
|
||||
|
||||
#[derive(Deserialize, Debug)]
|
||||
pub struct Config {
|
||||
pub struct Configuration {
|
||||
/** eg. meli-issues becomes [meli-issues] **/
|
||||
pub tag: String,
|
||||
/** your bot's authentication token from Gitea's Swagger **/
|
||||
|
@ -20,4 +36,6 @@ pub struct Config {
|
|||
pub bot_username: String,
|
||||
/** the command to pipe an email to **/
|
||||
pub mailer: String,
|
||||
/** file to write logs **/
|
||||
pub log_file: String,
|
||||
}
|
||||
|
|
59
src/cron.rs
59
src/cron.rs
|
@ -1,7 +1,24 @@
|
|||
use super::*;
|
||||
/* This file is part of issue-bot.
|
||||
*
|
||||
* issue-bot is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* issue-bot is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with issue-bot. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
pub fn check(conn: Connection, conf: Config) {
|
||||
let mut stmt = conn.prepare("SELECT * FROM issue").unwrap();
|
||||
use super::*;
|
||||
use melib::email::address::Address;
|
||||
|
||||
pub fn check(conn: Connection, conf: Configuration) -> Result<()> {
|
||||
let mut stmt = conn.prepare("SELECT * FROM issue")?;
|
||||
let mut results = stmt
|
||||
.query_map(NO_PARAMS, |row| {
|
||||
let submitter: String = row.get(1)?;
|
||||
|
@ -9,7 +26,7 @@ pub fn check(conn: Connection, conf: Config) {
|
|||
let last_update: Option<String> = row.get(7)?;
|
||||
Ok(Issue {
|
||||
id: row.get(0)?,
|
||||
submitter: new_address(submitter.as_str()),
|
||||
submitter: Address::new(None, submitter.as_str().to_string()),
|
||||
password: Password::from_slice(password.as_slice()).unwrap(),
|
||||
time_created: row.get(3)?,
|
||||
anonymous: row.get(4)?,
|
||||
|
@ -17,10 +34,8 @@ pub fn check(conn: Connection, conf: Config) {
|
|||
title: row.get(6)?,
|
||||
last_update: last_update.unwrap_or(String::new()),
|
||||
})
|
||||
})
|
||||
.unwrap()
|
||||
.map(|r| r.unwrap())
|
||||
.collect::<Vec<Issue>>();
|
||||
})?
|
||||
.collect::<std::result::Result<Vec<Issue>, _>>()?;
|
||||
for issue in &mut results {
|
||||
let mut update = false;
|
||||
let mut comments = api::comments(issue.id, &issue.last_update, &conf);
|
||||
|
@ -38,12 +53,10 @@ pub fn check(conn: Connection, conf: Config) {
|
|||
}
|
||||
});
|
||||
if update {
|
||||
let mut stmt = conn
|
||||
.prepare("UPDATE issue SET last_update = (:last_update) WHERE id = (:id)")
|
||||
.unwrap();
|
||||
let mut stmt =
|
||||
conn.prepare("UPDATE issue SET last_update = (:last_update) WHERE id = (:id)")?;
|
||||
assert_eq!(
|
||||
stmt.execute_named(&[(":last_update", &new_value), (":id", &issue.id),])
|
||||
.unwrap(),
|
||||
stmt.execute_named(&[(":last_update", &new_value), (":id", &issue.id),])?,
|
||||
1
|
||||
);
|
||||
if issue.subscribed {
|
||||
|
@ -63,16 +76,19 @@ pub fn check(conn: Connection, conf: Config) {
|
|||
.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
|
||||
))
|
||||
HeaderName::new_unchecked("From"),
|
||||
Address::new(
|
||||
None,
|
||||
format!(
|
||||
"{local_part}@{domain}",
|
||||
local_part = &conf.local_part,
|
||||
domain = &conf.domain
|
||||
),
|
||||
)
|
||||
.to_string(),
|
||||
);
|
||||
notice.headers_mut().insert(
|
||||
"Subject".to_string(),
|
||||
HeaderName::new_unchecked("Subject"),
|
||||
format!(
|
||||
"[{tag}] new replies in issue `{title}`",
|
||||
tag = &conf.tag,
|
||||
|
@ -82,11 +98,12 @@ pub fn check(conn: Connection, conf: Config) {
|
|||
);
|
||||
notice
|
||||
.headers_mut()
|
||||
.insert("To".to_string(), issue.submitter.to_string());
|
||||
.insert(HeaderName::new_unchecked("To"), issue.submitter.to_string());
|
||||
|
||||
notice.set_body(templates::reply_update(&issue, &conf, comments));
|
||||
send_mail(notice, &conf);
|
||||
}
|
||||
}
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
|
114
src/error.rs
114
src/error.rs
|
@ -1,92 +1,32 @@
|
|||
use std::borrow::Cow;
|
||||
use std::error::Error;
|
||||
use std::fmt;
|
||||
use std::io;
|
||||
use std::result;
|
||||
use std::str;
|
||||
use std::string;
|
||||
/* This file is part of issue-bot.
|
||||
*
|
||||
* issue-bot is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* issue-bot is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with issue-bot. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
pub type Result<T> = result::Result<T, IssueError>;
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct IssueError {
|
||||
details: String,
|
||||
error_chain! {
|
||||
foreign_links {
|
||||
Io(std::io::Error);
|
||||
Reqwest(reqwest::Error);
|
||||
Database(rusqlite::Error);
|
||||
Unicode(std::str::Utf8Error);
|
||||
UnicodeS(std::string::FromUtf8Error);
|
||||
Email(melib::error::MeliError);
|
||||
}
|
||||
}
|
||||
|
||||
impl IssueError {
|
||||
pub fn new<M>(msg: M) -> IssueError
|
||||
where
|
||||
M: Into<String>,
|
||||
{
|
||||
IssueError {
|
||||
details: msg.into(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl fmt::Display for IssueError {
|
||||
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
||||
write!(f, "{}", self.details)
|
||||
}
|
||||
}
|
||||
|
||||
impl Into<String> for IssueError {
|
||||
fn into(self) -> String {
|
||||
self.details
|
||||
}
|
||||
}
|
||||
|
||||
impl Error for IssueError {
|
||||
fn description(&self) -> &str {
|
||||
&self.details
|
||||
}
|
||||
}
|
||||
|
||||
impl From<io::Error> for IssueError {
|
||||
#[inline]
|
||||
fn from(kind: io::Error) -> IssueError {
|
||||
IssueError::new(kind.description())
|
||||
}
|
||||
}
|
||||
|
||||
impl Into<io::Error> for IssueError {
|
||||
#[inline]
|
||||
fn into(self) -> io::Error {
|
||||
io::Error::new(io::ErrorKind::Other, self.description())
|
||||
}
|
||||
}
|
||||
|
||||
impl<'a> From<Cow<'a, str>> for IssueError {
|
||||
#[inline]
|
||||
fn from(kind: Cow<'_, str>) -> IssueError {
|
||||
IssueError::new(format!("{:?}", kind))
|
||||
}
|
||||
}
|
||||
|
||||
impl From<string::FromUtf8Error> for IssueError {
|
||||
#[inline]
|
||||
fn from(kind: string::FromUtf8Error) -> IssueError {
|
||||
IssueError::new(format!("{:?}", kind))
|
||||
}
|
||||
}
|
||||
|
||||
impl From<str::Utf8Error> for IssueError {
|
||||
#[inline]
|
||||
fn from(kind: str::Utf8Error) -> IssueError {
|
||||
IssueError::new(format!("{:?}", kind))
|
||||
}
|
||||
}
|
||||
|
||||
impl From<rusqlite::Error> for IssueError {
|
||||
#[inline]
|
||||
fn from(kind: rusqlite::Error) -> IssueError {
|
||||
IssueError::new(format!("{}", kind.to_string()))
|
||||
}
|
||||
}
|
||||
|
||||
impl From<reqwest::Error> for IssueError {
|
||||
#[inline]
|
||||
fn from(kind: reqwest::Error) -> IssueError {
|
||||
IssueError::new(format!("{}", kind.to_string()))
|
||||
impl Error {
|
||||
pub fn new<S: Into<String>>(msg: S) -> Self {
|
||||
msg.into().into()
|
||||
}
|
||||
}
|
||||
|
|
473
src/main.rs
473
src/main.rs
|
@ -1,12 +1,38 @@
|
|||
use melib::Envelope;
|
||||
/* This file is part of issue-bot.
|
||||
*
|
||||
* issue-bot is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* issue-bot is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with issue-bot. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
#![recursion_limit = "1024"]
|
||||
|
||||
extern crate log;
|
||||
extern crate simplelog;
|
||||
#[macro_use]
|
||||
extern crate error_chain;
|
||||
|
||||
use log::{error, info, trace};
|
||||
use melib::email::headers::HeaderName;
|
||||
use melib::{Address, Envelope};
|
||||
use rusqlite::types::ToSql;
|
||||
use rusqlite::{Connection, NO_PARAMS};
|
||||
use simplelog::*;
|
||||
use std::fs::File;
|
||||
use std::io::{stdin, Read};
|
||||
use time::Timespec;
|
||||
use uuid::Uuid;
|
||||
|
||||
mod error;
|
||||
use error::*;
|
||||
pub use error::*;
|
||||
mod api;
|
||||
mod conf;
|
||||
use conf::*;
|
||||
|
@ -16,8 +42,6 @@ mod templates;
|
|||
type Password = Uuid;
|
||||
static PASSWORD_COMMANDS: &'static [&'static str] = &["reply", "unsubscribe", "subscribe", "close"];
|
||||
|
||||
use melib::{email::parser, Address};
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct Issue {
|
||||
id: i64,
|
||||
|
@ -30,11 +54,7 @@ pub struct Issue {
|
|||
last_update: String,
|
||||
}
|
||||
|
||||
fn new_address(s: &str) -> Address {
|
||||
parser::address(s.as_bytes()).to_full_result().unwrap()
|
||||
}
|
||||
|
||||
pub fn send_mail(d: melib::email::Draft, conf: &Config) {
|
||||
pub fn send_mail(d: melib::email::Draft, conf: &Configuration) {
|
||||
use std::io::Write;
|
||||
use std::process::Stdio;
|
||||
let parts = conf.mailer.split_whitespace().collect::<Vec<&str>>();
|
||||
|
@ -55,12 +75,229 @@ pub fn send_mail(d: melib::email::Draft, conf: &Config) {
|
|||
let output = mailer.wait().expect("Failed to wait on mailer");
|
||||
if !output.success() {
|
||||
// TODO: commit to database queue
|
||||
error!("mailer fail");
|
||||
eprintln!("mailer fail");
|
||||
std::process::exit(1);
|
||||
}
|
||||
}
|
||||
|
||||
fn main() -> std::result::Result<(), std::io::Error> {
|
||||
fn run_app(conn: Connection, conf: Configuration) -> Result<()> {
|
||||
let mut new_message_raw = vec![];
|
||||
stdin().lock().read_to_end(&mut new_message_raw)?;
|
||||
|
||||
trace!(
|
||||
"Received this raw message:\n{}",
|
||||
&String::from_utf8_lossy(&new_message_raw)
|
||||
);
|
||||
|
||||
let envelope = Envelope::from_bytes(new_message_raw.as_slice(), None)?;
|
||||
let mut reply = melib::Draft::new_reply(&envelope, new_message_raw.as_slice(), true);
|
||||
reply.headers_mut().insert(
|
||||
HeaderName::new_unchecked("From"),
|
||||
format!(
|
||||
"{local_part}@{domain}",
|
||||
local_part = &conf.local_part,
|
||||
domain = &conf.domain
|
||||
),
|
||||
);
|
||||
|
||||
let tags: Vec<String> = envelope.to()[0].get_tags('+');
|
||||
match tags.as_slice() {
|
||||
s if s.is_empty() || s == &["anonymous"] => {
|
||||
/* Assign new issue */
|
||||
let subject = envelope.subject().to_string();
|
||||
let body = envelope.body_bytes(new_message_raw.as_slice()).text();
|
||||
let from = envelope.from()[0].clone();
|
||||
info!("Assign new issue with subject {} from {}", &subject, &from);
|
||||
let mut reply = melib::Draft::new_reply(&envelope, new_message_raw.as_slice(), true);
|
||||
let anonymous = !tags.is_empty();
|
||||
reply.headers_mut().insert(
|
||||
HeaderName::new_unchecked("From"),
|
||||
format!(
|
||||
"{local_part}@{domain}",
|
||||
local_part = &conf.local_part,
|
||||
domain = &conf.domain
|
||||
),
|
||||
);
|
||||
match api::new_issue(&conn, subject.clone(), body, anonymous, from, &conf) {
|
||||
Ok((password, issue_id)) => {
|
||||
info!("Issue {} successfully created.", &subject);
|
||||
reply.headers_mut().insert(
|
||||
HeaderName::new_unchecked("Subject"),
|
||||
format!(
|
||||
"[{tag}] Issue `{}` successfully created",
|
||||
&subject,
|
||||
tag = &conf.tag
|
||||
),
|
||||
);
|
||||
reply.set_body(templates::new_issue_success(
|
||||
subject, password, issue_id, &conf,
|
||||
));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
Err(err) => {
|
||||
error!("Issue {} could not be created {}.", &subject, &err);
|
||||
reply.headers_mut().insert(
|
||||
HeaderName::new_unchecked("Subject"),
|
||||
format!(
|
||||
"[{tag}] Issue `{}` could not be created",
|
||||
&subject,
|
||||
tag = &conf.tag
|
||||
),
|
||||
);
|
||||
reply.set_body(templates::new_issue_failure(err, &conf));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
}
|
||||
}
|
||||
&[ref p, ref cmd]
|
||||
if Password::parse_str(p).is_ok() && PASSWORD_COMMANDS.contains(&cmd.as_str()) =>
|
||||
{
|
||||
trace!("Got command {} from {}", cmd.as_str(), &envelope.from()[0]);
|
||||
let p = Password::parse_str(p).unwrap();
|
||||
match cmd.as_str() {
|
||||
"reply" => {
|
||||
info!(
|
||||
"Got reply with subject {} from {}.",
|
||||
&envelope.subject(),
|
||||
&envelope.from()[0]
|
||||
);
|
||||
let body = envelope.body_bytes(new_message_raw.as_slice()).text();
|
||||
let from = envelope.from()[0].clone();
|
||||
match api::new_reply(&conn, body, p, from, &conf) {
|
||||
Ok((title, issue_id, is_subscribed)) => {
|
||||
info!("Reply successfully created.");
|
||||
reply.headers_mut().insert(
|
||||
HeaderName::new_unchecked("Subject"),
|
||||
format!(
|
||||
"[{tag}] Your reply on issue `{}` has been posted",
|
||||
&title,
|
||||
tag = &conf.tag,
|
||||
),
|
||||
);
|
||||
reply.set_body(templates::new_reply_success(
|
||||
title,
|
||||
p,
|
||||
issue_id,
|
||||
is_subscribed,
|
||||
&conf,
|
||||
));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
Err(err) => {
|
||||
error!(
|
||||
"Reply {} could not be created {}.",
|
||||
&envelope.subject(),
|
||||
&err
|
||||
);
|
||||
reply.headers_mut().insert(HeaderName::new_unchecked("Subject"),
|
||||
format!(
|
||||
"[{tag}] Your reply could not be created",
|
||||
tag = &conf.tag,
|
||||
),
|
||||
);
|
||||
reply.set_body(templates::new_reply_failure(err, &conf));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
}
|
||||
}
|
||||
"close" => match api::close(&conn, p, &conf) {
|
||||
Ok((title, issue_id, _)) => {
|
||||
reply.headers_mut().insert(
|
||||
HeaderName::new_unchecked("Subject"),
|
||||
format!(
|
||||
"[{tag}] issue `{}` has been closed",
|
||||
&title,
|
||||
tag = &conf.tag
|
||||
),
|
||||
);
|
||||
reply.set_body(templates::close_success(title, issue_id, &conf));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
Err(e) => {
|
||||
reply.headers_mut().insert(
|
||||
HeaderName::new_unchecked("Subject"),
|
||||
format!("[{tag}] issue could not be closed", tag = &conf.tag,),
|
||||
);
|
||||
reply.set_body(templates::close_failure(e, &conf));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
},
|
||||
"unsubscribe" => match api::change_subscription(&conn, p, false) {
|
||||
Ok((title, issue_id, _)) => {
|
||||
reply.headers_mut().insert(
|
||||
HeaderName::new_unchecked("Subject"),
|
||||
format!(
|
||||
"[{tag}] subscription removal to `{}` successful",
|
||||
&title,
|
||||
tag = &conf.tag
|
||||
),
|
||||
);
|
||||
reply.set_body(templates::change_subscription_success(
|
||||
title, p, issue_id, false, &conf,
|
||||
));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
Err(e) => {
|
||||
error!("unsubscribe error: {}", e.to_string());
|
||||
reply.headers_mut().insert(
|
||||
HeaderName::new_unchecked("Subject"),
|
||||
format!("[{tag}] could not unsubscribe", tag = &conf.tag,),
|
||||
);
|
||||
reply.set_body(templates::change_subscription_failure(false, &conf));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
},
|
||||
"subscribe" => match api::change_subscription(&conn, p, true) {
|
||||
Ok((title, issue_id, _)) => {
|
||||
reply.headers_mut().insert(
|
||||
HeaderName::new_unchecked("Subject"),
|
||||
format!(
|
||||
"[{tag}] subscription to `{}` successful",
|
||||
&title,
|
||||
tag = &conf.tag
|
||||
),
|
||||
);
|
||||
reply.set_body(templates::change_subscription_success(
|
||||
title, p, issue_id, true, &conf,
|
||||
));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
Err(e) => {
|
||||
error!("subscribe error: {}", e.to_string());
|
||||
reply.headers_mut().insert(
|
||||
HeaderName::new_unchecked("Subject"),
|
||||
format!("[{tag}] could not subscribe", tag = &conf.tag,),
|
||||
);
|
||||
reply.set_body(templates::change_subscription_failure(true, &conf));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
},
|
||||
|
||||
other => {
|
||||
reply.headers_mut().insert(
|
||||
HeaderName::new_unchecked("Subject"),
|
||||
format!("[{tag}] invalid action: `{}`", &other, tag = &conf.tag),
|
||||
);
|
||||
reply.set_body(templates::invalid_request(&conf));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
}
|
||||
}
|
||||
other => {
|
||||
reply.headers_mut().insert(
|
||||
HeaderName::new_unchecked("Subject"),
|
||||
format!("[{tag}] invalid request", tag = &conf.tag),
|
||||
);
|
||||
reply.set_body(templates::invalid_request(&conf));
|
||||
send_mail(reply, &conf);
|
||||
error!("invalid request: {:?}", other);
|
||||
}
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn main() -> Result<()> {
|
||||
let mut file = std::fs::File::open("./config.toml")?;
|
||||
let args = std::env::args().skip(1).collect::<Vec<String>>();
|
||||
let perform_cron: bool;
|
||||
|
@ -78,7 +315,17 @@ fn main() -> std::result::Result<(), std::io::Error> {
|
|||
|
||||
let mut contents = String::new();
|
||||
file.read_to_string(&mut contents)?;
|
||||
let conf: Config = toml::from_str(&contents).unwrap();
|
||||
let conf: Configuration = toml::from_str(&contents).unwrap();
|
||||
CombinedLogger::init(vec![
|
||||
TermLogger::new(LevelFilter::Error, Config::default(), TerminalMode::Mixed),
|
||||
WriteLogger::new(
|
||||
LevelFilter::Trace,
|
||||
Config::default(),
|
||||
File::create(&conf.log_file).unwrap(),
|
||||
),
|
||||
])
|
||||
.unwrap();
|
||||
|
||||
/* - read mail from stdin
|
||||
* - decide which case this mail falls to
|
||||
* a) error/junk
|
||||
|
@ -112,205 +359,17 @@ fn main() -> std::result::Result<(), std::io::Error> {
|
|||
.unwrap();
|
||||
|
||||
if perform_cron {
|
||||
cron::check(conn, conf);
|
||||
info!("Performing cron duties.");
|
||||
if let Err(err) = cron::check(conn, conf) {
|
||||
error!("Encountered an error: {}", &err);
|
||||
return Err(err);
|
||||
}
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
let mut new_message_raw = String::new();
|
||||
stdin().lock().read_to_string(&mut new_message_raw).unwrap();
|
||||
|
||||
let envelope = Envelope::from_bytes(new_message_raw.as_bytes(), None);
|
||||
if let Ok(envelope) = envelope {
|
||||
let mut reply = melib::Draft::new_reply(&envelope, &[]);
|
||||
reply.headers_mut().insert(
|
||||
"From".to_string(),
|
||||
format!(
|
||||
"{local_part}@{domain}",
|
||||
local_part = &conf.local_part,
|
||||
domain = &conf.domain
|
||||
),
|
||||
);
|
||||
|
||||
let tags: Vec<String> = envelope.to()[0].get_tags('+');
|
||||
match tags.as_slice() {
|
||||
s if s.is_empty() || s == &["anonymous"] => {
|
||||
/* Assign new issue */
|
||||
let subject = envelope.subject().to_string();
|
||||
let body = envelope.body_bytes(new_message_raw.as_bytes()).text();
|
||||
let from = envelope.from()[0].clone();
|
||||
let mut reply = melib::Draft::new_reply(&envelope, &[]);
|
||||
let anonymous = !tags.is_empty();
|
||||
reply.headers_mut().insert(
|
||||
"From".to_string(),
|
||||
format!(
|
||||
"{local_part}@{domain}",
|
||||
local_part = &conf.local_part,
|
||||
domain = &conf.domain
|
||||
),
|
||||
);
|
||||
match api::new_issue(&conn, subject.clone(), body, anonymous, from, &conf) {
|
||||
Ok((password, issue_id)) => {
|
||||
reply.headers_mut().insert(
|
||||
"Subject".to_string(),
|
||||
format!(
|
||||
"[{tag}] Issue `{}` successfully created",
|
||||
&subject,
|
||||
tag = &conf.tag
|
||||
),
|
||||
);
|
||||
reply.set_body(templates::new_issue_success(
|
||||
subject, password, issue_id, &conf,
|
||||
));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
Err(e) => {
|
||||
reply.headers_mut().insert(
|
||||
"Subject".to_string(),
|
||||
format!(
|
||||
"[{tag}] Issue `{}` could not be created",
|
||||
&subject,
|
||||
tag = &conf.tag
|
||||
),
|
||||
);
|
||||
reply.set_body(templates::new_issue_failure(e, &conf));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
}
|
||||
}
|
||||
&[ref p, ref cmd]
|
||||
if Password::parse_str(p).is_ok() && PASSWORD_COMMANDS.contains(&cmd.as_str()) =>
|
||||
{
|
||||
let p = Password::parse_str(p).unwrap();
|
||||
match cmd.as_str() {
|
||||
"reply" => {
|
||||
let body = envelope.body_bytes(new_message_raw.as_bytes()).text();
|
||||
let from = envelope.from()[0].clone();
|
||||
match api::new_reply(&conn, body, p, from, &conf) {
|
||||
Ok((title, issue_id, is_subscribed)) => {
|
||||
reply.headers_mut().insert(
|
||||
"Subject".to_string(),
|
||||
format!(
|
||||
"[{tag}] Your reply on issue `{}` has been posted",
|
||||
&title,
|
||||
tag = &conf.tag,
|
||||
),
|
||||
);
|
||||
reply.set_body(templates::new_reply_success(
|
||||
title,
|
||||
p,
|
||||
issue_id,
|
||||
is_subscribed,
|
||||
&conf,
|
||||
));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
Err(e) => {
|
||||
reply.headers_mut().insert(
|
||||
"Subject".to_string(),
|
||||
format!(
|
||||
"[{tag}] Your reply could not be created",
|
||||
tag = &conf.tag,
|
||||
),
|
||||
);
|
||||
reply.set_body(templates::new_reply_failure(e, &conf));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
}
|
||||
}
|
||||
"close" => match api::close(&conn, p, &conf) {
|
||||
Ok((title, issue_id, _)) => {
|
||||
reply.headers_mut().insert(
|
||||
"Subject".to_string(),
|
||||
format!(
|
||||
"[{tag}] issue `{}` has been closed",
|
||||
&title,
|
||||
tag = &conf.tag
|
||||
),
|
||||
);
|
||||
reply.set_body(templates::close_success(title, issue_id, &conf));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
Err(e) => {
|
||||
reply.headers_mut().insert(
|
||||
"Subject".to_string(),
|
||||
format!("[{tag}] issue could not be closed", tag = &conf.tag,),
|
||||
);
|
||||
reply.set_body(templates::close_failure(e, &conf));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
},
|
||||
"unsubscribe" => match api::change_subscription(&conn, p, false) {
|
||||
Ok((title, issue_id, _)) => {
|
||||
reply.headers_mut().insert(
|
||||
"Subject".to_string(),
|
||||
format!(
|
||||
"[{tag}] subscription removal to `{}` successful",
|
||||
&title,
|
||||
tag = &conf.tag
|
||||
),
|
||||
);
|
||||
reply.set_body(templates::change_subscription_success(
|
||||
title, p, issue_id, false, &conf,
|
||||
));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
Err(e) => {
|
||||
eprintln!("error: {}", e.to_string());
|
||||
reply.headers_mut().insert(
|
||||
"Subject".to_string(),
|
||||
format!("[{tag}] could not unsubscribe", tag = &conf.tag,),
|
||||
);
|
||||
reply.set_body(templates::change_subscription_failure(false, &conf));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
},
|
||||
"subscribe" => match api::change_subscription(&conn, p, true) {
|
||||
Ok((title, issue_id, _)) => {
|
||||
reply.headers_mut().insert(
|
||||
"Subject".to_string(),
|
||||
format!(
|
||||
"[{tag}] subscription to `{}` successful",
|
||||
&title,
|
||||
tag = &conf.tag
|
||||
),
|
||||
);
|
||||
reply.set_body(templates::change_subscription_success(
|
||||
title, p, issue_id, true, &conf,
|
||||
));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
Err(e) => {
|
||||
eprintln!("error: {}", e.to_string());
|
||||
reply.headers_mut().insert(
|
||||
"Subject".to_string(),
|
||||
format!("[{tag}] could not subscribe", tag = &conf.tag,),
|
||||
);
|
||||
reply.set_body(templates::change_subscription_failure(true, &conf));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
},
|
||||
|
||||
other => {
|
||||
reply.headers_mut().insert(
|
||||
"Subject".to_string(),
|
||||
format!("[{tag}] invalid action: `{}`", &other, tag = &conf.tag),
|
||||
);
|
||||
reply.set_body(templates::invalid_request(&conf));
|
||||
send_mail(reply, &conf);
|
||||
}
|
||||
}
|
||||
}
|
||||
other => {
|
||||
reply.headers_mut().insert(
|
||||
"Subject".to_string(),
|
||||
format!("[{tag}] invalid request", tag = &conf.tag),
|
||||
);
|
||||
reply.set_body(templates::invalid_request(&conf));
|
||||
send_mail(reply, &conf);
|
||||
println!("error: {:?}", other);
|
||||
}
|
||||
}
|
||||
if let Err(err) = run_app(conn, conf) {
|
||||
error!("Encountered an error: {}", &err);
|
||||
return Err(err);
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
|
|
@ -1,8 +1,24 @@
|
|||
/* This file is part of issue-bot.
|
||||
*
|
||||
* issue-bot is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* issue-bot is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with issue-bot. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
use super::*;
|
||||
|
||||
static BASE_ISSUE_URL: &'static str = "{base_url}/{repo}/issues";
|
||||
|
||||
pub fn new_issue_failure(e: IssueError, conf: &Config) -> String {
|
||||
pub fn new_issue_failure(e: Error, conf: &Configuration) -> String {
|
||||
format!("Hello,
|
||||
|
||||
Unfortunately we were not able to create your issue. The reason was: `{}`. Please contact the repository's owners for assistance.
|
||||
|
@ -14,7 +30,7 @@ pub fn new_issue_success(
|
|||
title: String,
|
||||
password: Password,
|
||||
issue_id: i64,
|
||||
conf: &Config,
|
||||
conf: &Configuration,
|
||||
) -> String {
|
||||
format!("Hello,
|
||||
|
||||
|
@ -33,7 +49,7 @@ Please keep this email in order to be able to keep in touch with your issue.
|
|||
This is an automated email from {bot_name} <{local_part}+help@{domain}>", title = title, password = password.to_string(), issue_id = issue_id, url = BASE_ISSUE_URL.replace("{base_url}", &conf.base_url).replace("{repo}", &conf.repo), local_part = &conf.local_part, domain = &conf.domain, bot_name = &conf.bot_name)
|
||||
}
|
||||
|
||||
pub fn new_reply_failure(e: IssueError, conf: &Config) -> String {
|
||||
pub fn new_reply_failure(e: Error, conf: &Configuration) -> String {
|
||||
format!("Hello,
|
||||
|
||||
Unfortunately we were not able to post your reply. The reason was: `{}`. Please contact the repository's owners for assistance.
|
||||
|
@ -46,7 +62,7 @@ pub fn new_reply_success(
|
|||
password: Password,
|
||||
issue_id: i64,
|
||||
is_subscribed: bool,
|
||||
conf: &Config,
|
||||
conf: &Configuration,
|
||||
) -> String {
|
||||
if is_subscribed {
|
||||
format!("Hello,
|
||||
|
@ -83,7 +99,7 @@ This is an automated email from {bot_name} <{local_part}+help@{domain}>", title
|
|||
}
|
||||
}
|
||||
|
||||
pub fn close_success(title: String, issue_id: i64, conf: &Config) -> String {
|
||||
pub fn close_success(title: String, issue_id: i64, conf: &Configuration) -> String {
|
||||
format!(
|
||||
"Hello,
|
||||
|
||||
|
@ -103,7 +119,7 @@ This is an automated email from {bot_name} <{local_part}+help@{domain}>",
|
|||
)
|
||||
}
|
||||
|
||||
pub fn close_failure(e: IssueError, conf: &Config) -> String {
|
||||
pub fn close_failure(e: Error, conf: &Configuration) -> String {
|
||||
format!("Hello,
|
||||
|
||||
Unfortunately we were not able to close this issue. The reason was: `{}`. Please contact the repository's owners for assistance.
|
||||
|
@ -111,7 +127,7 @@ Unfortunately we were not able to close this issue. The reason was: `{}`. Please
|
|||
This is an automated email from {bot_name} <{local_part}+help@{domain}>", e.to_string(), local_part = &conf.local_part, domain = &conf.domain, bot_name = &conf.bot_name)
|
||||
}
|
||||
|
||||
pub fn invalid_request(conf: &Config) -> String {
|
||||
pub fn invalid_request(conf: &Configuration) -> String {
|
||||
format!(
|
||||
"Hello,
|
||||
|
||||
|
@ -138,7 +154,7 @@ pub fn change_subscription_success(
|
|||
password: Password,
|
||||
issue_id: i64,
|
||||
is_subscribed: bool,
|
||||
conf: &Config,
|
||||
conf: &Configuration,
|
||||
) -> String {
|
||||
format!("Hello,
|
||||
|
||||
|
@ -157,7 +173,7 @@ Please keep this email in order to be able to keep in touch with your issue.
|
|||
This is an automated email from {bot_name} <{local_part}+help@{domain}>", title = title, password = password.to_string(), issue_id = issue_id, url = BASE_ISSUE_URL.replace("{base_url}", &conf.base_url).replace("{repo}", &conf.repo), local_part = &conf.local_part, domain = &conf.domain, bot_name = &conf.bot_name, not = if is_subscribed { "" }else {"not "}, un = if is_subscribed { "un" } else { "" } )
|
||||
}
|
||||
|
||||
pub fn change_subscription_failure(is_subscribed: bool, conf: &Config) -> String {
|
||||
pub fn change_subscription_failure(is_subscribed: bool, conf: &Configuration) -> String {
|
||||
format!(
|
||||
"Hello,
|
||||
|
||||
|
@ -171,7 +187,7 @@ This is an automated email from {bot_name} <{local_part}+help@{domain}>",
|
|||
)
|
||||
}
|
||||
|
||||
pub fn reply_update(issue: &Issue, conf: &Config, comments: Vec<String>) -> String {
|
||||
pub fn reply_update(issue: &Issue, conf: &Configuration, comments: Vec<String>) -> String {
|
||||
assert!(comments.len() > 0);
|
||||
format!(
|
||||
"Hello,
|
||||
|
|
Loading…
Reference in a new issue