1use git2::{FetchOptions, Repository};
82use log::debug;
83
84use crate::{
85 error::{PrError, Result},
86 get_remote_callbacks,
87};
88
89#[derive(Debug, Clone, PartialEq, Eq)]
91pub struct PullRequest {
92 pub number: u32,
94 pub remote: Option<String>,
96}
97
98#[derive(Debug, Clone)]
100pub struct PrMetadata {
101 pub number: u32,
103 pub title: String,
105 pub author: String,
107 pub head_ref: String,
109 pub base_ref: String,
111 pub is_fork: bool,
113 pub fork_owner: Option<String>,
115 pub fork_url: Option<String>,
117}
118
119pub fn parse_pr_reference(input: &str) -> Result<Option<PullRequest>> {
131 if let Some(num_str) = input.strip_prefix('#') {
133 return parse_number(num_str, input).map(|num| {
134 Some(PullRequest {
135 number: num,
136 remote: None,
137 })
138 });
139 }
140
141 if let Some(num_str) = input.strip_prefix("pr#") {
143 return parse_number(num_str, input).map(|num| {
144 Some(PullRequest {
145 number: num,
146 remote: None,
147 })
148 });
149 }
150
151 if let Some(num_str) = input.strip_prefix("pr-") {
153 return parse_number(num_str, input).map(|num| {
154 Some(PullRequest {
155 number: num,
156 remote: None,
157 })
158 });
159 }
160
161 if input.contains("github.com") && input.contains("/pull/") {
163 return parse_github_url(input);
164 }
165
166 if input.contains("/pull/") && input.ends_with("/head") {
168 return parse_remote_ref(input);
169 }
170
171 Ok(None)
173}
174
175fn parse_number(num_str: &str, original_input: &str) -> Result<u32> {
177 num_str.parse::<u32>().map_err(|_| {
178 PrError::InvalidReference {
179 input: original_input.to_string(),
180 }
181 .into()
182 })
183}
184
185fn parse_github_url(url: &str) -> Result<Option<PullRequest>> {
187 let parts: Vec<&str> = url.split('/').collect();
189
190 for (i, &part) in parts.iter().enumerate() {
192 if part == "pull" && i + 1 < parts.len() {
193 let num_str = parts[i + 1];
194 let number = parse_number(num_str, url)?;
195 return Ok(Some(PullRequest {
196 number,
197 remote: None,
198 }));
199 }
200 }
201
202 Err(PrError::InvalidReference {
203 input: url.to_string(),
204 }
205 .into())
206}
207
208fn parse_remote_ref(ref_str: &str) -> Result<Option<PullRequest>> {
210 let parts: Vec<&str> = ref_str.split('/').collect();
212
213 if parts.len() >= 4 && parts[parts.len() - 3] == "pull" && parts[parts.len() - 1] == "head" {
214 let num_str = parts[parts.len() - 2];
215 let number = parse_number(num_str, ref_str)?;
216 return Ok(Some(PullRequest {
217 number,
218 remote: None,
219 }));
220 }
221
222 Err(PrError::InvalidReference {
223 input: ref_str.to_string(),
224 }
225 .into())
226}
227
228pub fn check_gh_available() -> Result<()> {
232 std::process::Command::new("gh")
233 .arg("--version")
234 .output()
235 .map_err(|_| PrError::GhNotInstalled)?;
236 Ok(())
237}
238
239pub fn fetch_pr_metadata(pr_number: u32) -> Result<PrMetadata> {
244 check_gh_available()?;
246
247 let output = std::process::Command::new("gh")
249 .args([
250 "pr",
251 "view",
252 &pr_number.to_string(),
253 "--json",
254 "number,title,author,headRefName,baseRefName,isCrossRepository,headRepository",
255 ])
256 .output()
257 .map_err(|e| PrError::GhFetchFailed {
258 message: e.to_string(),
259 })?;
260
261 if !output.status.success() {
262 let stderr = String::from_utf8_lossy(&output.stderr);
263 return Err(PrError::GhFetchFailed {
264 message: stderr.to_string(),
265 }
266 .into());
267 }
268
269 let json_str = String::from_utf8_lossy(&output.stdout);
271 let json: serde_json::Value =
272 serde_json::from_str(&json_str).map_err(|e| PrError::GhJsonParseFailed {
273 message: e.to_string(),
274 })?;
275
276 let number = json["number"]
278 .as_u64()
279 .ok_or_else(|| PrError::GhJsonParseFailed {
280 message: "Missing 'number' field".to_string(),
281 })? as u32;
282
283 let title = json["title"]
284 .as_str()
285 .ok_or_else(|| PrError::GhJsonParseFailed {
286 message: "Missing 'title' field".to_string(),
287 })?
288 .to_string();
289
290 let author = json["author"]["login"]
291 .as_str()
292 .ok_or_else(|| PrError::GhJsonParseFailed {
293 message: "Missing 'author.login' field".to_string(),
294 })?
295 .to_string();
296
297 let head_ref = json["headRefName"]
298 .as_str()
299 .ok_or_else(|| PrError::GhJsonParseFailed {
300 message: "Missing 'headRefName' field".to_string(),
301 })?
302 .to_string();
303
304 let base_ref = json["baseRefName"]
305 .as_str()
306 .ok_or_else(|| PrError::GhJsonParseFailed {
307 message: "Missing 'baseRefName' field".to_string(),
308 })?
309 .to_string();
310
311 let is_fork = json["isCrossRepository"].as_bool().unwrap_or(false);
312
313 let (fork_owner, fork_url) = if is_fork {
314 let owner = json["headRepository"]["owner"]["login"]
315 .as_str()
316 .ok_or(PrError::MissingForkOwner)?
317 .to_string();
318 let url = json["headRepository"]["url"]
319 .as_str()
320 .map(|s| s.to_string());
321 (Some(owner), url)
322 } else {
323 (None, None)
324 };
325
326 Ok(PrMetadata {
327 number,
328 title,
329 author,
330 head_ref,
331 base_ref,
332 is_fork,
333 fork_owner,
334 fork_url,
335 })
336}
337
338pub fn find_merged_pr(branch: &str) -> Result<Option<u32>> {
350 let output = std::process::Command::new("gh")
351 .args([
352 "pr",
353 "list",
354 "--head",
355 branch,
356 "--state",
357 "merged",
358 "--json",
359 "number,mergedAt",
360 "--limit",
361 "1",
362 ])
363 .output()
364 .map_err(|e| PrError::GhFetchFailed {
365 message: e.to_string(),
366 })?;
367
368 if !output.status.success() {
369 let stderr = String::from_utf8_lossy(&output.stderr);
370 return Err(PrError::GhFetchFailed {
371 message: stderr.to_string(),
372 }
373 .into());
374 }
375
376 let json_str = String::from_utf8_lossy(&output.stdout);
377 Ok(parse_merged_pr(&json_str))
378}
379
380fn parse_merged_pr(json: &str) -> Option<u32> {
387 let json: serde_json::Value = serde_json::from_str(json).ok()?;
388 json.as_array()?
389 .first()?
390 .get("number")?
391 .as_u64()
392 .map(|n| n as u32)
393}
394
395fn sanitize_for_branch_name(s: &str) -> String {
397 let sanitized = s
398 .chars()
399 .map(|c| match c {
400 'a'..='z' | 'A'..='Z' | '0'..='9' | '-' | '_' => c,
401 ' ' | '/' => '-',
402 _ => '-',
403 })
404 .collect::<String>()
405 .to_lowercase();
406
407 let mut result = String::new();
409 let mut last_was_dash = false;
410 for c in sanitized.chars() {
411 if c == '-' {
412 if !last_was_dash {
413 result.push(c);
414 }
415 last_was_dash = true;
416 } else {
417 result.push(c);
418 last_was_dash = false;
419 }
420 }
421
422 result.trim_matches(|c| c == '-' || c == '_').to_string()
423}
424
425pub fn format_pr_name_with_metadata(format: &str, metadata: &PrMetadata) -> String {
430 format
431 .replace("{number}", &metadata.number.to_string())
432 .replace("{title}", &sanitize_for_branch_name(&metadata.title))
433 .replace("{author}", &sanitize_for_branch_name(&metadata.author))
434 .replace("{branch}", &sanitize_for_branch_name(&metadata.head_ref))
435}
436
437pub fn is_pr_reference(input: &str) -> bool {
441 parse_pr_reference(input).ok().flatten().is_some()
442}
443
444pub fn remote_priority(remote: &str) -> usize {
451 match remote {
452 "upstream" => 0,
453 "origin" => 1,
454 _ => 2,
455 }
456}
457
458pub fn preferred_remote_order(repo: &Repository) -> Vec<String> {
461 let Ok(remotes) = repo.remotes() else {
462 return vec![];
463 };
464 let mut all: Vec<String> = remotes
465 .iter()
466 .flatten()
467 .flatten()
468 .map(str::to_string)
469 .collect();
470 all.sort_by_key(|r| remote_priority(r));
471 all
472}
473
474pub fn has_github_remote(repo: &Repository) -> bool {
481 let Ok(remotes) = repo.remotes() else {
482 return false;
483 };
484 remotes.iter().flatten().flatten().any(|name| {
485 repo.find_remote(name)
486 .ok()
487 .and_then(|r| r.url().ok().map(str::to_string))
488 .and_then(|url| crate::ssh_config::extract_host_from_url(&url))
489 .is_some_and(|host| host.eq_ignore_ascii_case("github.com"))
490 })
491}
492
493pub fn detect_pr_remote(repo: &Repository) -> Result<String> {
498 preferred_remote_order(repo)
499 .into_iter()
500 .next()
501 .ok_or_else(|| PrError::NoRemoteConfigured.into())
502}
503
504pub fn setup_fork_remote(repo: &Repository, metadata: &PrMetadata) -> Result<String> {
510 if !metadata.is_fork {
511 return detect_pr_remote(repo);
513 }
514
515 let _fork_owner = metadata
517 .fork_owner
518 .as_ref()
519 .ok_or(PrError::MissingForkOwner)?;
520
521 let fork_url = metadata
522 .fork_url
523 .as_ref()
524 .ok_or(PrError::MissingForkOwner)?;
525
526 let fork_remote_name = format!("pr-{}-fork", metadata.number);
528
529 if repo.find_remote(&fork_remote_name).is_ok() {
530 debug!("Fork remote {} already exists", fork_remote_name);
531 return Ok(fork_remote_name);
532 }
533
534 debug!("Adding fork remote: {} -> {}", fork_remote_name, fork_url);
536 repo.remote(&fork_remote_name, fork_url)
537 .map_err(|e| PrError::FetchFailed {
538 remote: fork_remote_name.clone(),
539 message: format!("Failed to add fork remote: {}", e),
540 })?;
541
542 Ok(fork_remote_name)
543}
544
545pub fn fetch_branch(repo: &Repository, remote_name: &str, branch: &str) -> Result<()> {
552 let branch_ref = format!("refs/remotes/{}/{}", remote_name, branch);
554 if repo.find_reference(&branch_ref).is_ok() {
555 debug!("Branch ref {} already exists", branch_ref);
556 return Ok(());
557 }
558
559 debug!("Fetching branch {} from remote {}", branch, remote_name);
560
561 let refspec = format!(
562 "+refs/heads/{}:refs/remotes/{}/{}",
563 branch, remote_name, branch
564 );
565
566 let remote_url = repo
567 .find_remote(remote_name)
568 .ok()
569 .and_then(|r| r.url().ok().map(str::to_string));
570 let auth = get_remote_callbacks(repo, remote_url.as_deref())?;
571 let mut fetch_options = FetchOptions::new();
572 fetch_options.remote_callbacks(auth.callbacks());
573
574 repo.find_remote(remote_name)?
575 .fetch(
576 &[refspec.as_str()],
577 Some(&mut fetch_options),
578 Some("Fetching PR branch"),
579 )
580 .map_err(|e| PrError::FetchFailed {
581 remote: remote_name.to_string(),
582 message: e.message().to_string(),
583 })?;
584
585 debug!("Successfully fetched branch {}", branch);
586 Ok(())
587}
588
589pub fn format_pr_name(format: &str, pr_number: u32) -> String {
593 format.replace("{number}", &pr_number.to_string())
594}
595
596pub fn prepare_pr_worktree(
607 repo: &Repository,
608 pr_number: u32,
609 pr_format: &str,
610) -> Result<(String, String, String)> {
611 debug!("Preparing PR worktree for PR #{}", pr_number);
612
613 let metadata = fetch_pr_metadata(pr_number)?;
615 debug!(
616 "Fetched metadata: title='{}', author='{}', is_fork={}",
617 metadata.title, metadata.author, metadata.is_fork
618 );
619
620 let remote_name = if metadata.is_fork {
624 setup_fork_remote(repo, &metadata)?
625 } else {
626 detect_pr_remote(repo)?
627 };
628
629 fetch_branch(repo, &remote_name, &metadata.head_ref)?;
631
632 let worktree_name = format_pr_name_with_metadata(pr_format, &metadata);
634 debug!("Worktree name: {}", worktree_name);
635
636 let remote_ref = format!("{}/{}", remote_name, metadata.head_ref);
638 debug!("Remote ref: {}", remote_ref);
639
640 Ok((worktree_name, remote_ref, metadata.base_ref))
641}
642
643#[cfg(test)]
644mod tests {
645 use super::*;
646
647 #[test]
648 fn test_parse_hash_number() {
649 let pr = parse_pr_reference("#123").unwrap().unwrap();
650 assert_eq!(pr.number, 123);
651 assert_eq!(pr.remote, None);
652 }
653
654 #[test]
655 fn test_parse_pr_hash_number() {
656 let pr = parse_pr_reference("pr#456").unwrap().unwrap();
657 assert_eq!(pr.number, 456);
658 assert_eq!(pr.remote, None);
659 }
660
661 #[test]
662 fn test_parse_pr_dash_number() {
663 let pr = parse_pr_reference("pr-789").unwrap().unwrap();
664 assert_eq!(pr.number, 789);
665 assert_eq!(pr.remote, None);
666 }
667
668 #[test]
669 fn test_parse_github_url() {
670 let pr = parse_pr_reference("https://github.com/owner/repo/pull/999")
671 .unwrap()
672 .unwrap();
673 assert_eq!(pr.number, 999);
674 assert_eq!(pr.remote, None);
675 }
676
677 #[test]
678 fn test_parse_remote_ref() {
679 let pr = parse_pr_reference("origin/pull/111/head").unwrap().unwrap();
680 assert_eq!(pr.number, 111);
681 assert_eq!(pr.remote, None);
682 }
683
684 #[test]
685 fn test_parse_regular_branch_name() {
686 let result = parse_pr_reference("my-feature-branch").unwrap();
687 assert!(result.is_none());
688 }
689
690 #[test]
691 fn test_parse_invalid_number() {
692 let result = parse_pr_reference("#abc");
693 assert!(result.is_err());
694 }
695
696 #[test]
697 fn test_is_pr_reference_true() {
698 assert!(is_pr_reference("#123"));
699 assert!(is_pr_reference("pr#456"));
700 assert!(is_pr_reference("pr-789"));
701 assert!(is_pr_reference("https://github.com/owner/repo/pull/999"));
702 }
703
704 #[test]
705 fn test_is_pr_reference_false() {
706 assert!(!is_pr_reference("my-branch"));
707 assert!(!is_pr_reference("feature"));
708 }
709
710 #[test]
711 fn test_format_pr_name() {
712 assert_eq!(format_pr_name("pr-{number}", 123), "pr-123");
713 assert_eq!(format_pr_name("review-{number}", 456), "review-456");
714 assert_eq!(format_pr_name("{number}-test", 789), "789-test");
715 }
716
717 #[test]
718 fn test_sanitize_branch_name() {
719 assert_eq!(sanitize_for_branch_name("Fix Bug #123"), "fix-bug-123");
720 assert_eq!(
721 sanitize_for_branch_name("Add Feature (v2)"),
722 "add-feature-v2"
723 );
724 assert_eq!(sanitize_for_branch_name("john-smith"), "john-smith");
725 assert_eq!(
726 sanitize_for_branch_name("Fix: Authentication Issue"),
727 "fix-authentication-issue"
728 );
729 assert_eq!(sanitize_for_branch_name("Test@#$%"), "test");
730 }
731
732 #[test]
733 fn test_format_with_metadata() {
734 let metadata = PrMetadata {
735 number: 123,
736 title: "Fix Authentication Bug".to_string(),
737 author: "john-smith".to_string(),
738 head_ref: "feature/fix-auth".to_string(),
739 base_ref: "main".to_string(),
740 is_fork: false,
741 fork_owner: None,
742 fork_url: None,
743 };
744
745 assert_eq!(
746 format_pr_name_with_metadata("pr-{number}", &metadata),
747 "pr-123"
748 );
749 assert_eq!(
750 format_pr_name_with_metadata("{number}-{title}", &metadata),
751 "123-fix-authentication-bug"
752 );
753 assert_eq!(
754 format_pr_name_with_metadata("{author}/pr-{number}", &metadata),
755 "john-smith/pr-123"
756 );
757 assert_eq!(
758 format_pr_name_with_metadata("{branch}-{number}", &metadata),
759 "feature-fix-auth-123"
760 );
761 }
762
763 #[test]
764 fn test_parse_merged_pr_empty_array() {
765 assert_eq!(parse_merged_pr("[]"), None);
766 }
767
768 #[test]
769 fn test_parse_merged_pr_populated_array() {
770 let json = r#"[{"number":66,"mergedAt":"2024-01-01T00:00:00Z"}]"#;
771 assert_eq!(parse_merged_pr(json), Some(66));
772 }
773
774 #[test]
775 fn test_parse_merged_pr_number_missing() {
776 let json = r#"[{"mergedAt":"2024-01-01T00:00:00Z"}]"#;
777 assert_eq!(parse_merged_pr(json), None);
778 }
779
780 #[test]
781 fn test_parse_merged_pr_number_non_numeric() {
782 let json = r#"[{"number":"not-a-number","mergedAt":"2024-01-01T00:00:00Z"}]"#;
783 assert_eq!(parse_merged_pr(json), None);
784 }
785
786 #[test]
787 fn test_parse_merged_pr_non_array_payload() {
788 assert_eq!(parse_merged_pr(r#"{"number":66}"#), None);
789 }
790
791 #[test]
792 fn test_parse_merged_pr_malformed_json() {
793 assert_eq!(parse_merged_pr("not json"), None);
794 }
795
796 #[test]
798 #[ignore]
799 fn test_gh_cli_available() {
800 check_gh_available().expect("gh CLI should be installed");
801 }
802
803 #[test]
804 #[ignore]
805 fn test_fetch_real_pr_metadata() {
806 let metadata = fetch_pr_metadata(1).expect("Failed to fetch PR metadata");
810 assert_eq!(metadata.number, 1);
811 assert!(!metadata.title.is_empty());
812 assert!(!metadata.author.is_empty());
813 }
814}