-
Notifications
You must be signed in to change notification settings - Fork 5
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
4a1fe6d
commit e16c6e4
Showing
3 changed files
with
170 additions
and
174 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,161 @@ | ||
| #![allow(dead_code)] | ||
|
|
||
| use anyhow::Result; | ||
| use octocrab::{ | ||
| issues::IssueHandler, | ||
| models::{ | ||
| issues::Issue, | ||
| pulls::{self, PullRequest}, | ||
| IssueState, | ||
| }, | ||
| pulls::PullRequestHandler, | ||
| repos::RepoHandler, | ||
| GitHubError, Octocrab, | ||
| }; | ||
| use serde_json::json; | ||
| use std::sync::Arc; | ||
| use tokio::sync::OnceCell; | ||
|
|
||
| pub struct Repo { | ||
| user: String, | ||
| name: String, | ||
| gh: Arc<Octocrab>, | ||
| prs: OnceCell<Vec<PullRequest>>, | ||
| issues: OnceCell<Vec<Issue>>, | ||
| } | ||
|
|
||
| impl Repo { | ||
| pub fn new(user: &str, name: &str) -> Self { | ||
| Repo { | ||
| user: user.to_string(), | ||
| name: name.to_string(), | ||
| gh: octocrab::instance(), | ||
| prs: OnceCell::new(), | ||
| issues: OnceCell::new(), | ||
| } | ||
| } | ||
|
|
||
| pub async fn exists(&self) -> Result<bool> { | ||
| match self.gh.repos(&self.user, &self.name).get().await { | ||
| Ok(_) => Ok(true), | ||
| Err(octocrab::Error::GitHub { | ||
| source: | ||
| GitHubError { | ||
| status_code: http::StatusCode::NOT_FOUND, | ||
| .. | ||
| }, | ||
| .. | ||
| }) => Ok(false), | ||
| Err(e) => Err(e.into()), | ||
| } | ||
| } | ||
|
|
||
| pub async fn fork_from(&self, base: &Repo) -> Result<()> { | ||
| base.repo_handler().create_fork().send().await?; | ||
|
|
||
| let repo_route = format!("/repos/{}/{}", self.user, self.name); | ||
| let _response = self | ||
| .gh | ||
| .patch::<serde_json::Value, _, _>( | ||
| &repo_route, | ||
| Some(&json!({ | ||
| "has_issues": true | ||
| })), | ||
| ) | ||
| .await?; | ||
|
|
||
| // TODO: any way to enable actions? | ||
|
|
||
| Ok(()) | ||
| } | ||
|
|
||
| pub fn repo_handler(&self) -> RepoHandler { | ||
| self.gh.repos(&self.user, &self.name) | ||
| } | ||
|
|
||
| pub fn pr_handler(&self) -> PullRequestHandler { | ||
| self.gh.pulls(&self.user, &self.name) | ||
| } | ||
|
|
||
| pub async fn prs(&self) -> &[PullRequest] { | ||
| self | ||
| .prs | ||
| .get_or_init(|| async { | ||
| let pages = self.pr_handler().list().send().await.unwrap(); | ||
| pages.into_iter().collect::<Vec<_>>() | ||
| }) | ||
| .await | ||
| } | ||
|
|
||
| pub async fn pr(&self, ref_field: &str) -> Option<&PullRequest> { | ||
| let prs = self.prs().await; | ||
| prs | ||
| .iter() | ||
| .find(|pr| !matches!(pr.state, Some(IssueState::Closed)) && pr.head.ref_field == ref_field) | ||
| } | ||
|
|
||
| pub fn issue_handler(&self) -> IssueHandler { | ||
| self.gh.issues(&self.user, &self.name) | ||
| } | ||
|
|
||
| pub async fn issues(&self) -> &[Issue] { | ||
| self | ||
| .issues | ||
| .get_or_init(|| async { | ||
| let pages = self.issue_handler().list().send().await.unwrap(); | ||
| pages.into_iter().collect::<Vec<_>>() | ||
| }) | ||
| .await | ||
| } | ||
|
|
||
| pub async fn issue(&self, title: &str) -> Option<&Issue> { | ||
| let issues = self.issues().await; | ||
| issues | ||
| .iter() | ||
| .find(|issue| !matches!(issue.state, IssueState::Closed) && issue.title == title) | ||
| } | ||
|
|
||
| pub async fn copy_pr(&self, base: &Repo, base_pr: &PullRequest) -> Result<()> { | ||
| let pulls = self.pr_handler(); | ||
| let request = pulls | ||
| .create( | ||
| base_pr.title.as_ref().unwrap(), | ||
| &base_pr.head.ref_field, | ||
| &base_pr.base.ref_field, | ||
| ) | ||
| .body(base_pr.body.as_ref().unwrap()); | ||
| let self_pr = request.send().await?; | ||
|
|
||
| let comment_pages = base | ||
| .pr_handler() | ||
| .list_comments(Some(base_pr.number)) | ||
| .send() | ||
| .await?; | ||
| let comments = comment_pages.into_iter().collect::<Vec<_>>(); | ||
|
|
||
| for comment in &comments { | ||
| self.copy_pr_comment(self_pr.number, comment).await?; | ||
| } | ||
|
|
||
| Ok(()) | ||
| } | ||
|
|
||
| pub async fn copy_pr_comment(&self, pr: u64, comment: &pulls::Comment) -> Result<()> { | ||
| let route = format!("/repos/{}/{}/pulls/{pr}/comments", self.user, self.name); | ||
| let _response = self | ||
| .gh | ||
| .post::<_, serde_json::Value>(route, Some(&comment)) | ||
| .await?; | ||
| Ok(()) | ||
| } | ||
|
|
||
| pub async fn copy_issue(&self, issue: &Issue) -> Result<()> { | ||
| self | ||
| .issue_handler() | ||
| .create(&issue.title) | ||
| .body(issue.body.as_ref().unwrap()) | ||
| .send() | ||
| .await?; | ||
| Ok(()) | ||
| } | ||
| } |