feat(cmd/rofl): Add bundle-post script

oasisprotocol · Jan 16, 2025 · f51bc5f · f51bc5f
1 parent 1983c80
commit f51bc5f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/build/rofl/manifest.go b/build/rofl/manifest.go
@@ -34,8 +34,9 @@ const (
 
 // Well-known scripts.
 const (
-	ScriptBuildPre  = "build-pre"
-	ScriptBuildPost = "build-post"
+	ScriptBuildPre   = "build-pre"
+	ScriptBuildPost  = "build-post"
+	ScriptBundlePost = "bundle-post"
 )
 
 // Manifest is the ROFL app manifest that configures various aspects of the app in a single place.

diff --git a/cmd/rofl/build/build.go b/cmd/rofl/build/build.go
@@ -77,7 +77,7 @@ var (
 
 			// Setup some build environment variables.
 			os.Setenv("ROFL_MANIFEST", manifest.SourceFileName())
-			os.Setenv("ROFL_DEPLOYMENT", deploymentName)
+			os.Setenv("ROFL_DEPLOYMENT_NAME", deploymentName)
 			os.Setenv("ROFL_DEPLOYMENT_NETWORK", deployment.Network)
 			os.Setenv("ROFL_DEPLOYMENT_PARATIME", deployment.ParaTime)
 			os.Setenv("ROFL_TMPDIR", tmpDir)
@@ -132,6 +132,15 @@ var (
 				return
 			}
 
+			// Setup some post-bundle environment variables.
+			os.Setenv("ROFL_BUNDLE", outFn)
+			for idx, enclaveID := range eids {
+				data, _ := enclaveID.MarshalText()
+				os.Setenv(fmt.Sprintf("ROFL_ENCLAVE_ID_%d", idx), string(data))
+			}
+
+			runScript(manifest, buildRofl.ScriptBundlePost)
+
 			// Override the update manifest flag in case the policy does not exist.
 			if deployment.Policy == nil {
 				doUpdate = false