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
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
use crate::{GithubClient, SortDirection, Team, User};
use anyhow::{bail, Context, Result};
use async_trait::async_trait;
use jacklog::{debug, trace};
use serde::Deserialize;

#[derive(Debug, Eq, PartialEq, Deserialize)]
pub struct PullRequest {
    pub url: String,
    pub id: u64,
    pub node_id: String,
    pub html_url: String,
    pub diff_url: String,
    pub patch_url: String,
    pub number: u64,
    pub state: String,
    pub locked: bool,
    pub title: String,
    pub user: User,
    pub body: String,
    pub labels: Vec<Label>,
    pub created_at: String,
    pub updated_at: String,
    pub closed_at: Option<String>,
    pub assignee: Option<User>,
    pub assignees: Vec<User>,
    pub requested_reviewers: Vec<User>,
    pub requested_teams: Vec<Team>,
    pub author_association: String,
    pub draft: bool,
}

#[derive(Debug, Deserialize)]
pub enum PullRequestState {
    Open,
    Closed,
    All,
}

#[derive(Debug, PartialEq, Eq, Deserialize)]
pub struct Label {
    pub id: u64,
    pub node_id: String,
    pub url: String,
    pub name: String,
    pub description: Option<String>,
    pub color: String,
    pub default: bool,
}

#[derive(Debug, PartialEq, Eq, Deserialize)]
pub struct Review {
    pub id: u64,
    pub node_id: String,
    pub user: User,
    pub body: String,
    pub state: String,
    pub submitted_at: String,
    pub commit_id: String,
    pub author_association: String,
}

#[derive(Debug, Deserialize)]
pub enum ListPullRequestsSort {
    Created,
    Updated,
    Popularity,
    #[serde(rename = "long-running")]
    LongRunning,
}

#[derive(Default, Debug)]
pub struct ListPullRequestsRequest {
    pub owner: String,
    pub repo: String,
    pub state: Option<PullRequestState>,
    pub head: Option<String>,
    pub base: Option<String>,
    pub sort: Option<ListPullRequestsSort>,
    pub direction: Option<SortDirection>,
    pub per_page: Option<usize>, // max 100
    pub page: Option<usize>,
}

#[derive(Debug)]
pub struct ListPullRequestsResponse {
    pub pull_requests: Vec<PullRequest>,
}

#[derive(Default, Debug)]
pub struct ListReviewsForPullRequestRequest {
    pub owner: String,
    pub repo: String,
    pub pull_number: u64,
    pub per_page: Option<usize>, // max 100
    pub page: Option<usize>,
}

#[derive(Debug)]
pub struct ListReviewsForPullRequestResponse {
    pub reviews: Vec<Review>,
}

#[async_trait]
pub trait Pulls {
    async fn list_pull_requests(
        &self,
        input: ListPullRequestsRequest,
    ) -> Result<ListPullRequestsResponse>;

    async fn list_reviews_for_pull_request(
        &self,
        input: ListReviewsForPullRequestRequest,
    ) -> Result<ListReviewsForPullRequestResponse>;
}

#[async_trait]
impl Pulls for GithubClient {
    async fn list_pull_requests(
        &self,
        input: ListPullRequestsRequest,
    ) -> Result<ListPullRequestsResponse> {
        // Make the request.
        let res = self
            .client()
            .get(&format!(
                "https://api.github.com/repos/{}/{}/pulls",
                input.owner, input.repo
            ))
            .send()
            .await?;

        // Check the response code.
        if !res.status().is_success() {
            bail!("{}", res.status().canonical_reason().unwrap_or(&"unknown"));
        }

        // Get the response text so we can dump it for debugging.
        let res = res.text().await?;
        trace!("{}", &res);
        std::fs::write("/tmp/res.json", &res)?;

        //let res: Response = res.json().await?;
        let pull_requests =
            serde_json::from_str(&res).context("error parsing results from list_pull_requests")?;
        debug!("{:?}", &pull_requests);

        Ok(ListPullRequestsResponse { pull_requests })
    }

    async fn list_reviews_for_pull_request(
        &self,
        input: ListReviewsForPullRequestRequest,
    ) -> Result<ListReviewsForPullRequestResponse> {
        // Make the request.
        let res = self
            .client()
            .get(&format!(
                "https://api.github.com/repos/{}/{}/pulls/{}/reviews",
                input.owner, input.repo, input.pull_number
            ))
            .send()
            .await?;

        // Check the response code.
        if !res.status().is_success() {
            bail!("{}", res.status().canonical_reason().unwrap_or(&"unknown"));
        }

        // Get the response text so we can dump it for debugging.
        let res = res.text().await?;
        trace!("{}", &res);
        std::fs::write("/tmp/res.json", &res)?;

        //let res: Response = res.json().await?;
        let reviews = serde_json::from_str(&res)
            .context("error parsing reviews from list_reviews_for_pull_request")?;
        debug!("{:?}", &reviews);

        Ok(ListReviewsForPullRequestResponse { reviews })
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::{client, org, repo};
    use std::env;

    #[cfg(test)]
    fn team() -> String {
        env::var("GITHUB_TEST_TEAM_SLUG").expect("error reading GITHUB_TEST_TEAM_SLUG")
    }

    #[tokio::test]
    async fn test_list_pull_requests() {
        let client = client();
        let res = client
            .list_pull_requests(ListPullRequestsRequest {
                owner: org(),
                repo: repo(),
                ..Default::default()
            })
            .await
            .unwrap();

        assert_eq!(res.pull_requests, vec![]);
    }
}