From 513fce723c7e37aa353844303f36022517f2db43 Mon Sep 17 00:00:00 2001 From: DanConwayDev Date: Fri, 25 Jul 2025 21:02:26 +0100 Subject: refactor: move `utils` and `list` helpers to lib to enable forthcoming `ngit sync` cmd --- src/bin/git_remote_nostr/utils.rs | 448 -------------------------------------- 1 file changed, 448 deletions(-) delete mode 100644 src/bin/git_remote_nostr/utils.rs (limited to 'src/bin/git_remote_nostr/utils.rs') diff --git a/src/bin/git_remote_nostr/utils.rs b/src/bin/git_remote_nostr/utils.rs deleted file mode 100644 index 2cb85bf..0000000 --- a/src/bin/git_remote_nostr/utils.rs +++ /dev/null @@ -1,448 +0,0 @@ -use core::str; -use std::{ - collections::HashMap, - fmt, - io::{self, Stdin}, - str::FromStr, -}; - -use anyhow::{Context, Result, bail}; -use git2::Repository; -use ngit::{ - client::{ - get_all_proposal_patch_pr_pr_update_events_from_cache, get_events_from_local_cache, - get_proposals_and_revisions_from_cache, - }, - git::{ - Repo, RepoActions, - nostr_url::{CloneUrl, NostrUrlDecoded, ServerProtocol}, - }, - git_events::{ - event_is_revision_root, get_pr_tip_event_or_most_recent_patch_with_ancestors, get_status, - is_event_proposal_root_for_branch, status_kinds, - }, - repo_ref::RepoRef, -}; -use nostr_sdk::{Event, EventId, Kind, PublicKey, Url}; - -pub fn get_short_git_server_name(git_repo: &Repo, url: &str) -> std::string::String { - if let Ok(name) = get_remote_name_by_url(&git_repo.git_repo, url) { - return name; - } - if let Ok(url) = Url::parse(url) { - if let Some(domain) = url.domain() { - return domain.to_string(); - } - } - url.to_string() -} - -pub fn get_remote_name_by_url(git_repo: &Repository, url: &str) -> Result { - let remotes = git_repo.remotes()?; - Ok(remotes - .iter() - .find(|r| { - if let Some(name) = r { - if let Some(remote_url) = git_repo.find_remote(name).unwrap().url() { - url == remote_url - } else { - false - } - } else { - false - } - }) - .context("could not find remote with matching url")? - .context("remote with matching url must be named")? - .to_string()) -} - -pub fn get_oids_from_fetch_batch( - stdin: &Stdin, - initial_oid: &str, - initial_refstr: &str, -) -> Result> { - let mut line = String::new(); - let mut batch = HashMap::new(); - batch.insert(initial_refstr.to_string(), initial_oid.to_string()); - loop { - let tokens = read_line(stdin, &mut line)?; - match tokens.as_slice() { - ["fetch", oid, refstr] => { - batch.insert((*refstr).to_string(), (*oid).to_string()); - } - [] => break, - _ => bail!( - "after a `fetch` command we are only expecting another fetch or an empty line" - ), - } - } - Ok(batch) -} - -/// Read one line from stdin, and split it into tokens. -pub fn read_line<'a>(stdin: &io::Stdin, line: &'a mut String) -> io::Result> { - line.clear(); - - let read = stdin.read_line(line)?; - if read == 0 { - return Ok(vec![]); - } - let line = line.trim(); - let tokens = line.split(' ').filter(|t| !t.is_empty()).collect(); - - Ok(tokens) -} - -pub async fn get_open_or_draft_proposals( - git_repo: &Repo, - repo_ref: &RepoRef, -) -> Result)>> { - let git_repo_path = git_repo.get_path()?; - let proposals: Vec = - get_proposals_and_revisions_from_cache(git_repo_path, repo_ref.coordinates()) - .await? - .iter() - .filter(|e| - // If we wanted to treat to list Pull Requests that revise a Patch we would do this: - // e.kind.eq(&KIND_PULL_REQUEST) || - !event_is_revision_root(e)) - .cloned() - .collect(); - - let statuses: Vec = { - let mut statuses = get_events_from_local_cache( - git_repo_path, - vec![ - nostr::Filter::default() - .kinds(status_kinds().clone()) - .events(proposals.iter().map(|e| e.id)), - ], - ) - .await?; - statuses.sort_by_key(|e| e.created_at); - statuses.reverse(); - statuses - }; - let mut open_or_draft_proposals = HashMap::new(); - - for proposal in &proposals { - let status = get_status(proposal, repo_ref, &statuses, &proposals); - if [Kind::GitStatusOpen, Kind::GitStatusDraft].contains(&status) { - if let Ok(commits_events) = get_all_proposal_patch_pr_pr_update_events_from_cache( - git_repo_path, - repo_ref, - &proposal.id, - ) - .await - { - if let Ok(most_recent_proposal_patch_chain) = - get_pr_tip_event_or_most_recent_patch_with_ancestors(commits_events.clone()) - { - open_or_draft_proposals.insert( - proposal.id, - (proposal.clone(), most_recent_proposal_patch_chain), - ); - } - } - } - } - Ok(open_or_draft_proposals) -} - -pub async fn get_all_proposals( - git_repo: &Repo, - repo_ref: &RepoRef, -) -> Result)>> { - let git_repo_path = git_repo.get_path()?; - let proposals: Vec = - get_proposals_and_revisions_from_cache(git_repo_path, repo_ref.coordinates()) - .await? - .iter() - .filter(|e| - // If we wanted to treat to list Pull Requests that revise a Patch we would do this: - // e.kind.eq(&KIND_PULL_REQUEST) || - !event_is_revision_root(e)) - .cloned() - .collect(); - - let mut all_proposals = HashMap::new(); - - for proposal in proposals { - if let Ok(commits_events) = get_all_proposal_patch_pr_pr_update_events_from_cache( - git_repo_path, - repo_ref, - &proposal.id, - ) - .await - { - if let Ok(most_recent_proposal_patch_chain) = - get_pr_tip_event_or_most_recent_patch_with_ancestors(commits_events.clone()) - { - all_proposals.insert(proposal.id, (proposal, most_recent_proposal_patch_chain)); - } - } - } - Ok(all_proposals) -} - -pub fn find_proposal_and_patches_by_branch_name<'a>( - refstr: &'a str, - proposals: &'a HashMap)>, - current_user: Option<&PublicKey>, -) -> Option<(&'a EventId, &'a (Event, Vec))> { - proposals.iter().find(|(_, (proposal, _))| { - is_event_proposal_root_for_branch(proposal, refstr, current_user).unwrap_or(false) - }) -} - -pub fn join_with_and(items: &[T]) -> String { - match items.len() { - 0 => String::new(), - 1 => items[0].to_string(), - _ => { - let last_item = items.last().unwrap().to_string(); - let rest = &items[..items.len() - 1]; - format!( - "{} and {}", - rest.iter() - .map(std::string::ToString::to_string) - .collect::>() - .join(", "), - last_item - ) - } - } -} - -/// get an ordered vector of server protocols to attempt -pub fn get_read_protocols_to_try( - git_repo: &Repo, - server_url: &CloneUrl, - decoded_nostr_url: &NostrUrlDecoded, - is_grasp_server: bool, -) -> Vec { - if is_grasp_server { - if server_url.protocol() == ServerProtocol::Http { - vec![(ServerProtocol::UnauthHttp)] - } else { - vec![(ServerProtocol::UnauthHttps)] - } - } else if server_url.protocol() == ServerProtocol::Filesystem { - vec![(ServerProtocol::Filesystem)] - } else if let Some(protocol) = &decoded_nostr_url.protocol { - vec![protocol.clone()] - } else { - let mut list = if server_url.protocol() == ServerProtocol::Http { - vec![ - ServerProtocol::UnauthHttp, - ServerProtocol::Ssh, - // note: list and fetch stop here if ssh was authenticated - ServerProtocol::Http, - ] - } else if server_url.protocol() == ServerProtocol::Ftp { - vec![ServerProtocol::Ftp, ServerProtocol::Ssh] - } else if server_url.protocol() == ServerProtocol::Git { - vec![ - ServerProtocol::Git, - ServerProtocol::UnauthHttps, - ServerProtocol::Ssh, - // note: list and fetch stop here if ssh was authenticated - ServerProtocol::Https, - ] - } else { - vec![ - ServerProtocol::UnauthHttps, - ServerProtocol::Ssh, - // note: list and fetch stop here if ssh was authenticated - ServerProtocol::Https, - ] - }; - if let Some(protocol) = get_protocol_preference(git_repo, server_url, &Direction::Fetch) { - if let Some(pos) = list.iter().position(|p| *p == protocol) { - list.remove(pos); - list.insert(0, protocol); - } - } - list - } -} - -/// get an ordered vector of server protocols to attempt -pub fn get_write_protocols_to_try( - git_repo: &Repo, - server_url: &CloneUrl, - decoded_nostr_url: &NostrUrlDecoded, - is_grasp_server: bool, -) -> Vec { - if is_grasp_server { - if server_url.protocol() == ServerProtocol::Http { - vec![(ServerProtocol::UnauthHttp)] - } else { - vec![(ServerProtocol::UnauthHttps)] - } - } else if server_url.protocol() == ServerProtocol::Filesystem { - vec![(ServerProtocol::Filesystem)] - } else if let Some(protocol) = &decoded_nostr_url.protocol { - vec![protocol.clone()] - } else { - let mut list = if server_url.protocol() == ServerProtocol::Http { - vec![ - ServerProtocol::Ssh, - // note: list and fetch stop here if ssh was authenticated - ServerProtocol::Http, - ] - } else if server_url.protocol() == ServerProtocol::Ftp { - vec![ServerProtocol::Ssh, ServerProtocol::Ftp] - } else if server_url.protocol() == ServerProtocol::Git { - vec![ - ServerProtocol::Ssh, - ServerProtocol::Git, - // note: list and fetch stop here if ssh was authenticated - ServerProtocol::Https, - ] - } else { - vec![ - ServerProtocol::Ssh, - // note: list and fetch stop here if ssh was authenticated - ServerProtocol::Https, - ] - }; - if let Some(protocol) = get_protocol_preference(git_repo, server_url, &Direction::Push) { - if let Some(pos) = list.iter().position(|p| *p == protocol) { - list.remove(pos); - list.insert(0, protocol); - } - } - - list - } -} - -#[derive(Debug, PartialEq)] -pub enum Direction { - Push, - Fetch, -} -impl fmt::Display for Direction { - fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { - match self { - Direction::Push => write!(f, "push"), - Direction::Fetch => write!(f, "fetch"), - } - } -} - -pub fn get_protocol_preference( - git_repo: &Repo, - server_url: &CloneUrl, - direction: &Direction, -) -> Option { - let server_short_name = server_url.short_name(); - if let Ok(Some(list)) = - git_repo.get_git_config_item(format!("nostr.protocol-{direction}").as_str(), Some(false)) - { - for item in list.split(';') { - let pair = item.split(',').collect::>(); - if let Some(url) = pair.get(1) { - if *url == server_short_name { - if let Some(protocol) = pair.first() { - if let Ok(protocol) = ServerProtocol::from_str(protocol) { - return Some(protocol); - } - } - } - } - } - } - None -} - -pub fn set_protocol_preference( - git_repo: &Repo, - protocol: &ServerProtocol, - server_url: &CloneUrl, - direction: &Direction, -) -> Result<()> { - let server_short_name = server_url.short_name(); - let mut new = String::new(); - if let Some(list) = - git_repo.get_git_config_item(format!("nostr.protocol-{direction}").as_str(), Some(false))? - { - for item in list.split(';') { - let pair = item.split(',').collect::>(); - if let Some(url) = pair.get(1) { - if *url != server_short_name && !item.is_empty() { - new.push_str(format!("{item};").as_str()); - } - } - } - } - new.push_str(format!("{protocol},{server_short_name};").as_str()); - - git_repo.save_git_config_item( - format!("nostr.protocol-{direction}").as_str(), - new.as_str(), - false, - ) -} - -pub fn error_might_be_authentication_related(error: &anyhow::Error) -> bool { - let error_str = error.to_string(); - for s in [ - "no ssh keys found", - "invalid or unknown remote ssh hostkey", - "authentication", - "Permission", - "permission", - "not found", - ] { - if error_str.contains(s) { - return true; - } - } - false -} - -#[cfg(test)] -mod tests { - use super::*; - mod join_with_and { - use super::*; - #[test] - fn test_empty() { - let items: Vec<&str> = vec![]; - assert_eq!(join_with_and(&items), ""); - } - - #[test] - fn test_single_item() { - let items = vec!["apple"]; - assert_eq!(join_with_and(&items), "apple"); - } - - #[test] - fn test_two_items() { - let items = vec!["apple", "banana"]; - assert_eq!(join_with_and(&items), "apple and banana"); - } - - #[test] - fn test_three_items() { - let items = vec!["apple", "banana", "cherry"]; - assert_eq!(join_with_and(&items), "apple, banana and cherry"); - } - - #[test] - fn test_four_items() { - let items = vec!["apple", "banana", "cherry", "date"]; - assert_eq!(join_with_and(&items), "apple, banana, cherry and date"); - } - - #[test] - fn test_multiple_items() { - let items = vec!["one", "two", "three", "four", "five"]; - assert_eq!(join_with_and(&items), "one, two, three, four and five"); - } - } -} -- cgit v1.2.3