use std::borrow::Cow; use crate::{ bstr::{BStr, ByteSlice, ByteVec}, remote, Reference, }; /// The name of a remote, either interpreted as symbol like `origin` or as url as returned by [`Reference::remote_name()`]. #[derive(Debug, PartialEq, Eq, Clone)] pub enum Name<'repo> { /// A symbolic name, like `origin` Symbol(Cow<'repo, str>), /// A url pointing to the remote host directly. Url(Cow<'repo, BStr>), } impl Name<'_> { /// Return this instance as a symbolic name, if it is one. pub fn as_symbol(&self) -> Option<&str> { match self { Name::Symbol(n) => n.as_ref().into(), Name::Url(_) => None, } } /// Return this instance as url, if it is one. pub fn as_url(&self) -> Option<&BStr> { match self { Name::Url(n) => n.as_ref().into(), Name::Symbol(_) => None, } } } /// Remotes impl<'repo> Reference<'repo> { /// Find the name of our remote for `direction` as configured in `branch..remote|pushRemote` respectively. /// If `Some()` it can be used in [`Repository::find_remote(…)`][crate::Repository::find_remote()], or if `None` then /// [Repository::remote_default_name()][crate::Repository::remote_default_name()] could be used in its place. /// /// Return `None` if no remote is configured. /// /// # Note /// /// - it's recommended to use the [`remote(…)`][Self::remote()] method as it will configure the remote with additional /// information. /// - `branch..pushRemote` falls back to `branch..remote`. pub fn remote_name(&self, direction: remote::Direction) -> Option> { let name = self.name().shorten().to_str().ok()?; let config = &self.repo.config.resolved; (direction == remote::Direction::Push) .then(|| { config .string("branch", Some(name), "pushRemote") .or_else(|| config.string("remote", None, "pushDefault")) }) .flatten() .or_else(|| config.string("branch", Some(name), "remote")) .and_then(|name| { if name.contains(&b'/') { Some(Name::Url(name)) } else { match name { Cow::Borrowed(n) => n.to_str().ok().map(Cow::Borrowed), Cow::Owned(n) => Vec::from(n).into_string().ok().map(Cow::Owned), } .map(Name::Symbol) } }) } /// Like [`remote_name(…)`][Self::remote_name()], but configures the returned `Remote` with additional information like /// /// - `branch..merge` to know which branch on the remote side corresponds to this one for merging when pulling. pub fn remote( &self, direction: remote::Direction, ) -> Option, remote::find::existing::Error>> { // TODO: use `branch..merge` self.remote_name(direction).map(|name| match name { Name::Symbol(name) => self.repo.find_remote(name.as_ref()).map_err(Into::into), Name::Url(url) => git_url::parse(url.as_ref()).map_err(Into::into).and_then(|url| { self.repo .remote_at(url) .map_err(|err| remote::find::existing::Error::Find(remote::find::Error::Init(err))) }), }) } }