Source code
Revision control
Copy as Markdown
Other Tools
// 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
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 {
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(),
);
}
}