From e8234b335363842c6e330e2725dd077c7f6a5815 Mon Sep 17 00:00:00 2001 From: Cyborus Date: Wed, 10 Jul 2024 13:48:28 -0400 Subject: [PATCH] fix: read detected pr from correct repo --- src/prs.rs | 87 +++++++++++++++++++++++++++++++---------------------- src/repo.rs | 5 ++- 2 files changed, 55 insertions(+), 37 deletions(-) diff --git a/src/prs.rs b/src/prs.rs index 487f320..67de9cb 100644 --- a/src/prs.rs +++ b/src/prs.rs @@ -288,11 +288,11 @@ impl PrCommand { match command.unwrap_or(ViewCommand::Body) { ViewCommand::Body => view_pr(&repo, &api, id).await?, ViewCommand::Comment { idx } => { - let id = try_get_pr_number(&repo, &api, id).await?; + let (repo, id) = try_get_pr_number(&repo, &api, id).await?; crate::issues::view_comment(&repo, &api, id, idx).await? } ViewCommand::Comments => { - let id = try_get_pr_number(&repo, &api, id).await?; + let (repo, id) = try_get_pr_number(&repo, &api, id).await?; crate::issues::view_comments(&repo, &api, id).await? } ViewCommand::Labels => view_pr_labels(&repo, &api, id).await?, @@ -318,16 +318,16 @@ impl PrCommand { let pr = pr.map(|pr| pr.number); match command { EditCommand::Title { new_title } => { - let pr = try_get_pr_number(&repo, &api, pr).await?; - crate::issues::edit_title(&repo, &api, pr, new_title).await? + let (repo, id) = try_get_pr_number(&repo, &api, pr).await?; + crate::issues::edit_title(&repo, &api, id, new_title).await? } EditCommand::Body { new_body } => { - let pr = try_get_pr_number(&repo, &api, pr).await?; - crate::issues::edit_body(&repo, &api, pr, new_body).await? + let (repo, id) = try_get_pr_number(&repo, &api, pr).await?; + crate::issues::edit_body(&repo, &api, id, new_body).await? } EditCommand::Comment { idx, new_body } => { - let pr = try_get_pr_number(&repo, &api, pr).await?; - crate::issues::edit_comment(&repo, &api, pr, idx, new_body).await? + let (repo, id) = try_get_pr_number(&repo, &api, pr).await?; + crate::issues::edit_comment(&repo, &api, id, idx, new_body).await? } EditCommand::Labels { add, rm } => { edit_pr_labels(&repo, &api, pr, add, rm).await? @@ -335,16 +335,16 @@ impl PrCommand { } } Close { pr, with_msg } => { - let pr = try_get_pr_number(&repo, &api, pr.map(|pr| pr.number)).await?; + let (repo, pr) = try_get_pr_number(&repo, &api, pr.map(|pr| pr.number)).await?; crate::issues::close_issue(&repo, &api, pr, with_msg).await? } Checkout { pr, branch_name } => checkout_pr(&repo, &api, pr, branch_name).await?, Browse { id } => { - let id = try_get_pr_number(&repo, &api, id.map(|id| id.number)).await?; + let (repo, id) = try_get_pr_number(&repo, &api, id.map(|pr| pr.number)).await?; browse_pr(&repo, &api, id).await? } Comment { pr, body } => { - let pr = try_get_pr_number(&repo, &api, pr.map(|pr| pr.number)).await?; + let (repo, pr) = try_get_pr_number(&repo, &api, pr.map(|pr| pr.number)).await?; crate::issues::add_comment(&repo, &api, pr, body).await? } } @@ -414,6 +414,7 @@ pub async fn view_pr(repo: &RepoName, api: &Forgejo, id: Option) -> eyre::R } = crate::special_render(); let pr = try_get_pr(repo, api, id).await?; let id = pr.number.ok_or_eyre("pr does not have number")?; + let repo = repo_name_from_pr(&pr)?; let mut additions = 0; let mut deletions = 0; @@ -588,6 +589,7 @@ fn darken(r: u8, g: u8, b: u8) -> (u8, u8, u8) { async fn view_pr_status(repo: &RepoName, api: &Forgejo, id: Option) -> eyre::Result<()> { let pr = try_get_pr(repo, api, id).await?; + let repo = repo_name_from_pr(&pr)?; let SpecialRender { bright_magenta, @@ -708,7 +710,9 @@ async fn edit_pr_labels( add: Vec, rm: Vec, ) -> eyre::Result<()> { - let pr_number = try_get_pr_number(repo, api, pr).await?; + let pr = try_get_pr(repo, api, pr).await?; + let pr_number = pr.number.ok_or_eyre("pr does not have number")?; + let repo = repo_name_from_pr(&pr)?; let query = forgejo_api::structs::IssueListLabelsQuery { limit: Some(u32::MAX), @@ -859,6 +863,7 @@ async fn merge_pr( let repo_info = api.repo_get(repo.owner(), repo.name()).await?; let pr_info = try_get_pr(repo, api, pr).await?; + let repo = repo_name_from_pr(&pr_info)?; let pr_html_url = pr_info .html_url .as_ref() @@ -945,17 +950,7 @@ async fn checkout_pr( PrNumber::This(_) => api.repo_get(repo.owner(), repo.name()).await?, }; - let repo_owner = remote_repo - .owner - .as_ref() - .ok_or_eyre("repo does not have owner")? - .login - .as_deref() - .ok_or_eyre("owner does not have login")?; - let repo_name = remote_repo - .name - .as_ref() - .ok_or_eyre("repo does not have name")?; + let (repo_owner, repo_name) = repo_name_from_repo(&remote_repo)?; let pull_data = api .repo_get_pull_request(repo_owner, repo_name, pr.number()) @@ -1082,13 +1077,15 @@ async fn view_diff( patch: bool, editor: bool, ) -> eyre::Result<()> { - let pr = try_get_pr_number(repo, api, pr).await?; + let pr = try_get_pr(repo, api, pr).await?; + let pr_number = pr.number.ok_or_eyre("pr does not have number")?; + let repo = repo_name_from_pr(&pr)?; let diff_type = if patch { "patch" } else { "diff" }; let diff = api .repo_download_pull_diff_or_patch( repo.owner(), repo.name(), - pr, + pr_number, diff_type, forgejo_api::structs::RepoDownloadPullDiffOrPatchQuery::default(), ) @@ -1106,7 +1103,9 @@ async fn view_diff( } async fn view_pr_files(repo: &RepoName, api: &Forgejo, pr: Option) -> eyre::Result<()> { - let pr = try_get_pr_number(repo, api, pr).await?; + let pr = try_get_pr(repo, api, pr).await?; + let pr_number = pr.number.ok_or_eyre("pr does not have number")?; + let repo = repo_name_from_pr(&pr)?; let crate::SpecialRender { bright_red, bright_green, @@ -1119,7 +1118,7 @@ async fn view_pr_files(repo: &RepoName, api: &Forgejo, pr: Option) -> eyre: ..Default::default() }; let (_, files) = api - .repo_get_pull_request_files(repo.owner(), repo.name(), pr, query) + .repo_get_pull_request_files(repo.owner(), repo.name(), pr_number, query) .await?; let max_additions = files .iter() @@ -1150,14 +1149,16 @@ async fn view_pr_commits( pr: Option, oneline: bool, ) -> eyre::Result<()> { - let pr = try_get_pr_number(repo, api, pr).await?; + let pr = try_get_pr(repo, api, pr).await?; + let pr_number = pr.number.ok_or_eyre("pr does not have number")?; + let repo = repo_name_from_pr(&pr)?; let query = RepoGetPullRequestCommitsQuery { limit: Some(u32::MAX), files: Some(false), ..Default::default() }; let (_headers, commits) = api - .repo_get_pull_request_commits(repo.owner(), repo.name(), pr, query) + .repo_get_pull_request_commits(repo.owner(), repo.name(), pr_number, query) .await?; let max_additions = commits @@ -1251,14 +1252,17 @@ async fn try_get_pr_number( repo: &RepoName, api: &Forgejo, number: Option, -) -> eyre::Result { +) -> eyre::Result<(RepoName, u64)> { let pr = match number { - Some(number) => number, - None => guess_pr(repo, api) - .await - .wrap_err("could not guess pull request number, please specify")? - .number - .ok_or_eyre("pr does not have number")?, + Some(number) => (repo.clone(), number), + None => { + let pr = guess_pr(repo, api) + .await + .wrap_err("could not guess pull request number, please specify")?; + let number = pr.number.ok_or_eyre("pr does not have number")?; + let repo = repo_name_from_pr(&pr)?; + (repo, number) + } }; Ok(pr) } @@ -1421,6 +1425,17 @@ fn repo_name_from_repo(repo: &forgejo_api::structs::Repository) -> eyre::Result< Ok((owner, name)) } +fn repo_name_from_pr(pr: &forgejo_api::structs::PullRequest) -> eyre::Result { + let base_branch = pr.base.as_ref().ok_or_eyre("pr does not have base")?; + let repo = base_branch + .repo + .as_ref() + .ok_or_eyre("branch does not have repo")?; + let (owner, name) = repo_name_from_repo(repo)?; + let repo_name = RepoName::new(owner.to_owned(), name.to_owned()); + Ok(repo_name) +} + //async fn guess_pr( // repo: &RepoName, // api: &Forgejo, diff --git a/src/repo.rs b/src/repo.rs index 0e12f01..4a418c9 100644 --- a/src/repo.rs +++ b/src/repo.rs @@ -213,13 +213,16 @@ fn url_strip_repo_name(mut url: Url) -> eyre::Result<(Url, RepoName)> { Ok((url, RepoName { owner, name })) } -#[derive(Debug)] +#[derive(Clone, Debug)] pub struct RepoName { owner: String, name: String, } impl RepoName { + pub fn new(owner: String, name: String) -> Self { + Self { owner, name } + } pub fn owner(&self) -> &str { &self.owner }