use std::error;
use std::io;
use std::io::prelude::*;
use std::sync::{Arc, Mutex, Once};
use ureq::Agent;
use ureq::AgentBuilder;
use ureq::Proxy;
use url::Url;
use log::{debug, info};
use git2::transport::{Service, SmartSubtransport, SmartSubtransportStream, Transport};
use git2::Error;
#[derive(Default)]
struct UreqTransport {
base_url: Arc<Mutex<String>>,
proxy: Option<Proxy>,
}
struct UreqSubtransport {
service: &'static str,
url_path: &'static str,
base_url: Arc<Mutex<String>>,
method: &'static str,
reader: Option<Box<dyn Read + Send>>,
sent_request: bool,
client: Agent,
}
pub unsafe fn register(proxy: Option<String>) {
static INIT: Once = Once::new();
let proxy = proxy.map(|s| Proxy::new(s).ok()).flatten();
let p = proxy.clone();
INIT.call_once(move || {
git2::transport::register("http", move |remote| factory(remote, proxy.as_ref())).unwrap();
git2::transport::register("https", move |remote| factory(remote, p.as_ref())).unwrap();
});
}
fn factory(remote: &git2::Remote<'_>, proxy: Option<&Proxy>) -> Result<Transport, Error> {
Transport::smart(remote, true, UreqTransport::new(proxy.cloned()))
}
impl UreqTransport {
pub(crate) fn new(proxy: Option<Proxy>) -> Self {
Self {
proxy,
..Default::default()
}
}
}
impl SmartSubtransport for UreqTransport {
fn action(
&self,
url: &str,
action: Service,
) -> Result<Box<dyn SmartSubtransportStream>, Error> {
let mut base_url = self.base_url.lock().unwrap();
if base_url.len() == 0 {
*base_url = url.to_string();
}
let (service, path, method) = match action {
Service::UploadPackLs => ("upload-pack", "/info/refs?service=git-upload-pack", "GET"),
Service::UploadPack => ("upload-pack", "/git-upload-pack", "POST"),
Service::ReceivePackLs => {
("receive-pack", "/info/refs?service=git-receive-pack", "GET")
}
Service::ReceivePack => ("receive-pack", "/git-receive-pack", "POST"),
};
info!("action {} {}", service, path);
Ok(Box::new(UreqSubtransport {
service,
url_path: path,
base_url: self.base_url.clone(),
method,
reader: None,
sent_request: false,
client: self
.proxy
.clone()
.map(|p| AgentBuilder::new().proxy(p))
.unwrap_or_else(AgentBuilder::new)
.build(),
}))
}
fn close(&self) -> Result<(), Error> {
Ok(())
}
}
impl UreqSubtransport {
fn err<E: Into<Box<dyn error::Error + Send + Sync>>>(&self, err: E) -> io::Error {
io::Error::new(io::ErrorKind::Other, err)
}
fn execute(&mut self, data: &[u8]) -> io::Result<()> {
if self.sent_request {
return Err(self.err("already sent HTTP request"));
}
let agent = format!("git/1.0 (git2-ureq {})", env!("CARGO_PKG_VERSION"));
let url = format!("{}{}", self.base_url.lock().unwrap(), self.url_path);
let parsed = Url::parse(&url).map_err(|_| self.err("invalid url, failed to parse"))?;
let host = match parsed.host_str() {
Some(host) => host,
None => return Err(self.err("invalid url, did not have a host")),
};
debug!("request to {}", url);
let request = self
.client
.request(self.method, &url)
.set("User-Agent", &agent)
.set("Host", &host)
.set("Expect", "");
let request = if data.is_empty() {
request.set("Accept", "*/*")
} else {
request
.set(
"Accept",
&format!("application/x-git-{}-result", self.service),
)
.set(
"Conent-Type",
&format!("application/x-git-{}-request", self.service),
)
};
let response = request.send(data).unwrap();
let content_type = response.header("Content-Type");
let code = response.status();
if code != 200 {
return Err(self.err(&format!("failed to receive HTTP 200 response: got {code}",)[..]));
}
let expected = match self.method {
"GET" => format!("application/x-git-{}-advertisement", self.service),
_ => format!("application/x-git-{}-result", self.service),
};
if let Some(content_type) = content_type {
if content_type != expected {
return Err(self.err(
&format!(
"expected a Content-Type header with `{expected}` but found `{content_type}`",
)[..],
));
}
} else {
return Err(
self.err(
&format!(
"expected a Content-Type header with `{expected}` but didn't find one"
)[..],
),
);
}
self.reader = Some(Box::new(response.into_reader()));
Ok(())
}
}
impl Read for UreqSubtransport {
fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
if self.reader.is_none() {
self.execute(&[])?;
}
self.reader.as_mut().unwrap().read(buf)
}
}
impl Write for UreqSubtransport {
fn write(&mut self, data: &[u8]) -> io::Result<usize> {
if self.reader.is_none() {
self.execute(data)?;
}
Ok(data.len())
}
fn flush(&mut self) -> io::Result<()> {
Ok(())
}
}