Add custom docker image support

Cargo.toml

@@ -43,7 +43,7 @@ schemamama = "0.3"
 schemamama_postgres = "0.3"
 systemstat = "0.1.4"
 prometheus = { version = "0.10.0", default-features = false }
-rustwide = "0.11"
+rustwide = { git = "https://github.com/devsnek/rustwide.git", branch = "sandbox-image-override" }
 mime_guess = "2"
 dotenv = "0.15"
 zstd = "0.5"

crates/metadata/lib.rs

@@ -95,6 +95,7 @@ pub enum MetadataError {
 /// targets = [ "x86_64-apple-darwin", "x86_64-pc-windows-msvc" ]
 /// rustc-args = [ "--example-rustc-arg" ]
 /// rustdoc-args = [ "--example-rustdoc-arg" ]
+/// docker-image = "rustops/crates-build-env"
 /// ```
 ///
 /// You can define one or more fields in your `Cargo.toml`.
@@ -128,6 +129,9 @@ pub struct Metadata {
     /// List of command line arguments for `rustdoc`.
     #[serde(default)]
     rustdoc_args: Vec<String>,
+
+    /// Custom docker image.
+    docker_image: Option<String>,
 }
 
 /// The targets that should be built for a crate.
@@ -277,6 +281,11 @@ impl Metadata {
         map.insert("DOCS_RS", "1".into());
         map
     }
+
+    /// Return the custom docker image, if provided.
+    pub fn docker_image(&self) -> Option<&String> {
+        self.docker_image.as_ref()
+    }
 }
 
 impl std::str::FromStr for Metadata {

src/docbuilder/rustwide_builder.rs

@@ -106,11 +106,15 @@ impl RustwideBuilder {
         self.skip_build_if_exists = should;
     }
 
-    fn prepare_sandbox(&self, limits: &Limits) -> SandboxBuilder {
-        SandboxBuilder::new()
+    fn prepare_sandbox(&self, metadata: &Metadata, limits: &Limits) -> Result<SandboxBuilder> {
+        let mut builder = SandboxBuilder::new()
             .cpu_limit(self.config.build_cpu_limit.map(|limit| limit as f32))
             .memory_limit(Some(limits.memory()))
-            .enable_networking(limits.networking())
+            .enable_networking(limits.networking());
+        if let Some(image) = metadata.docker_image() {
+            builder = builder.image(SandboxImage::remote(&image)?)
+        }
+        Ok(builder)
     }
 
     pub fn update_toolchain(&mut self) -> Result<()> {
@@ -208,11 +212,15 @@ impl RustwideBuilder {
         let krate = Crate::crates_io(DUMMY_CRATE_NAME, DUMMY_CRATE_VERSION);
         krate.fetch(&self.workspace)?;
 
+        let metadata = Metadata::from_crate_root(&build_dir.get_source_dir(&krate)?)?;
+
         build_dir
-            .build(&self.toolchain, &krate, self.prepare_sandbox(&limits))
+            .build(
+                &self.toolchain,
+                &krate,
+                self.prepare_sandbox(&metadata, &limits)?,
+            )
             .run(|build| {
-                let metadata = Metadata::from_crate_root(&build.host_source_dir())?;
-
                 let res = self.execute_build(HOST_TARGET, true, build, &limits, &metadata)?;
                 if !res.result.successful {
                     failure::bail!("failed to build dummy crate for {}", self.rustc_version);
@@ -322,14 +330,18 @@ impl RustwideBuilder {
 
         let local_storage = tempfile::Builder::new().prefix("docsrs-docs").tempdir()?;
 
+        let metadata = Metadata::from_crate_root(&build_dir.get_source_dir(&krate)?)?;
         let res = build_dir
-            .build(&self.toolchain, &krate, self.prepare_sandbox(&limits))
+            .build(
+                &self.toolchain,
+                &krate,
+                self.prepare_sandbox(&metadata, &limits)?,
+            )
             .run(|build| {
                 use docsrs_metadata::BuildTargets;
 
                 let mut has_docs = false;
                 let mut successful_targets = Vec::new();
-                let metadata = Metadata::from_crate_root(&build.host_source_dir())?;
                 let BuildTargets {
                     default_target,
                     other_targets,
@@ -424,6 +436,11 @@ impl RustwideBuilder {
         build_dir.purge()?;
         krate.purge_from_cache(&self.workspace)?;
         local_storage.close()?;
+        if let Some(image) = metadata.docker_image() {
+            std::process::Command::new("docker")
+                .args(&["image", "rm", image])
+                .status()?;
+        }
         Ok(res.result.successful)
     }