feat(cmd/rofl): Add support for build scripts

Author: kostko

build/rofl/manifest.go

@@ -32,6 +32,12 @@ const (
 	TEETypeTDX = "tdx"
 )
 
+// Well-known scripts.
+const (
+	ScriptBuildPre  = "build-pre"
+	ScriptBuildPost = "build-post"
+)
+
 // Manifest is the ROFL app manifest that configures various aspects of the app in a single place.
 type Manifest struct {
 	// Name is the human readable ROFL app name.
@@ -50,6 +56,9 @@ type Manifest struct {
 	// Deployments are the ROFL app deployments.
 	Deployments map[string]*Deployment `yaml:"deployments" json:"deployments"`
 
+	// Scripts are custom scripts that are executed by the build system at specific stages.
+	Scripts map[string]string `yaml:"scripts,omitempty" json:"scripts,omitempty"`
+
 	// sourceFn is the filename from which the manifest has been loaded.
 	sourceFn string
 }

cmd/rofl/build/build.go

@@ -76,6 +76,15 @@ var (
 				return
 			}
 
+			// Setup some build environment variables.
+			os.Setenv("ROFL_MANIFEST", manifest.SourceFileName())
+			os.Setenv("ROFL_DEPLOYMENT", deploymentName)
+			os.Setenv("ROFL_DEPLOYMENT_NETWORK", deployment.Network)
+			os.Setenv("ROFL_DEPLOYMENT_PARATIME", deployment.ParaTime)
+			os.Setenv("ROFL_TMPDIR", tmpDir)
+
+			runScript(manifest, buildRofl.ScriptBuildPre)
+
 			switch manifest.TEE {
 			case buildRofl.TEETypeSGX:
 				// SGX.
@@ -102,6 +111,8 @@ var (
 				return
 			}
 
+			runScript(manifest, buildRofl.ScriptBuildPost)
+
 			// Write the bundle out.
 			outFn := fmt.Sprintf("%s.%s.orc", manifest.Name, deploymentName)
 			if outputFn != "" {

cmd/rofl/build/scripts.go

@@ -0,0 +1,33 @@
+package build
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+
+	"github.com/spf13/cobra"
+
+	buildRofl "github.com/oasisprotocol/cli/build/rofl"
+)
+
+// runScripts executes the specified build script using the current build environment.
+func runScript(manifest *buildRofl.Manifest, name string) {
+	script, ok := manifest.Scripts[name]
+	if !ok {
+		return
+	}
+
+	fmt.Printf("Running script '%s'...\n", name)
+
+	cmd := exec.Command( //nolint: gosec
+		os.Getenv("SHELL"),
+		"-c",
+		script,
+	)
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+	err := cmd.Run()
+	if err != nil {
+		cobra.CheckErr(fmt.Errorf("script '%s' failed to execute: %w", name, err))
+	}
+}