DragonOS-Community
/
triagebot
镜像来自 https://github.com/DragonOS-Community/triagebot.git


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132
							use crate::handlers::Context;
use chrono::{Duration, Utc};
use hyper::{Body, Response, StatusCode};
use serde::Serialize;
use serde_json::value::{to_value, Value};
use std::sync::Arc;
use url::Url;

const YELLOW_DAYS: i64 = 7;
const RED_DAYS: i64 = 14;

pub fn index() -> Result<Response<Body>, hyper::Error> {
    Ok(Response::builder()
        .header("Content-Type", "text/html")
        .status(StatusCode::OK)
        .body(Body::from(include_str!("../templates/triage/index.html")))
        .unwrap())
}

pub async fn pulls(
    ctx: Arc<Context>,
    owner: &str,
    repo: &str,
) -> Result<Response<Body>, hyper::Error> {
    let octocrab = &ctx.octocrab;
    let res = octocrab
        .pulls(owner, repo)
        .list()
        .sort(octocrab::params::pulls::Sort::Updated)
        .direction(octocrab::params::Direction::Ascending)
        .per_page(100)
        .send()
        .await;
    let mut page = match res {
        Ok(page) => page,
        Err(_) => {
            return Ok(Response::builder()
                .status(StatusCode::NOT_FOUND)
                .body(Body::from("The repository is not found."))
                .unwrap());
        }
    };
    let mut base_pulls = page.take_items();
    let mut next_page = page.next;
    while let Some(mut page) = octocrab
        .get_page::<octocrab::models::pulls::PullRequest>(&next_page)
        .await
        .unwrap()
    {
        base_pulls.extend(page.take_items());
        next_page = page.next;
    }

    let mut pulls: Vec<Value> = Vec::new();
    for base_pull in base_pulls.into_iter() {
        let assignee = base_pull.assignee.map_or("".to_string(), |v| v.login);
        let updated_at = base_pull
            .updated_at
            .map_or("".to_string(), |v| v.format("%Y-%m-%d").to_string());

        let yellow_line = Utc::now() - Duration::days(YELLOW_DAYS);
        let red_line = Utc::now() - Duration::days(RED_DAYS);
        let need_triage = match base_pull.updated_at {
            Some(updated_at) if updated_at <= red_line => "red".to_string(),
            Some(updated_at) if updated_at <= yellow_line => "yellow".to_string(),
            _ => "green".to_string(),
        };
        let days_from_last_updated_at = if let Some(updated_at) = base_pull.updated_at {
            (Utc::now() - updated_at).num_days()
        } else {
            (Utc::now() - base_pull.created_at.unwrap()).num_days()
        };

        let labels = base_pull.labels.map_or("".to_string(), |labels| {
            labels
                .iter()
                .map(|label| label.name.clone())
                .collect::<Vec<_>>()
                .join(", ")
        });
        let wait_for_author = labels.contains("S-waiting-on-author");
        let wait_for_review = labels.contains("S-waiting-on-review");
        let html_url = base_pull.html_url.unwrap();
        let number = base_pull.number;
        let title = base_pull.title.unwrap();
        let author = base_pull.user.unwrap().login;

        let pull = PullRequest {
            html_url,
            number,
            title,
            assignee,
            updated_at,
            need_triage,
            labels,
            author,
            wait_for_author,
            wait_for_review,
            days_from_last_updated_at,
        };
        pulls.push(to_value(pull).unwrap());
    }

    let mut context = tera::Context::new();
    context.insert("pulls", &pulls);
    context.insert("owner", &owner);
    context.insert("repo", &repo);

    let tera = tera::Tera::new("templates/triage/**/*").unwrap();
    let body = Body::from(tera.render("pulls.html", &context).unwrap());

    Ok(Response::builder()
        .header("Content-Type", "text/html")
        .status(StatusCode::OK)
        .body(body)
        .unwrap())
}

#[derive(Serialize)]
struct PullRequest {
    pub html_url: Url,
    pub number: u64,
    pub title: String,
    pub assignee: String,
    pub updated_at: String,
    pub need_triage: String,
    pub labels: String,
    pub author: String,
    pub wait_for_author: bool,
    pub wait_for_review: bool,
    pub days_from_last_updated_at: i64,
}