55 lines
1.9 KiB
Rust
55 lines
1.9 KiB
Rust
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
use anyhow::{Context, Result};
|
|
|
|
pub struct TaskCluster {
|
|
root_url: url::Url,
|
|
client: reqwest::blocking::Client,
|
|
}
|
|
|
|
impl TaskCluster {
|
|
pub fn from_env() -> Result<Self> {
|
|
std::env::var("TASKCLUSTER_ROOT_URL")
|
|
.context("TASKCLUSTER_ROOT_URL not set.")
|
|
.and_then(|var| var.parse().context("Couldn't parse TASKCLUSTER_ROOT_URL."))
|
|
.map(|root_url| TaskCluster {
|
|
root_url,
|
|
client: reqwest::blocking::Client::new(),
|
|
})
|
|
}
|
|
|
|
/// Return the root URL as suitable for passing to other processes.
|
|
///
|
|
/// In particular, any trailing slashes are removed.
|
|
pub fn root_url(&self) -> String {
|
|
self.root_url.as_str().trim_end_matches("/").to_string()
|
|
}
|
|
|
|
pub fn task_artifact_url(&self, task_id: &str, path: &str) -> url::Url {
|
|
let mut url = self.root_url.clone();
|
|
url.set_path(&format!("api/queue/v1/task/{}/artifacts/{}", task_id, path));
|
|
url
|
|
}
|
|
|
|
pub fn stream_artifact(&self, task_id: &str, path: &str) -> Result<impl std::io::Read> {
|
|
let url = self.task_artifact_url(task_id, path);
|
|
Ok(self.client.get(url).send()?.error_for_status()?)
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
#[test]
|
|
fn test_url() {
|
|
let cluster = super::TaskCluster {
|
|
root_url: url::Url::parse("http://taskcluster.example").unwrap(),
|
|
client: reqwest::blocking::Client::new(),
|
|
};
|
|
assert_eq!(
|
|
cluster.task_artifact_url("QzDLgP4YRwanIvgPt6ClfA","public/docker-contexts/decision.tar.gz"),
|
|
url::Url::parse("http://taskcluster.example/api/queue/v1/task/QzDLgP4YRwanIvgPt6ClfA/artifacts/public/docker-contexts/decision.tar.gz").unwrap(),
|
|
);
|
|
}
|
|
}
|