ghtool 0.1.0

A command-line tool for interacting with Github API with some specialized features oriented around Checks
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

use eyre::Result;
use http::Uri;
use hyper::body;
use octocrab::models::pulls::PullRequest;
use tracing::info;

use crate::{cache, git::Repository};

pub mod graphql;
mod pull_request_status_checks;

pub use pull_request_status_checks::*;

pub async fn get_pr_for_branch(repo: &Repository, branch: &str) -> Result<PullRequest> {
    info!(?branch, ?repo, "getting pr for branch");

    let client = octocrab::instance();
    let page = client
        .pulls(&repo.owner, &repo.name)
        .list()
        .state(octocrab::params::State::Open)
        .head(&format!("{}:{}", repo.owner, branch))
        .send()
        .await?;
    let pr = page
        .items
        .into_iter()
        .next()
        .ok_or_else(|| eyre::eyre!("No PR found for branch {}", branch))?;
    Ok(pr)
}

pub async fn get_pr_for_branch_memoized(repo: &Repository, branch: &str) -> Result<PullRequest> {
    let key = format!("pr_for_branch_{}_{}", repo, branch);
    cache::memoize(key, || get_pr_for_branch(repo, branch)).await
}

pub async fn get_job_logs(repo: &Repository, check_run: &CheckRun) -> Result<hyper::body::Bytes> {
    info!(?check_run, "getting job logs");

    let client = octocrab::instance();
    let route = format!(
        "/repos/{owner}/{repo}/actions/jobs/{job_id}/logs",
        owner = repo.owner,
        repo = repo.name,
        job_id = check_run.id,
    );
    let uri = Uri::builder().path_and_query(route).build()?;
    let data_response = client
        .follow_location_to_data(client._get(uri).await?)
        .await?;
    let body = data_response.into_body();
    body::to_bytes(body).await.map_err(Into::into)
}