Add delivery_export.c

3 files changed, 83 insertions, 0 deletions

diff --git a/src/lib/delivery/CMakeLists.txt b/src/lib/delivery/CMakeLists.txt
index 78ed20f..559b2dc 100644
--- a/src/lib/delivery/CMakeLists.txt
+++ b/src/lib/delivery/CMakeLists.txt
@@ -1,4 +1,5 @@
 add_library(stasis_delivery STATIC
+        delivery_export.c
         delivery_postprocess.c
         delivery_conda.c
         delivery_docker.c
diff --git a/src/lib/delivery/delivery_export.c b/src/lib/delivery/delivery_export.c
new file mode 100644
index 0000000..d982ad5
--- /dev/null
+++ b/src/lib/delivery/delivery_export.c
@@ -0,0 +1,58 @@
+#include "delivery.h"
+
+static void delivery_export_configuration(const struct Delivery *ctx) {
+    msg(STASIS_MSG_L2, "Exporting delivery configuration\n");
+    if (!pushd(ctx->storage.cfgdump_dir)) {
+        char filename[PATH_MAX] = {0};
+        sprintf(filename, "%s.ini", ctx->info.release_name);
+        FILE *spec = fopen(filename, "w+");
+        if (!spec) {
+            msg(STASIS_MSG_ERROR | STASIS_MSG_L2, "failed %s\n", filename);
+            exit(1);
+        }
+        ini_write(ctx->_stasis_ini_fp.delivery, &spec, INI_WRITE_RAW);
+        fclose(spec);
+
+        memset(filename, 0, sizeof(filename));
+        sprintf(filename, "%s-rendered.ini", ctx->info.release_name);
+        spec = fopen(filename, "w+");
+        if (!spec) {
+            msg(STASIS_MSG_ERROR | STASIS_MSG_L2, "failed %s\n", filename);
+            exit(1);
+        }
+        ini_write(ctx->_stasis_ini_fp.delivery, &spec, INI_WRITE_PRESERVE);
+        fclose(spec);
+        popd();
+    } else {
+        SYSERROR("Failed to enter directory: %s", ctx->storage.delivery_dir);
+        exit(1);
+    }
+}
+
+void delivery_export(const struct Delivery *ctx, char *envs[]) {
+    delivery_export_configuration(ctx);
+
+    for (size_t i = 0; envs[i] != NULL; i++) {
+        char *name = envs[i];
+        msg(STASIS_MSG_L2, "Exporting %s\n", name);
+        if (conda_env_export(name, ctx->storage.delivery_dir, name)) {
+            msg(STASIS_MSG_ERROR | STASIS_MSG_L2, "failed %s\n", name);
+            exit(1);
+        }
+    }
+}
+
+void delivery_rewrite_stage1(struct Delivery *ctx, char *specfile) {
+    // Rewrite release environment output (i.e. set package origin(s) to point to the deployment server, etc.)
+    msg(STASIS_MSG_L3, "Rewriting release spec file (stage 1): %s\n", path_basename(specfile));
+    delivery_rewrite_spec(ctx, specfile, DELIVERY_REWRITE_SPEC_STAGE_1);
+
+    msg(STASIS_MSG_L1, "Rendering mission templates\n");
+    delivery_mission_render_files(ctx);
+}
+
+void delivery_rewrite_stage2(struct Delivery *ctx, char *specfile) {
+    msg(STASIS_MSG_L3, "Rewriting release spec file (stage 2): %s\n", path_basename(specfile));
+    delivery_rewrite_spec(ctx, specfile, DELIVERY_REWRITE_SPEC_STAGE_2);
+}
+
diff --git a/src/lib/delivery/include/delivery.h b/src/lib/delivery/include/delivery.h
index 26a5499..69ec089 100644
--- a/src/lib/delivery/include/delivery.h
+++ b/src/lib/delivery/include/delivery.h
@@ -459,4 +459,28 @@ int delivery_series_sync(struct Delivery *ctx);
  */
 int delivery_purge_packages(struct Delivery *ctx, const char *env_name, int use_pkg_manager);
 
+/**
+ * Export delivery environments
+ *
+ * @param ctx Delivery context
+ * @param envs array of conda environment names
+ */
+void delivery_export(const struct Delivery *ctx, char *envs[]);
+
+/**
+ * STAGE 1: Rewrite delivery-related strings in specfile
+ *
+ * @param ctx Delivery context
+ * @param specfile path to YAML spec file
+ */
+void delivery_rewrite_stage1(struct Delivery *ctx, char *specfile);
+
+/**
+ * STAGE 2: Rewrite delivery-related strings in specfile
+ *
+ * @param ctx Delivery context
+ * @param specfile path to YAML spec file
+ */
+void delivery_rewrite_stage2(struct Delivery *ctx, char *specfile);
+
 #endif //STASIS_DELIVERY_H