Add wheel_builder CLI arguments

4 files changed, 15 insertions, 1 deletions

diff --git a/src/cli/stasis/args.c b/src/cli/stasis/args.c
index 172981a..dbc9c2f 100644
--- a/src/cli/stasis/args.c
+++ b/src/cli/stasis/args.c
@@ -15,6 +15,8 @@ struct option long_options[] = {
     {"fail-fast", no_argument, 0, OPT_FAIL_FAST},
     {"task-timeout", required_argument, 0, OPT_TASK_TIMEOUT},
     {"overwrite", no_argument, 0, OPT_OVERWRITE},
+    {"wheel-builder", required_argument, 0, OPT_WHEEL_BUILDER},
+    {"wheel-builder-manylinux-image", required_argument, 0, OPT_WHEEL_BUILDER_MANYLINUX_IMAGE},
     {"no-docker", no_argument, 0, OPT_NO_DOCKER},
     {"no-artifactory", no_argument, 0, OPT_NO_ARTIFACTORY},
     {"no-artifactory-build-info", no_argument, 0, OPT_NO_ARTIFACTORY_BUILD_INFO},
@@ -40,6 +42,8 @@ const char *long_options_help[] = {
     "On error, immediately terminate all tasks",
     "Terminate task after timeout is reached (#s, #m, #h)",
     "Overwrite an existing release",
+    "Wheel building backend (build, cibuildwheel, manylinux)",
+    "Manylinux image name",
     "Do not build docker images",
     "Do not upload artifacts to Artifactory",
     "Do not upload build info objects to Artifactory",
diff --git a/src/cli/stasis/include/args.h b/src/cli/stasis/include/args.h
index 5536735..e789261 100644
--- a/src/cli/stasis/include/args.h
+++ b/src/cli/stasis/include/args.h
@@ -19,6 +19,8 @@
 #define OPT_POOL_STATUS_INTERVAL 1011
 #define OPT_NO_TASK_LOGGING 1012
 #define OPT_TASK_TIMEOUT 1013
+#define OPT_WHEEL_BUILDER 1014
+#define OPT_WHEEL_BUILDER_MANYLINUX_IMAGE 1015
 
 extern struct option long_options[];
 void usage(char *progname);
diff --git a/src/cli/stasis/stasis_main.c b/src/cli/stasis/stasis_main.c
index 633d014..48504b9 100644
--- a/src/cli/stasis/stasis_main.c
+++ b/src/cli/stasis/stasis_main.c
@@ -605,6 +605,12 @@ int main(int argc, char *argv[]) {
             case OPT_NO_TASK_LOGGING:
                 globals.enable_task_logging = false;
                 break;
+            case OPT_WHEEL_BUILDER:
+                globals.wheel_builder = strdup(optarg);
+                break;
+            case OPT_WHEEL_BUILDER_MANYLINUX_IMAGE:
+                globals.wheel_builder_manylinux_image = strdup(optarg);
+                break;
             case '?':
             default:
                 exit(1);
diff --git a/src/lib/core/include/core.h b/src/lib/core/include/core.h
index 5a3fa85..c895267 100644
--- a/src/lib/core/include/core.h
+++ b/src/lib/core/include/core.h
@@ -51,7 +51,9 @@ struct STASIS_GLOBAL {
     char *tmpdir; //!< Path to temporary storage directory
     char *conda_install_prefix; //!< Path to install conda
     char *sysconfdir; //!< Path where STASIS reads its configuration files (mission directory, etc)
-    int task_timeout; ///< Time in seconds before task is terminated
+    int task_timeout; ///!< Time in seconds before task is terminated
+    char *wheel_builder; ///!< Backend to build wheels (build, cibuildwheel, manylinux)
+    char *wheel_builder_manylinux_image; ///!< Image to use for a Manylinux build
     struct {
         char *tox_posargs;
         char *conda_reactivate;