diff --git a/Cargo.lock b/Cargo.lock index 61464a5..ba933cb 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -169,6 +169,7 @@ checksum = "00b0228411908ca8685dba7fc2cdd70ec9990a6e753e89b6ac91a84c40fbaf4b" name = "forgejo-api" version = "0.1.0" dependencies = [ + "bytes", "eyre", "reqwest", "serde", diff --git a/Cargo.toml b/Cargo.toml index eb21559..83ad83c 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -14,6 +14,7 @@ url = { version = "2.4.0", features = ["serde"] } serde = { version = "1.0.168", features = ["derive"] } time = { version = "0.3.22", features = ["parsing", "serde", "formatting"] } serde_json = "1.0.108" +bytes = "1.5.0" [dev-dependencies] eyre = "0.6.9" diff --git a/src/lib.rs b/src/lib.rs index 21b0a45..4743e91 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -180,6 +180,24 @@ impl Forgejo { status => Err(ForgejoError::UnexpectedStatusCode(status)), } } + + /// Like `execute`, but returns a `String`. + async fn execute_opt_raw(&self, request: Request) -> Result, ForgejoError> { + let response = self.client.execute(request).await?; + match response.status() { + status if status.is_success() => Ok(Some(response.bytes().await?)), + StatusCode::NOT_FOUND => Ok(None), + status if status.is_client_error() => Err(ForgejoError::ApiError( + status, + response + .json::() + .await? + .message + .unwrap_or_else(|| String::from("[no message]")), + )), + status => Err(ForgejoError::UnexpectedStatusCode(status)), + } + } /// Like `execute`, but returns a `String`. async fn execute_str(&self, request: Request) -> Result { diff --git a/src/repository.rs b/src/repository.rs index 966b5e3..992f78a 100644 --- a/src/repository.rs +++ b/src/repository.rs @@ -207,6 +207,34 @@ impl Forgejo { .await } + pub async fn download_release_zip(&self, owner: &str, repo: &str, release: u64) -> Result, ForgejoError> { + let release = self.get_release(owner, repo, release).await; + let release = match release { + Ok(Some(release)) => release, + Ok(None) => return Ok(None), + Err(e) => return Err(e), + }; + let request = self.client.get(release.zipball_url).build()?; + self.execute_opt_raw(request).await + } + + pub async fn download_release_tarball(&self, owner: &str, repo: &str, release: u64) -> Result, ForgejoError> { + let release = self.get_release(owner, repo, release).await; + let release = match release { + Ok(Some(release)) => release, + Ok(None) => return Ok(None), + Err(e) => return Err(e), + }; + let request = self.client.get(release.tarball_url).build()?; + self.execute_opt_raw(request).await + } + + pub async fn download_release_attachment(&self, owner: &str, repo: &str, release: u64, attach: u64) -> Result, ForgejoError> { + let release = self.get_release_attachment(owner, repo, release, attach).await?; + let request = self.client.get(release.browser_download_url).build()?; + self.execute_opt_raw(request).await + } + pub async fn get_tags( &self, owner: &str,