Handle wheel_builder configuration items

1 files changed, 34 insertions, 0 deletions

diff --git a/src/lib/delivery/delivery_populate.c b/src/lib/delivery/delivery_populate.c
index 15ab6bd..48ea7b3 100644
--- a/src/lib/delivery/delivery_populate.c
+++ b/src/lib/delivery/delivery_populate.c
@@ -85,6 +85,40 @@ int populate_delivery_cfg(struct Delivery *ctx, int render_mode) {
     }
     globals.pip_packages = ini_getval_strlist(cfg, "default", "pip_packages", LINE_SEP, render_mode, &err);
 
+    err = 0;
+    if (!globals.wheel_builder) {
+        globals.wheel_builder = ini_getval_str(cfg, "default", "wheel_builder", render_mode, &err);
+        if (err) {
+            msg(STASIS_MSG_WARN, "wheel_builder is undefined. Falling back to system toolchain: 'build'.\n");
+            globals.wheel_builder = strdup("build");
+            if (!globals.wheel_builder) {
+                SYSERROR("%s", "unable to allocate memory for default wheel_builder value");
+                return -1;
+            }
+        }
+    }
+
+    err = 0;
+    if (!globals.wheel_builder_manylinux_image) {
+        globals.wheel_builder_manylinux_image = ini_getval_str(cfg, "default", "wheel_builder_manylinux_image", render_mode, &err);
+    }
+
+    if (err && globals.wheel_builder && strcmp(globals.wheel_builder, "manylinux") == 0) {
+        SYSERROR("%s", "default:wheel_builder is set to 'manylinux', however default:wheel_builder_manylinux_image is not configured");
+        return -1;
+    }
+
+    char *manifest_inspect_cmd = NULL;
+    if (asprintf(&manifest_inspect_cmd, "manifest inspect '%s'", globals.wheel_builder_manylinux_image) < 0) {
+        SYSERROR("%s", "unable to allocate memory for docker command");
+        return -1;
+    }
+    if (ctx->deploy.docker.capabilities.usable && docker_exec(manifest_inspect_cmd, STASIS_DOCKER_QUIET_STDOUT)) {
+        SYSERROR("Image provided by default:wheel_builder_manylinux_image does not exist: %s", globals.wheel_builder_manylinux_image);
+        return -1;
+    }
+    guard_free(manifest_inspect_cmd);
+
     if (globals.jfrog.jfrog_artifactory_base_url) {
         guard_free(globals.jfrog.jfrog_artifactory_base_url);
     }