use std::{io::Write, ops::Add}; use anyhow::{bail, Context, Result}; use ngit::{ client::{get_all_proposal_patch_events_from_cache, get_proposals_and_revisions_from_cache}, git_events::{ get_commit_id_from_patch, get_most_recent_patch_with_ancestors, status_kinds, tag_value, }, }; use nostr_sdk::Kind; use crate::{ cli_interactor::{Interactor, InteractorPrompt, PromptChoiceParms, PromptConfirmParms}, client::{ fetching_with_report, get_events_from_local_cache, get_repo_ref_from_cache, Client, Connect, }, git::{str_to_sha1, Repo, RepoActions}, git_events::{ commit_msg_from_patch_oneliner, event_is_revision_root, event_to_cover_letter, patch_supports_commit_ids, }, repo_ref::get_repo_coordinates_when_remote_unknown, }; #[allow(clippy::too_many_lines)] pub async fn launch() -> Result<()> { let git_repo = Repo::discover().context("failed to find a git repository")?; let git_repo_path = git_repo.get_path()?; // TODO: check for empty repo // TODO: check for existing maintaiers file // TODO: check for other claims let client = Client::default(); let repo_coordinates = get_repo_coordinates_when_remote_unknown(&git_repo, &client).await?; fetching_with_report(git_repo_path, &client, &repo_coordinates).await?; let repo_ref = get_repo_ref_from_cache(Some(git_repo_path), &repo_coordinates).await?; let proposals_and_revisions: Vec = get_proposals_and_revisions_from_cache(git_repo_path, repo_ref.coordinates()).await?; if proposals_and_revisions.is_empty() { println!("no proposals found... create one? try `ngit send`"); return Ok(()); } let statuses: Vec = { let mut statuses = get_events_from_local_cache( git_repo_path, vec![ nostr::Filter::default() .kinds(status_kinds().clone()) .events(proposals_and_revisions.iter().map(|e| e.id)), ], ) .await?; statuses.sort_by_key(|e| e.created_at); statuses.reverse(); statuses }; let mut open_proposals: Vec<&nostr::Event> = vec![]; let mut draft_proposals: Vec<&nostr::Event> = vec![]; let mut closed_proposals: Vec<&nostr::Event> = vec![]; let mut applied_proposals: Vec<&nostr::Event> = vec![]; let proposals: Vec = proposals_and_revisions .iter() .filter(|e| !event_is_revision_root(e)) .cloned() .collect(); for proposal in &proposals { let status = if let Some(e) = statuses .iter() .filter(|e| { status_kinds().contains(&e.kind) && e.tags.iter().any(|t| { t.as_slice().len() > 1 && t.as_slice()[1].eq(&proposal.id.to_string()) }) }) .collect::>() .first() { e.kind } else { Kind::GitStatusOpen }; if status.eq(&Kind::GitStatusOpen) { open_proposals.push(proposal); } else if status.eq(&Kind::GitStatusClosed) { closed_proposals.push(proposal); } else if status.eq(&Kind::GitStatusDraft) { draft_proposals.push(proposal); } else if status.eq(&Kind::GitStatusApplied) { applied_proposals.push(proposal); } } let mut selected_status = Kind::GitStatusOpen; loop { let proposals_for_status = if selected_status == Kind::GitStatusOpen { &open_proposals } else if selected_status == Kind::GitStatusDraft { &draft_proposals } else if selected_status == Kind::GitStatusClosed { &closed_proposals } else if selected_status == Kind::GitStatusApplied { &applied_proposals } else { &open_proposals }; let prompt = if proposals.len().eq(&open_proposals.len()) { "all proposals" } else if selected_status == Kind::GitStatusOpen { if open_proposals.is_empty() { "proposals menu" } else { "open proposals" } } else if selected_status == Kind::GitStatusDraft { "draft proposals" } else if selected_status == Kind::GitStatusClosed { "closed proposals" } else { "applied proposals" }; let mut choices: Vec = proposals_for_status .iter() .map(|e| { if let Ok(cl) = event_to_cover_letter(e) { cl.title } else if let Ok(msg) = tag_value(e, "description") { msg.split('\n').collect::>()[0].to_string() } else { e.id.to_string() } }) .collect(); if !selected_status.eq(&Kind::GitStatusOpen) && open_proposals.len().gt(&0) { choices.push(format!("({}) Open proposals...", open_proposals.len())); } if !selected_status.eq(&Kind::GitStatusDraft) && draft_proposals.len().gt(&0) { choices.push(format!("({}) Draft proposals...", draft_proposals.len())); } if !selected_status.eq(&Kind::GitStatusClosed) && closed_proposals.len().gt(&0) { choices.push(format!("({}) Closed proposals...", closed_proposals.len())); } if !selected_status.eq(&Kind::GitStatusApplied) && applied_proposals.len().gt(&0) { choices.push(format!( "({}) Applied proposals...", applied_proposals.len() )); } let selected_index = Interactor::default().choice( PromptChoiceParms::default() .with_prompt(prompt) .with_default(0) .with_choices(choices.clone()), )?; if (selected_index + 1).gt(&proposals_for_status.len()) { if choices[selected_index].contains("Open") { selected_status = Kind::GitStatusOpen; } else if choices[selected_index].contains("Draft") { selected_status = Kind::GitStatusDraft; } else if choices[selected_index].contains("Closed") { selected_status = Kind::GitStatusClosed; } else if choices[selected_index].contains("Applied") { selected_status = Kind::GitStatusApplied; } continue; } let cover_letter = event_to_cover_letter(proposals_for_status[selected_index]) .context("failed to extract proposal details from proposal root event")?; let commits_events: Vec = get_all_proposal_patch_events_from_cache( git_repo_path, &repo_ref, &proposals_for_status[selected_index].id, ) .await?; let Ok(most_recent_proposal_patch_chain) = get_most_recent_patch_with_ancestors(commits_events.clone()) else { if Interactor::default().confirm( PromptConfirmParms::default() .with_default(true) .with_prompt( "failed to find any patches on this proposal. choose another proposal?", ), )? { continue; } return Ok(()); }; // for commit in &most_recent_proposal_patch_chain { // println!("recent_event: {:?}", commit.as_json()); // } let binding_patch_text_ref = format!("{} commits", most_recent_proposal_patch_chain.len()); let patch_text_ref = if most_recent_proposal_patch_chain.len().gt(&1) { binding_patch_text_ref.as_str() } else { "1 commit" }; let no_support_for_patches_as_branch = most_recent_proposal_patch_chain .iter() .any(|event| !patch_supports_commit_ids(event)); if no_support_for_patches_as_branch { println!("{patch_text_ref}"); return match Interactor::default().choice( PromptChoiceParms::default() .with_default(0) .with_choices(vec![ "learn why 'patch only' proposals can't be checked out".to_string(), format!("apply to current branch with `git am`"), format!("download to ./patches"), "back".to_string(), ]), )? { 0 => { println!("Some proposals are posted as 'patch only'\n"); println!( "they are not anchored against a particular state of the code base like a standard proposal or a GitHub Pull Request can be\n" ); println!( "they are designed to reviewed by studying the diff (in a tool like gitworkshop.dev) and if acceptable by a maintainer, applied to the latest version of master with any conflicts resolved as the do so\n" ); println!( "this has proven to be a smoother workflow for large scale projects with a high frequency of changes, even when patches are exchanged via email\n" ); println!( "by default ngit posts proposals that support both the branch and patch model so either workflow can be used" ); Interactor::default().choice( PromptChoiceParms::default() .with_default(0) .with_choices(vec!["back".to_string()]), )?; continue; } 1 => launch_git_am_with_patches(most_recent_proposal_patch_chain), 2 => save_patches_to_dir(most_recent_proposal_patch_chain, &git_repo), 3 => continue, _ => { bail!("unexpected choice") } }; } let branch_exists = git_repo .get_local_branch_names() .context("gitlib2 will not show a list of local branch names")? .iter() .any(|n| n.eq(&cover_letter.get_branch_name().unwrap())); let checked_out_proposal_branch = git_repo .get_checked_out_branch_name()? .eq(&cover_letter.get_branch_name()?); let proposal_base_commit = str_to_sha1(&tag_value( most_recent_proposal_patch_chain.last().context( "there should be at least one patch as we have already checked for this", )?, "parent-commit", )?) .context("failed to get valid parent commit id from patch")?; let (main_branch_name, master_tip) = git_repo.get_main_or_master_branch()?; if !git_repo.does_commit_exist(&proposal_base_commit.to_string())? { println!("your '{main_branch_name}' branch may not be up-to-date."); println!("the proposal parent commit doesnt exist in your local repository."); return match Interactor::default().choice(PromptChoiceParms::default().with_default(0).with_choices( vec![ format!( "manually run `git pull` on '{main_branch_name}' and select proposal again" ), format!("apply to current branch with `git am`"), format!("download to ./patches"), "back".to_string(), ], ))? { 0 | 3 => continue, 1 => launch_git_am_with_patches(most_recent_proposal_patch_chain), 2 => save_patches_to_dir(most_recent_proposal_patch_chain, &git_repo), _ => { bail!("unexpected choice") } }; } let proposal_tip = str_to_sha1( &get_commit_id_from_patch(most_recent_proposal_patch_chain.first().context( "there should be at least one patch as we have already checked for this", )?) .context("failed to get valid commit_id from patch")?, ) .context("failed to get valid commit_id from patch")?; let (_, proposal_behind_main) = git_repo.get_commits_ahead_behind(&master_tip, &proposal_base_commit)?; // branch doesnt exist if !branch_exists { return match Interactor::default() .choice(PromptChoiceParms::default().with_default(0).with_choices(vec![ format!( "create and checkout proposal branch ({} ahead {} behind '{main_branch_name}')", most_recent_proposal_patch_chain.len(), proposal_behind_main.len(), ), format!("apply to current branch with `git am`"), format!("download to ./patches"), "back".to_string(), ]))? { 0 => { check_clean(&git_repo)?; let _ = git_repo .apply_patch_chain( &cover_letter.get_branch_name()?, most_recent_proposal_patch_chain, ) .context("failed to apply patch chain")?; println!( "checked out proposal as '{}' branch", cover_letter.get_branch_name()? ); Ok(()) } 1 => launch_git_am_with_patches(most_recent_proposal_patch_chain), 2 => save_patches_to_dir(most_recent_proposal_patch_chain, &git_repo), 3 => continue, _ => { bail!("unexpected choice") } }; } let local_branch_tip = git_repo.get_tip_of_branch(&cover_letter.get_branch_name()?)?; // up-to-date if proposal_tip.eq(&local_branch_tip) { if checked_out_proposal_branch { println!("branch checked out and up-to-date"); return match Interactor::default().choice( PromptChoiceParms::default() .with_default(0) .with_choices(vec!["exit".to_string(), "back".to_string()]), )? { 0 => Ok(()), 1 => continue, _ => { bail!("unexpected choice") } }; } return match Interactor::default().choice( PromptChoiceParms::default() .with_default(0) .with_choices(vec![ format!( "checkout proposal branch ({} ahead {} behind '{main_branch_name}')", most_recent_proposal_patch_chain.len(), proposal_behind_main.len(), ), format!("apply to current branch with `git am`"), format!("download to ./patches"), "back".to_string(), ]), )? { 0 => { check_clean(&git_repo)?; git_repo.checkout(&cover_letter.get_branch_name()?)?; println!( "checked out proposal as '{}' branch", cover_letter.get_branch_name()? ); Ok(()) } 1 => launch_git_am_with_patches(most_recent_proposal_patch_chain), 2 => save_patches_to_dir(most_recent_proposal_patch_chain, &git_repo), 3 => continue, _ => { bail!("unexpected choice") } }; } let (local_ahead_of_main, local_beind_main) = git_repo.get_commits_ahead_behind(&master_tip, &local_branch_tip)?; // new appendments to proposal if let Some(index) = most_recent_proposal_patch_chain.iter().position(|patch| { get_commit_id_from_patch(patch) .unwrap_or_default() .eq(&local_branch_tip.to_string()) }) { return match Interactor::default().choice( PromptChoiceParms::default() .with_default(0) .with_choices(vec![ format!("checkout proposal branch and apply {} appendments", &index,), format!("apply to current branch with `git am`"), format!("download to ./patches"), "back".to_string(), ]), )? { 0 => { check_clean(&git_repo)?; git_repo.checkout(&cover_letter.get_branch_name()?)?; let _ = git_repo .apply_patch_chain( &cover_letter.get_branch_name()?, most_recent_proposal_patch_chain, ) .context("failed to apply patch chain")?; println!( "checked out proposal branch and applied {} appendments ({} ahead {} behind '{main_branch_name}')", &index, local_ahead_of_main.len().add(&index), local_beind_main.len(), ); Ok(()) } 1 => launch_git_am_with_patches(most_recent_proposal_patch_chain), 2 => save_patches_to_dir(most_recent_proposal_patch_chain, &git_repo), 3 => continue, _ => { bail!("unexpected choice") } }; } // new proposal revision / rebase // tip of local in proposal history (new, amended or rebased version but no // local changes) if commits_events.iter().any(|patch| { get_commit_id_from_patch(patch) .unwrap_or_default() .eq(&local_branch_tip.to_string()) }) { println!( "updated proposal available ({} ahead {} behind '{main_branch_name}'). existing version is {} ahead {} behind '{main_branch_name}'", most_recent_proposal_patch_chain.len(), proposal_behind_main.len(), local_ahead_of_main.len(), local_beind_main.len(), ); return match Interactor::default().choice( PromptChoiceParms::default() .with_default(0) .with_choices(vec![ format!("checkout and overwrite existing proposal branch"), format!("checkout existing outdated proposal branch"), format!("apply to current branch with `git am`"), format!("download to ./patches"), "back".to_string(), ]), )? { 0 => { check_clean(&git_repo)?; git_repo.create_branch_at_commit( &cover_letter.get_branch_name()?, &proposal_base_commit.to_string(), )?; git_repo.checkout(&cover_letter.get_branch_name()?)?; let chain_length = most_recent_proposal_patch_chain.len(); let _ = git_repo .apply_patch_chain( &cover_letter.get_branch_name()?, most_recent_proposal_patch_chain, ) .context("failed to apply patch chain")?; println!( "checked out new version of proposal ({} ahead {} behind '{main_branch_name}'), replacing old version ({} ahead {} behind '{main_branch_name}')", chain_length, proposal_behind_main.len(), local_ahead_of_main.len(), local_beind_main.len(), ); Ok(()) } 1 => { check_clean(&git_repo)?; git_repo.checkout(&cover_letter.get_branch_name()?)?; println!( "checked out old proposal in existing branch ({} ahead {} behind '{main_branch_name}')", local_ahead_of_main.len(), local_beind_main.len(), ); Ok(()) } 2 => launch_git_am_with_patches(most_recent_proposal_patch_chain), 3 => save_patches_to_dir(most_recent_proposal_patch_chain, &git_repo), 4 => continue, _ => { bail!("unexpected choice") } }; } // tip of proposal in branch in history (local appendments made to up-to-date // proposal) else if git_repo.ancestor_of(&local_branch_tip, &proposal_tip)? { let (local_ahead_of_proposal, _) = git_repo .get_commits_ahead_behind(&proposal_tip, &local_branch_tip) .context( "failed to get commits ahead behind for propsal_top and local_branch_tip", )?; println!( "local proposal branch exists with {} unpublished commits on top of the most up-to-date version of the proposal ({} ahead {} behind '{main_branch_name}')", local_ahead_of_proposal.len(), local_ahead_of_main.len(), proposal_behind_main.len(), ); return match Interactor::default().choice( PromptChoiceParms::default() .with_default(0) .with_choices(vec![ format!( "checkout proposal branch with {} unpublished commits", local_ahead_of_proposal.len(), ), "back".to_string(), ]), )? { 0 => { git_repo.checkout(&cover_letter.get_branch_name()?)?; println!( "checked out proposal branch with {} unpublished commits ({} ahead {} behind '{main_branch_name}')", local_ahead_of_proposal.len(), local_ahead_of_main.len(), proposal_behind_main.len(), ); Ok(()) } 1 => continue, _ => { bail!("unexpected choice") } }; } println!("you have an amended/rebase version the proposal that is unpublished"); // user probably has a unpublished amended or rebase version of the latest // proposal version // if tip of proposal commits exist (were once part of branch but have been // amended and git clean up job hasn't removed them) if git_repo.does_commit_exist(&proposal_tip.to_string())? { println!( "you have previously applied the latest version of the proposal ({} ahead {} behind '{main_branch_name}') but your local proposal branch has amended or rebased it ({} ahead {} behind '{main_branch_name}')", most_recent_proposal_patch_chain.len(), proposal_behind_main.len(), local_ahead_of_main.len(), local_beind_main.len(), ); } // user probably has a unpublished amended or rebase version of an older // proposal version else { println!( "your local proposal branch ({} ahead {} behind '{main_branch_name}') has conflicting changes with the latest published proposal ({} ahead {} behind '{main_branch_name}')", local_ahead_of_main.len(), local_beind_main.len(), most_recent_proposal_patch_chain.len(), proposal_behind_main.len(), ); println!( "its likely that you have rebased / amended an old proposal version because git has no record of the latest proposal commit." ); println!( "it is possible that you have been working off the latest version and git has delete this commit as part of a clean up" ); } println!("to view the latest proposal but retain your changes:"); println!(" 1) create a new branch off the tip commit of this one to store your changes"); println!(" 2) run `ngit list` and checkout the latest published version of this proposal"); println!("if you are confident in your changes consider running `ngit push --force`"); return match Interactor::default().choice( PromptChoiceParms::default() .with_default(0) .with_choices(vec![ format!("checkout local branch with unpublished changes"), format!("discard unpublished changes and checkout new revision",), format!("apply to current branch with `git am`"), format!("download to ./patches"), "back".to_string(), ]), )? { 0 => { check_clean(&git_repo)?; git_repo.checkout(&cover_letter.get_branch_name()?)?; println!( "checked out old proposal in existing branch ({} ahead {} behind '{main_branch_name}')", local_ahead_of_main.len(), local_beind_main.len(), ); Ok(()) } 1 => { check_clean(&git_repo)?; git_repo.create_branch_at_commit( &cover_letter.get_branch_name()?, &proposal_base_commit.to_string(), )?; let chain_length = most_recent_proposal_patch_chain.len(); let _ = git_repo .apply_patch_chain( &cover_letter.get_branch_name()?, most_recent_proposal_patch_chain, ) .context("failed to apply patch chain")?; git_repo.checkout(&cover_letter.get_branch_name()?)?; println!( "checked out latest version of proposal ({} ahead {} behind '{main_branch_name}'), replacing unpublished version ({} ahead {} behind '{main_branch_name}')", chain_length, proposal_behind_main.len(), local_ahead_of_main.len(), local_beind_main.len(), ); Ok(()) } 2 => launch_git_am_with_patches(most_recent_proposal_patch_chain), 3 => save_patches_to_dir(most_recent_proposal_patch_chain, &git_repo), 4 => continue, _ => { bail!("unexpected choice") } }; } } fn launch_git_am_with_patches(mut patches: Vec) -> Result<()> { println!("applying to current branch with `git am`"); // TODO: add PATCH x/n to appended patches patches.reverse(); let mut am = std::process::Command::new("git") .arg("am") .stdin(std::process::Stdio::piped()) .stdout(std::process::Stdio::inherit()) .stderr(std::process::Stdio::inherit()) .spawn() .context("failed to spawn git am")?; let stdin = am .stdin .as_mut() .context("git am process failed to take stdin")?; for patch in patches { stdin .write(format!("{}\n\n", patch.content).as_bytes()) .context("failed to write patch content into git am stdin buffer")?; } stdin.flush()?; let output = am .wait_with_output() .context("failed to read git am stdout")?; print!("{:?}", output.stdout); Ok(()) } fn event_id_extra_shorthand(event: &nostr::Event) -> String { event.id.to_string()[..5].to_string() } fn save_patches_to_dir(mut patches: Vec, git_repo: &Repo) -> Result<()> { // TODO: add PATCH x/n to appended patches patches.reverse(); let path = git_repo.get_path()?.join("patches"); std::fs::create_dir_all(&path)?; let id = event_id_extra_shorthand( patches .first() .context("there must be at least one patch to save")?, ); for (i, patch) in patches.iter().enumerate() { let path = path.join(format!( "{}-{:0>4}-{}.patch", &id, i.add(&1), commit_msg_from_patch_oneliner(patch)? )); let mut file = std::fs::OpenOptions::new() .create(true) .write(true) .truncate(true) .open(path) .context("open new patch file with write and truncate options")?; file.write_all(patch.content.as_bytes())?; file.write_all("\n\n".as_bytes())?; file.flush()?; } println!("created {} patch files in ./patches/{id}-*", patches.len()); Ok(()) } fn check_clean(git_repo: &Repo) -> Result<()> { if git_repo.has_outstanding_changes()? { bail!( "failed to pull proposal branch when repository is not clean. discard or stash (un)staged changes and try again." ); } Ok(()) }