triagebot/src/main.rs

#![feature(proc_macro_hygiene, decl_macro)]

#[macro_use]
extern crate rocket;

use reqwest::Client;
use rocket::request;
use rocket::State;
use rocket::{http::Status, Outcome, Request};
use std::env;

mod handlers;
mod registry;

mod github;
mod payload;
mod team;

static BOT_USER_NAME: &str = "rust-highfive";

use github::{Comment, GithubClient, Issue};
use payload::SignedPayload;
use registry::HandleRegistry;

#[derive(PartialEq, Eq, Debug, serde::Deserialize)]
#[serde(rename_all = "lowercase")]
pub enum IssueCommentAction {
    Created,
    Edited,
    Deleted,
}

#[derive(Debug, serde::Deserialize)]
pub struct IssueCommentEvent {
    action: IssueCommentAction,
    issue: Issue,
    comment: Comment,
}

enum Event {
    IssueComment,
    Other,
}

impl<'a, 'r> request::FromRequest<'a, 'r> for Event {
    type Error = String;
    fn from_request(req: &'a Request<'r>) -> request::Outcome<Self, Self::Error> {
        let ev = if let Some(ev) = req.headers().get_one("X-GitHub-Event") {
            ev
        } else {
            return Outcome::Failure((Status::BadRequest, format!("Needs a X-GitHub-Event")));
        };
        let ev = match ev {
            "issue_comment" => Event::IssueComment,
            _ => Event::Other,
        };
        Outcome::Success(ev)
    }
}

#[post("/github-hook", data = "<payload>")]
fn webhook(event: Event, payload: SignedPayload, reg: State<HandleRegistry>) -> Result<(), String> {
    match event {
        Event::IssueComment => {
            let payload = payload
                .deserialize::<IssueCommentEvent>()
                .map_err(|e| format!("IssueCommentEvent failed to deserialize: {:?}", e))?;

            let event = registry::Event::IssueComment(payload);
            reg.handle(&event).map_err(|e| format!("{:?}", e))?;
        }
        // Other events need not be handled
        Event::Other => {}
    }
    Ok(())
}

#[catch(404)]
fn not_found(_: &Request) -> &'static str {
    "Not Found"
}

fn main() {
    env_logger::init();
    dotenv::dotenv().ok();
    let client = Client::new();
    let gh = GithubClient::new(client.clone(), env::var("GITHUB_API_TOKEN").unwrap());
    let mut registry = HandleRegistry::new();
    handlers::register_all(&mut registry, gh.clone());
    rocket::ignite()
        .manage(gh)
        .manage(registry)
        .mount("/", routes![webhook])
        .register(catchers![not_found])
        .launch();
}